massgen 0.0.3__py3-none-any.whl → 0.1.0__py3-none-any.whl

massgen/configs/tools/mcp/claude_mcp_test.yaml

@@ -0,0 +1,27 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/claude_mcp_test "Test the MCP tools by calling mcp_echo with text 'Hello massgen' and add_numbers with 46 and 52"
+agents:
+  - id: "claude_mcp_test"
+    backend:
+      type: "claude"
+      model: "claude-sonnet-4-20250514"  # Model name
+      mcp_servers:
+        - name: "test_server"
+          type: "stdio"
+          command: "python"
+          args: ["-u", "-m", "massgen.tests.mcp_test_server"]
+    system_message: |
+      You are testing the MCP integration with Google Gemini.
+
+      MCP tools from the "test_server" will be available via sessions and auto-called by the system when needed.
+      Do not output function-call syntax. Simply perform the tasks and present clear, concise results.
+
+      Tasks to verify:
+      - Echo: return the text given in the question.
+      - Add numbers: return the sum of a and b.
+      - Current time: return the current timestamp.
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/five_agents_travel_mcp_test.yaml

@@ -0,0 +1,157 @@
+# Five Agent Travel Research MCP Test Configuration
+# This configuration uses 5 different backends (Gemini, OpenAI, Claude Code, Claude, Grok) with Multi-MCP capability
+# Each agent has access to Airbnb search and Brave Search for comprehensive travel research
+#
+# Setup Requirements:
+# 1. Install MCP servers:
+#    npm install -g @openbnb/mcp-server-airbnb
+#    npm install -g @modelcontextprotocol/server-brave-search
+# 2. Set environment variables in .env file:
+#    BRAVE_API_KEY="your_brave_search_api_key"
+#    OPENAI_API_KEY="your_openai_key"
+#    GOOGLE_API_KEY="your_gemini_key"
+#    ANTHROPIC_API_KEY="your_claude_key"
+#    XAI_API_KEY="your_grok_key"
+#
+# Example usage:
+# massgen --config @examples/tools/mcp/five_agents_travel_mcp_test "Plan a 5-day trip to Barcelona for a couple on a mid-range budget. Find the best neighborhoods, suggest 3 Airbnb options, and provide a complete travel guide with attractions and dining recommendations."
+
+agents:
+  - id: "gemini_travel_agent"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      mcp_servers:
+        # Airbnb server for accommodations
+        - name: "airbnb_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@openbnb/mcp-server-airbnb", "--ignore-robots-txt"]
+          security:
+            level: "moderate"
+        # Brave Search API server
+        - name: "brave_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-brave-search"]
+          env:
+            BRAVE_API_KEY: "${BRAVE_API_KEY}"
+          security:
+            level: "moderate"
+    system_message: |
+      Available Tools:
+      - mcp__brave_web_search: Research destinations, attractions, and local insights
+      - mcp__airbnb_search: Find accommodations with filtering capabilities
+      - mcp__airbnb_search__airbnb_listing_details: Get detailed property information
+
+  - id: "openai_travel_agent"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      mcp_servers:
+        # Airbnb server for accommodations
+        - name: "airbnb_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@openbnb/mcp-server-airbnb", "--ignore-robots-txt"]
+          security:
+            level: "moderate"
+        # Brave Search API server
+        - name: "brave_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-brave-search"]
+          env:
+            BRAVE_API_KEY: "${BRAVE_API_KEY}"
+          security:
+            level: "moderate"
+    system_message: |
+      Available Tools:
+      - mcp__brave_web_search: Research destinations, attractions, and local insights
+      - mcp__airbnb_search: Find accommodations with filtering capabilities
+      - mcp__airbnb_search__airbnb_listing_details: Get detailed property information
+
+  - id: "claude_code_travel_agent"
+    backend:
+      type: "claude_code"
+      cwd: "claude_code_workspace_travel_mcp"
+      permission_mode: "bypassPermissions"
+      mcp_servers:
+        airbnb_search:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@openbnb/mcp-server-airbnb", "--ignore-robots-txt"]
+        brave_search:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-brave-search"]
+          env:
+            BRAVE_API_KEY: "${BRAVE_API_KEY}"
+    system_message: |
+      Available Tools:
+      - mcp__brave_web_search: Research destinations, attractions, and local insights
+      - mcp__airbnb_search: Find accommodations with filtering capabilities
+      - mcp__airbnb_search__airbnb_listing_details: Get detailed property information
+
+  - id: "claude_travel_agent"
+    backend:
+      type: "claude"
+      model: "claude-sonnet-4-20250514"
+      mcp_servers:
+        # Airbnb server for accommodations
+        - name: "airbnb_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@openbnb/mcp-server-airbnb", "--ignore-robots-txt"]
+          security:
+            level: "moderate"
+        # Brave Search API server
+        - name: "brave_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-brave-search"]
+          env:
+            BRAVE_API_KEY: "${BRAVE_API_KEY}"
+          security:
+            level: "moderate"
+    system_message: |
+      Available Tools:
+      - mcp__brave_web_search: Research destinations, attractions, and local insights
+      - mcp__airbnb_search: Find accommodations with filtering capabilities
+      - mcp__airbnb_search__airbnb_listing_details: Get detailed property information
+
+  - id: "grok_travel_agent"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      mcp_servers:
+        # Airbnb server for accommodations
+        - name: "airbnb_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@openbnb/mcp-server-airbnb", "--ignore-robots-txt"]
+          security:
+            level: "moderate"
+        # Brave Search API server
+        - name: "brave_search"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-brave-search"]
+          env:
+            BRAVE_API_KEY: "${BRAVE_API_KEY}"
+          security:
+            level: "moderate"
+    system_message: |
+      Available Tools:
+      - mcp__brave_web_search: Research destinations, attractions, and local insights
+      - mcp__airbnb_search: Find accommodations with filtering capabilities
+      - mcp__airbnb_search__airbnb_listing_details: Get detailed property information
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+
+# Orchestrator Settings with Coordination Configuration
+orchestrator:
+  snapshot_storage: "massgen_logs/snapshots"  # Directory for workspace snapshots
+  agent_temporary_workspace: "massgen_logs/temp_workspaces"  # Directory for temporary agent workspaces

