scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-02 02:11:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

Update test_ollama_chat_transformation.py

Browse source
This commit is contained in:
Jared Everett 2026-04-15 19:18:31 -06:00 • committed by GitHub
parent e29502660a
commit be66aa2bd1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/test_litellm/llms/ollama/test_ollama_chat_transformation.py
View file

@ -15,6 +15,12 @@ from litellm.llms.ollama.chat.transformation import OllamaChatConfig, OllamaChat
from litellm.types.llms.openai import AllMessageValues
from litellm.utils import get_optional_params
import json
from unittest.mock import MagicMock
import litellm
from litellm.types.utils import Choices, Message, ModelResponse
class TestEvent(BaseModel):
name: str
@ -427,12 +433,6 @@ class TestOllamaToolCalling:
def test_finish_reason_stop_when_no_tool_calls(self):
"""Test that finish_reason remains 'stop' when no tool_calls present."""
import json
from unittest.mock import MagicMock
import litellm
from litellm.types.utils import Choices, Message, ModelResponse
config = OllamaChatConfig()
# Simulated Ollama response without tool_calls
@ -486,11 +486,6 @@ class TestOllamaFinishReasonLength:
def test_finish_reason_length_non_streaming(self):
"""Non-streaming: done_reason='length' must propagate as finish_reason='length'."""
import json
from unittest.mock import MagicMock
from litellm.types.utils import Choices, Message, ModelResponse
config = OllamaChatConfig()
ollama_response = {
@ -535,11 +530,6 @@ class TestOllamaFinishReasonLength:
def test_finish_reason_stop_non_streaming(self):
"""Non-streaming: done_reason='stop' (natural finish) must stay 'stop'."""
import json
from unittest.mock import MagicMock
from litellm.types.utils import Choices, Message, ModelResponse
config = OllamaChatConfig()
ollama_response = {