diff --git a/backend/src/repository/rag/chat.py b/backend/src/repository/rag/chat.py index 21ea4f0..89fc3cf 100644 --- 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)