auto-coder 0.1.217__py3-none-any.whl → 0.1.219__py3-none-any.whl

{auto_coder-0.1.217.dist-info → auto_coder-0.1.219.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.217
+Version: 0.1.219
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto_coder-0.1.217.dist-info → auto_coder-0.1.219.dist-info}/RECORD

@@ -1,14 +1,14 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=fnm1dhTj5AidkYf65qt4XjrXZz0D3OrkHW8y7QOMjmQ,41765
+autocoder/auto_coder.py,sha256=xWEEptvSMPu0tEtqOo2V-GoCzL1402NveoxjyYrsw_g,41929
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=illKgzP2bv-Tq50ujsofJnOHdI4pzr0ALtfR8NHHWdQ,22351
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
-autocoder/chat_auto_coder.py,sha256=9YKP4SvQjPrkNN_rCakPaXWCQWFXPvrR7DKH4gw-oIg,93871
+autocoder/chat_auto_coder.py,sha256=NTCWQKBQJluEhay5NGoTl5tdv00Zmu8oFioCjCpJCd8,94294
 autocoder/chat_auto_coder_lang.py,sha256=ReWukXKVvuzVvpbYk5O9kc1ev7XNmAv3DnuQhmpLmnc,8717
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
-autocoder/version.py,sha256=0-vnhNuH2D1UmiWFmbqXgCF2VQY1f7PpLtBG-hBuxTc,24
+autocoder/version.py,sha256=CiJSNMnvgzsdXFL0TOFxCpTE04WDBfpfA19PCdyUupg,24
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_demand_organizer.py,sha256=NWSAEsEk94vT3lGjfo25kKLMwYdPcpy9e-i21txPasQ,6942
 autocoder/agent/auto_filegroup.py,sha256=CW7bqp0FW1GIEMnl-blyAc2UGT7O9Mom0q66ITz1ckM,6635
@@ -38,7 +38,7 @@ autocoder/common/code_auto_merge_diff.py,sha256=5SI6ggklJ0QDHvsS0cpNXFuIkFRQxp1i
 autocoder/common/code_auto_merge_editblock.py,sha256=l6yEiZqXyIlUNIIPXvkHOnLCIInXR78TzSjF-jtJkkg,17035
 autocoder/common/code_auto_merge_strict_diff.py,sha256=ABYOTDUQYA4Bn4BwT1Rw812y49cHW3UH_JSpM9uJ6ig,9399
 autocoder/common/code_modification_ranker.py,sha256=DFlbwgdg8GK47zVcvfZSzkyniEKmTVLTOWejjcVIgaw,5121
-autocoder/common/command_completer.py,sha256=iFTrKlSYeF6YCm9VMwRiuLVOMx7V-zWSJMGyHwOrgrg,9121
+autocoder/common/command_completer.py,sha256=10nNnazkzZwya-jScPLlhpCBR2oYsOiVfwofqGvIUEM,9145
 autocoder/common/command_generator.py,sha256=v4LmU7sO-P7jEZIXCWHUC6P-vT7AvBi_x_PTwCqBAE8,1323
 autocoder/common/command_templates.py,sha256=3G-pCNbL6iHbnkG6v1JZpbIK3Mc9d373_RYGmCcDPMY,8548
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
@@ -46,8 +46,8 @@ autocoder/common/git_utils.py,sha256=btK45sxvfm4tX3fBRNUPRZoGQuZuOEQrWSAwLy1yoLw
 autocoder/common/image_to_page.py,sha256=O0cNO_vHHUP-fP4GXiVojShmNqkPnZXeIyiY1MRLpKg,13936
 autocoder/common/interpreter.py,sha256=62-dIakOunYB4yjmX8SHC0Gdy2h8NtxdgbpdqRZJ5vk,2833
 autocoder/common/llm_rerank.py,sha256=FbvtCzaR661Mt2wn0qsuiEL1Y3puD6jeIJS4zg_e7Bs,3260
