Update interfaces/interface.py

interfaces/interface.py

@@ -8,33 +8,22 @@ def create_interface():
     
     graph = StateGraph(AgentState)
     graph.add_node("agent", agent_node)
+    graph.add_node("tool", tool_node)
+    graph.add_node("format", format_output)
+    
+    # Conditional routing
     graph.add_conditional_edges(
         "agent",
         lambda state: "tool" if state["messages"][-1].tool_calls else "format",
         {"tool": "tool", "format": "format"}
     )
-    graph.add_node("tool", tool_node)
-    graph.add_node("format", format_output)
-    
-    graph.set_entry_point("agent")
-    graph.add_edge("agent", "tool")
     graph.add_edge("tool", "format")
-    graph.add_conditional_edges(
-        "tool",
-        lambda state: "format",
-        {"format": "format"}
-    )    
     graph.add_edge("format", END)
+
+
     
     app = graph.compile()
 
-    # def process_query(query: str) -> dict:
-    #     try:
-    #         inputs = {"messages": [HumanMessage(content=query)]}
-    #         result = app.invoke(inputs)  
-    #         return messages_to_dict(result['messages'])[2]['data']['content']
-    #     except Exception as e:
-    #         return {"error": f"Execution error: {str(e)}"}
 
     def process_query(query: str) -> dict:
         try: