fixed app.py
Browse files
app.py
CHANGED
|
@@ -39,11 +39,9 @@ def respond(
|
|
| 39 |
history: [{"role": "user", "content": ...}, {"role": "assistant", "content": ...}, ...]
|
| 40 |
system_message: 自定义 System Prompt
|
| 41 |
"""
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
]
|
| 45 |
-
chat_history.extend(history)
|
| 46 |
-
chat_history.append({"role": "user", "content": message})
|
| 47 |
|
| 48 |
query_vec = encode_query(encoder, message)
|
| 49 |
results = retrieve_docs(collection, query_vec, top_k=30)
|
|
|
|
| 39 |
history: [{"role": "user", "content": ...}, {"role": "assistant", "content": ...}, ...]
|
| 40 |
system_message: 自定义 System Prompt
|
| 41 |
"""
|
| 42 |
+
default_system_message = "你是BangDream知识问答助手, 只能基于提供资料作答。"
|
| 43 |
+
system_msg = (system_message or default_system_message).strip()
|
| 44 |
+
chat_history = [{"role": "system", "content": system_msg}]
|
|
|
|
|
|
|
| 45 |
|
| 46 |
query_vec = encode_query(encoder, message)
|
| 47 |
results = retrieve_docs(collection, query_vec, top_k=30)
|