From 3ffa896c39f120beaccfa332a67a1c3e45f8817a Mon Sep 17 00:00:00 2001 From: JGalego Date: Mon, 20 May 2024 17:35:55 +0100 Subject: [PATCH] Fixed model ID -> model name conversion --- scrapegraphai/nodes/robots_node.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/scrapegraphai/nodes/robots_node.py b/scrapegraphai/nodes/robots_node.py index cce8fdaa..e807fcf1 100644 --- a/scrapegraphai/nodes/robots_node.py +++ b/scrapegraphai/nodes/robots_node.py @@ -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: