Clean up code

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

mcp_client.py

@@ -85,49 +85,6 @@ class MCPClientWrapper:
         self.anthropic = Anthropic()
         self.tools = []
 
-    # def connect(self, server_path: str) -> str:
-    #     return loop.run_until_complete(self._connect(server_path))
-
-    # async def _connect(self, server_path: str) -> str:
-    #     if self.exit_stack:
-    #         await self.exit_stack.aclose()
-
-    #     self.exit_stack = AsyncExitStack()
-
-    #     is_python = server_path.endswith(".py")
-    #     command = "python" if is_python else "node"
-
-    #     server_params = StdioServerParameters(
-    #         command=command,
-    #         args=[server_path],
-    #         env={"PYTHONIOENCODING": "utf-8", "PYTHONUNBUFFERED": "1"},
-    #     )
-
-    #     stdio_transport = await self.exit_stack.enter_async_context(
-    #         stdio_client(server_params)
-    #     )
-    #     self.stdio, self.write = stdio_transport
-
-    #     self.session = await self.exit_stack.enter_async_context(
-    #         ClientSession(self.stdio, self.write)
-    #     )
-    #     await self.session.initialize()
-
-    #     response = await self.session.list_tools()
-    #     self.tools = [
-    #         {
-    #             "name": tool.name,
-    #             "description": tool.description,
-    #             "input_schema": tool.inputSchema,
-    #         }
-    #         for tool in response.tools
-    #     ]
-
-    #     print(self.tools)
-
-    #     tool_names = [tool["name"] for tool in self.tools]
-    #     return f"Connected to MCP server. Available tools: {', '.join(tool_names)}"
-
     async def connect(self, server_path: str) -> str:
         # If there's an existing session, close it
         if self.exit_stack:
@@ -182,11 +139,6 @@ class MCPClientWrapper:
             ]
             yield messages, gr.Textbox(value="")
         else:
-            # new_messages = loop.run_until_complete(
-            #     self._process_query(message, history)
-            # )
-            # messages = history + [{"role": "user", "content": message}] + new_messages
-
             messages = history + [{"role": "user", "content": message}]
 
             yield messages, gr.Textbox(value="")
@@ -250,36 +202,6 @@ class MCPClientWrapper:
                 yield [result_messages[-1]]
                 partial_messages = []
 
-                # if (
-                #     auto_calls < MAX_CALLS
-                #     and "TOOL_USE" in content.text
-                #     and len(contents) == 0
-                # ):
-                #     # Call the LLM automatically.
-                #     claude_messages.append({"role": "user", "content": "ok"})
-                #     try:
-                #         next_response = self.anthropic.messages.create(
-                #             model=LLM_MODEL,
-                #             system=SYSTEM_PROMPT,
-                #             max_tokens=1000,
-                #             messages=claude_messages,
-                #             tools=self.tools,
-                #         )
-                #     except OverloadedError:
-                #         return [
-                #             {
-                #                 "role": "assistant",
-                #                 "content": "The LLM API is overloaded now, try again later...",
-                #             }
-                #         ]
-
-                #     print("next_response", next_response.content)
-
-                #     contents.extend(next_response.content)
-                #     auto_calls += 1
-
-                #     # continue
-
             elif content.type == "tool_use":
                 tool_id = content.id
                 tool_name = content.name
@@ -350,7 +272,6 @@ class MCPClientWrapper:
                         except:
                             pass
 
-                    # result_content = "\n".join(str(item) for item in result_content)
                     print("result_content", result_content)
 
                 result_messages.append(
@@ -370,12 +291,6 @@ class MCPClientWrapper:
                 yield [result_messages[-1]]
                 partial_messages = []
 
-                # claude_messages.append(
-                #     {
-                #         "role": "user",
-                #         "content": f"Tool result for {tool_name}: {result_content}",
-                #     }
-                # )
                 claude_messages.append(
                     {"role": "assistant", "content": [content.model_dump()]}
                 )
@@ -413,13 +328,6 @@ class MCPClientWrapper:
 
                 contents.extend(next_response.content)
 
-                # if next_response.content and next_response.content[0].type == "text":
-                #     result_messages.append(
-                #         {"role": "assistant", "content": next_response.content[0].text}
-                #     )
-
-        # yield result_messages
-
 
 client = MCPClientWrapper()