AFischer1985 commited on
Commit
f359740
·
verified ·
1 Parent(s): 05ce570

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -3
run.py CHANGED
@@ -198,9 +198,7 @@ def response(
198
  seed=42,
199
  )
200
  addon=""
201
- first_message=message
202
- if(len(history)>0):
203
- first_message=history[0]["user_message"]
204
  results=collection.query(
205
  query_texts=[first_message],
206
  n_results=2,
 
198
  seed=42,
199
  )
200
  addon=""
201
+ first_message = history[0][0] if history else message
 
 
202
  results=collection.query(
203
  query_texts=[first_message],
204
  n_results=2,