Upload 4 files

examples/cmd/agent_deploy/demo_agent/agent.py

@@ -0,0 +1,67 @@
+import logging
+import os
+import json
+from aworld.cmd import BaseAWorldAgent, ChatCompletionRequest
+from aworld.config.conf import AgentConfig, TaskConfig
+from aworld.agents.llm_agent import Agent
+from aworld.core.task import Task
+from aworld.runner import Runners
+
+logger = logging.getLogger(__name__)
+
+
+class AWorldAgent(BaseAWorldAgent):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def name(self):
+        return "Demo Agent"
+
+    def description(self):
+        return "Demo Agent with fetch and time mcp server"
+
+    async def run(self, prompt: str = None, request: ChatCompletionRequest = None):
+        llm_provider = os.getenv("LLM_PROVIDER_DEMO", "openai")
+        llm_model_name = os.getenv("LLM_MODEL_NAME_DEMO")
+        llm_api_key = os.getenv("LLM_API_KEY_DEMO")
+        llm_base_url = os.getenv("LLM_BASE_URL_DEMO")
+        llm_temperature = os.getenv("LLM_TEMPERATURE_DEMO", 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"
+            )
+
+        agent_config = AgentConfig(
+            llm_provider=llm_provider,
+            llm_model_name=llm_model_name,
+            llm_api_key=llm_api_key,
+            llm_base_url=llm_base_url,
+            llm_temperature=llm_temperature,
+        )
+
+        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:
+            mcp_config = json.load(f)
+
+        super_agent = Agent(
+            conf=agent_config,
+            name="🙋🏻‍♂️ Demo Agent",
+            system_prompt="You are a demo agent, you can query current time and fetch data from the internet. you must use search engine to get url, then fetch data from the url, don't use url don't exist.",
+            mcp_config=mcp_config,
+            mcp_servers=mcp_config.get("mcpServers", {}).keys(),
+        )
+
+        if prompt is None and request is not None:
+            prompt = request.messages[-1].content
+
+        task = Task(
+            input=prompt,
+            agent=super_agent,
+            conf=TaskConfig(max_steps=20),
+        )
+
+        async for output in Runners.streamed_run_task(task).stream_events():
+            logger.info(f"Agent Ouput: {output}")
+            yield output

examples/cmd/agent_deploy/demo_agent/mcp.json

@@ -0,0 +1,32 @@
+{
+    "mcpServers": {
+        "google-pse-search": {
+            "command": "npx",
+            "args": [
+                "-y",
+                "@adenot/mcp-google-search"
+            ],
+            "env": {
+                "GOOGLE_API_KEY": "${GOOGLE_API_KEY}",
+                "GOOGLE_SEARCH_ENGINE_ID": "${GOOGLE_SEARCH_ENGINE_ID}"
+            }
+        },
+        "fetch": {
+            "command": "uvx",
+            "args": [
+                "-i",
+                "https://mirrors.aliyun.com/pypi/simple/",
+                "mcp-server-fetch",
+                "--ignore-robots-txt"
+            ]
+        },
+        "time": {
+            "command": "uvx",
+            "args": [
+                "mcp-server-time",
+                "--local-timezone",
+                "Asia/Shanghai"
+            ]
+        }
+    }
+}