diff --git a/scrapegraphai/graphs/abstract_graph.py b/scrapegraphai/graphs/abstract_graph.py index 04dbf9d7..63ccda45 100644 --- a/scrapegraphai/graphs/abstract_graph.py +++ b/scrapegraphai/graphs/abstract_graph.py @@ -148,7 +148,7 @@ class AbstractGraph(ABC): except KeyError as exc: raise KeyError("Model not supported") from exc return Gemini(llm_params) - elif "claude" in llm_params["model"]: + elif llm_params["model"].startswith("claude"): try: self.model_token = models_tokens["claude"][llm_params["model"]] except KeyError as exc: diff --git a/scrapegraphai/models/anthropic.py b/scrapegraphai/models/anthropic.py deleted file mode 100644 index 5b6e21e7..00000000 --- a/scrapegraphai/models/anthropic.py +++ /dev/null @@ -1,17 +0,0 @@ -""" -Anthropic Module -""" -from langchain_anthropic import ChatAnthropic - - -class Anthropic(ChatAnthropic): - """ - A wrapper for the ChatAnthropic class that provides default configuration - and could be extended with additional methods if needed. - - Args: - llm_config (dict): Configuration parameters for the language model. - """ - - def __init__(self, llm_config: dict): - super().__init__(**llm_config)