1class MCPClient:
2    def __init__(self):
3        # Initialize session and client objects
4        self.session: Optional[ClientSession] = None
5        self.exit_stack = AsyncExitStack()
6        self.anthropic = Anthropic()
7    # methods will go here
8
9async def connect_to_server(self, server_script_path: str):
10    is_python = server_script_path.endswith('.py')
11    is_js = server_script_path.endswith('.js')
12    if not (is_python or is_js):
13        raise ValueError("Server script must be a .py or .js file")
14
15    command = "python" if is_python else "node"
16    server_params = StdioServerParameters(
17        command=command,
18        args=[server_script_path],
19        env=None
20    )
21
22    stdio_transport = await self.exit_stack.enter_async_context(stdio_client(server_params))
23    self.stdio, self.write = stdio_transport
24    self.session = await self.exit_stack.enter_async_context(ClientSession(self.stdio, self.write))
25
26    await self.session.initialize()
27
28    # List available tools
29    response = await self.session.list_tools()
30    tools = response.tools
31    print("\nConnected to server with tools:", [tool.name for tool in tools])
32    async def process_query(self, query: str) -> str:
33    """Process a query using Claude and available tools"""
34    messages = [
35        {
36            "role": "user",
37            "content": query
38        }
39    ]
40
41    response = await self.session.list_tools()
42    available_tools = [{
43        "name": tool.name,
44        "description": tool.description,
45        "input_schema": tool.inputSchema
46    } for tool in response.tools]
47
48    # Initial Claude API call
49    response = self.anthropic.messages.create(
50        model="claude-3-5-sonnet-20241022",
51        max_tokens=1000,
52        messages=messages,
53        tools=available_tools
54    )
55
56    # Process response and handle tool calls
57    final_text = []
58
59    assistant_message_content = []
60    for content in response.content:
61        if content.type == 'text':
62            final_text.append(content.text)
63            assistant_message_content.append(content)
64        elif content.type == 'tool_use':
65            tool_name = content.name
66            tool_args = content.input
67
68            # Execute tool call
69            result = await self.session.call_tool(tool_name, tool_args)
70            final_text.append(f"[Calling tool {tool_name} with args {tool_args}]")
71
72            assistant_message_content.append(content)
73            messages.append({
74                "role": "assistant",
75                "content": assistant_message_content
76            })
77            messages.append({
78                "role": "user",
79                "content": [
80                    {
81                        "type": "tool_result",
82                        "tool_use_id": content.id,
83                        "content": result.content
84                    }
85                ]
86            })
87
88            # Get next response from Claude
89            response = self.anthropic.messages.create(
90                model="claude-3-5-sonnet-20241022",
91                max_tokens=1000,
92                messages=messages,
93                tools=available_tools
94            )
95
96            final_text.append(response.content[0].text)
97
98    return "\n".join(final_text)