2MoOn24mOoN4 commited on
Commit
a5f60d5
·
verified ·
1 Parent(s): 0d58a9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -5
app.py CHANGED
@@ -20,11 +20,21 @@ class BasicAgent:
20
  def __call__(self, question: str) -> str:
21
  print(f"Agent received question (first 50 chars): {question[:50]}...")
22
  messages = [{"role": "system", "content": sys_prompt},{"role": "user", "content": question}]
23
- response = self.agent.run(messages)
24
- answer = response['messages'][-1].content
25
- if isinstance(answer, list):
26
- answer = answer[0] if answer else '' # 提取第一个元素或返回空字符串
27
- return answer.strip() if isinstance(answer, str) else str(answer).strip()
 
 
 
 
 
 
 
 
 
 
28
 
29
  def run_and_submit_all( profile: gr.OAuthProfile | None):
30
  """
 
20
  def __call__(self, question: str) -> str:
21
  print(f"Agent received question (first 50 chars): {question[:50]}...")
22
  messages = [{"role": "system", "content": sys_prompt},{"role": "user", "content": question}]
23
+
24
+ try:
25
+ answer = self.agent.run(messages)
26
+ #answer = response['body'][-1].content
27
+ print(f"Answer type: {type(answer)}, value: {answer}")
28
+ # 处理 answer 的类型
29
+ if isinstance(answer, list):
30
+ answer = answer[0].get('body', '') if answer else ''
31
+ elif not isinstance(answer, str):
32
+ answer = str(answer)
33
+ return answer.strip()
34
+ except Exception as e:
35
+ print(f"Error processing question: {e}")
36
+ return f"Error: {e}"
37
+
38
 
39
  def run_and_submit_all( profile: gr.OAuthProfile | None):
40
  """