fix model catalog typing and known-model helper

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

View File

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