Update app.py

app.py

@@ -1,11 +1,11 @@
-# app.py
 import gradio as gr
 import asyncio, json, os, sys
 from dotenv import load_dotenv
 from langchain_core.tools import Tool
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.messages import HumanMessage, AIMessage
-from langchain.agents import AgentExecutor, create_tool_calling_agent
+from langchain.agents import create_tool_calling_agent
+from langchain.agents.agent_executor import AgentExecutor
 from langchain_groq import ChatGroq
 
 load_dotenv()
@@ -29,9 +29,22 @@ class MCPClient:
             raise ConnectionError("Server process closed unexpectedly.")
     
     async def get_tools(self) -> list[Tool]:
-        self.process = await asyncio.create_subprocess_exec(*self._cmd, stdin=asyncio.subprocess.PIPE, stdout=asyncio.subprocess.PIPE)
+        self.process = await asyncio.create_subprocess_exec(
+            *self._cmd,
+            stdin=asyncio.subprocess.PIPE,
+            stdout=asyncio.subprocess.PIPE
+        )
         tool_schemas = await self._send_request("discover")
-        return [Tool(name=s['name'], description=s['description'], func=None, coroutine=self._create_tool_coro(s['name']), args_schema=s['args_schema']) for s in tool_schemas]
+        return [
+            Tool(
+                name=s['name'],
+                description=s['description'],
+                func=None,
+                coroutine=self._create_tool_coro(s['name']),
+                args_schema=s['args_schema']
+            )
+            for s in tool_schemas
+        ]
     
     def _create_tool_coro(self, tool_name: str):
         async def _tool_coro(tool_input):
@@ -44,27 +57,42 @@ _agent_executor = None
 async def get_agent_executor():
     global _agent_executor
     if _agent_executor is None:
-        if not os.getenv("GROQ_API_KEY"): raise ValueError("GROQ_API_KEY not set.")
+        if not os.getenv("GROQ_API_KEY"):
+            raise ValueError("GROQ_API_KEY not set.")
         client = MCPClient(command=sys.executable, args=["server.py"])
         tools = await client.get_tools()
         model = ChatGroq(model="llama3-groq-70b-8192-tool-use-preview", temperature=0)
+
         prompt = ChatPromptTemplate.from_messages([
             ("system", "You are a helpful assistant for gold and silver price information and forecasts."),
             MessagesPlaceholder(variable_name="chat_history", optional=True),
             ("human", "{input}"),
             MessagesPlaceholder(variable_name="agent_scratchpad"),
         ])
+
+        # 🔧 only this line changed from create_openai_functions_agent(...)
         agent = create_tool_calling_agent(llm=model, tools=tools, prompt=prompt)
-        _agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=False, handle_parsing_errors=True, max_iterations=5)
+
+        _agent_executor = AgentExecutor(
+            agent=agent,
+            tools=tools,
+            verbose=False,
+            handle_parsing_errors=True,
+            max_iterations=5,
+        )
     return _agent_executor
 
 # --- Chat Handler ---
 async def respond_to_chat(message: str, history: list):
     agent = await get_agent_executor()
-    chat_history = [msg for human, ai in history for msg in (HumanMessage(content=human), AIMessage(content=ai))]
+    chat_history = [
+        msg
+        for human, ai in history
+        for msg in (HumanMessage(content=human), AIMessage(content=ai))
+    ]
     try:
         response = await agent.ainvoke({"input": message, "chat_history": chat_history})
-        return response['output']
+        return response["output"]
     except Exception as e:
         print(f"ERROR: {e}", file=sys.stderr)
         return "Sorry, an error occurred while processing your request."
@@ -78,4 +106,4 @@ demo = gr.ChatInterface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()