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

Fixed model ID -> model name conversion

Browse Source
This commit is contained in:
JGalego 2024-05-20 17:35:55 +01:00
parent d0a301d3c3
commit 3ffa896c39
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scrapegraphai/nodes/robots_node.py
View File

@ -100,8 +100,7 @@ class RobotsNode(BaseNode):
self.llm_model.model_name = self.llm_model.model_name.split("/")[-1]
model = self.llm_model.model_name.split("/")[-1]
elif hasattr(self.llm_model, "model_id"): # Bedrock uses model IDs, not model names
self.llm_model.model_name = self.llm_model.model_id.split("/")[-1]
model = self.llm_model.model_name
model = self.llm_model.model_id.split("/")[-1]
else:
model = self.llm_model.model_name
try: