Merge pull request #228 from JGalego/fix/bedrock-support

fix: Amazon Bedrock Support

scrapegraphai/graphs/abstract_graph.py

@@ -3,11 +3,12 @@
 """
 from abc import ABC, abstractmethod
 from typing import Optional
+from langchain_aws import BedrockEmbeddings
 from langchain_openai import AzureOpenAIEmbeddings, OpenAIEmbeddings
-from langchain_community.embeddings import HuggingFaceHubEmbeddings, OllamaEmbeddings, BedrockEmbeddings
+from langchain_community.embeddings import HuggingFaceHubEmbeddings, OllamaEmbeddings
 from langchain_google_genai import GoogleGenerativeAIEmbeddings
 from ..helpers import models_tokens
-from ..models import AzureOpenAI, Bedrock, Gemini, Groq, HuggingFace, Ollama, OpenAI, Anthropic, Claude, DeepSeek
+from ..models import AzureOpenAI, Bedrock, Gemini, Groq, HuggingFace, Ollama, OpenAI, Anthropic, DeepSeek
 
 
 class AbstractGraph(ABC):
@@ -153,7 +154,7 @@ def _create_llm(self, llm_config: dict, chat=False) -> object:
                 self.model_token = models_tokens["claude"][llm_params["model"]]
             except KeyError as exc:
                 raise KeyError("Model not supported") from exc
-            return Claude(llm_params)
+            return Anthropic(llm_params)
         elif "ollama" in llm_params["model"]:
             llm_params["model"] = llm_params["model"].split("/")[-1]
 

scrapegraphai/models/__init__.py

@@ -12,5 +12,4 @@
 from .groq import Groq
 from .bedrock import Bedrock
 from .anthropic import Anthropic
-from .claude import Claude
 from .deepseek import DeepSeek