ppasd000 commited on
Commit
d3c0253
·
verified ·
1 Parent(s): 2a389ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -19,7 +19,7 @@ CONFIG = {
19
  }
20
 
21
  # 是否启用上下文
22
- ENABLE_CONTEXT = False
23
 
24
  app = Flask(__name__)
25
  CORS(app)
@@ -31,10 +31,10 @@ def process_messages(messages):
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,7 +52,9 @@ def process_messages(messages):
52
  current_message = messages[-1]["content"]
53
  full_message = f"{full_context}{current_message}"
54
 
55
- return full_message
 
 
56
 
57
 
58
 
 
19
  }
20
 
21
  # 是否启用上下文
22
+ ENABLE_CONTEXT = True
23
 
24
  app = Flask(__name__)
25
  CORS(app)
 
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
  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
 
60