diff --git a/litellm/llms/ollama_chat.py b/litellm/llms/ollama_chat.py index b84705998b..45aab0f959 100644 --- a/litellm/llms/ollama_chat.py +++ b/litellm/llms/ollama_chat.py @@ -298,6 +298,7 @@ def get_ollama_response( ], ) model_response["choices"][0]["message"] = message + model_response["choices"][0]["finish_reason"] = "tool_calls" else: model_response["choices"][0]["message"] = response_json["message"] model_response["created"] = int(time.time()) @@ -480,6 +481,7 @@ async def ollama_acompletion( ], ) model_response["choices"][0]["message"] = message + model_response["choices"][0]["finish_reason"] = "tool_calls" else: model_response["choices"][0]["message"] = response_json["message"]