Upload run.py

examples/search/run.py

@@ -0,0 +1,66 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+
+from aworld.agents.llm_agent import Agent
+from aworld.config.conf import AgentConfig
+from aworld.core.agent.swarm import Swarm
+from aworld.runner import Runners
+from examples.tools.common import Tools
+
+search_sys_prompt = "You are a helpful search agent."
+search_prompt = """
+    Please act as a search agent, constructing appropriate keywords and searach terms, using search toolkit to collect relevant information, including urls, webpage snapshots, etc.
+
+    Here are the question: {task}
+
+    pleas only use one action complete this task, at least results 6 pages.
+    """
+
+summary_sys_prompt = "You are a helpful general summary agent."
+
+summary_prompt = """
+Summarize the following text in one clear and concise paragraph, capturing the key ideas without missing critical points. 
+Ensure the summary is easy to understand and avoids excessive detail.
+
+Here are the content: 
+{task}
+"""
+
+# search and summary
+if __name__ == "__main__":
+    # need to set GOOGLE_API_KEY and GOOGLE_ENGINE_ID to use Google search.
+    # os.environ['GOOGLE_API_KEY'] = ""
+    # os.environ['GOOGLE_ENGINE_ID'] = ""
+
+    agent_config = AgentConfig(
+        llm_provider="openai",
+        llm_model_name="gpt-4o",
+        llm_temperature=1,
+        # need to set llm_api_key for use LLM
+    )
+
+    search = Agent(
+        conf=agent_config,
+        name="search_agent",
+        system_prompt=search_sys_prompt,
+        agent_prompt=search_prompt,
+        tool_names=[Tools.SEARCH_API.value]
+    )
+
+    summary = Agent(
+        conf=agent_config,
+        name="summary_agent",
+        system_prompt=summary_sys_prompt,
+        agent_prompt=summary_prompt
+    )
+    # default is workflow swarm
+    swarm = Swarm(search, summary, max_steps=1)
+
+    prefix = ""
+    # can special search google, wiki, duck go, or baidu. such as:
+    # prefix = "search wiki: "
+    res = Runners.sync_run(
+        input=prefix + """What is an agent.""",
+        swarm=swarm
+    )
+    print(res.answer)