From 4b890e91bf7e18a4a9bd54c77d62e4956ccb4573 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 4 Jan 2025 15:54:09 +0400
Subject: [PATCH] fix: openai mock
---
tests/unit/test_router.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index 61953d87..4ceebac6 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
--
GitLab