massgen/configs/tools/mcp/five_agents_weather_mcp_test.yaml

@@ -0,0 +1,103 @@
+# Five Agent Weather MCP Test Configuration
+# This configuration uses 5 different backends (Gemini, OpenAI, Claude Code, Claude, Grok) with Weather MCP capability
+# Each agent can get weather information, forecasts, and weather alerts
+#
+# Setup Requirements:
+# 1. Install Weather MCP server: npm install -g @fak111/weather-mcp
+# 2. Set environment variables in .env file:
+#    OPENAI_API_KEY="your_openai_key"
+#    GOOGLE_API_KEY="your_gemini_key"
+#    ANTHROPIC_API_KEY="your_claude_key"
+#    XAI_API_KEY="your_grok_key"
+#
+# Example usage:
+# massgen --config @examples/tools/mcp/five_agents_weather_mcp_test "What's the weather like in Tokyo, London, and New York today? Compare the conditions and give me travel recommendations."
+
+agents:
+  - id: "gemini_weather_agent"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+          security:
+            level: "moderate"
+    system_message: |
+      Available Weather Tools:
+      - Weather information via MCP integration
+      - Current conditions, forecasts, and alerts
+
+  - id: "openai_weather_agent"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+          security:
+            level: "moderate"
+    system_message: |
+      Available Weather Tools:
+      - Weather information via MCP integration
+      - Current conditions, forecasts, and alerts
+
+  - id: "claude_code_weather_agent"
+    backend:
+      type: "claude_code"
+      cwd: "claude_code_workspace_weather_mcp"
+      permission_mode: "bypassPermissions"
+      mcp_servers:
+        weather:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+    system_message: |
+      Available Weather Tools:
+      - Weather information via MCP integration
+      - Current conditions, forecasts, and alerts
+
+  - id: "claude_weather_agent"
+    backend:
+      type: "claude"
+      model: "claude-sonnet-4-20250514"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+          security:
+            level: "moderate"
+    system_message: |
+      Available Weather Tools:
+      - Weather information via MCP integration
+      - Current conditions, forecasts, and alerts
+
+  - id: "grok_weather_agent"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+          security:
+            level: "moderate"
+    system_message: |
+      Available Weather Tools:
+      - Weather information via MCP integration
+      - Current conditions, forecasts, and alerts
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+
+# Orchestrator Settings with Coordination Configuration
+orchestrator:
+  snapshot_storage: "massgen_logs/snapshots"  # Directory for workspace snapshots
+  agent_temporary_workspace: "massgen_logs/temp_workspaces"  # Directory for temporary agent workspaces

