Upload 2 files

examples/cmd/agent_deploy/gaia_agent/agent.py

@@ -0,0 +1,47 @@
+import logging
+import os
+import json
+from aworld.cmd import BaseAWorldAgent, ChatCompletionRequest
+from examples.gaia.gaia_agent_runner import GaiaAgentRunner
+
+logger = logging.getLogger(__name__)
+
+
+class AWorldAgent(BaseAWorldAgent):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        path_cwd = os.path.dirname(os.path.abspath(__file__))
+        mcp_path = os.path.join(path_cwd, "mcp.json")
+        with open(mcp_path, "r") as f:
+            self.mcp_config = json.load(f)
+        os.makedirs(os.path.join(os.getcwd(), "static"), exist_ok=True)
+
+    async def run(self, prompt: str = None, request: ChatCompletionRequest = None):
+        llm_provider = os.getenv("LLM_PROVIDER_GAIA", "openai")
+        llm_model_name = os.getenv("LLM_MODEL_NAME_GAIA")
+        llm_api_key = os.getenv("LLM_API_KEY_GAIA")
+        llm_base_url = os.getenv("LLM_BASE_URL_GAIA")
+        llm_temperature = os.getenv("LLM_TEMPERATURE_GAIA", 0.0)
+
+        if not llm_model_name or not llm_api_key or not llm_base_url:
+            raise ValueError(
+                "LLM_MODEL_NAME, LLM_API_KEY, LLM_BASE_URL must be set in your envrionment variables"
+            )
+
+        runner = GaiaAgentRunner(
+            llm_provider=llm_provider,
+            llm_model_name=llm_model_name,
+            llm_base_url=llm_base_url,
+            llm_api_key=llm_api_key,
+            llm_temperature=llm_temperature,
+            mcp_config=self.mcp_config,
+        )
+
+        if prompt is None and request is not None:
+            prompt = request.messages[-1].content
+
+        logger.info(f">>> Gaia Agent: prompt={prompt}, runner={runner}")
+
+        async for line in runner.run(prompt):
+            logger.info(f">>> Gaia Agent Line: {line}")
+            yield line

examples/cmd/agent_deploy/gaia_agent/mcp.json

@@ -0,0 +1,139 @@
+{
+    "mcpServers": {
+        "e2b-server": {
+            "command": "npx",
+            "args": [
+                "-y",
+                "@e2b/mcp-server"
+            ],
+            "env": {
+                "E2B_API_KEY": "${E2B_API_KEY}",
+                "SESSION_REQUEST_CONNECT_TIMEOUT": "120"
+            }
+        },
+        "filesystem": {
+            "command": "npx",
+            "args": [
+                "-y",
+                "@modelcontextprotocol/server-filesystem",
+                "${FILESYSTEM_SERVER_WORKDIR}"
+            ]
+        },
+        "terminal-controller": {
+            "command": "python",
+            "args": [
+                "-m",
+                "terminal_controller"
+            ]
+        },
+        "calculator": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_server_calculator"
+            ]
+        },
+        "excel": {
+            "command": "npx",
+            "args": [
+                "--yes",
+                "@negokaz/excel-mcp-server"
+            ],
+            "env": {
+                "EXCEL_MCP_PAGING_CELLS_LIMIT": "4000"
+            }
+        },
+        "playwright": {
+            "command": "npx",
+            "args": [
+                "-y",
+                "@executeautomation/playwright-mcp-server"
+            ]
+        },
+        "google-search": {
+            "command": "npx",
+            "args": [
+                "-y",
+                "@adenot/mcp-google-search"
+            ],
+            "env": {
+                "GOOGLE_API_KEY": "${GOOGLE_API_KEY}",
+                "GOOGLE_SEARCH_ENGINE_ID": "${GOOGLE_SEARCH_ENGINE_ID}"
+            }
+        },
+        "ms-playwright": {
+            "command": "npx",
+            "args": [
+                "@playwright/mcp@latest",
+                "--isolated"
+            ],
+            "env": {
+                "PLAYWRIGHT_TIMEOUT": "120000",
+                "SESSION_REQUEST_CONNECT_TIMEOUT": "120"
+            }
+        },
+        "audio_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.audio_server"
+            ]
+        },
+        "image_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.image_server"
+            ]
+        },
+        "youtube_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.youtube_server"
+            ]
+        },
+        "video_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.video_server"
+            ]
+        },
+        "search_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.search_server"
+            ]
+        },
+        "download_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.download_server"
+            ]
+        },
+        "document_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.document_server"
+            ]
+        },
+        "browser_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.browser_server"
+            ]
+        },
+        "reasoning_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.reasoning_server"
+            ]
+        }
+    }
+}