Merge pull request #412 from oviniciusfeitosa/patch-1

🐛 Rename user_prompt parameter to prompt

Author: VinciGit00

scrapegraphai/builders/graph_builder.py

@@ -40,11 +40,11 @@ class GraphBuilder:
         ValueError: If 'api_key' is not included in llm_config.
     """
 
-    def __init__(self, user_prompt: str, config: dict):
+    def __init__(self, prompt: str, config: dict):
         """
         Initializes the GraphBuilder with a user prompt and language model configuration.
         """
-        self.user_prompt = user_prompt
+        self.prompt = prompt
         self.config = config
         self.llm = self._create_llm(config["llm"])
         self.nodes_description = self._generate_nodes_description()
@@ -122,7 +122,7 @@ def build_graph(self):
         Returns:
             dict: A JSON representation of the graph configuration.
         """
-        return self.chain.invoke(self.user_prompt)
+        return self.chain.invoke(self.prompt)
 
     @staticmethod
     def convert_json_to_graphviz(json_data, format: str = 'pdf'):

scrapegraphai/helpers/models_tokens.py

@@ -81,7 +81,8 @@
         "claude_instant": 100000,
         "claude2": 9000,
         "claude2.1": 200000,
-        "claude3": 200000
+        "claude3": 200000,
+        "claude3.5": 200000
     },
     "bedrock": {
         "anthropic.claude-3-haiku-20240307-v1:0": 200000,