From ca80574ed1873ae1c56071d99186ed66b90004df Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=98=B3=E8=99=8E?= Date: Mon, 16 Mar 2026 16:20:59 +0800 Subject: [PATCH] refactor: extract AZURE_MODELS constant to reduce code duplication --- cli/utils.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/cli/utils.py b/cli/utils.py index 3bfc80a6..a8dee8e0 100644 --- a/cli/utils.py +++ b/cli/utils.py @@ -14,6 +14,22 @@ ANALYST_ORDER = [ ("Fundamentals Analyst", AnalystType.FUNDAMENTALS), ] +# Shared model lists for providers with identical options in shallow/deep selections +# Azure: deployment names are user-specific, so we provide common defaults +AZURE_MODELS = [ + ("GPT-4o (your deployment)", "gpt-4o"), + ("GPT-4o-mini (your deployment)", "gpt-4o-mini"), + ("GPT-4 Turbo (your deployment)", "gpt-4-turbo"), + ("Custom deployment name", "__custom__"), +] + +# Ollama: same models for both shallow and deep (local inference) +OLLAMA_MODELS = [ + ("Qwen3:latest (8B, local)", "qwen3:latest"), + ("GPT-OSS:latest (20B, local)", "gpt-oss:latest"), + ("GLM-4.7-Flash:latest (30B, local)", "glm-4.7-flash:latest"), +] + def get_ticker() -> str: """Prompt the user to enter a ticker symbol."""