优化:改进DiseaseAnalyst模型配置和输出格式

主要改进:
- 更新默认模型:从deepseek-v3改为gpt-oss:latest统一配置
- 优化prompt模板:添加明确的输出指令提升JSON解析成功率
- 统一模型配置:保持与项目整体配置一致性
- 完善文档注释:更新模型类型说明信息

技术细节:
- 调整constructor默认参数适配新模型配置
- 在prompt末尾添加"输出内容为:"引导符
- 保持向后兼容性支持自定义模型配置

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
iomgaa 2025-08-11 00:03:00 +08:00
parent bc471256f4
commit ba0e04362a

View File

@ -18,16 +18,16 @@ class DiseaseContextAnalyst(BaseAgent):
4. 为后续子任务确定针对性的评估重点 4. 为后续子任务确定针对性的评估重点
Attributes: Attributes:
model_type (str): 使用的大语言模型类型默认为 deepseek-v3 model_type (str): 使用的大语言模型类型默认为 gpt-oss:latest
llm_config (dict): LLM模型配置参数 llm_config (dict): LLM模型配置参数
""" """
def __init__(self, model_type: str = "deepseek-v3", llm_config: dict = None): def __init__(self, model_type: str = "gpt-oss:latest", llm_config: dict = None):
""" """
初始化疾病上下文分析智能体 初始化疾病上下文分析智能体
Args: Args:
model_type (str): 大语言模型类型默认使用 deepseek-v3 model_type (str): 大语言模型类型默认使用 gpt-oss:latest
llm_config (dict): LLM模型的配置参数如果为None则使用默认配置 llm_config (dict): LLM模型的配置参数如果为None则使用默认配置
""" """
super().__init__( super().__init__(
@ -148,7 +148,8 @@ class DiseaseContextAnalyst(BaseAgent):
输出格式示例 输出格式示例
{example_output} {example_output}
请严格按照上述JSON格式输出""" 请严格按照上述JSON格式输出
输出内容为:"""
return prompt return prompt