CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-06-25 21:11:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: bug for claude

Browse Source
This commit is contained in:
VinciGit00 2024-05-13 13:43:42 +02:00
parent 7c91f9f94c
commit d0167dee71
2 changed files with 1 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scrapegraphai/graphs/abstract_graph.py
View File

@ -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:

17
scrapegraphai/models/anthropic.py
View File

@ -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)