Skip to content

Commit 3ffa896

Browse files
committed
Fixed model ID -> model name conversion
1 parent d0a301d commit 3ffa896

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

scrapegraphai/nodes/robots_node.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,7 @@ def execute(self, state: dict) -> dict:
100100
self.llm_model.model_name = self.llm_model.model_name.split("/")[-1]
101101
model = self.llm_model.model_name.split("/")[-1]
102102
elif hasattr(self.llm_model, "model_id"): # Bedrock uses model IDs, not model names
103-
self.llm_model.model_name = self.llm_model.model_id.split("/")[-1]
104-
model = self.llm_model.model_name
103+
model = self.llm_model.model_id.split("/")[-1]
105104
else:
106105
model = self.llm_model.model_name
107106
try:

0 commit comments

Comments
 (0)