-autocoder/common/mcp_hub.py,sha256=bs9ZEsuN3CKs37tLKADtBA_Mm-TcYkFkXsUmGS2hljQ,13721
-autocoder/common/mcp_server.py,sha256=OgR6dceCqTnO2g-64eQ6TOqKucjt-kkG4IXlrp37XN8,11368
+autocoder/common/mcp_hub.py,sha256=9BqNWjVG8polG-bG8sXIqze-ZzzNJxg2OjfOqgxeM3U,14174
+autocoder/common/mcp_server.py,sha256=tZfjUjBvEp8gaEusD2SaH5Mh0FdbP2MpnyqAWuSgrwo,12652
 autocoder/common/mcp_tools.py,sha256=KsLvRrB6pvmebqd-lDaSH6IBJR0AIxWRE-dtCEG_w9k,12485
 autocoder/common/recall_validation.py,sha256=Avt9Q9dX3kG6Pf2zsdlOHmsjd-OeSj7U1PFBDp_Cve0,1700
 autocoder/common/screenshots.py,sha256=_gA-z1HxGjPShBrtgkdideq58MG6rqFB2qMUJKjrycs,3769
@@ -122,9 +122,9 @@ autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
 autocoder/utils/rest.py,sha256=HawagAap3wMIDROGhY1730zSZrJR_EycODAA5qOj83c,8807
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
-auto_coder-0.1.217.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.217.dist-info/METADATA,sha256=6WH7sysYNE75MfeufxCCQCJGmjI6oq5iRwQ_3mdDeH0,2615
-auto_coder-0.1.217.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.217.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.217.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.217.dist-info/RECORD,,
+auto_coder-0.1.219.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.219.dist-info/METADATA,sha256=qsE3dhB3-5DCw0cflbsa5zCFs5DqDnyXhzSDum8RNgU,2615
+auto_coder-0.1.219.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.219.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.219.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.219.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -786,7 +786,7 @@ def main(input_args: Optional[List[str]] = None):
                     pre_conversations.append(
                         {
                             "role": "user",
-                            "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n{file_content}",
+                            "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n <files>\n{file_content}</files>",
                         },
                     )
                     pre_conversations.append(
@@ -813,7 +813,7 @@ def main(input_args: Optional[List[str]] = None):
                 pre_conversations.append(
                     {
                         "role": "user",
-                        "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n{s}",
+                        "content": f"下面是一些文档和源码，如果用户的问题和他们相关，请参考他们：\n <files>{s}</files>",
                     }
                 )
                 pre_conversations.append(
@@ -831,15 +831,19 @@ def main(input_args: Optional[List[str]] = None):
                     source_codes, pre_conversations, last_conversation
                 ):
                     """
+                    <files>
                     {% if source_codes %}
                     {{ source_codes }}
                     {% endif %}
+                    </files>
 
                     {% if pre_conversations %}
                     下面是我们之间的历史对话，假设我是A，你是B。
+                    <conversations>
                     {% for conv in pre_conversations %}
                     {{ "A" if conv.role == "user" else "B" }}: {{ conv.content }}
                     {% endfor %}
+                    </conversations>
                     {% endif %}
 
 

autocoder/chat_auto_coder.py

@@ -44,9 +44,10 @@ from autocoder.common import git_utils
 from autocoder.chat_auto_coder_lang import get_message
 from autocoder.utils import operate_config_api
 from autocoder.agent.auto_guess_query import AutoGuessQuery
-from autocoder.common.mcp_server import get_mcp_server, McpRequest, McpInstallRequest, McpRemoveRequest, McpListRequest, McpListRunningRequest
+from autocoder.common.mcp_server import get_mcp_server, McpRequest, McpInstallRequest, McpRemoveRequest, McpListRequest, McpListRunningRequest, McpRefreshRequest
 import byzerllm
 
+
 class SymbolItem(BaseModel):
     symbol_name: str
     symbol_type: SymbolType
@@ -1443,6 +1444,16 @@ def mcp(query: str):
             print(response.result)
         return
 
+    # Handle refresh command
+    if query.startswith("/refresh"):
+        server_name = query.replace("/refresh", "", 1).strip()    
+        response = mcp_server.send_request(McpRefreshRequest(name=server_name or None))
+        if response.error:
+            print(f"Error refreshing MCP servers: {response.error}")
+        else:
+            print("Successfully refreshed MCP servers")
+        return
+
     # Handle add command
     if query.startswith("/add"):
         query = query.replace("/add", "", 1).strip()

autocoder/common/command_completer.py

@@ -18,7 +18,8 @@ COMMANDS = {
         "/add": "",         
         "/remove": "",
         "/list": "",
-        "/list_running": ""
+        "/list_running": "",
+        "/refresh": ""
     },
     "/lib": {
         "/add": "",

autocoder/common/mcp_hub.py

@@ -240,6 +240,19 @@ class McpHub:
                 if name in self.connections:
                     del self.connections[name]
 
+    async def refresh_server_connection(self, name: str) -> None:
+        """
+        Refresh a server connection
+        """
+        try:
+            config = self._read_settings()
+            await self.delete_connection(name)
+            await self.connect_to_server(name, config.get("mcpServers", {}).get(name, {}))
+        except Exception as e:
+            logger.error(f"Failed to refresh MCP server {name}: {e}")
+            raise
+        
+
     async def update_server_connections(self, new_servers: Dict[str, Any]) -> None:
         """
         Update server connections based on new configuration
@@ -309,7 +322,7 @@ class McpHub:
                 for resource in response.resources
             ]
         except Exception as e:
