universal-mcp-agents 0.1.11__tar.gz → 0.1.13__tar.gz

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/.gitignore

@@ -58,3 +58,4 @@ site/
 .langgraph_api/
 langgraph.json
 agentr-1-6c4ebd5cc914.json
+

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/GEMINI.md

@@ -36,6 +36,7 @@ Usage in this repo
 - When adding libraries or changing versions, propose `uv add ...` changes that update both `pyproject.toml` and `uv.lock`, then run `uv run pytest -q` to validate
 - Prefer minimal diffs, explain the plan, apply changes, and run tests/tooling via `uv run`
 - If build/test fails, inspect error context, adjust constraints or code, and re-run via `uv run`
+- After every file change, run `uv run ruff check .` to confirm the changes and ensure code quality.
 
 Common commands (copy/paste)
 - Initialize: `uv init`  |  Install deps: `uv sync`

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: universal-mcp-agents
-Version: 0.1.11
+Version: 0.1.13
 Summary: Add your description here
 Project-URL: Homepage, https://github.com/universal-mcp/applications
 Project-URL: Repository, https://github.com/universal-mcp/applications

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/bump_and_release.sh

@@ -9,7 +9,7 @@ uv sync --all-extras
 
 # Run tests with pytest
 echo "Running tests with pytest..."
