ChoCoFly commited on
Commit
4626bd2
·
verified ·
1 Parent(s): 41955ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -18
app.py CHANGED
@@ -373,26 +373,9 @@ class SpyAgent(BasicAgent):
373
 
374
  result = completion.choices[0].message.content.lstrip('\n\t\r')
375
  logger.info("initial llm result: {}".format(result))
376
-
377
- # 提取真正需要的内容,去除分析和额外内容
378
- session_data = [{'role': 'assistant', 'content': result}]
379
- extract_prompt = '请从你的上述回答中提取出最核心的答案部分,不要包含任何分析过程、附加说明或其他内容。请确保提取的内容简洁明了,直接给出答案。'
380
-
381
- session_data.append({'role': 'user', 'content': extract_prompt})
382
 
383
- # 第二次调用获取提取后的纯答案
384
- extract_completion = client.chat.completions.create(
385
- model=self.model_name,
386
- messages=session_data
387
- )
388
-
389
- final_result = extract_completion.choices[0].message.content.lstrip('\n\t\r')
390
- logger.info("extracted llm result: {}".format(final_result))
391
-
392
- # 进一步清理结果,去除多余的空格、换行和引号
393
- final_result = final_result.strip(' \n\r\t"\'').split('\n', 1)[0]
394
 
395
- return final_result
396
 
397
  except Exception as e:
398
  logger.error(f"LLM调用错误: {e}")
 
373
 
374
  result = completion.choices[0].message.content.lstrip('\n\t\r')
375
  logger.info("initial llm result: {}".format(result))
 
 
 
 
 
 
376
 
 
 
 
 
 
 
 
 
 
 
 
377
 
378
+ return result
379
 
380
  except Exception as e:
381
  logger.error(f"LLM调用错误: {e}")