client

mcp-flight-agent-http.py

@@ -0,0 +1,62 @@
+import os
+import dspy
+import asyncio
+import mlflow
+
+from mcp import ClientSession
+from mcp.client.streamable_http import streamablehttp_client
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+lm = dspy.LM(
+    model='openai/gpt-4o-mini', 
+    temperature=0,
+    api_key=os.environ['OPENAI_API_KEY'], 
+    api_base=os.environ['OPENAI_BASE_URL']
+)
+
+mcp_url = "https://pgurazada1-airline-ops-mcp-server.hf.space/mcp/"
+# mcp_url = "http://localhost:8000/mcp" # local server testing
+
+dspy.configure(lm=lm)
+mlflow.dspy.autolog()
+mlflow.set_experiment('flight-agent-http')
+
+
+class DSPyAirlineCustomerService(dspy.Signature):
+    """
+    You are an airline customer service agent. 
+    You are given a list of tools to handle user requests. 
+    You should decide the right tool to use in order to fulfill users' requests.
+    """
+
+    user_request: str = dspy.InputField()
+    process_result: str = dspy.OutputField(
+        desc=(
+            "Message that summarizes the process result, and the information users need, "
+            "e.g., the confirmation_number if it's a flight booking request."
+        )
+    )
+
+async def run(user_request):
+    async with streamablehttp_client(url=mcp_url) as (read, write, _):
+        async with ClientSession(read, write) as session:
+            # Initialize the connection
+            await session.initialize()
+            # List available tools
+            tools = await session.list_tools()
+
+            # Convert MCP tools to DSPy tools
+            dspy_tools = []
+            for tool in tools.tools:
+                dspy_tools.append(dspy.Tool.from_mcp_tool(session, tool))
+
+            # Create the agent
+            react_agent = dspy.ReAct(DSPyAirlineCustomerService, tools=dspy_tools)
+
+            output = await react_agent.acall(user_request=user_request)
+            print(output.process_result)
+
+asyncio.run(run("please help me book a flight from SFO to JFK on 09/01/2025, my name is Adam"))