scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

TestDockerModelRunnerIntegration

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-21 17:39:33 -08:00
parent 3296ffd3ca
commit 1f36fad94b
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/llms/docker_model_runner/test_docker_model_runner_chat_transformation.py
View file

@ -15,6 +15,7 @@ sys.path.insert(
import json
from unittest.mock import Mock, patch
import httpx
import pytest
import litellm
@ -32,7 +33,7 @@ class TestDockerModelRunnerIntegration:
"""
with patch("litellm.llms.custom_httpx.http_handler.HTTPHandler.post") as mock_post:
# Mock the response
mock_response = Mock()
mock_response = Mock(spec=httpx.Response)
mock_response.json.return_value = {
"id": "chatcmpl-123",
"object": "chat.completion",
@ -53,7 +54,7 @@ class TestDockerModelRunnerIntegration:
}
}
mock_response.status_code = 200
mock_response.headers = {"content-type": "application/json"}
mock_response.headers = httpx.Headers({"content-type": "application/json"})
mock_post.return_value = mock_response
# Make the completion call with engine in api_base
@ -103,7 +104,7 @@ class TestDockerModelRunnerIntegration:
"""
with patch("litellm.llms.custom_httpx.http_handler.HTTPHandler.post") as mock_post:
# Mock the response
mock_response = Mock()
mock_response = Mock(spec=httpx.Response)
mock_response.json.return_value = {
"id": "chatcmpl-456",
"object": "chat.completion",
@ -124,7 +125,7 @@ class TestDockerModelRunnerIntegration:
}
}
mock_response.status_code = 200
mock_response.headers = {"content-type": "application/json"}
mock_response.headers = httpx.Headers({"content-type": "application/json"})
mock_post.return_value = mock_response
# Make the completion call with custom engine and host