Fix extra Anthropic import error.

eli64s · Oct 7, 2024 · 0793a71 · 0793a71
1 parent 2337e4b
commit 0793a71
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "readmeai"
-version = "0.5.91"
+version = "0.5.92"
 description = "Automated README file generator, powered by AI."
 authors = ["Eli <egsalamie@gmail.com>"]
 license = "MIT"

diff --git a/readmeai/models/factory.py b/readmeai/models/factory.py
@@ -4,8 +4,8 @@
 from readmeai.config.settings import ConfigLoader
 from readmeai.errors import UnsupportedServiceError
 from readmeai.ingestion.models import RepositoryContext
-from readmeai.models.anthropic import AnthropicHandler
 from readmeai.models.base import BaseModelHandler
+from readmeai.models.claude import AnthropicHandler
 from readmeai.models.gemini import GeminiHandler
 from readmeai.models.offline import OfflineHandler
 from readmeai.models.openai import OpenAIHandler

diff --git a/repo-banner.svg b/repo-banner.svg
diff --git a/tests/models/test_claude.py b/tests/models/test_claude.py
@@ -6,7 +6,7 @@
 
 from readmeai.config.settings import ConfigLoader
 from readmeai.ingestion.models import RepositoryContext
-from readmeai.models.anthropic import AnthropicHandler
+from readmeai.models.claude import AnthropicHandler
 
 
 @pytest.fixture
@@ -34,7 +34,7 @@ async def test_build_payload(anthropic_handler):
 
 
 @pytest.mark.asyncio
-@patch("readmeai.models.anthropic.token_handler", new_callable=AsyncMock)
+@patch("readmeai.models.claude.token_handler", new_callable=AsyncMock)
 @patch("anthropic.AsyncAnthropic", new_callable=AsyncMock)
 async def test_make_request_success(
     mock_create, mock_token_handler, anthropic_handler: AnthropicHandler
@@ -57,7 +57,7 @@ async def test_make_request_success(
 
 
 @pytest.mark.asyncio
-@patch("readmeai.models.anthropic.token_handler", new_callable=AsyncMock)
+@patch("readmeai.models.claude.token_handler", new_callable=AsyncMock)
 @patch("anthropic.AsyncAnthropic", new_callable=AsyncMock)
 async def test_make_request_api_error(
     mock_create, mock_token_handler, anthropic_handler: AnthropicHandler
@@ -78,7 +78,7 @@ async def test_make_request_api_error(
 
 
 @pytest.mark.asyncio
-@patch("readmeai.models.anthropic.token_handler", new_callable=AsyncMock)
+@patch("readmeai.models.claude.token_handler", new_callable=AsyncMock)
 @patch("anthropic.AsyncAnthropic", new_callable=AsyncMock)
 async def test_make_request_unexpected_error(
     mock_create, mock_token_handler, anthropic_handler: AnthropicHandler