massgen 0.0.3__py3-none-any.whl

massgen/configs/README.md

@@ -0,0 +1,293 @@
+# MassGen Configuration Examples
+
+This directory contains sample configuration files for MassGen CLI usage. Each configuration is optimized for specific use cases and demonstrates different agent collaboration patterns.
+
+## 📁 Available Configurations
+
+### 🤖 Single Agent Configurations
+
+- **`single_agent.yaml`** - Basic single agent setup with Gemini
+  - Uses Gemini 2.5 Flash model
+  - Web search enabled for current information
+  - Rich terminal display with token usage tracking
+
+### 👥 Multi-Agent Configurations
+
+#### General Purpose Teams
+
+- **`three_agents_default.yaml`** - Default three-agent setup with frontier models
+  - **Gemini 2.5 Flash**: Web search enabled
+  - **GPT-4o-mini**: Web search and code interpreter enabled
+  - **Grok-3-mini**: Web search with citations
+  - Best for general questions requiring diverse perspectives
+
+- **`gemini_4o_claude.yaml`** - Premium three-agent configuration
+  - **Gemini 2.5 Flash**: Web search enabled
+  - **GPT-4o**: Full GPT-4o with web search and code interpreter
+  - **Claude 3.5 Haiku**: Web search with citations
+  - Higher quality responses for complex tasks
+
+- **`two_agents.yaml`** - Focused two-agent collaboration
+  - **Primary Agent (GPT-4o)**: Comprehensive research and analysis
+  - **Secondary Agent (GPT-4o-mini)**: Review and refinement
+  - Efficient for tasks needing depth with validation
+
+#### Specialized Teams
+
+- **`research_team.yaml`** - Academic/technical research configuration
+  - **Information Gatherer (Grok)**: Web search specialist
+  - **Domain Expert (GPT-4o)**: Deep analysis with code interpreter
+  - **Synthesizer (GPT-4o-mini)**: Integration and summarization
+  - Low temperature (0.3) for accuracy
+
+- **`creative_team.yaml`** - Creative writing and storytelling
+  - **Storyteller (GPT-4o)**: Narrative creation
+  - **Editor (GPT-4o-mini)**: Structure and flow refinement
+  - **Critic (Grok-3-mini)**: Literary analysis
+  - High temperature (0.8) for creativity
+
+- **`news_analysis.yaml`** - Current events and news synthesis
+  - **News Gatherer (GPT-4o)**: Finding current events
+  - **Trend Analyst (Grok-3-mini)**: Pattern identification
+  - **News Synthesizer (GPT-4o-mini)**: Balanced summaries
+  - Medium temperature (0.4) for balanced analysis
+
+- **`technical_analysis.yaml`** - Technical queries and cost estimation
+  - **Technical Researcher (GPT-4o)**: Specifications and documentation
+  - **Cost Analyst (Grok-3-mini)**: Pricing and cost calculations
+  - **Technical Advisor (GPT-4o-mini)**: Practical recommendations
+  - Low temperature (0.2) for precision
+
+- **`travel_planning.yaml`** - Travel recommendations and planning
+  - **Travel Researcher (GPT-4o)**: Destination information
+  - **Local Expert (Grok-3-mini)**: Insider knowledge
+  - **Travel Planner (GPT-4o-mini)**: Itinerary organization
+  - Medium temperature (0.6) for balanced suggestions
+
+## 🚀 Usage Examples
+
+### Single Agent Mode
+```bash
+# Using configuration file
+uv run python -m massgen.cli --config single_agent.yaml "What is machine learning?"
+
+# Quick setup without config file
+uv run python -m massgen.cli --model gemini-2.5-flash "Explain quantum computing"
+```
+
+### Multi-Agent Mode
+```bash
+# Default three agents for general questions
+uv run python -m massgen.cli --config three_agents_default.yaml "Compare renewable energy technologies"
+
+# Premium agents for complex analysis
+uv run python -m massgen.cli --config gemini_4o_claude.yaml "Analyze the implications of quantum computing on cryptography"
+
+# Specialized teams for specific tasks
+uv run python -m massgen.cli --config research_team.yaml "Latest developments in CRISPR gene editing"
+uv run python -m massgen.cli --config creative_team.yaml "Write a short story about AI consciousness"
+uv run python -m massgen.cli --config news_analysis.yaml "What happened in tech news this week?"
+uv run python -m massgen.cli --config technical_analysis.yaml "Cost analysis for running LLMs at scale"
+uv run python -m massgen.cli --config travel_planning.yaml "Plan a 5-day trip to Tokyo in spring"
+```
+
+### Interactive Mode
+```bash
+# Start interactive session with any configuration
+uv run python -m massgen.cli --config gemini_4o_claude.yaml
+
+# Commands in interactive mode:
+# /clear - Clear conversation history
+# /quit, /exit, /q - Exit the session
+```
+
+## 📋 Configuration Structure
+
+**Single Agent Configuration:**
+
+Use the `agent` field to define a single agent with its backend and settings:
+
+```yaml
+agent: 
+  id: "<agent_name>"
+  backend:
+    type: "claude" | "gemini" | "grok" | "openai" #Type of backend (Optional because we can infer backend type through model.)
+    model: "<model_name>" # Model name
+    api_key: "<optional_key>"  # API key for backend. Uses env vars by default.
+  system_message: "..."    # System Message for Single Agent
+```
+
+**Multi-Agent Configuration:**
+
+Use the `agents` field to define multiple agents, each with its own backend and config:
+
+```yaml
+agents:  # Multiple agents (alternative to 'agent')
+  - id: "<agent1 name>"
+    backend: 
+      type: "claude" | "gemini" | "grok" | "openai" #Type of backend (Optional because we can infer backend type through model.)
+      model: "<model_name>" # Model name
+      api_key: "<optional_key>"  # API key for backend. Uses env vars by default.
+    system_message: "..."    # System Message for Single Agent
+  - id: "..."
+    backend:
+      type: "..."
+      model: "..."
+      ...
+    system_message: "..."
+```
+
+**Backend Configuration:**
+
+Detailed parameters for each agent's backend can be specified using the following configuration formats:
+
+#### Claude
+
+```yaml
+backend:
+  type: "claude"
+  model: "claude-sonnet-4-20250514"  # Model name
+  api_key: "<optional_key>"          # API key for backend. Uses env vars by default.
+  temperature: 0.7                   # Creativity vs consistency (0.0-1.0)
+  max_tokens: 2500                   # Maximum response length
+  enable_web_search: true            # Web search capability
+  enable_code_execution: true        # Code execution capability
+```
+
+#### Gemini
+
+```yaml
+backend:
+  type: "gemini"
+  model: "gemini-2.5-flash"          # Model name
+  api_key: "<optional_key>"          # API key for backend. Uses env vars by default.
+  temperature: 0.7                   # Creativity vs consistency (0.0-1.0)
+  max_tokens: 2500                   # Maximum response length
+  enable_web_search: true            # Web search capability
+  enable_code_execution: true        # Code execution capability
+```
+
+#### Grok
+
+```yaml
+backend:
+  type: "grok"
+  model: "grok-3-mini"               # Model name
+  api_key: "<optional_key>"          # API key for backend. Uses env vars by default.
+  temperature: 0.7                   # Creativity vs consistency (0.0-1.0)
+  max_tokens: 2500                   # Maximum response length
+  enable_web_search: true            # Web search capability
+  return_citations: true             # Include search result citations
+  max_search_results: 10             # Maximum search results to use 
+  search_mode: "auto"                # Search strategy: "auto", "fast", "thorough" 
+```
+
+#### OpenAI
+
+```yaml
+backend:
+  type: "openai"
+  model: "gpt-4o"                    # Model name
+  api_key: "<optional_key>"          # API key for backend. Uses env vars by default.
+  temperature: 0.7                   # Creativity vs consistency (0.0-1.0, o-series models don't support this)
+  max_tokens: 2500                   # Maximum response length (o-series models don't support this)
+  enable_web_search: true            # Web search capability
+  enable_code_interpreter: true      # Code interpreter capability
+```
+
+**UI Configuration:**
+
+Configure how MassGen displays information and handles logging during execution:
+
+```yaml
+ui:
+  display_type: "rich_terminal" | "terminal" | "simple"  # Display format for agent interactions
+  logging_enabled: true | false                          # Enable/disable real-time logging 
+```
+
+- `display_type`: Controls the visual presentation of agent interactions
+  - `"rich_terminal"`: Full-featured display with multi-region layout, live status updates, and colored output
+  - `"terminal"`: Standard terminal display with basic formatting and sequential output
+  - `"simple"`: Plain text output without any formatting or special display features
+- `logging_enabled`: When `true`, saves detailed timestamp, agent outputs and system status
+
+**Advanced Parameters:**
+```yaml
+# Global backend parameters
+backend_params:
+  temperature: 0.7
+  max_tokens: 2000
+  enable_web_search: true  # Web search capability (all backends)
+  enable_code_interpreter: true  # OpenAI only
+  enable_code_execution: true    # Gemini/Claude only
+```
+
+## 🔧 Environment Variables
+
+Set these in your `.env` file:
+
+```bash
+# API Keys
+ANTHROPIC_API_KEY="your-anthropic-key"
+GEMINI_API_KEY="your-gemini-key"
+OPENAI_API_KEY="your-openai-key"
+XAI_API_KEY="your-xai-key"
+```
+
+## 💡 Backend Capabilities
+
+| Backend | Live Search | Code Execution |
+|---------|:-----------:|:--------------:|
+| **Claude** | ✅ | ✅ |
+| **OpenAI** | ✅ | ✅ |
+| **Grok** | ✅ | ❌ |
+| **Gemini** | ✅ | ✅ |
+
+## 📚 Best Practices
+
+1. **Choose the Right Configuration**
+   - Use `three_agents_default.yaml` for general questions
+   - Use specialized teams for domain-specific tasks
+   - Use `single_agent.yaml` for quick, simple queries
+
+2. **Temperature Settings**
+   - Low (0.1-0.3): Technical analysis, factual information
+   - Medium (0.4-0.6): Balanced tasks, general questions
+   - High (0.7-0.9): Creative writing, brainstorming
+
+3. **Cost Optimization**
+   - Use mini models (gpt-4o-mini, grok-3-mini) for routine tasks
+   - Reserve premium models (gpt-4o, claude-3-5-sonnet) for complex analysis
+   - Single agent mode is most cost-effective for simple queries
+
+4. **Tool Usage**
+   - Enable web search for current events and real-time information
+   - Use code interpreter for data analysis and calculations
+   - Combine tools for comprehensive research tasks
+
+## 🛠️ Creating Custom Configurations
+
+1. Copy an existing configuration as a template
+2. Modify agent roles and system messages for your use case
+3. Adjust temperature and max_tokens based on task requirements
+4. Enable/disable tools based on agent needs
+5. Test with sample queries to refine the configuration
+
+Example custom configuration for code review:
+```yaml
+agents:
+  - id: "code_analyzer"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      temperature: 0.2
+      enable_code_interpreter: true
+    system_message: "Analyze code for bugs, security issues, and best practices"
+    
+  - id: "refactoring_expert"
+    backend:
+      type: "claude"
+      model: "claude-3-5-sonnet-20250514"
+      temperature: 0.3
+    system_message: "Suggest code improvements and refactoring opportunities"
+```

