Simplify

Author: rlancemartin

examples/research/src/agent/agent.py

@@ -1,4 +1,3 @@
-
 # Imports
 from contextlib import asynccontextmanager
 
@@ -31,11 +30,6 @@
 async def make_graph():
     async with MultiServerMCPClient(
     {
-        "research-server": {
-            "command": "npx",
-            "args": ["@playwright/mcp"],
-            "transport": "stdio",
-            },
         "planning-server": {
             "command": "uvx",
             "args": [
@@ -66,13 +60,9 @@ async def make_graph():
         # Researcher agent
         researcher_agent = create_react_agent(model, 
                                             prompt=researcher_prompt, 
-                                            tools=client.server_name_to_tools["research-server"].append(transfer_to_planner_agent),
+                                            tools=client.server_name_to_tools["planning-server"].append(transfer_to_planner_agent),
                                             name="researcher_agent") 
 
         # Swarm
-        agent_swarm = create_swarm([planner_agent, researcher_agent], default_active_agent="planner_agent")
-
-        # app = agent_swarm.compile(config_schema=Configuration)
-        agent = agent_swarm.compile()
-        
-        yield agent
+        agent_swarm = create_swarm([planner_agent, researcher_agent], default_active_agent="planner_agent")        
+        yield agent_swarm

examples/research/src/agent/prompts.py

@@ -6,7 +6,7 @@
 
 <Instructions>
 1. Reflect on the user's request. 
-2. Use the list_doc_sources tool to fetch and read the llms.txt file.
+2. Use the list_doc_sources tool to fetch and the fetch_docs tool to read the llms.txt file.
 3. Identify documents that are relevant to the user's request.
 4. Ask follow-up questions to help refine the project scope and narrow the set of documents to be used for the project.
 5. When the project scope is clear produce a short description of the project with relevant URLs.
@@ -21,7 +21,7 @@
 
 <Instructions>
 1. Reflect on the project scope and provided URLs from the planner.
-2. Use the browser_navigate tool to fetch and read each URL.
+2. Use the fetch_docs tool to fetch and read each URL.
 3. Use the information in these URLs to implement the solution to the user's request.
 4. If you need further clarification or additional sources to implement the solution, transfer to transfer_to_planner_agent.
 </Instructions>