model providers polishing
frdel committed
Jul 21, 2025 at 20:45 UTC
a4f13fc26f0ee99ced366b9e8cbe5f5c0bc183e3
3 files changed
+8
-20
agent.py
+1
-1
@@ -671,7 +671,7 @@ class Agent:
671
type="util",
672
update_progress="none",
673
heading=msg,
674
- model=f"{model_config.provider.value}\\{model_config.name}",
674
+ model=f"{model_config.provider}\\{model_config.name}",
675
)
676
wait_log.update(heading=msg, key=key, value=total, limit=limit)
677
if not background:
conf/model_providers.yaml
+3
@@ -70,6 +70,9 @@ embedding:
70
huggingface:
71
name: HuggingFace
72
litellm_provider: huggingface
73
+ google:
74
+ name: Google
75
+ litellm_provider: gemini
76
lm_studio:
77
name: LM Studio
78
litellm_provider: lm_studio
models.py
+4
-19
@@ -469,11 +469,6 @@ def _merge_provider_defaults(
469
for k, v in extra_kwargs.items():
470
kwargs.setdefault(k, v)
471
472
- # Copy any additional top-level fields except metadata keys
473
- for k, v in cfg.items():
474
- if k not in ("id", "name", "value", "litellm_provider", "kwargs"):
475
- kwargs.setdefault(k, v)
476
-
472
# Inject API key based on the *original* provider id if still missing
473
if "api_key" not in kwargs:
474
key = get_api_key(original_provider)
@@ -483,19 +478,7 @@ def _merge_provider_defaults(
478
return provider_name, kwargs
479
480
486
-def get_model(type: ModelType, provider: str, name: str, **kwargs: Any):
487
- provider_name = provider.lower()
488
- if type == ModelType.CHAT:
489
- return _get_litellm_chat(LiteLLMChatWrapper, name, provider_name, **kwargs)
490
- elif type == ModelType.EMBEDDING:
491
- return _get_litellm_embedding(name, provider_name, **kwargs)
492
- else:
493
- raise ValueError(f"Unsupported model type: {type}")
494
-
495
-
496
-def get_chat_model(
497
- provider: str, name: str, **kwargs: Any
498
-) -> LiteLLMChatWrapper:
481
+def get_chat_model(provider: str, name: str, **kwargs: Any) -> LiteLLMChatWrapper:
482
orig = provider.lower()
483
provider_name, kwargs = _merge_provider_defaults("chat", orig, kwargs)
484
return _get_litellm_chat(LiteLLMChatWrapper, name, provider_name, **kwargs)
@@ -506,7 +489,9 @@ def get_browser_model(
489
) -> BrowserCompatibleChatWrapper:
490
orig = provider.lower()
491
provider_name, kwargs = _merge_provider_defaults("chat", orig, kwargs)
509
- return _get_litellm_chat(BrowserCompatibleChatWrapper, name, provider_name, **kwargs)
492
+ return _get_litellm_chat(
493
+ BrowserCompatibleChatWrapper, name, provider_name, **kwargs
494
+ )
495
496
497
def get_embedding_model(