massgen/configs/creative_team.yaml

@@ -0,0 +1,53 @@
+# MassGen Creative Team Configuration
+# Optimized for creative writing and storytelling tasks
+
+agents:
+  - id: "storyteller"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      temperature: 0.8
+      max_tokens: 3000
+    system_message: |
+      You are a creative storyteller who excels at crafting engaging narratives 
+      with rich characters, vivid descriptions, and compelling plot structures. 
+      Focus on:
+      - Character development and emotional depth
+      - Vivid imagery and sensory details
+      - Strong narrative arc and pacing
+      - Creative and original concepts
+
+  - id: "editor"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      temperature: 0.8
+      max_tokens: 3000
+    system_message: |
+      You are a skilled editor who evaluates creative works for structure, 
+      flow, and impact. You excel at identifying what makes a story compelling. 
+      Focus on:
+      - Narrative structure and coherence
+      - Character consistency and development
+      - Emotional resonance and engagement
+      - Overall polish and readability
+
+  - id: "critic"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      temperature: 0.8
+      max_tokens: 3000
+    system_message: |
+      You are a literary critic who analyzes creative works with a discerning eye. 
+      You evaluate originality, thematic depth, and artistic merit. Focus on:
+      - Originality and creativity
+      - Thematic depth and meaning
+      - Technical craft and execution
+      - Overall artistic impact
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+
+# Creative-optimized settings

