Skip to content

Commit d0167de

Browse files
committed
fix: bug for claude
1 parent 7c91f9f commit d0167de

File tree

2 files changed

+1
-18
lines changed

2 files changed

+1
-18
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def _create_llm(self, llm_config: dict, chat=False) -> object:
148148
except KeyError as exc:
149149
raise KeyError("Model not supported") from exc
150150
return Gemini(llm_params)
151-
elif "claude" in llm_params["model"]:
151+
elif llm_params["model"].startswith("claude"):
152152
try:
153153
self.model_token = models_tokens["claude"][llm_params["model"]]
154154
except KeyError as exc:

scrapegraphai/models/anthropic.py

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)