Update app.py

app.py

@@ -32,7 +32,7 @@ class BasicAgent:
         return fixed_answer
 
 class ZephyrAPI:
-    def __init__(self):
+    def __init__(self, tools=None):
         self.api_url = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
         self.headers = {
             "Authorization": f"Bearer {os.getenv('HF_TOKEN')}"
@@ -43,7 +43,7 @@ class ZephyrAPI:
     def format_tools(self, tools):
         return "\n".join([f"- {tool.name}: {tool.description}" for tool in tools])
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str, scratchpad: str = "") -> str:
         prompt = f"""<|system|>
                 You are a smart agent that can reason step-by-step and use tools when necessary.
                 
@@ -94,32 +94,32 @@ class LangGraphAgent:
             user_msg = next((m for m in messages if isinstance(m, HumanMessage)), None)
             if not user_msg:
                 return {"messages": messages + [AIMessage(content="❌ No user input found.")]}
-            
+
             content = user_msg.content.strip()
             raw_response = self.model(content)
 
-            # Check if model issued a tool call
-            match = re.search(r"Action:\s*(\w+)\s*Action Input:\s*\"(.+?)\"", response, re.DOTALL)
+            # Detect tool usage
+            match = re.search(r"Action:\s*(\w+)\s*Action Input:\s*\"(.+?)\"", raw_response, re.DOTALL)
             if match:
                 tool_name, tool_input = match.groups()
                 tool_fn = self.tools.get(tool_name)
                 if tool_fn:
                     try:
                         tool_output = tool_fn(tool_input)
-                        # Append tool call and result to scratchpad
-                        scratchpad = f"{response}\nObservation: {tool_output}"
+                        scratchpad = f"{raw_response}\nObservation: {tool_output}"
                         follow_up = self.model(content, scratchpad)
                         return {"messages": messages + [
-                            AIMessage(content=response),
+                            AIMessage(content=raw_response),
                             AIMessage(content=f"Observation: {tool_output}"),
                             AIMessage(content=follow_up),
                         ]}
                     except Exception as e:
                         return {"messages": messages + [AIMessage(content=f"⚠️ Tool error: {e}")]}
+
                 else:
                     return {"messages": messages + [AIMessage(content=f"⚠️ Unknown tool: {tool_name}")]}
-        
-            return {"messages": messages + [AIMessage(content=response)]}
+
+            return {"messages": messages + [AIMessage(content=raw_response)]}
 
         builder.add_node("chat", call_model)
         builder.set_entry_point("chat")