From e7a32c1e8f40c15bfcc9bfbaa225865bb8a740a9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 21 Nov 2025 16:52:58 -0800 Subject: [PATCH] docker test fixes --- .../test_docker_model_runner_chat_transformation.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/llms/docker_model_runner/test_docker_model_runner_chat_transformation.py b/tests/test_litellm/llms/docker_model_runner/test_docker_model_runner_chat_transformation.py index 9cd76c3ef6a..bcddb33dbb0 100644 --- a/tests/test_litellm/llms/docker_model_runner/test_docker_model_runner_chat_transformation.py +++ b/tests/test_litellm/llms/docker_model_runner/test_docker_model_runner_chat_transformation.py @@ -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