Upload 5 files

examples/cmd/agent_deploy/team_agent/agent.py

@@ -0,0 +1,94 @@
+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.agent.swarm import Swarm
+from aworld.core.task import Task
+from aworld.runner import Runners
+from examples.plan_execute.agent import PlanAgent
+from .prompt import *
+
+logger = logging.getLogger(__name__)
+
+
+class AWorldAgent(BaseAWorldAgent):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def name(self):
+        return "Team Agent"
+
+    def description(self):
+        return "Team Agent with fetch and time mcp server"
+
+    async def run(self, prompt: str = None, request: ChatCompletionRequest = None):
+        llm_provider = os.getenv("LLM_PROVIDER_TEAM", "openai")
+        llm_model_name = os.getenv("LLM_MODEL_NAME_TEAM")
+        llm_api_key = os.getenv("LLM_API_KEY_TEAM")
+        llm_base_url = os.getenv("LLM_BASE_URL_TEAM")
+        llm_temperature = os.getenv("LLM_TEMPERATURE_TEAM", 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)
+
+        plan_agent = PlanAgent(name="🙋🏻‍♂️ Team Agent Demo", conf=agent_config)
+
+        search_agent = Agent(
+            conf=agent_config,
+            name="🔎 Search Agent",
+            system_prompt=search_sys_prompt,
+            agent_prompt=search_agent_prompt,
+            mcp_config=mcp_config,
+            mcp_servers=mcp_config.get("mcpServers", {}).keys(),
+        )
+
+        summary_agent = Agent(
+            conf=agent_config,
+            name="💬 Summary Agent",
+            system_prompt=summary_sys_prompt,
+            agent_prompt=summary_agent_prompt,
+        )
+
+        output_agent = Agent(
+            conf=agent_config,
+            name="💬 Output Agent",
+            system_prompt=output_sys_prompt,
+            agent_prompt=output_agent_prompt,
+            mcp_config=mcp_config,
+            mcp_servers=mcp_config.get("mcpServers", {}).keys(),
+        )
+
+        # default is sequence swarm mode
+        swarm = Swarm(
+            plan_agent, search_agent, summary_agent, output_agent, max_steps=10
+        )
+
+        if prompt is None and request is not None:
+            prompt = request.messages[-1].content
+
+        task = Task(
+            input=prompt,
+            swarm=swarm,
+            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/team_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"
+            ]
+        }
+    }
+}

examples/cmd/agent_deploy/team_agent/prompt.py

@@ -0,0 +1,13 @@
+search_sys_prompt = "YOU MUST SAY: 'Search Agent System Prompt'"
+
+search_agent_prompt = "YOU MUST SAY: 'Search Agent Prompt'"
+
+summary_sys_prompt = "YOU MUST SAY: 'Summary Agent System Prompt'"
+
+summary_agent_prompt = "YOU MUST SAY: 'Summary Agent Prompt'"
+
+
+output_sys_prompt = "YOU MUST SAY: 'Output Agent System Prompt'"
+
+
+output_agent_prompt = "YOU MUST SAY: 'Output Agent Prompt'"