fix import in test_anthropic_client.py

uripeled2 · Jun 30, 2023 · 4370e01 · 4370e01
1 parent 213ca8e
commit 4370e01
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/llm_api_client/anthropic_client/test_anthropic_client.py b/tests/llm_api_client/anthropic_client/test_anthropic_client.py
@@ -1,6 +1,7 @@
 import pytest
 
-from llm_client import LLMAPIClientFactory, LLMAPIClientType, AnthropicClient
+from llm_client import LLMAPIClientFactory, LLMAPIClientType
+from llm_client.llm_api_client.anthropic_client import AnthropicClient
 from llm_client.consts import PROMPT_KEY, MODEL_KEY
 from llm_client.llm_api_client.anthropic_client import AUTH_HEADER, COMPLETIONS_KEY, MAX_TOKENS_KEY, ACCEPT_HEADER, \
     ACCEPT_VALUE, VERSION_HEADER