Cudd1es commited on
Commit
ebd4559
·
1 Parent(s): c881ef3

fixed app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -39,11 +39,9 @@ def respond(
39
  history: [{"role": "user", "content": ...}, {"role": "assistant", "content": ...}, ...]
40
  system_message: 自定义 System Prompt
41
  """
42
- chat_history = [
43
- {"role": "system", "content": system_message.strip() or "你是BangDream知识问答助手, 只能基于提供资料作答。"}
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)