scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-16 23:41:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

Fixup ollama model listing (again) (#13008)

Browse source
This commit is contained in:
Matthias Dittrich 2025-07-26 20:07:49 +02:00 • committed by GitHub
parent 10595ea077
commit 9211158e73
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
litellm/__init__.py
View file

@ -797,6 +797,7 @@ models_by_provider: dict = {
"bedrock": bedrock_models + bedrock_converse_models,
"petals": petals_models,
"ollama": ollama_models,
"ollama_chat": ollama_models,
"deepinfra": deepinfra_models,
"perplexity": perplexity_models,
"maritalk": maritalk_models,

3
litellm/types/router.py
View file

@ -342,7 +342,8 @@ class LiteLLM_Params(GenericLiteLLMParams):
args.pop("__class__", None)
if max_retries is not None and isinstance(max_retries, str):
max_retries = int(max_retries) # cast to int
super().__init__(max_retries=max_retries, **args, **params)
args["max_retries"] = max_retries
super().__init__(**{ **args, **params })
def __contains__(self, key):
# Define custom behavior for the 'in' operator