从向量数据库中检索图片并集成到chat-robot

2024-10-29 16:50:46 +08:00
parent bf8b3b417b
commit 9d11e995dd
7 changed files with 217 additions and 25 deletions
--- a/app/service/chat_robot/script/main.py
+++ b/app/service/chat_robot/script/main.py
@@ -100,7 +100,7 @@ def chat(post_data):
    #     session_key=f"buffer:{user_id}:{session_id}",
    # )

-    final_outputs = CallQWen.call_with_messages(input_message)
+    final_outputs = CallQWen.call_with_messages(input_message, gender)
    # api_response = {
    #     'user_id': user_id,
    #     'session_id': session_id,