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

massgen/configs/tools/mcp/grok3_mini_mcp_test.yaml

@@ -0,0 +1,27 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/grok3_mini_mcp_test "Test the MCP tools by calling mcp_echo with text 'Hello massgen' and add_numbers with 46 and 52"
+agents:
+  - id: "grok3_mini_mcp_test"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      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 Grok.
+
+      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/multimcp_gemini.yaml

@@ -0,0 +1,111 @@
+# Multi-MCP Server Configuration - Airbnb + Brave Search
+# Test multiple API-based MCP servers with proper security configuration
+#
+# Setup:
+# Place it in your .env file, BRAVE_API_KEY="your_brave_key_here"
+#
+# Example usage - Travel Research:
+# massgen --config @examples/tools/mcp/multimcp_gemini "Find the best neighborhoods in New York, suggest safe and affordable Airbnb stays for 2 people from October 10–15, and give me a travel summary with highlights and costs"
+
+agents:
+  - id: "travel_research_assistant"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      mcp_servers:
+        # Airbnb server for accommodations (web scraping based)
+        - 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: |
+      You are a comprehensive travel research assistant with access to multiple data sources.
+      Your goal is to create personalized, actionable, and well-structured travel plans by
+      combining neighborhood insights with cost-effective accommodations.
+
+      ## Available Tools
+
+      ### 1. Web Search (mcp__brave_web_search)
+      - Purpose: Research neighborhoods, safety, attractions, events, and transportation.
+      - Use for: travel guides, reviews, safety info, current events, local insights.
+
+      ### 2. Airbnb Search (mcp__airbnb_search)
+      - Purpose: Find accommodations with filtering by location, dates, guests, and budget.
+      - Parameters: location, checkin/checkout, adults/children/pets, minPrice/maxPrice.
+      - Returns: Property listings with price, amenities, reviews, and booking links.
+
+      ### 3. Airbnb Details (mcp__airbnb_search__airbnb_listing_details)
+      - Purpose: Retrieve full details about a specific property.
+      - Parameters: listing ID, checkin/checkout, guest details.
+      - Returns: Property info, rules, location, and amenities.
+
+      ## Research Workflow
+
+      ### Step 1: Gather User Context
+      Always ask or infer the following details if not explicitly provided:
+      - Trip duration (number of days/nights).
+      - Traveler type: solo, couple, family, group.
+      - Budget preference: economical, mid-range, or luxury.
+      - Special requirements: e.g., pet-friendly, kid-friendly, business trip.
+
+      ### Step 2: Neighborhood Research
+      - Search queries:
+        - “best neighborhoods in [city] for tourists”
+        - “safest areas in [city] for families”
+        - “budget-friendly neighborhoods in [city]”
+        - “up-and-coming neighborhoods in [city]”
+      - Extract: pros/cons, safety, vibe, attractions, and accessibility.
+
+      ### Step 3: Accommodation Search
+      - Use Airbnb search for shortlisted neighborhoods.
+      - Apply filters: price range (based on budget), guest count (based on traveler type),
+        and duration (check-in/check-out).
+      - Compare options by price, amenities, and reviews.
+
+      ### Step 4: Create Comprehensive Travel Summary
+      Include:
+      - **Neighborhood Overview**: highlights, vibe, pros/cons, safety info.
+      - **Local Attractions**: key sights, dining, cultural experiences.
+      - **Transportation Tips**: nearest transit, ease of access to city center.
+      - **Airbnb Options**: 2 or 3 listings with price/night, amenities, and links.
+      - **Budget Estimate**: total accommodation cost for trip duration.
+      - **Tailored Advice**: safety for families, nightlife for couples, space for groups, etc.
+      - **Booking Recommendations**: most practical or best value options.
+
+      ## Example Workflows
+
+      **Barcelona Trip (Budget Traveler, 5 Nights, Couple)**
+      1. Research neighborhoods good for couples, affordable dining, and safety.
+      2. Search Airbnb with filters: 2 adults, 5 nights, price range $80–120/night.
+      3. Summarize with pros/cons, attractions, and Airbnb picks.
+      4. Provide estimated total cost for stay + recommendations.
+
+      **Tokyo Family Travel (Mid-Range, 7 Nights, 2 Adults + 2 Kids)**
+      1. Research “family-friendly neighborhoods in Tokyo” with good schools/parks.
+      2. Airbnb search: 4 guests, 7 nights, price range $150–200/night, family amenities.
+      3. Summarize: neighborhoods, attractions, transportation, Airbnb options.
+      4. Highlight kid-friendly features and total accommodation budget.
+
+      ## Output Guidelines
+      - Use clear headings and bullet points.
+      - Present neighborhood recommendations with pros/cons.
+      - Provide 2 or 3 Airbnb listings with prices and amenities.
+      - Show total accommodation cost for stay duration.
+      - Tailor advice to traveler type and budget.
+      - Always include practical travel tips.
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/tools/mcp/qwen_api_mcp_example.yaml

