Revert "chore: refactor app MCP configuration and implement AI chat functionality"

This reverts commit bf9ae94.

Author: liuruibin

apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py

@@ -269,8 +269,14 @@ def _handle_mcp_request(self, mcp_enable, tool_enable, mcp_source, mcp_servers,
                 self.context['application_ids'] = application_ids
                 for application_id in application_ids:
                     app = QuerySet(Application).filter(id=application_id).first()
+                    app_key = QuerySet(ApplicationApiKey).filter(application_id=application_id, is_active=True).first()
+                    # TODO 处理api
+                    if app_key is not None:
+                        api_key = app_key.secret_key
+                    else:
+                        continue
                     executor = ToolExecutor()
-                    app_config = executor.get_app_mcp_config(app.id, app.name, app.desc)
+                    app_config = executor.get_app_mcp_config(api_key, app.name, app.desc)
                     mcp_servers_config[str(app.id)] = app_config
 
         if len(mcp_servers_config) > 0:

apps/common/utils/app_mcp.py

@@ -233,18 +233,59 @@ def get_tool_mcp_config(self, code, params, name, description):
         }
         return tool_config
 
-    def get_app_mcp_config(self, application_id, name, description):
-        cwd = os.path.dirname(os.path.abspath(__file__))
+    def get_app_mcp_config(self, api_key, name, description):
+        chat_path = CONFIG.get_chat_path()
+        _code = f'''
+import requests
+from typing import Optional
+
+def _get_chat_id() -> Optional[str]:
+    url = f"http://127.0.0.1:8080{chat_path}/api/open"
+    headers = {{
+        'accept': '*/*',
+        'Authorization': f'Bearer {api_key}'
+    }}
+    try:
+        resp = requests.get(url, headers=headers, timeout=10)
+        resp.raise_for_status()
+        return resp.json().get("data")
+    except Exception as e:
+        raise e
+
+
+def _chat_with_ai(chat_id: str, message: str) -> Optional[str]:
+    url = f"http://127.0.0.1:8080{chat_path}/api/chat_message/{{chat_id}}"
+    headers = {{"Content-Type": "application/json", "Authorization": f'Bearer {api_key}'}}
+    payload = {{
+        "message": message,
+        "re_chat": False,
+        "stream": False
+    }}
+    try:
+        resp = requests.post(url, json=payload, headers=headers, timeout=600)
+        resp.raise_for_status()
+        data = resp.json()
+        return str(data.get("data", {{}}).get("content") or data.get("response"))
+    except Exception as e:
+        raise e
+
+def ai_chat(message: str) -> str:
+    chat_id = _get_chat_id()
+    reply = _chat_with_ai(chat_id, message)
+    return reply or "AI 未能生成回复"
+
+        '''
+        _code = self.generate_mcp_server_code(_code, {}, name, description)
+        # print(_code)
+        maxkb_logger.debug(f"Python code of mcp app: {_code}")
+        compressed_and_base64_encoded_code_str = base64.b64encode(gzip.compress(_code.encode())).decode()
         app_config = {
             'command': sys.executable,
             'args': [
-                f'{cwd}/app_mcp.py',
-                BASE_DIR,
-                str(application_id),
-                name,
-                description,
+                '-c',
+                f'import base64,gzip; exec(gzip.decompress(base64.b64decode(\'{compressed_and_base64_encoded_code_str}\')).decode())',
             ],
-            'cwd': BASE_DIR,
+            'cwd': _sandbox_path,
             'env': {
                 'LD_PRELOAD': f'{_sandbox_path}/lib/sandbox.so',
             },