-            logger.error(f"Failed to fetch resources for {server_name}: {e}")
+            logger.warning(f"Failed to fetch resources for {server_name}: {e}")
             return []
 
     async def _fetch_resource_templates(
@@ -346,7 +359,7 @@ class McpHub:
                 for template in response.resourceTemplates
             ]
         except Exception as e:
-            logger.error(
+            logger.warning(
                 f"Failed to fetch resource templates for {server_name}: {e}")
             return []
 

autocoder/common/mcp_server.py

@@ -14,7 +14,6 @@ from pydantic import BaseModel
 import sys
 from loguru import logger
 
-
 @dataclass
 class McpRequest:
     query: str
@@ -42,13 +41,24 @@ class McpListRunningRequest:
     """Request to list all running MCP servers"""
     pass
 
-
 @dataclass
 class McpResponse:
     result: str
     error: Optional[str] = None
 
 
+@dataclass
+class McpRefreshRequest:
+    """Request to refresh MCP server connections"""
+    name: Optional[str] = None
+
+
+@dataclass
+class McpExternalServer(BaseModel):
+    result: str
+    error: Optional[str] = None
+
+
 class McpExternalServer(BaseModel):
     """Represents an external MCP server configuration"""
     name: str
@@ -118,7 +128,7 @@ class McpServer:
         self._task = self._loop.create_task(self._process_request())
         self._loop.run_forever()
 
-    async def _install_server(self, request: McpInstallRequest) -> McpResponse:
+    async def _install_server(self, request: McpInstallRequest, hub: McpHub) -> McpResponse:
         """Install an MCP server with module dependency check"""
         name = ""
         config = {}
@@ -134,7 +144,7 @@ class McpServer:
                 config = list(raw_config.values())[0]
                 name = list(raw_config.keys())[0]
             except json.JSONDecodeError:
-                name = server_name_or_config
+                name = server_name_or_config.strip()
                 if name not in MCP_BUILD_IN_SERVERS:
                     # 查找外部server
                     external_servers = get_mcp_external_servers()
@@ -144,7 +154,8 @@ class McpServer:
                                 # Check if module exists
                                 try:
                                     import importlib
-                                    importlib.import_module(name)
+                                    importlib.import_module(
+                                        name.replace("-", "_"))
                                 except ImportError:
                                     # Install missing module
                                     import subprocess
@@ -152,17 +163,20 @@ class McpServer:
                                         subprocess.run(
                                             [sys.executable, "-m", "pip", "install", name], check=True)
                                     except subprocess.CalledProcessError:
-                                        print(f"\n\033[93mFailed to automatically install {name}. Please manually install it using:\n")
+                                        print(
+                                            f"\n\033[93mFailed to automatically install {name}. Please manually install it using:\n")
                                         print(f"    pip install {name}\n")
-                                        print(f"We have already updated the server configuration in ~/.autocoder/mcp/settings.json.\n")
-                                        print(f"After installation, you can restart the auto-coder.chat using the server.\033[0m\n")
-
-                                    config = {
-                                        "command": "python",
-                                        "args": [
-                                            "-m", name.replace("-", "_")
-                                        ],
-                                    }
+                                        print(
+                                            f"We have already updated the server configuration in ~/.autocoder/mcp/settings.json.\n")
+                                        print(
+                                            f"After installation, you can restart the auto-coder.chat using the server.\033[0m\n")
+
+                                config = {
+                                    "command": "python",
+                                    "args": [
+                                        "-m", name.replace("-", "_")
+                                    ],
+                                }
                             elif s.runtime == "node":
                                 # Check if package exists
                                 try:
@@ -174,24 +188,32 @@ class McpServer:
                                         subprocess.run(
                                             ["npm", "install", "-y", "-g", name], check=True)
                                     except subprocess.CalledProcessError:
-                                        print(f"\n\033[93mFailed to automatically install {name}. Please manually install it using:\n")
+                                        print(
+                                            f"\n\033[93mFailed to automatically install {name}. Please manually install it using:\n")
                                         print(f"    npm install -g {name}\n")
-                                        print(f"We have already updated the server configuration in ~/.autocoder/mcp/settings.json.\n")
-                                        print(f"After installation, you can restart the auto-coder.chat using the server.\033[0m\n")
-
-                                    config = {
-                                        "command": "npx",
-                                        "args": [
-                                            "-y",
-                                            name
-                                        ]
-                                    }
+                                        print(
+                                            f"We have already updated the server configuration in ~/.autocoder/mcp/settings.json.\n")
+                                        print(
+                                            f"After installation, you can restart the auto-coder.chat using the server.\033[0m\n")
+
+                                config = {
+                                    "command": "npx",
+                                    "args": [
+                                        "-y",
+                                        name
+                                    ]
+                                }
                             break
                 else:
                     config = MCP_BUILD_IN_SERVERS[name]
             if not name:
-                raise ValueError("MCP server name is not available in MCP_BUILD_IN_SERVERS or external servers")
-            hub = McpHub()
+                raise ValueError(
+                    "MCP server name is not available in MCP_BUILD_IN_SERVERS or external servers")
+
+            logger.info(f"Installing MCP server: {name} with config: {config}")
+            if not config:
+                raise ValueError(f"MCP server {name} config is not available")
+
             await hub.add_server_config(name, config)
             return McpResponse(result=f"Successfully installed MCP server: {request.server_name_or_config}")
         except Exception as e:
@@ -208,7 +230,7 @@ class McpServer:
                     break
 
                 if isinstance(request, McpInstallRequest):
-                    response = await self._install_server(request)
+                    response = await self._install_server(request, hub)
                     await self._response_queue.put(response)
 
                 elif isinstance(request, McpRemoveRequest):
@@ -246,6 +268,16 @@ class McpServer:
                     except Exception as e:
                         await self._response_queue.put(McpResponse(result="", error=f"Failed to list running servers: {str(e)}"))
 
+                elif isinstance(request, McpRefreshRequest):
+                    try:
+                        if request.name:
+                            await hub.refresh_server_connection(request.name)
+                        else:
+                            await hub.initialize()
+                        await self._response_queue.put(McpResponse(result="Successfully refreshed MCP server connections"))
+                    except Exception as e:
+                        await self._response_queue.put(McpResponse(result="", error=f"Failed to refresh MCP servers: {str(e)}"))
+
                 else:
                     llm = byzerllm.ByzerLLM.from_default_model(
                         model=request.model)

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.217"
+__version__ = "0.1.219"