massgen/configs/gemini_4o_claude.yaml

@@ -0,0 +1,31 @@
+# MassGen Three Agent Configuration
+# Gemini-2.5-flash, GPT-4o-mini, and Grok-3-mini with builtin tools enabled
+
+agents:
+  - id: "gemini2.5flash"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      enable_web_search: true
+      # enable_code_execution: true
+    # system_message: "You are a helpful AI assistant with web search and code execution capabilities. For any question involving current events, recent information, or real-time data, ALWAYS use web search first."
+
+  - id: "gpt-4o"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      enable_web_search: true
+      enable_code_interpreter: true
+    # system_message: "You are a helpful AI assistant with web search and code execution capabilities. For any question involving current events, recent information, or real-time data, ALWAYS use web search first."
+
+  - id: "claude-3-5-haiku"
+    backend:
+      type: "claude"
+      model: "claude-3-5-haiku-20241022"
+      enable_web_search: true
+      return_citations: true
+    # system_message: "You are a helpful AI assistant with web search capabilities. For any question involving current events, recent information, or real-time data, ALWAYS use web search first."
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/news_analysis.yaml

@@ -0,0 +1,51 @@
+# MassGen News Analysis Configuration
+# Specialized for current events and news synthesis
+
+agents:
+  - id: "news_gatherer"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      temperature: 0.4
+      max_tokens: 2500
+    system_message: |
+      You are a news researcher who specializes in finding and curating current 
+      events and breaking news. Focus on:
+      - Finding the most recent and relevant news
+      - Identifying authoritative news sources
+      - Covering diverse perspectives and angles
+      - Prioritizing high-impact and significant developments
+
+  - id: "trend_analyst"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      temperature: 0.4
+      max_tokens: 2500
+    system_message: |
+      You are a trend analyst who identifies patterns, significance, and 
+      implications of current events. Focus on:
+      - Analyzing the broader context and implications
+      - Identifying emerging trends and patterns
+      - Connecting related events and developments
+      - Assessing long-term significance
+
+  - id: "news_synthesizer"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      temperature: 0.4
+      max_tokens: 2500
+    system_message: |
+      You are a news editor who synthesizes multiple news sources into 
+      comprehensive, balanced summaries. Focus on:
+      - Creating coherent narratives from multiple sources
+      - Balancing different perspectives
+      - Organizing information by importance and relevance
+      - Providing clear, accessible summaries
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+
+# News-optimized settings

