Update agents/agents_nodes.py

agents/agents_nodes.py

@@ -21,10 +21,32 @@ text_generator = pipeline(
 llm = HuggingFacePipeline(pipeline=text_generator)
 
 llm = HuggingFacePipeline(pipeline=text_generator)
-llm_instantiated = llm.bind_tools(
-    [time_value_tool],
-    tool_choice={"type": "function", "function": {"name": "time_value_tool"}}
+# llm_instantiated = llm.bind_tools(
+#     [time_value_tool],
+#     tool_choice={"type": "function", "function": {"name": "time_value_tool"}}
+# )
+
+llm_instantiated = llm.bind(
+    tools=[{
+        "type": "function",
+        "function": {
+            "name": time_value_tool.name,
+            "description": time_value_tool.description,
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "CF": {"type": "number"},
+                    "F": {"type": "string"},
+                    "i": {"type": "number"},
+                    "n": {"type": "number"},
+                    "g": {"type": "number", "nullable": True}
+                },
+                "required": ["CF", "F", "i", "n"]
+            }
+        }
+    }]
 )
+
 def agent_node(state: AgentState):
     response = llm_instantiated.invoke(state["messages"])
     if not (hasattr(response, 'tool_calls') and response.tool_calls):