charieswei
/
TradingAgents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: extract shared model lists (AZURE_MODELS, OLLAMA_MODELS) to reduce duplication

This commit is contained in:
阳虎 2026-03-16 16:23:51 +08:00
parent ca80574ed1
commit 760f084a6a
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
cli/utils.py
View File

@ -180,12 +180,8 @@ def select_shallow_thinking_agent(provider) -> str:
("GPT-OSS:latest (20B, local)", "gpt-oss:latest"), ("GPT-OSS:latest (20B, local)", "gpt-oss:latest"),
("GLM-4.7-Flash:latest (30B, local)", "glm-4.7-flash:latest"), ("GLM-4.7-Flash:latest (30B, local)", "glm-4.7-flash:latest"),
], ],
"azure": [ "azure": AZURE_MODELS,
("GPT-4o (your deployment)", "gpt-4o"), "ollama": OLLAMA_MODELS,
("GPT-4o-mini (your deployment)", "gpt-4o-mini"),
("GPT-4 Turbo (your deployment)", "gpt-4-turbo"),
("Custom deployment name", "__custom__"),
],
} }
choice = questionary.select( choice = questionary.select(
@ -253,12 +249,8 @@ def select_deep_thinking_agent(provider) -> str:
("GPT-OSS:latest (20B, local)", "gpt-oss:latest"), ("GPT-OSS:latest (20B, local)", "gpt-oss:latest"),
("Qwen3:latest (8B, local)", "qwen3:latest"), ("Qwen3:latest (8B, local)", "qwen3:latest"),
], ],
"azure": [ "azure": AZURE_MODELS,
("GPT-4o (your deployment)", "gpt-4o"), "ollama": OLLAMA_MODELS,
("GPT-4o-mini (your deployment)", "gpt-4o-mini"),
("GPT-4 Turbo (your deployment)", "gpt-4-turbo"),
("Custom deployment name", "__custom__"),
],
} }
choice = questionary.select( choice = questionary.select(