try to fix input as a list

Signed-off-by: Aisuko <[email protected]>
SkywardAI · Jul 21, 2024 · ab4e0ec · ab4e0ec
1 parent 22c10c2
commit ab4e0ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/src/repository/rag/chat.py b/backend/src/repository/rag/chat.py
@@ -137,7 +137,7 @@ async def get_context_by_question(input_msg: str):
             except Exception as e:
                 loguru.logger.error(e)
             # collection name for testing
-            context = vector_db.search(embedd_input, 1, collection_name="aisuko_squad01")
+            context = vector_db.search(list(embedd_input), 1, collection_name="aisuko_squad01")
             return context or InferenceHelper.instruction
 
         current_context = await get_context_by_question(input_msg)