Update app.py
Browse files
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
|
| 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}")
|