iomgaa
ab88c50827
- 修复基础代理类的导入路径问题和配置引用 - 添加疾病分析器智能体模块 - 实现疾病风险评估和个性化建议功能 - 包含结构化输出模型和智能提示模板 - 新增项目配置文件和包初始化文件 - 更新依赖：添加packaging和sqlalchemy包 - 添加Claude指导文档 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
import os
|
|
|
|
API_KEY = "sk-263038d4bf4e46a0bed16532587cff40"
|
|
|
|
# {project_root}/medsynthai
|
|
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
# LLM model configuration based on agno
|
|
LLM_CONFIG = {
|
|
"deepseek": {
|
|
"class": "DeepSeek",
|
|
"params": {
|
|
"id": "deepseek-chat",
|
|
"api_key": API_KEY,
|
|
"base_url": "https://api.deepseek.com"
|
|
}
|
|
},
|
|
"ollama": {
|
|
"class": "Ollama",
|
|
"params": {
|
|
"id": "qwen2.5:latest",
|
|
"host": "127.0.0.1"
|
|
}
|
|
},
|
|
"deepseek-v3": {
|
|
"class": "OpenAILike",
|
|
"params": {
|
|
"id": "deepseek-v3",
|
|
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
"api_key": API_KEY
|
|
}
|
|
},
|
|
"deepseek-r1": {
|
|
"class": "OpenAILike",
|
|
"params": {
|
|
"id": "deepseek-r1",
|
|
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
"api_key": API_KEY
|
|
}
|
|
},
|
|
"qwen-max": {
|
|
"class": "OpenAILike",
|
|
"params": {
|
|
"id": "qwen-max",
|
|
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
"api_key": API_KEY
|
|
}
|
|
},
|
|
"qwen-vl-max": {
|
|
"class": "OpenAILike", # 使用OpenAI兼容类
|
|
"params": {
|
|
"id": "qwen-vl-max",
|
|
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1", # OpenAI兼容端点 # 降低随机性，提高结果一致性
|
|
"api_key": API_KEY
|
|
}
|
|
},
|
|
"aliyun": {
|
|
"class": "OpenAILike",
|
|
"params": {
|
|
"id": "qwen-max", # 默认使用qwen-max模型
|
|
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
"api_key": API_KEY
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
AGENT_CONFIG = {
|
|
"medical_image_analysis_agent": "qwen-vl-max"
|
|
}
|
|
|
|
RAG_CONFIG = {
|
|
"lightrag": {
|
|
"working_dir": "./Vector_DB_Med",
|
|
"tokenizer_name": "trueto/medbert-base-chinese",
|
|
"model_name": "trueto/medbert-base-chinese",
|
|
"embedding_dim": 768,
|
|
"max_token_size": 512
|
|
},
|
|
"chroma_db": {
|
|
"api_key": API_KEY,
|
|
"base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
"collection_name": "doctor",
|
|
"batch_size": 100,
|
|
"chroma_db_path": os.path.join(BASE_DIR, "static/rag/chroma_db"),
|
|
"csv_path": os.path.join(BASE_DIR, "static/files/zhongkang_doctor_list.csv")
|
|
}
|
|
}
|