@@ -0,0 +1,25 @@
+# MassGen Configuration: Gemini with MCP Integration
+# Usage:
+#   massgen --config @examples/tools/mcp/qwen_api_mcp_example "whats the weather of Tokyo"
+agents:
+  - id: "qwen-mcp_weather" # Cerebras AI
+    backend:
+      type: "chatcompletion"
+      model: "qwen-3-235b-a22b-instruct-2507"
+      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/qwen_api_mcp_test.yaml

@@ -0,0 +1,28 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/qwen_api_mcp_test "Test the MCP tools by calling mcp_echo with text 'Hello massgen' and add_numbers with 46 and 52"
+agents:
+  - id: "qwen_mcp_weather"
+    backend:
+      type: "chatcompletion"
+      model: "qwen-3-235b-a22b-instruct-2507"
+      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 Qwen API.
+
+      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/qwen_local_mcp_example.yaml

@@ -0,0 +1,24 @@
+# MassGen Configuration: Gemini with MCP Integration
+# Usage:
+#   massgen --config @examples/tools/mcp/qwen_local_mcp_example "whats the weather of Tokyo"
+agents:
+  - id: "qwen3-4b-mcp_weather" # Cerebras AI
+    backend:
+      type: "lmstudio"
+      model: "qwen/qwen3-4b-2507"
+      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: "simple"
+  logging_enabled: true

massgen/configs/tools/mcp/qwen_local_mcp_test.yaml

@@ -0,0 +1,27 @@
+# MassGen Configuration
+# Usage:
+# massgen --config @examples/tools/mcp/qwen_local_mcp_test "Test the MCP tools by calling mcp_echo with text 'Hello massgen' and add_numbers with 46 and 52"
+agents:
+  - id: "qwen_mcp_weather"
+    backend:
+      type: "lmstudio"
+      model: "qwen/qwen3-4b-2507"
+      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 Qwen API.
+
+      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/planning/five_agents_discord_mcp_planning_mode.yaml