massgen/configs/research_team.yaml

@@ -0,0 +1,51 @@
+# MassGen Research Team Configuration
+# Specialized configuration for research and analysis tasks
+
+agents:
+  - id: "information_gatherer"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      enable_web_search: true
+    system_message: |
+      You are an information gathering specialist with access to web search. 
+      Your role is to find the most current and relevant information on any topic. 
+      Focus on:
+      - Finding authoritative sources
+      - Gathering diverse perspectives
+      - Identifying recent developments
+      - Providing comprehensive coverage of the topic
+
+  - id: "domain_expert"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      enable_code_interpreter: true
+    system_message: |
+      You are a domain expert with analytical capabilities. Your role is to 
+      provide deep subject matter expertise and analytical insights. Focus on:
+      - Technical accuracy and depth
+      - Identifying key concepts and relationships
+      - Providing expert-level analysis
+      - Validating information for accuracy
+
+  - id: "synthesizer"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+    system_message: |
+      You are a synthesis specialist who excels at combining multiple perspectives 
+      into coherent, actionable insights. Your role is to:
+      - Integrate findings from multiple sources
+      - Identify patterns and themes
+      - Create clear, structured summaries
+      - Highlight practical implications and recommendations
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+
+# Research-optimized settings
+backend_params:
+  temperature: 0.3  # Lower temperature for more focused research
+  max_tokens: 3000  # Longer responses for detailed research

massgen/configs/single_agent.yaml

@@ -0,0 +1,18 @@
+# Example Gemini configuration for MassGen
+# Usage: python -m massgen.cli --config example_gemini_config.yaml "Your question here"
+
+# Single agent configuration
+agent:
+  id: "gemini_agent"
+  # system_message: "You are a helpful AI assistant powered by Google Gemini."
+  backend:
+    type: "gemini"
+    model: "gemini-2.5-flash"
+    enable_web_search: true
+    # enable_code_execution: true
+  system_message: "You are a helpful assistant"
+
+# Display configuration
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/single_flash2.5.yaml

@@ -0,0 +1,44 @@
+# Example Gemini configuration for MassGen 
+# Usage: python -m massgen.cli --config example_gemini_config.yaml "Your question here"
+
+# Single agent configuration
+agent:
+  id: "gemini_agent"
+  # system_message: "You are a helpful AI assistant powered by Google Gemini."
+  backend:
+    type: "gemini"
+    model: "gemini-2.5-flash"
+    enable_web_search: true
+    # enable_code_execution: true
+
+# Alternative: Multi-agent configuration
+# agents:
+#   research_agent:
+#     system_message: "You are a research specialist. Use web search for current information."
+#     backend:
+#       type: "gemini"
+#       model: "gemini-2.5-flash"
+#       enable_web_search: true
+#       enable_code_execution: false
+#   
+#   compute_agent:
+#     system_message: "You are a computational specialist. Use code execution for calculations."
+#     backend:
+#       type: "gemini"  
+#       model: "gemini-2.5-flash"
+#       enable_web_search: false
+#       enable_code_execution: true
+#   
+#   analyst_agent:
+#     system_message: "You are a general analyst providing clear reasoning."
+#     backend:
+#       type: "gemini"
+#       model: "gemini-2.5-flash"
+#       enable_web_search: false
+#       enable_code_execution: false
+
+# Display configuration
+display:
+  type: "rich_terminal"
+  show_agent_details: true
+  show_token_usage: true