massgen/configs/tools/mcp/gemini_mcp_example.yaml

@@ -0,0 +1,24 @@
+# MassGen Configuration: Gemini with MCP Integration
+# Usage:
+#   massgen --config @examples/tools/mcp/gemini_mcp_example "whats the weather of Tokyo"
+agents:
+  - id: "gemini2.5flash_mcp_weather"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+    system_message: |
+      You are an AI assistant with access to weather information through MCP (Model Context Protocol) integration.
+
+      Weather tools are available via MCP sessions and will be called automatically by the system when needed.
+      Do not output tool call syntax or function declarations. Focus on answering the user's question clearly.
+
+      When users ask about weather conditions, forecasts, or alerts, include the location and time frame in your
+      response and provide concise, up-to-date information using the integrated tools.
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gemini_mcp_filesystem_test.yaml

@@ -0,0 +1,23 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/gemini_mcp_filesystem_test " Please create a Python script with a hello world function in various fictional languages and save it directly into a file in your workspace."
+agents:
+  - id: "gemini_agent1"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-pro"
+      cwd: "workspace1"  # Working directory for file operations
+
+  - id: "gemini_agent2"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-pro"
+      cwd: "workspace2"  # Working directory for file operations
+
+orchestrator:
+    snapshot_storage: "snapshots"  # Directory to store workspace snapshots
+    agent_temporary_workspace: "temp_workspaces"  # Directory for temporary agent workspaces
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gemini_mcp_filesystem_test_sharing.yaml

@@ -0,0 +1,23 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/gemini_mcp_filesystem_test_sharing "Please create a txt file with all the current directories you can read files from."
+agents:
+  - id: "gemini_agent1"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      cwd: "workspace1"  # Working directory for file operations. Generic name so no hint of content.
+
+  - id: "gemini_agent2"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      cwd: "workspace2"  # Working directory for file operations. Generic name so no hint of content.
+
+orchestrator:
+    snapshot_storage: "snapshots"  # Directory to store workspace snapshots
+    agent_temporary_workspace: "workspaces"  # Directory for temporary agent workspaces
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gemini_mcp_filesystem_test_single_agent.yaml

@@ -0,0 +1,17 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/gemini_mcp_filesystem_test_single_agent "Please create a Python script with a hello world function in various fictional languages and save it directly into a file in your workspace."
+agents:
+  - id: "gemini_agent1"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      cwd: "workspace1"  # Working directory for file operations
+
+orchestrator:
+    snapshot_storage: "snapshots"  # Directory to store workspace snapshots
+    agent_temporary_workspace: "workspaces"  # Directory for temporary agent workspaces
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gemini_mcp_filesystem_test_with_claude_code.yaml

@@ -0,0 +1,24 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/gemini_mcp_filesystem_test_with_claude_code "Create a presentation that teaches a reinforcement learning algorithm and output it in LaTeX Beamer format. No figures should be added."
+agents:
+  - id: "gemini_agent"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-pro"
+      cwd: "workspace1"  # Working directory for file operations
+      enable_web_search: true
+
+  - id: "claude_code"
+    backend:
+      type: "claude_code"
+      model: "claude-sonnet-4-20250514"
+      cwd: "workspace2"  # Working directory for file operations
+
+orchestrator:
+    snapshot_storage: "snapshots"  # Directory to store workspace snapshots
+    agent_temporary_workspace: "temp_workspaces"  # Directory for temporary agent workspaces
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gemini_mcp_test.yaml

@@ -0,0 +1,27 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/gemini_mcp_test "Test the MCP tools by calling mcp_echo with text 'Hello massgen' and add_numbers with 46 and 52"
+agents:
+  - id: "gemini_mcp_test"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      mcp_servers:
+        - name: "test_server"
+          type: "stdio"
+          command: "python"
+          args: ["-u", "-m", "massgen.tests.mcp_test_server"]
+    system_message: |
+      You are testing the MCP integration with Google Gemini.
+
+      MCP tools from the "test_server" will be available via sessions and auto-called by the system when needed.
+      Do not output function-call syntax. Simply perform the tasks and present clear, concise results.
+
+      Tasks to verify:
+      - Echo: return the text given in the question.
+      - Add numbers: return the sum of a and b.
+      - Current time: return the current timestamp.
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gemini_notion_mcp.yaml