@@ -0,0 +1,140 @@
+# Five Agent Discord MCP Test Configuration with Planning Mode
+# This configuration demonstrates planning vs execution mode to prevent duplicate tool calls
+# during coordination while allowing full tool execution during final presentation
+#
+# Setup Requirements:
+# 1. Install Discord MCP server: npm install -g mcp-discord
+# 2. Set environment variables in .env file:
+#    DISCORD_TOKEN="your_discord_bot_token"
+#    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/planning/five_agents_discord_mcp_planning_mode "Check recent messages in our development channel, summarize the discussion, and post a helpful response about the current topic."
+
+agents:
+  - id: "gemini_discord_agent"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      mcp_servers:
+        - name: "discord"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "mcp-discord", "--config", "${DISCORD_TOKEN}"]
+          env:
+            DISCORD_TOKEN: "${DISCORD_TOKEN}"
+          security:
+            level: "high"
+    system_message: |
+      Available Discord Tools:
+      - Discord server interaction via MCP integration
+      - Message reading, sending, and management
+      - Channel and server information access
+
+  - id: "openai_discord_agent"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      mcp_servers:
+        - name: "discord"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "mcp-discord", "--config", "${DISCORD_TOKEN}"]
+          env:
+            DISCORD_TOKEN: "${DISCORD_TOKEN}"
+          security:
+            level: "high"
+      exclude_tools:
+        - mcp__discord__discord_send_webhook_message
+        - mcp__discord__discord_edit_webhook_message
+    system_message: |
+      Available Discord Tools:
+      - Discord server interaction via MCP integration
+      - Message reading, sending, and management
+      - Channel and server information access
+
+  - id: "claude_code_discord_agent"
+    backend:
+      type: "claude_code"
+      cwd: "claude_code_workspace_discord_mcp"
+      permission_mode: "bypassPermissions"
+      mcp_servers:
+        discord:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "mcp-discord", "--config", "${DISCORD_TOKEN}"]
+          env:
+            DISCORD_TOKEN: "${DISCORD_TOKEN}"
+    system_message: |
+      Available Discord Tools:
+      - Discord server interaction via MCP integration
+      - Message reading, sending, and management
+      - Channel and server information access
+
+  - id: "claude_discord_agent"
+    backend:
+      type: "claude"
+      model: "claude-sonnet-4-20250514"
+      mcp_servers:
+        - name: "discord"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "mcp-discord", "--config", "${DISCORD_TOKEN}"]
+          env:
+            DISCORD_TOKEN: "${DISCORD_TOKEN}"
+          security:
+            level: "high"
+      exclude_tools:
+        - mcp__discord__discord_send_webhook_message
+        - mcp__discord__discord_edit_webhook_message
+    system_message: |
+      Available Discord Tools:
+      - Discord server interaction via MCP integration
+      - Message reading, sending, and management
+      - Channel and server information access
+
+  - id: "grok_discord_agent"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      mcp_servers:
+        - name: "discord"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "mcp-discord", "--config", "${DISCORD_TOKEN}"]
+          env:
+            DISCORD_TOKEN: "${DISCORD_TOKEN}"
+          security:
+            level: "high"
+      exclude_tools:
+        - mcp__discord__discord_send_webhook_message
+        - mcp__discord__discord_edit_webhook_message
+    system_message: |
+      Available Discord Tools:
+      - Discord server interaction via MCP integration
+      - Message reading, sending, and management
+      - Channel and server information access
+
+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
+  coordination:
+    enable_planning_mode: true
+    planning_mode_instruction: |
+      PLANNING MODE ACTIVE: You are currently in the coordination phase where agents discuss and plan their approach.
+      During this phase, you should:
+      1. Describe your intended actions and reasoning
+      2. Analyze other agents' proposals
+      3. Use only the 'vote' or 'new_answer' tools for coordination
+      4. DO NOT execute any actual Discord commands or API calls
+      5. Save tool execution for the final presentation phase when the winning agent will implement the plan
+
+      Focus on planning, analysis, and coordination rather than execution.

massgen/configs/tools/planning/five_agents_filesystem_mcp_planning_mode.yaml

