github-mirrors/tradingagents
1
0
Fork 0
mirror of https://github.com/TauricResearch/TradingAgents.git synced 2026-05-01 14:33:10 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix model catalog typing and known-model helper

Browse Source
This commit is contained in:
CadeYu
2026-03-25 21:46:56 +08:00
parent 8793336dad
commit bd6a5b75b5
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tradingagents/llm_clients/model_catalog.py
View File

@@ -5,7 +5,7 @@ from __future__ import annotations
from typing import Dict, List, Tuple from typing import Dict, List, Tuple
ModelOption = Tuple[str, str] ModelOption = Tuple[str, str]
ProviderModeOptions = Dict[str, List[ModelOption]] ProviderModeOptions = Dict[str, Dict[str, List[ModelOption]]]
MODEL_OPTIONS: ProviderModeOptions = { MODEL_OPTIONS: ProviderModeOptions = {
@@ -95,12 +95,13 @@ def get_model_options(provider: str, mode: str) -> List[ModelOption]:
def get_known_models() -> Dict[str, List[str]]: def get_known_models() -> Dict[str, List[str]]:
"""Build known model names from the shared CLI catalog.""" """Build known model names from the shared CLI catalog."""
known_models: Dict[str, List[str]] = {} return {
for provider, mode_options in MODEL_OPTIONS.items(): provider: sorted(
model_names = { {
value value
for options in mode_options.values() for options in mode_options.values()
for _, value in options for _, value in options
} }
known_models[provider] = sorted(model_names) )
return known_models for provider, mode_options in MODEL_OPTIONS.items()
}