@@ -0,0 +1,52 @@
+# Dual agent configuration using Notion MCP with Gemini-2.5-pro.
+# To avoid connection issues, we give each agent its own Notion token with access to a different page (we call them `LLM Agent Research (1)` and `LLM Agent Research (2)`) and use NOTION_TOKEN_ONE and NOTION_TOKEN_TWO in the .env file to represent them, respectively.
+# To set this up, you must do the below twice, since we create one page for each agent (the process for doing it once is detailed in depth [here](https://github.com/makenotion/notion-mcp-server)):
+#   1) create a new page titled `LLM Agent Research ({1 or 2})` in Notion.
+#   2) generate an integration token (Internal Integration Secret) in Notion by visiting [the integrations page](https://www.notion.so/profile/integrations), and visit the Access tab to add the integration to your new page.
+#   3) place it in your .env file: NOTION_TOKEN_{ONE or TWO}="your_notion_token_here"
+#
+# cmd: massgen --config @examples/tools/mcp/gemini_notion_mcp "Generate and refine a structured Todo list for learning about LLM multi-agent systems, complete with exciting objectives and fun activities. Each time you have a new version, create a new Notion page with a title and the current date and time (including hours, minutes, seconds, and milliseconds) to store the list. Then, verify that you can access the page and read back the content. Create this page as a subpage under an existing notion page called 'LLM Agent Research (x)', where x is either 1 or 2 depending on which you have access to."
+agents:
+  - id: "gemini-2.5-pro1"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-pro"
+      enable_web_search: true
+
+
+      # MCP tools configuration
+      mcp_servers:
+        notionApi:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@notionhq/notion-mcp-server"]
+          env:
+            NOTION_TOKEN: "${NOTION_TOKEN_ONE}"
+      # There are errors with how this tool functions, so we exclude it for now
+      exclude_tools:
+        - post_search
+
+
+  - id: "gemini-2.5-pro2"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-pro"
+      enable_web_search: true
+
+
+      # MCP tools configuration
+      mcp_servers:
+        notionApi:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@notionhq/notion-mcp-server"]
+          env:
+            NOTION_TOKEN: "${NOTION_TOKEN_TWO}"
+      exclude_tools:
+        - post_search
+
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gpt5_nano_mcp_example.yaml

@@ -0,0 +1,24 @@
+# MassGen Configuration: Gemini with MCP Integration
+# Usage:
+#   massgen --config @examples/tools/mcp/gpt5_nano_mcp_example "whats the weather of Tokyo"
+agents:
+  - id: "gpt5_nano_mcp_weather"
+    backend:
+      type: "openai"
+      model: "gpt-5-nano"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+    system_message: |
+      You are an AI assistant with access to weather information through MCP (Model Context Protocol) integration.
+
+      Weather tools are available via MCP sessions and will be called automatically by the system when needed.
+      Do not output tool call syntax or function declarations. Focus on answering the user's question clearly.
+
+      When users ask about weather conditions, forecasts, or alerts, include the location and time frame in your
+      response and provide concise, up-to-date information using the integrated tools.
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gpt5_nano_mcp_test.yaml

@@ -0,0 +1,27 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/gpt5_nano_mcp_test "Test the MCP tools by calling mcp_echo with text 'Hello massgen' and add_numbers with 46 and 52"
+agents:
+  - id: "gpt5_nano_mcp_test"
+    backend:
+      type: "openai"
+      model: "gpt-5-nano"
+      mcp_servers:
+        - name: "test_server"
+          type: "stdio"
+          command: "python"
+          args: ["-u", "-m", "massgen.tests.mcp_test_server"]
+    system_message: |
+      You are testing the MCP integration with OpenAI GPT.
+
+      MCP tools from the "test_server" will be available via sessions and auto-called by the system when needed.
+      Do not output function-call syntax. Simply perform the tasks and present clear, concise results.
+
+      Tasks to verify:
+      - Echo: return the text given in the question.
+      - Add numbers: return the sum of a and b.
+      - Current time: return the current timestamp.
+
+ui:
+  display_type: "simple"
+  logging_enabled: true

