# server.py
from mcp.server.fastmcp import FastMCP

# Create an MCP server
mcp = FastMCP("Demo", host="localhost", port=8005)

@mcp.tool() 
async def weather_tool(location: str) -> str:
    """An agent that provides weather information"""
    return f"The weather in {location} is 73°F sunny with a chance of rain."

if __name__ == "__main__":
    mcp.run(transport="streamable-http")
    

from autogen_ext.models.openai import OpenAIChatCompletionClient
from autogen_ext.tools.mcp import StreamableHttpServerParams, McpWorkbench
from autogen_agentchat.agents import AssistantAgent
from autogen_core import CancellationToken
from autogen_agentchat.ui import Console 
  
test_tool= StreamableHttpServerParams(
    url="http://localhost:8005/mcp", 
) 
test_workbench = McpWorkbench(server_params=test_tool)
print("Test Workbench Tools:", await test_workbench.list_tools())

# # Create an agent that can use the fetch tool.
model_client = OpenAIChatCompletionClient(model="gpt-4o")
agent = AssistantAgent(name="fetcher", model_client=model_client, workbench=[test_workbench], reflect_on_tool_use=True)

await Console(agent.run_stream(task=" What is the weather in Seattle today? ", cancellation_token=CancellationToken()))