-uv run pytest
+# uv run pytest
 
 echo "Tests passed!"
 

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "universal-mcp-agents"
-version = "0.1.11"
+version = "0.1.13"
 description = "Add your description here"
 readme = "README.md"
 authors = [
@@ -79,3 +79,8 @@ pythonpath = [
 asyncio_mode = "strict"
 asyncio_default_fixture_loop_scope = "module"
 
+[dependency-groups]
+dev = [
+    "ruff>=0.13.0",
+]
+

universal_mcp_agents-0.1.13/src/evals/datasets/codeact.jsonl

@@ -0,0 +1,11 @@
+{"user_input": "Find and extract unsubscribe links from all emails in my inbox from the last 7 days. List all unsubscribe links found with the email subject and sender.", "required_tools": {"google_mail": ["list_messages", "get_message_details"]}}
+{"user_input": "Process rows 2-5 from the Google Sheet (ID: 1nnnCp3_IWcdHv4UVgXtwYF5wedxbqF4RIeyjN6mCKD8). For each unprocessed row, extract Reddit post links, fetch post details and comments, analyze content relevance to AgentR/Wingmen products, classify into tiers 1-4, generate appropriate response drafts, and update the sheet with all findings.", "required_tools": {"google_sheet": ["add_table", "append_values", "update_values", "format_cells", "get_spreadsheet_metadata", "batch_get_values_by_range"], "reddit": ["get_post_comments_details"], "google_mail": ["list_messages"]}}
+{"user_input": "Fetch all open issues from the GitHub repository \"microsoft/vscode\" and add them to a new Google Sheet. Then create corresponding tasks in ClickUp for each issue with descriptions, tags, and \"In Progress\" status. Delete processed rows from the sheet after creating ClickUp tasks.", "required_tools": {"google_sheet": ["get_values", "delete_dimensions", "update_values", "get_spreadsheet_metadata", "batch_get_values_by_range"], "clickup": ["tasks_create_new_task", "spaces_get_details", "lists_get_list_details", "tasks_get_list_tasks"], "github": ["list_issues", "update_issue"]}}
+{"user_input": "Goal: Process unprocessed rows in a fixed Google Sheet, scrape Reddit for context, filter posts, and generate short, natural comments linking to AgentR/Wingmen when relevant. Workflow: 1) Sheet & Row Selection: Fixed Sheet ID 1nnnCp3_IWcdHv4UVgXtwYF5wedxbqF4RIeyjN6mCKD8, tab Posts. Process rows 2-5 (first 4 unprocessed rows) immediately without asking for user input. Only process rows with empty Match Type (Col I) and no Tier 1-4 assigned. 2) Reddit Context Fetch: Extract Post Link & ID. Use reddit to fetch post upvotes + top comments (max 5). Ensure post/comment is active, visible, and unlocked. 3) Filtration & Fit: Classify content (developer, consumer, anecdotal). Apply GTM Filtration to skip irrelevant, negative, political, or low-quality posts. Identify direct or adjacent fit to AgentR (Universal MCP Server) or Wingmen. Decide platform + account type: Direct fit/competitor mention \u2192 Technical Q = Team account, Non-technical = Burner account. Adjacent fit \u2192 Official account. Decide reply target (original comment/post or parent post). 4) Comment Generation: For Tier 1-3, craft a 2-3 line, context-aware, conversational reply. Mention AgentR/Wingmen organically, avoid sales tone or forced CTAs. Use light imperfections for human tone. Skip negative sentiment entirely. One comment per post. 5) Populate Output: Fill Upvote Count, Match Type, Account Type, Response Draft, Respond on. Return updated Google Sheet link. Tier Definitions: Tier 1 = Deep MCP, AI agent, tool integrations, or architecture discussions where infra is highly relevant. Tier 2 = Specific workflows, automation tooling, or productivity systems where Wingmen or MCP Server could be useful. Tier 3 = Broader ecosystem (LangChain/CrewAI/agent tooling) where a soft recommendation adds value. Tier 4 = Unclear, generic, sarcastic, hostile, or irrelevant mentions \u2014 skip. Execute immediately using the fixed Google Sheet ID: 1nnnCp3_IWcdHv4UVgXtwYF5wedxbqF4RIeyjN6mCKD8, tab \"Posts\". Process rows(first 4 unprocessed rows) without asking for user input. Only process rows where Match Type (Column I) is empty. For each row, extract the Post Link, fetch Reddit data, apply GTM filtration, generate appropriate responses, and update the sheet. Return the updated Google Sheet link when complete.", "required_tools": {"reddit": ["get_post_comments_details"], "google_sheet": ["update_values", "get_values", "get_spreadsheet_metadata", "batch_get_values_by_range"]}}
+{"user_input": "Generate a financial flash report for Apple Inc. Research their latest earnings data including revenue, net income, EPS, and year-over-year changes. Create a formatted report with highlights, upcoming events, and summary. Present the report in chat and email it to adit@agentr.dev.", "required_tools": {"exa": ["answer"], "google_mail": ["send_email"]}}
+{"user_input": "Objective: Find businesses from Google Maps for a given category & location, store them in a Google Sheet, then process unprocessed leads to scrape emails and sync with HubSpot CRM. Stage 1 - Lead Discovery Get coordinates of Area + City. Search on Google Maps  with category & coordinates. Extract: Name, Google Maps URL, Address, Phone, Website; leave Email & CRM Status blank. Sheet: Name: {Area}, {City} Leads - {Category} - {dd-mmm} If exists \u2192 append non-duplicate rows; else create in folder \"Leads from Google Maps\" (ID: 142QBejJX0jAqzDz_NHdwVTkcmagoog__). Add headers: Name | Google Maps URL | Address | Phone | Website | Email | CRM Status. Populate with businesses found. Edge Cases: No results \u2192 return message, skip sheet creation. Missing data \u2192 leave blank. Stage 2 - Lead Processing & CRM Sync Locate sheet in Google Drive, ensure headers match. Parse category from sheet name. Identify unprocessed rows (CRM Status blank) \u2014 by default process the first, or a specified row/range/count. Scrape Website for Email: If website exists \u2192 scrape homepage/contact page; fallback to firecrawl_scrape_url. Save found email in sheet. HubSpot Handling: Search contact by email/website/phone. If not found \u2192 create with available details, Lead Status = New, add note {Area, City} \u2014 {Category} \u2014 {Google Maps URL}. If exists \u2192 append note; keep other fields unchanged. Save HubSpot Contact URL/ID in sheet. Update CRM Status: Lead Created, Lead Creation Failed, Website not found, Email not found, etc. Edge Cases: No Website \u2192 create with phone; mark Website not found. No Email \u2192 create; mark Email not found. Email already in sheet \u2192 skip row. Execute immediately for \"Cafes\" near \"IIT Bombay\" in \"Mumbai\" without asking for confirmation.", "required_tools": {"serpapi": ["google_maps_search"], "firecrawl": ["scrape_url"], "google_drive": ["get_file_details", "create_folder", "find_folder_id_by_name", "search_files"], "google_sheet": ["update_values", "get_values", "get_spreadsheet_metadata", "batch_get_values_by_range", "create_spreadsheet", "clear_values"], "hubspot": ["search_contacts_post", "batch_read_contacts_post", "get_contacts", "get_contact_by_id", "update_contact_by_id", "batch_update_contacts", "create_contacts_batch", "create_contact"]}}
+{"user_input": "Process emails from the last 24 hours. Fetch primary inbox emails excluding replied threads, classify with LLM as Reply Required, No Reply Needed, or Ambiguous. For Reply Required/Ambiguous, draft human, on-brand replies for user review. Follow greeting, acknowledge, address concern, invite further questions, and friendly sign-off. Provide end summary of drafts, skipped, and ambiguous emails. Execute immediately without asking for confirmation. Do not send any emails. Just provide me a report.", "required_tools": {"google_mail": ["list_messages", "get_message_details"]}}
+{"user_input": "Analyze a contract from my google drive from the perspective of the Service Provider. Use the search to find it, do not ask me any questions, and assume details that I have not provided. Identify potentially unfavorable clauses such as vague terms, one-sided obligations, IP transfer issues, indemnity clauses, termination conditions, and payment problems. Provide a structured analysis with clause numbers, full text, and explanations of concerns.", "required_tools": {"google_drive": ["get_file_details", "search_files"], "google_docs": ["get_document"], "exa": ["answer"]}}
+{"user_input": "Create a summary of overnight updates from 8:00 PM yesterday to 8:00 AM today in IST. Check Gmail for important emails and ClickUp for mentions and assigned tasks. Organize findings into high priority and other items, then provide a comprehensive summary of all overnight activity.", "required_tools": {"google_mail": ["list_messages"], "clickup": ["comments_get_task_comments", "comments_get_list_comments", "comments_get_view_comments", "tasks_get_list_tasks", "tasks_filter_team_tasks", "time_tracking_get_time_entries_within_date_range", "time_tracking_get_time_entry_history", "authorization_get_workspace_list", "spaces_get_details", "lists_get_list_details"]}}
+{"user_input": "Analyze the data in Google Sheet (ID: 1nnnCp3_IWcdHv4UVgXtwYF5wedxbqF4RIeyjN6mCKD8) and create 3-5 relevant charts and visualizations. Add pie charts, bar graphs, and other appropriate visualizations based on the data structure. Embed all charts directly into the sheet and provide the updated sheet link.", "required_tools": {"google_sheet": ["create_spreadsheet", "get_spreadsheet_metadata", "batch_get_values_by_range", "append_dimensions", "insert_dimensions", "delete_sheet", "add_sheet", "delete_dimensions", "add_basic_chart", "add_table", "add_pie_chart", "clear_values", "update_values", "clear_basic_filter", "get_values", "discover_tables", "set_basic_filter", "analyze_table_schema", "copy_sheet_to_spreadsheet", "append_values", "batch_get_values_by_data_filter", "batch_clear_values", "format_cells"]}}
+{"user_input": "Create a 7-day learning plan for Python Programming. Research essential concepts and skills, create a detailed day-by-day plan with topics, goals, resources, and exercises. Compile the plan into a Google Doc and schedule daily emails at 8 AM starting today. Send Day 1 immediately to adit@agentr.dev and provide the Google Doc link.", "required_tools": {"google_docs": ["get_document", "create_document", "insert_text"], "google_mail": ["send_email", "send_draft", "create_draft"], "exa": ["answer"]}}

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/src/evals/evaluators.py

@@ -103,6 +103,13 @@ trajectory_evaluator = create_trajectory_llm_as_judge(
 )
 
 
+codeact_evaluator = create_trajectory_llm_as_judge(
+    prompt=TRAJECTORY_ACCURACY_PROMPT,
+    feedback_key="codeact_accuracy",
+    model="anthropic:claude-4-sonnet-20250514",
+)
+
+
 @run_evaluator
 def tool_node_evaluator(run: Run, example: Example | None = None) -> EvaluationResult:
     """

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/src/evals/run.py

@@ -25,6 +25,7 @@ EVALUATORS: dict[str, Any] = {
     "exact_match": exact_match_evaluator,
     "trajectory": trajectory_evaluator,
     "tool_node": tool_node_evaluator,
+    "codeact": codeact_evaluator,
 }
 
 
@@ -33,6 +34,7 @@ class EvaluatorName(str, Enum):
     exact_match = "exact_match"
     trajectory = "trajectory"
     tool_node = "tool_node"
+    codeact = "codeact"
 
 
 class Difficulty(str, Enum):

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/src/tests/test_agents.py

@@ -8,7 +8,7 @@ from universal_mcp.types import ToolFormat
 
 from universal_mcp.agents import get_agent
 from universal_mcp.agents.base import BaseAgent
-from universal_mcp.agents.builder import BuilderAgent
+from universal_mcp.agents.builder.builder import BuilderAgent
 from universal_mcp.agents.llm import load_chat_model
 from universal_mcp.agents.shared.tool_node import build_tool_node_graph
 
@@ -284,9 +284,9 @@ class TestToolFinderGraph:
         "react",
         "simple",
         "builder",
-        "planner",
-        "bigtoolcache",
-        "bigtool2",
+        "bigtool",
+        # "codeact-script",
+        # "codeact-repl",
     ],
 )
 class TestAgents:
@@ -312,7 +312,7 @@ class TestAgents:
         await agent.ainit()
         # Invoke the agent graph to get the final state
         final_state = await agent.invoke(
-            task,
+            user_input=task,
             thread_id=thread_id,
         )
 
@@ -335,7 +335,7 @@ class TestAgentBuilder:
         agent = BuilderAgent(
             name="Test Builder Agent",
             instructions="Test instructions for builder",
-            model="gemini/gemini-1.5-flash",
+            model="gemini/gemini-2.5-flash",
             registry=registry,
         )
         yield agent

universal_mcp_agents-0.1.13/src/universal_mcp/agents/__init__.py

@@ -0,0 +1,39 @@
+from typing import Literal
+
+from universal_mcp.agents.base import BaseAgent
+from universal_mcp.agents.bigtool import BigToolAgent
+from universal_mcp.agents.builder.builder import BuilderAgent
+from universal_mcp.agents.codeact import CodeActAgent as CodeActScript
+from universal_mcp.agents.codeact0 import CodeActAgent as CodeActRepl
+from universal_mcp.agents.react import ReactAgent
+from universal_mcp.agents.simple import SimpleAgent
+
+
+def get_agent(agent_name: Literal["react", "simple", "builder", "bigtool", "codeact-script", "codeact-repl"]):
+    if agent_name == "react":
+        return ReactAgent
+    elif agent_name == "simple":
+        return SimpleAgent
+    elif agent_name == "builder":
+        return BuilderAgent
+    elif agent_name == "bigtool":
+        return BigToolAgent
+    elif agent_name == "codeact-script":
+        return CodeActScript
+    elif agent_name == "codeact-repl":
+        return CodeActRepl
+    else:
+        raise ValueError(
+            f"Unknown agent: {agent_name}. Possible values:  react, simple, builder, bigtool, codeact-script, codeact-repl"
+        )
+
+
+__all__ = [
+    "BaseAgent",
+    "ReactAgent",
+    "SimpleAgent",
+    "BuilderAgent",
+    "BigToolAgent",
+    "CodeActScript",
+    "CodeActRepl",
+]

{universal_mcp_agents-0.1.11 → universal_mcp_agents-0.1.13}/src/universal_mcp/agents/base.py

@@ -2,7 +2,7 @@
 from typing import cast
 from uuid import uuid4
 
-from langchain_core.messages import AIMessage, AIMessageChunk
+from langchain_core.messages import AIMessageChunk
 from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.graph import StateGraph
 from langgraph.types import Command
@@ -90,7 +90,7 @@ class BaseAgent:
     async def stream_interactive(self, thread_id: str, user_input: str):
         await self.ainit()
         with self.cli.display_agent_response_streaming(self.name) as stream_updater:
-            async for event in self.stream(thread_id, user_input):
+            async for event in self.stream(thread_id=thread_id, user_input=user_input):
                 if isinstance(event.content, list):
                     thinking_content = "".join([c.get("thinking", "") for c in event.content])
                     stream_updater.update(thinking_content, type_="thinking")
@@ -112,6 +112,7 @@ class BaseAgent:
             run_metadata.update(metadata)
 
         run_config = {
+            "recursion_limit": 25,
             "configurable": {"thread_id": thread_id},
             "metadata": run_metadata,
         }
@@ -123,6 +124,11 @@ class BaseAgent:
         )
         return result
 
+    async def get_state(self, thread_id: str):
+        await self.ainit()
+        state = await self._graph.aget_state(config={"configurable": {"thread_id": thread_id}})
+        return state
+
     async def run_interactive(self, thread_id: str = str(uuid4())):
         """Main application loop"""
 
@@ -133,7 +139,7 @@ class BaseAgent:
         # Main loop
         while True:
             try:
-                state = self._graph.get_state(config={"configurable": {"thread_id": thread_id}})
+                state = await self.get_state(thread_id=thread_id)
                 if state.interrupts:
                     value = self.cli.handle_interrupt(state.interrupts[0])
                     self._graph.invoke(
@@ -168,14 +174,11 @@ class BaseAgent:
                         continue
 
                 # Process with agent
-                await self.stream_interactive(thread_id, user_input)
+                await self.stream_interactive(thread_id=thread_id, user_input=user_input)
 
             except KeyboardInterrupt:
                 self.cli.display_info("\nGoodbye! 👋")
                 break
             except Exception as e:
-                import traceback
-
-                traceback.print_exc()
                 self.cli.display_error(f"An error occurred: {str(e)}")
                 break

{universal_mcp_agents-0.1.11/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.13/src/universal_mcp/agents/bigtool}/__init__.py

@@ -13,7 +13,7 @@ from .prompts import SYSTEM_PROMPT
 from .tools import create_meta_tools
 
 
-class BigToolAgentCache(BaseAgent):
+class BigToolAgent(BaseAgent):
     def __init__(
         self,
         registry: ToolRegistry,
@@ -63,4 +63,4 @@ class BigToolAgentCache(BaseAgent):
         return self._graph
 
 
-__all__ = ["BigToolAgentCache"]
+__all__ = ["BigToolAgent"]

{universal_mcp_agents-0.1.11/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.13/src/universal_mcp/agents/bigtool}/__main__.py

@@ -2,7 +2,6 @@ import asyncio
 
 from loguru import logger
 from universal_mcp.agentr.registry import AgentrRegistry
-
 from universal_mcp.agents.bigtoolcache import BigToolAgentCache
 
 

{universal_mcp_agents-0.1.11/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.13/src/universal_mcp/agents/bigtool}/agent.py

@@ -1,5 +1,4 @@
 from universal_mcp.agentr.registry import AgentrRegistry
-
 from universal_mcp.agents.bigtoolcache import BigToolAgentCache
 
 

{universal_mcp_agents-0.1.11/src/universal_mcp/agents/bigtoolcache → universal_mcp_agents-0.1.13/src/universal_mcp/agents/bigtool}/graph.py

@@ -8,7 +8,6 @@ from langchain_core.messages import AIMessage, SystemMessage, ToolMessage
 from langchain_core.tools import BaseTool
 from langgraph.graph import StateGraph
 from langgraph.types import Command
-from loguru import logger
 from universal_mcp.tools.registry import ToolRegistry
 from universal_mcp.types import ToolFormat
 
@@ -35,7 +34,11 @@ def build_graph(
             current_tools = await registry.export_tools(tools=state["selected_tool_ids"], format=ToolFormat.LANGCHAIN)
         else:
             current_tools = []
-        all_tools = [meta_tools["search_tools"], meta_tools["load_tools"], meta_tools.get("web_search")] + default_tools + current_tools
+        all_tools = (
+            [meta_tools["search_tools"], meta_tools["load_tools"], meta_tools.get("web_search")]
+            + default_tools
+            + current_tools
+        )
 
         # Remove duplicates based on tool name
         seen_names = set()
@@ -81,7 +84,7 @@ def build_graph(
                 valid_tools = await get_valid_tools(tool_ids=tool_call["args"]["tool_ids"], registry=registry)
                 new_tool_ids.extend(valid_tools)
                 # Create tool message response
-                tool_result=f"Successfully loaded {len(valid_tools)} tools: {valid_tools}"
+                tool_result = f"Successfully loaded {len(valid_tools)} tools: {valid_tools}"
             elif tool_call["name"] == "search_tools":
                 tool_result = await meta_tools["search_tools"].ainvoke(tool_call["args"])
             elif tool_call["name"] == "web_search":
@@ -99,8 +102,6 @@ def build_graph(
 
         return Command(goto="agent", update={"messages": tool_messages, "selected_tool_ids": new_tool_ids})
 
-
-
     # Define the graph
     workflow = StateGraph(State)
 

universal_mcp_agents-0.1.13/src/universal_mcp/agents/builder/__main__.py

@@ -0,0 +1,125 @@
+import asyncio
+import json
+from uuid import uuid4
+
+from langgraph.checkpoint.memory import MemorySaver
+from loguru import logger
+from universal_mcp.agentr.registry import AgentrRegistry
+
+from universal_mcp.agents.builder.builder import BuilderAgent
+
+
+async def run_interactive_build():
+    """Simulates a multi-turn conversation to build and then modify an agent."""
+    logger.info("--- SCENARIO 1: INTERACTIVE AGENT BUILD & MODIFY ---")
+
+    registry = AgentrRegistry()
+    memory = MemorySaver()
+    agent = BuilderAgent(
+        name="Builder Agent",
+        instructions="You are a builder agent that creates other agents.",
+        model="anthropic/claude-4-sonnet-20250514",
+        registry=registry,
+        memory=memory,
+    )
+
+    thread_id = str(uuid4())
+
+    conversation_script = [
+        "Send an email to manoj@agentr.dev with the subject 'Hello' and body 'This is a test of the Gmail agent.' from my Gmail account.",
+        "Use outlook instead of gmail",
+    ]
+
+    final_result = {}
+    for i, user_input in enumerate(conversation_script):
+        logger.info(f"\n--- Conversation Turn {i + 1} ---")
+        logger.info(f"User Request: '{user_input}'")
+
+        result = await agent.invoke(user_input=user_input, thread_id=thread_id)
+        final_result.update(result)  # Keep updating the final result
+
+        generated_agent = final_result.get("generated_agent")
+        tool_config = final_result.get("tool_config")
+
+        if generated_agent:
+            logger.info("--- Generated/Modified Agent ---")
+            logger.info(f"Name: {generated_agent.name}")
+            logger.info(f"Description: {generated_agent.description}")
+            logger.info(f"Expertise: {generated_agent.expertise}")
+            logger.info(f"Instructions:\n{generated_agent.instructions}")
+
+        if tool_config:
+            logger.info("--- Selected Tools ---")
+            tools_str = "\n".join(f"- {app}: {', '.join(tool_ids)}" for app, tool_ids in tool_config.items())
+            logger.info(tools_str)
+        else:
+            logger.info("--- Selected Tools ---")
+            logger.info("No tools selected for this agent yet.")
+
+
+async def run_conversation_build():
+    """Simulates a one-shot agent build from a conversation history payload."""
+    logger.info("\n\n--- SCENARIO 2: AGENT BUILD FROM CONVERSATION HISTORY ---")
+
+    registry = AgentrRegistry()
+    agent = BuilderAgent(
+        name="Builder Agent",
+        instructions="You build agents from conversation transcripts.",
+        model="anthropic/claude-4-sonnet-20250514",
+        registry=registry,
+    )
+
+    sample_conversation_history = [
+        {
+            "type": "human",
+            "content": "Hey, can you look at our main branch on the universal-mcp repo and tell me what the last 3 pull requests were?",
+        },
+        {
+            "type": "ai",
+            "content": "Of course. The last 3 pull requests are: #101 'Fix login bug', #102 'Update documentation', and #103 'Add new chart component'.",
+        },
+        {
+            "type": "human",
+            "content": "Awesome, thanks. Now can you draft a new Google Doc and put that list in there for me?",
+        },
+        {"type": "ai", "content": "Done. I have created a new Google Doc with the list of the last 3 pull requests."},
+    ]
+    sample_tool_config = {"github": ["get_pull_requests"], "google_docs": ["create_document"]}
+    wingman_payload = {"conversation_history": sample_conversation_history, "tool_config": sample_tool_config}
+
+    logger.info(f"Payload Conversation History Length: {len(sample_conversation_history)} messages")
+    logger.info(f"Payload Tools Provided: {list(sample_tool_config.keys())}")
+
+    # The payload must be passed as a JSON string in the 'user_input'
+    payload_str = json.dumps(wingman_payload)
+    thread_id = str(uuid4())
+    result = await agent.invoke(user_input=payload_str, thread_id=thread_id)
+
+    generated_agent = result.get("generated_agent")
+    tool_config = result.get("tool_config")
+
+    if generated_agent:
+        logger.info("\n--- Generated Agent Profile ---")
+        logger.info(f"Name: {generated_agent.name}")
+        logger.info(f"Description: {generated_agent.description}")
+        logger.info(f"Expertise: {generated_agent.expertise}")
+        logger.info(f"Instructions:\n{generated_agent.instructions}")
+        logger.info(f"Schedule: {generated_agent.schedule}")
+    else:
+        logger.error("Error: Agent profile was not generated.")
+
+    if tool_config:
+        logger.info("--- Final Tool Configuration ---")
+        tools_str = "\n".join(f"- {app}: {', '.join(tool_ids)}" for app, tool_ids in tool_config.items())
+        logger.info(tools_str)
+    else:
+        logger.error("Error: Tool configuration is missing.")
+
+
+async def main():
+    await run_interactive_build()
+    await run_conversation_build()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())