chore: move anthropic as optional (#1069)

camel-ai · Oct 17, 2024 · 2466015 · 2466015
1 parent 523ebae
commit 2466015
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/camel/configs/anthropic_config.py b/camel/configs/anthropic_config.py
@@ -15,7 +15,7 @@
 
 from typing import List, Union
 
-from anthropic import NOT_GIVEN, NotGiven
+from openai._types import NOT_GIVEN, NotGiven
 
 from camel.configs.base_config import BaseConfig
 

diff --git a/camel/models/anthropic_model.py b/camel/models/anthropic_model.py
@@ -14,7 +14,7 @@
 import os
 from typing import Any, Dict, List, Optional, Union
 
-from anthropic import NOT_GIVEN, Anthropic
+from openai import NOT_GIVEN
 
 from camel.configs import ANTHROPIC_API_PARAMS, AnthropicConfig
 from camel.messages import OpenAIMessage
@@ -54,6 +54,8 @@ def __init__(
         url: Optional[str] = None,
         token_counter: Optional[BaseTokenCounter] = None,
     ) -> None:
+        from anthropic import Anthropic
+
         if model_config_dict is None:
             model_config_dict = AnthropicConfig().as_dict()
         api_key = api_key or os.environ.get("ANTHROPIC_API_KEY")

diff --git a/camel/utils/token_counting.py b/camel/utils/token_counting.py
@@ -20,7 +20,6 @@
 from math import ceil
 from typing import TYPE_CHECKING, List, Optional
 
-from anthropic import Anthropic
 from PIL import Image
 
 from camel.types import (
@@ -220,6 +219,7 @@ def _count_tokens_from_image(
 class AnthropicTokenCounter(BaseTokenCounter):
     def __init__(self):
         r"""Constructor for the token counter for Anthropic models."""
+        from anthropic import Anthropic
 
         self.client = Anthropic()
         self.tokenizer = self.client.get_tokenizer()