From 2337e4bfe2af2d6337d09e4005d5d56e42f68382 Mon Sep 17 00:00:00 2001 From: Eli Date: Sun, 6 Oct 2024 20:49:04 -0500 Subject: [PATCH] Fix extra Anthropic import error. --- pyproject.toml | 2 +- readmeai/models/__init__.py | 1 - readmeai/models/{anthropic.py => claude.py} | 9 +++++++++ tests/models/{test_anthropic.py => test_claude.py} | 0 4 files changed, 10 insertions(+), 2 deletions(-) rename readmeai/models/{anthropic.py => claude.py} (93%) rename tests/models/{test_anthropic.py => test_claude.py} (100%) diff --git a/pyproject.toml b/pyproject.toml index 9c6ab4fc..511c01a3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "readmeai" -version = "0.5.90" +version = "0.5.91" description = "Automated README file generator, powered by AI." authors = ["Eli "] license = "MIT" diff --git a/readmeai/models/__init__.py b/readmeai/models/__init__.py index c53129b8..e69de29b 100644 --- a/readmeai/models/__init__.py +++ b/readmeai/models/__init__.py @@ -1 +0,0 @@ -"""Initialize the llms module.""" diff --git a/readmeai/models/anthropic.py b/readmeai/models/claude.py similarity index 93% rename from readmeai/models/anthropic.py rename to readmeai/models/claude.py index b7352f6d..e029a12a 100644 --- a/readmeai/models/anthropic.py +++ b/readmeai/models/claude.py @@ -1,5 +1,6 @@ """Anthropic API service implementation.""" +import logging from typing import Any import anthropic @@ -15,6 +16,14 @@ from readmeai.models.base import BaseModelHandler from readmeai.models.tokens import token_handler +_logger = logging.getLogger(__name__) + +try: + import anthropic +except ImportError: + _logger.warning("Anthropic API not found, please install 'anthropic'.") + anthropic = None + class AnthropicHandler(BaseModelHandler): """ diff --git a/tests/models/test_anthropic.py b/tests/models/test_claude.py similarity index 100% rename from tests/models/test_anthropic.py rename to tests/models/test_claude.py