massgen/configs/technical_analysis.yaml

@@ -0,0 +1,51 @@
+# MassGen Technical Analysis Configuration
+# Optimized for technical queries and cost estimation
+
+agents:
+  - id: "technical_researcher"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      temperature: 0.2
+      max_tokens: 3000
+    system_message: |
+      You are a technical researcher who specializes in finding accurate 
+      technical specifications, pricing, and system requirements. Focus on:
+      - Finding official documentation and specifications
+      - Gathering accurate pricing and cost information
+      - Understanding technical requirements and constraints
+      - Providing detailed technical analysis
+
+  - id: "cost_analyst"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      temperature: 0.2
+      max_tokens: 3000
+    system_message: |
+      You are a cost analyst who excels at calculating and estimating 
+      technical costs and resource requirements. Focus on:
+      - Accurate cost calculations and estimations
+      - Understanding pricing models and structures
+      - Identifying cost variables and factors
+      - Providing detailed cost breakdowns
+
+  - id: "technical_advisor"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      temperature: 0.2
+      max_tokens: 3000
+    system_message: |
+      You are a technical advisor who synthesizes technical information 
+      into practical recommendations and insights. Focus on:
+      - Translating technical details into practical advice
+      - Identifying key considerations and trade-offs
+      - Providing clear recommendations
+      - Highlighting important caveats and limitations
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+
+# Technical analysis settings

massgen/configs/three_agents_default.yaml

@@ -0,0 +1,31 @@
+# MassGen Three Agent Configuration
+# Gemini-2.5-flash, GPT-4o-mini, and Grok-3-mini with builtin tools enabled
+
+agents:
+  - id: "gemini2.5flash"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      enable_web_search: true
+      # enable_code_execution: true
+    # system_message: "You are a helpful AI assistant with web search and code execution capabilities. For any question involving current events, recent information, or real-time data, ALWAYS use web search first."
+
+  - id: "4omini"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      enable_web_search: true
+      enable_code_interpreter: true
+    # system_message: "You are a helpful AI assistant with web search and code execution capabilities. For any question involving current events, recent information, or real-time data, ALWAYS use web search first."
+
+  - id: "grok3mini"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      enable_web_search: true
+      return_citations: true
+    # system_message: "You are a helpful AI assistant with web search capabilities. For any question involving current events, recent information, or real-time data, ALWAYS use web search first."
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true

massgen/configs/travel_planning.yaml

@@ -0,0 +1,51 @@
+# MassGen Travel Planning Configuration
+# Specialized for travel recommendations and destination guides
+
+agents:
+  - id: "travel_researcher"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      temperature: 0.6
+      max_tokens: 3000
+    system_message: |
+      You are a travel researcher who excels at finding comprehensive 
+      information about destinations, attractions, and travel logistics. Focus on:
+      - Finding current attractions, events, and activities
+      - Gathering practical travel information
+      - Identifying seasonal considerations and timing
+      - Researching local culture and customs
+
+  - id: "local_expert"
+    backend:
+      type: "grok"
+      model: "grok-3-mini"
+      temperature: 0.6
+      max_tokens: 3000
+    system_message: |
+      You are a local travel expert who provides insider knowledge 
+      and authentic recommendations for destinations. Focus on:
+      - Hidden gems and local favorites
+      - Practical tips and insider knowledge
+      - Cultural insights and local customs
+      - Authentic experiences beyond tourist attractions
+
+  - id: "travel_planner"
+    backend:
+      type: "openai"
+      model: "gpt-4o-mini"
+      temperature: 0.6
+      max_tokens: 3000
+    system_message: |
+      You are a travel planner who organizes information into practical, 
+      actionable travel itineraries and recommendations. Focus on:
+      - Creating well-structured travel plans
+      - Organizing activities by location and timing
+      - Providing practical logistics and tips
+      - Balancing must-see attractions with local experiences
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+
+# Travel planning settings