From 8a6b69096c51a7bfdd2f38415a8b755ce1cbc1ef Mon Sep 17 00:00:00 2001 From: Yazhou Cao Date: Wed, 15 May 2024 17:47:51 -0700 Subject: [PATCH] Fix unit test --- tests/fixtures.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/fixtures.py b/tests/fixtures.py index 602096ac..75013c9f 100644 --- a/tests/fixtures.py +++ b/tests/fixtures.py @@ -9,7 +9,7 @@ def openai_llm_mock(request): content = request.param # Note the path here is adjusted to where OpenAI is used, not where it's defined - with patch("vision_agent.llm.llm.OpenAI") as mock: + with patch("vision_agent.llm.llm.wrap_openai") as mock: # Setup a mock response structure that matches what your code expects mock_instance = mock.return_value mock_instance.chat.completions.create.return_value = MagicMock(