diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py index 61953d8751c50ecc739516702e9cd024ee29a096..4ceebac69fa35138bde09c1c7c2086eefc7bb21f 100644 --- a/tests/unit/test_router.py +++ b/tests/unit/test_router.py @@ -138,7 +138,7 @@ def openai_encoder(mocker): mocker.patch.object(OpenAIEncoder, "acall", side_effect=async_mock_encoder_call) # Create and return the mocked encoder encoder = OpenAIEncoder( - name="text-embedding-3-small", openai_api_key="test_api_key" + name="text-embedding-3-small" ) return encoder @@ -154,7 +154,7 @@ def mock_openai_llm(mocker): mocker.patch.object(OpenAILLM, "acall", side_effect=async_mock_llm_call) - return OpenAILLM(name="fake-model-v1", openai_api_key="test_llm_api_key") + return OpenAILLM(name="fake-model-v1") @pytest.fixture