massgen/configs/tools/mcp/gpt5mini_claude_code_discord_mcp_example.yaml

@@ -0,0 +1,38 @@
+agents:
+  - id: "claude_code_discord_mcp"
+    backend:
+      type: "claude_code"
+      cwd: "claude_code_workspace_discord_mcp"
+      permission_mode: "bypassPermissions"
+      
+      # Discord MCP server 
+      mcp_servers:
+        discord:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "mcp-discord", "--config", "YOUR_DISCORD_TOKEN"]
+
+      allowed_tools:
+        - "Read"
+        - "Write"
+        - "Bash"
+        - "LS"
+        - "WebSearch"
+        # MCP tools will be auto-discovered from the server
+    system_message: |
+      You are a helpful assistant with access to built-in tools and MCP servers. Before using any tools, check if the task is already completed by previous steps or agents. Only use tools for unfinished tasks. Never duplicate actions, especially communications (messages, emails) or workspace modifications (file edits, Discord/Slack posts). For these critical operations, first provide the content as your answer, then execute the tools only during final presentation.
+
+  - id: "gpt-5-mini"
+    backend:
+      type: "openai"
+      model: "gpt-5-mini"
+      reasoning:
+        effort: "medium"
+        summary: "auto"
+      enable_web_search: true
+      enable_code_interpreter: true
+    system_message: "You are a helpful AI assistant with web search and code execution capabilities. You should always complete tasks that you can do. If you cannot complete a task, just omit it."
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gpt_oss_mcp_example.yaml

@@ -0,0 +1,25 @@
+# MassGen Configuration: Gemini with MCP Integration
+# Usage:
+#   massgen --config @examples/tools/mcp/gpt_oss_mcp_example "whats the weather of Tokyo"
+agents:
+  - id: "gpt-oss-mcp_weather" # Cerebras AI
+    backend:
+      type: "chatcompletion"
+      model: "gpt-oss-120b"
+      base_url: "https://api.cerebras.ai/v1"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+    system_message: |
+      You are an AI assistant with access to weather information through MCP (Model Context Protocol) integration.
+
+      Weather tools are available via MCP sessions and will be called automatically by the system when needed.
+      Do not output tool call syntax or function declarations. Focus on answering the user's question clearly.
+
+      When users ask about weather conditions, forecasts, or alerts, include the location and time frame in your
+      response and provide concise, up-to-date information using the integrated tools.
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/gpt_oss_mcp_test.yaml

@@ -0,0 +1,28 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/gpt_oss_mcp_test "Test the MCP tools by calling mcp_echo with text 'Hello massgen' and add_numbers with 46 and 52"
+agents:
+  - id: "gpt_oss_mcp_weather"
+    backend:
+      type: "chatcompletion"
+      model: "gpt-oss-120b"
+      base_url: "https://api.cerebras.ai/v1"
+      mcp_servers:
+        - name: "test_server"
+          type: "stdio"
+          command: "python"
+          args: ["-u", "-m", "massgen.tests.mcp_test_server"]
+    system_message: |
+      You are testing the MCP integration with OpenAI GPT.
+
+      MCP tools from the "test_server" will be available via sessions and auto-called by the system when needed.
+      Do not output function-call syntax. Simply perform the tasks and present clear, concise results.
+
+      Tasks to verify:
+      - Echo: return the text given in the question.
+      - Add numbers: return the sum of a and b.
+      - Current time: return the current timestamp.
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/grok3_mini_mcp_example.yaml

@@ -0,0 +1,24 @@
+# MassGen Configuration: Gemini with MCP Integration
+# Usage:
+#   massgen --config @examples/tools/mcp/grok3_mini_mcp_example "whats the weather of Tokyo"
+agents:
+  - id: "grok3_mini_mcp_weather"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      mcp_servers:
+        - name: "weather"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@fak111/weather-mcp"]
+    system_message: |
+      You are an AI assistant with access to weather information through MCP (Model Context Protocol) integration.
+
+      Weather tools are available via MCP sessions and will be called automatically by the system when needed.
+      Do not output tool call syntax or function declarations. Focus on answering the user's question clearly.
+
+      When users ask about weather conditions, forecasts, or alerts, include the location and time frame in your
+      response and provide concise, up-to-date information using the integrated tools.
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true