AI-RAG-Interface-to-Hub

Running

AFischer1985 commited on May 27, 2024

Commit

56c713d

verified ·

1 Parent(s): ebe03ef

Update run.py

Files changed (1) hide show

run.py CHANGED Viewed

@@ -198,7 +198,8 @@ def response(
         seed=42,
     )
     addon=""
-    first_message=history[1]["user_message"]
     results=collection.query(
       query_texts=[first_message],
       n_results=2,

         seed=42,
     )
     addon=""
+    if(len(history)==0)first_message=message
+    first_message=history[0]["user_message"]
     results=collection.query(
       query_texts=[first_message],
       n_results=2,