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

docker test fixes

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-21 16:52:58 -08:00
parent 473fec8a60
commit e7a32c1e8f
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -24,8 +24,7 @@ from litellm import completion
class TestDockerModelRunnerIntegration:
"""Integration test for Docker Model Runner"""
@pytest.mark.asyncio
async def test_completion_hits_correct_url_and_body(self):
def test_completion_hits_correct_url_and_body(self):
"""
Test that litellm.completion with docker_model_runner provider:
1. Hits the correct URL: {api_base}/v1/chat/completions where api_base includes engine path
@ -95,8 +94,7 @@ class TestDockerModelRunnerIntegration:
# Verify response
assert response.choices[0].message.content == "Hello! How can I help you today?"
@pytest.mark.asyncio
async def test_completion_with_custom_engine_and_host(self):
def test_completion_with_custom_engine_and_host(self):
"""
Test that litellm.completion works with custom engine and host:
1. Uses model-runner.docker.internal as host