ppasd000 commited on
Commit
37c78e1
·
verified ·
1 Parent(s): d3c0253

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -31,10 +31,10 @@ def process_messages(messages):
31
  Args:
32
  messages: 消息列表,每个消息包含 role 和 content
33
  Returns:
34
- list: 处理后的消息列表,保持原有的消息结构
35
  """
36
  if not messages:
37
- return [{"role": "user", "content": CONFIG["DEFAULT_MESSAGE"]}]
38
 
39
  # 添加引导语
40
  instruction = "我们接下来的讨论是基于我们在使用Claude时遇到的一些问题。你只需要回答问题就行了,回答内容不要引用文档,明白吗?\n\n"
@@ -52,8 +52,7 @@ def process_messages(messages):
52
  current_message = messages[-1]["content"]
53
  full_message = f"{full_context}{current_message}"
54
 
55
- # 返回保持原有结构的消息列表,只是更新了最后一条消息的内容
56
- return [{"role": messages[-1]["role"], "content": full_message}]
57
 
58
 
59
 
@@ -217,8 +216,7 @@ def chat_completions():
217
 
218
  messages = req.get("messages", [])
219
  if ENABLE_CONTEXT:
220
- processed = process_messages(messages)
221
- message_input = processed[-1] if processed else CONFIG["DEFAULT_MESSAGE"]
222
  else:
223
  message_input = (
224
  messages[-1]["content"] if messages else CONFIG["DEFAULT_MESSAGE"]
 
31
  Args:
32
  messages: 消息列表,每个消息包含 role 和 content
33
  Returns:
34
+ str: 处理后的消息内容
35
  """
36
  if not messages:
37
+ return CONFIG["DEFAULT_MESSAGE"]
38
 
39
  # 添加引导语
40
  instruction = "我们接下来的讨论是基于我们在使用Claude时遇到的一些问题。你只需要回答问题就行了,回答内容不要引用文档,明白吗?\n\n"
 
52
  current_message = messages[-1]["content"]
53
  full_message = f"{full_context}{current_message}"
54
 
55
+ return full_message
 
56
 
57
 
58
 
 
216
 
217
  messages = req.get("messages", [])
218
  if ENABLE_CONTEXT:
219
+ message_input = process_messages(messages)
 
220
  else:
221
  message_input = (
222
  messages[-1]["content"] if messages else CONFIG["DEFAULT_MESSAGE"]