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

chore: Improved OllamaConfig get_required_params and ollama_acompletion and ollama_async_streaming functions

Browse source
This commit is contained in:
corrm 2024-06-24 05:55:22 +03:00
parent b2c1a3ad79
commit 423a60c8bc
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/ollama.py
View file

@ -126,7 +126,7 @@ class OllamaConfig:
)
and v is not None
}
def get_required_params(self) -> List[ProviderField]:
"""For a given provider, return it's required fields with a description"""
return [
@ -451,7 +451,7 @@ async def ollama_acompletion(url, data, model_response, encoding, logging_obj):
{
"id": f"call_{str(uuid.uuid4())}",
"function": {
"name": function_call["name"],
"name": function_call.get("name", function_call.get("function", None)),
"arguments": json.dumps(function_call["arguments"]),
},
"type": "function",