@@ -0,0 +1,151 @@
+# Five Agent Filesystem MCP Test Configuration with Planning Mode
+# This configuration demonstrates planning vs execution mode using local filesystem tools
+# that require NO external API keys - perfect for testing and demonstration!
+#
+# Setup Requirements:
+# 1. No external API keys needed! Uses local filesystem MCP servers
+# 2. Set environment variables in .env file:
+#    OPENAI_API_KEY="your_openai_key" (for OpenAI agent)
+#    GOOGLE_API_KEY="your_gemini_key" (for Gemini agent)
+#    ANTHROPIC_API_KEY="your_claude_key" (for Claude agent)
+#    XAI_API_KEY="your_grok_key" (for Grok agent)
+#
+# Example usage:
+# massgen --config @examples/tools/planning/five_agents_filesystem_mcp_planning_mode "Create a project structure for a Python web app with directories for src, tests, docs, and config. Add some example files and then analyze the structure."
+
+agents:
+  - id: "gemini_filesystem_agent"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      mcp_servers:
+        - name: "filesystem"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-filesystem", "."]
+          security:
+            level: "high"
+        - name: "workspace_tools"
+          type: "stdio"
+          command: "fastmcp"
+          args: ["run", "massgen.filesystem_manager._workspace_tools_server:create_server", "--", "--allowed-paths", "."]
+          env:
+            FASTMCP_SHOW_CLI_BANNER: "false"
+    system_message: |
+      Available Filesystem Tools:
+      - Filesystem operations: read_file, write_file, create_directory, list_directory
+      - Workspace tools: copy_file, compare_files, delete_file, batch operations
+      - File analysis and management capabilities
+
+  - id: "openai_filesystem_agent"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      mcp_servers:
+        - name: "filesystem"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-filesystem", "."]
+          security:
+            level: "high"
+        - name: "workspace_tools"
+          type: "stdio"
+          command: "fastmcp"
+          args: ["run", "massgen.filesystem_manager._workspace_tools_server:create_server", "--", "--allowed-paths", "."]
+          env:
+            FASTMCP_SHOW_CLI_BANNER: "false"
+    system_message: |
+      Available Filesystem Tools:
+      - Filesystem operations: read_file, write_file, create_directory, list_directory
+      - Workspace tools: copy_file, compare_files, delete_file, batch operations
+      - File analysis and management capabilities
+
+  - id: "claude_code_filesystem_agent"
+    backend:
+      type: "claude_code"
+      cwd: "claude_code_workspace_filesystem_mcp"
+      permission_mode: "bypassPermissions"
+      mcp_servers:
+        filesystem:
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-filesystem", "."]
+        workspace_tools:
+          type: "stdio"
+          command: "fastmcp"
+          args: ["run", "massgen.filesystem_manager._workspace_tools_server:create_server", "--", "--allowed-paths", "."]
+          env:
+            FASTMCP_SHOW_CLI_BANNER: "false"
+    system_message: |
+      Available Filesystem Tools:
+      - Filesystem operations: read_file, write_file, create_directory, list_directory
+      - Workspace tools: copy_file, compare_files, delete_file, batch operations
+      - File analysis and management capabilities
+
+  - id: "claude_filesystem_agent"
+    backend:
+      type: "claude"
+      model: "claude-sonnet-4-20250514"
+      mcp_servers:
+        - name: "filesystem"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-filesystem", "."]
+          security:
+            level: "high"
+        - name: "workspace_tools"
+          type: "stdio"
+          command: "fastmcp"
+          args: ["run", "massgen.filesystem_manager._workspace_tools_server:create_server", "--", "--allowed-paths", "."]
+          env:
+            FASTMCP_SHOW_CLI_BANNER: "false"
+    system_message: |
+      Available Filesystem Tools:
+      - Filesystem operations: read_file, write_file, create_directory, list_directory
+      - Workspace tools: copy_file, compare_files, delete_file, batch operations
+      - File analysis and management capabilities
+
+  - id: "grok_filesystem_agent"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      mcp_servers:
+        - name: "filesystem"
+          type: "stdio"
+          command: "npx"
+          args: ["-y", "@modelcontextprotocol/server-filesystem", "."]
+          security:
+            level: "high"
+        - name: "workspace_tools"
+          type: "stdio"
+          command: "fastmcp"
+          args: ["run", "massgen.filesystem_manager._workspace_tools_server:create_server", "--", "--allowed-paths", "."]
+          env:
+            FASTMCP_SHOW_CLI_BANNER: "false"
+    system_message: |
+      Available Filesystem Tools:
+      - Filesystem operations: read_file, write_file, create_directory, list_directory
+      - Workspace tools: copy_file, compare_files, delete_file, batch operations
+      - File analysis and management capabilities
+
+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
+  coordination:
+    enable_planning_mode: true
+    planning_mode_instruction: |
+      PLANNING MODE ACTIVE: You are currently in the coordination phase where agents discuss and plan their approach.
+      During this phase, you should:
+      1. Describe your intended file operations and reasoning
+      2. Analyze other agents' proposals for the filesystem tasks
+      3. Use only the 'vote' or 'new_answer' tools for coordination
+      4. DO NOT execute any actual filesystem operations, file creation, or directory management
+      5. Save all file operations for the final presentation phase when the winning agent will implement the plan
+
+      Focus on planning, analysis, and coordination rather than execution.
+      Example: "I would create a 'src' directory and write a main.py file..." rather than actually creating them.