charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor: remove magic string

This commit is contained in:
mogita 2025-08-16 17:05:47 +08:00
parent 5c7c5e091e
commit 937036331e
No known key found for this signature in database
GPG Key ID: A0AA1B9C57A48ECF
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cli/utils.py
View File

@ -10,6 +10,8 @@ from cli.models import AnalystType
console = Console() console = Console()
CUSTOM_MODEL_IDENTIFIER = "__CUSTOM_MODEL__"
ANALYST_ORDER = [ ANALYST_ORDER = [
("Market Analyst", AnalystType.MARKET), ("Market Analyst", AnalystType.MARKET),
("Social Media Analyst", AnalystType.SOCIAL), ("Social Media Analyst", AnalystType.SOCIAL),
@ -215,7 +217,7 @@ def _get_all_models_for_custom_provider(model_type: str) -> list:
all_models.append((labeled_description, model_value)) all_models.append((labeled_description, model_value))
# Add custom model option at the end # Add custom model option at the end
all_models.append(("Custom Model - Enter your own model name", "__CUSTOM_MODEL__")) all_models.append(("Custom Model - Enter your own model name", CUSTOM_MODEL_IDENTIFIER))
return all_models return all_models
@ -253,7 +255,7 @@ def _select_custom_provider_model(model_type: str, title: str, default_model: st
raise ValueError(f"No {model_type} thinking model selected") raise ValueError(f"No {model_type} thinking model selected")
# Handle custom model input # Handle custom model input
if choice == "__CUSTOM_MODEL__": if choice == CUSTOM_MODEL_IDENTIFIER:
custom_model = questionary.text( custom_model = questionary.text(
f"Enter your custom {model_type} thinking model name:", f"Enter your custom {model_type} thinking model name:",
default=default_model, default=default_model,