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

massgen/configs/ag2/ag2_case_study.yaml

@@ -0,0 +1,27 @@
+# Example configuration for using AG2 with MassGen
+# Single AG2 agent example
+# uv run python -m massgen.cli --config ag2/ag2_case_study.yaml "Compare the differences between AG2 and MassGen for LLM agents"
+agents:
+  - id: "ag2_assistant"
+    backend:
+      type: ag2
+      agent_config:
+        type: assistant
+        name: "AG2_Assistant"
+        system_message: "You are a helpful AI assistant powered by AG2."
+        llm_config:
+          - api_type: "anthropic"
+            model: "claude-sonnet-4-20250514"
+            temperature: 0.3
+            cache_seed: 42
+
+  - id: "gemini2.5pro"
+    backend:
+        type: "gemini"
+        model: "gemini-2.5-pro"
+        enable_web_search: true
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/ag2/ag2_coder.yaml

@@ -0,0 +1,34 @@
+# Example configuration for using AG2 with MassGen
+# Single AG2 agent example
+# uv run python -m massgen.cli --config ag2/ag2_coder.yaml "Create a factorial function and calculate the factorial of 8. Show the result?"
+agents:
+  - id: "ag2_coder"
+    backend:
+      type: ag2
+      agent_config:
+        type: assistant
+        name: "AG2_coder"
+        system_message: |
+          You are a helpful coding assistant. When asked to create and run code:
+          1. Write Python code in a markdown code block (```python ... ```)
+          2. The code will be automatically executed
+          3. Always print the results so they are visible
+
+          For factorial calculations:
+          - Define the factorial function
+          - Call it with the provided number
+          - Print the result clearly
+        llm_config:
+          api_type: "openai"
+          model: "gpt-5"
+        code_execution_config:
+            executor:
+              type: "LocalCommandLineCodeExecutor"
+              timeout: 60
+              work_dir: "./code_execution_workspace"
+
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/ag2/ag2_coder_case_study.yaml

@@ -0,0 +1,36 @@
+# Example configuration for using AG2 with MassGen
+# Single AG2 agent example
+# uv run python -m massgen.cli --config ag2/ag2_coder_case_study.yaml "Output a summary comparing the differences between AG2 (https://github.com/ag2ai/ag2) and MassGen (https://github.com/Leezekun/MassGen) for LLM agents."
+agents:
+  - id: "ag2_coder"
+    backend:
+      type: ag2
+      agent_config:
+        type: assistant
+        name: "AG2_coder"
+        system_message: |
+          You are a helpful coding assistant. When asked to create and run code:
+          1. Write Python code in a markdown code block (```python ... ```)
+          2. The code will be automatically executed
+          3. Always print the results so they are visible
+
+          If you need to access information from the web, create and run a web scraping script.
+        llm_config:
+          api_type: "openai"
+          model: "gpt-5"
+        code_execution_config:
+            executor:
+              type: "LocalCommandLineCodeExecutor"
+              timeout: 60
+              work_dir: "./code_execution_workspace"
+
+  - id: "gemini2.5pro"
+    backend:
+        type: "gemini"
+        model: "gemini-2.5-pro"
+        enable_web_search: true
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/ag2/ag2_gemini.yaml

@@ -0,0 +1,27 @@
+# Example configuration for using AG2 with MassGen
+# Single AG2 agent example
+# uv run python -m massgen.cli --config ag2/ag2_gemini.yaml "what is quantum computing?"
+agents:
+  - id: "ag2_assistant"
+    backend:
+      type: ag2
+      agent_config:
+        type: assistant
+        name: "AG2_Assistant"
+        system_message: "You are a helpful AI assistant powered by AG2."
+        llm_config:
+          - api_type: "anthropic"
+            model: "claude-sonnet-4-20250514"
+            temperature: 0.7
+            cache_seed: 42
+
+  - id: "gemini2.5flash"
+    backend:
+        type: "gemini"
+        model: "gemini-2.5-flash"
+        enable_web_search: true
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/ag2/ag2_groupchat.yaml

@@ -0,0 +1,108 @@
+# Example configuration for using AG2 GroupChat with MassGen
+# The entire GroupChat acts as a single agent in MassGen's orchestration
+ # uv run python -m massgen.cli --config ag2/ag2_groupchat.yaml "Write a Python function to calculate factorial."
+agents:
+  - id: "ag2_team"
+    backend:
+      type: ag2
+      group_config:
+        # Default llm_config for all agents (REQUIRED)
+        # Individual agents can override this if needed
+        llm_config:
+            api_type: "openai"
+            model: "gpt-5"
+
+        agents:
+          - type: assistant
+            name: "Coder"
+            system_message: "You are an expert programmer who writes clean, efficient code."
+            # Uses default llm_config from group_config
+
+          - type: assistant
+            name: "Reviewer"
+            description: "Code reviewer who provides constructive feedback. Should be selected after Coder"
+            system_message: |
+              You are a code reviewer who provides constructive feedback. You role is to:
+                1. Review the code for correctness, efficiency, and style
+                2. Suggest improvements and optimizations
+                3. DO NOT write complete to address original request
+                4. DO NOT write test code
+                5. When you think the code is good enough to submit, say "LGTM" (Looks Good To Me)
+            # Override with different model
+            llm_config:
+                api_type: "google"
+                model: "gemini-2.5-flash"
+
+          - type: assistant
+            name: "Tester"
+            description: "QA engineer who writes and runs tests. Should be selected if new code has not been validated."
+            system_message: |
+              You are a QA engineer. Your role is to:
+              1. Write test code in Python that validates the provided function
+              2. Test edge cases (empty input, negative numbers, large values, etc.)
+              3. Execute the tests using code blocks
+              4. Report results and any failures
+
+              IMPORTANT: Always write executable Python code in markdown code blocks like:
+              ```python
+              # your test code here
+              ```
+
+              Do NOT just repeat the function code. Write NEW test code that calls the function.
+            llm_config:
+                api_type: "google"
+                model: "gemini-2.5-flash"
+            code_execution_config:
+              executor:
+                type: "LocalCommandLineCodeExecutor"
+                timeout: 60
+                work_dir: "./code_execution_workspace"
+
+        # Pattern configuration (REQUIRED)
+        # Determines how agents are selected to speak in the group chat
+        pattern:
+          type: "auto"  # Currently supported: "auto" only
+          # Required: Name of the agent that starts the conversation
+          initial_agent: "Coder"
+
+          # Optional: Additional pattern-specific arguments
+          # For AutoPattern, supported arguments include:
+          #   - exclude_transit_message (bool): Hide transit messages
+          #   - summary_method (str): How to summarize conversation
+          # Any additional arguments are passed directly to the pattern constructor
+
+          # Group manager configuration
+          group_manager_args:
+            # Optional: override llm_config for the group manager
+            # If not provided, uses default llm_config from group_config
+            llm_config:
+                api_type: "google"
+                model: "gemini-2.5-flash"
+
+        # Optional: Override the default user agent behavior. Not recommended to change.
+        # By default, a minimal user_agent is created automatically
+        # You can provide custom configuration here.
+        # However, it plays an important role in terminating the chat, so it shouldn't be changed ideally.
+        # user_agent:
+        #   name: "User"  # Must be "User" for termination to work
+        #   system_message: "MUST say 'TERMINATE' when the original request is well answered. Do NOT do anything else."
+        #   description: "MUST ONLY be selected when the original request is well answered and the conversation should terminate."
+        #   llm_config:  # Optional: override llm_config
+        #     api_type: "openai"
+        #     model: "gpt-4o"
+
+        # Maximum rounds of conversation before termination
+        max_rounds: 10
+
+#  # Optional: Mix with native MassGen agents
+#  - id: "claude_architect"
+#    backend:
+#      type: claude
+#      model: claude-3-opus-20240229
+#      temperature: 0.7
+#      system_message: "You are a software architect who designs scalable systems."
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/ag2/ag2_groupchat_gpt.yaml

@@ -0,0 +1,118 @@
+# Example configuration for using AG2 GroupChat with MassGen
+# The entire GroupChat acts as a single agent in MassGen's orchestration
+ # uv run python -m massgen.cli --config ag2/ag2_groupchat_gpt.yaml "Write a Python function to calculate factorial."
+agents:
+  - id: "gpt-5-nano"
+    system_message: "You are an expert programmer who writes clean, efficient code."
+    backend:
+      type: "openai"
+      model: "gpt-5-nano"
+      text:
+        verbosity: "medium"
+      reasoning:
+        effort: "medium"
+
+  - id: "ag2_team"
+    backend:
+      type: ag2
+      group_config:
+        # Default llm_config for all agents (REQUIRED)
+        # Individual agents can override this if needed
+        llm_config:
+            api_type: "openai"
+            model: "gpt-5"
+
+        agents:
+          - type: assistant
+            name: "Coder"
+            system_message: "You are an expert programmer who writes clean, efficient code."
+            # Uses default llm_config from group_config
+
+          - type: assistant
+            name: "Reviewer"
+            description: "Code reviewer who provides constructive feedback. Should be selected after Coder"
+            system_message: |
+              You are a code reviewer who provides constructive feedback. You role is to:
+                1. Review the code for correctness, efficiency, and style
+                2. Suggest improvements and optimizations
+                3. DO NOT write complete to address original request
+                4. DO NOT write test code
+                5. When you think the code is good enough to submit, say "LGTM" (Looks Good To Me)
+            # Override with different model
+            llm_config:
+                api_type: "google"
+                model: "gemini-2.5-flash"
+
+          - type: assistant
+            name: "Tester"
+            description: "QA engineer who writes and runs tests. Should be selected if new code has not been validated."
+            system_message: |
+              You are a QA engineer. Your role is to:
+              1. Write test code in Python that validates the provided function
+              2. Test edge cases (empty input, negative numbers, large values, etc.)
+              3. Execute the tests using code blocks
+              4. Report results and any failures
+
+              IMPORTANT: Always write executable Python code in markdown code blocks like:
+              ```python
+              # your test code here
+              ```
+
+              Do NOT just repeat the function code. Write NEW test code that calls the function.
+            llm_config:
+                api_type: "google"
+                model: "gemini-2.5-flash"
+            code_execution_config:
+              executor:
+                type: "LocalCommandLineCodeExecutor"
+                timeout: 60
+                work_dir: "./code_execution_workspace"
+
+        # Pattern configuration (REQUIRED)
+        # Determines how agents are selected to speak in the group chat
+        pattern:
+          type: "auto"  # Currently supported: "auto" only
+          # Required: Name of the agent that starts the conversation
+          initial_agent: "Coder"
+
+          # Optional: Additional pattern-specific arguments
+          # For AutoPattern, supported arguments include:
+          #   - exclude_transit_message (bool): Hide transit messages
+          #   - summary_method (str): How to summarize conversation
+          # Any additional arguments are passed directly to the pattern constructor
+
+          # Group manager configuration
+          group_manager_args:
+            # Optional: override llm_config for the group manager
+            # If not provided, uses default llm_config from group_config
+            llm_config:
+                api_type: "google"
+                model: "gemini-2.5-flash"
+
+        # Optional: Override the default user agent behavior
+        # By default, a minimal user_agent is created automatically
+        # You can provide custom configuration here.- agents:
+        # However, it plays an important role in terminating the chat, so it shouldn't be changed ideally1
+        # user_agent:
+        #   name: "User"  # Must be "User" for termination to work
+        #   system_message: "MUST say 'TERMINATE' when the original request is well answered. Do NOT do anything else."
+        #   description: "MUST ONLY be selected when the original request is well answered and the conversation should terminate."
+        #   llm_config:  # Optional: override llm_config
+        #     api_type: "openai"
+        #     model: "gpt-4o"
+
+        # Maximum rounds of conversation before termination
+        max_rounds: 10
+
+#  # Optional: Mix with native MassGen agents
+#  - id: "claude_architect"
+#    backend:
+#      type: claude
+#      model: claude-3-opus-20240229
+#      temperature: 0.7
+#      system_message: "You are a software architect who designs scalable systems."
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/ag2/ag2_single_agent.yaml

@@ -0,0 +1,21 @@
+# Example configuration for using AG2 with MassGen
+# Single AG2 agent example
+# uv run python -m massgen.cli --config ag2/ag2_single_agent.yaml "what is quantum computing?"
+agents:
+  - id: "ag2_assistant"
+    backend:
+      type: ag2
+      agent_config:
+        type: assistant
+        name: "AG2_Assistant"
+        system_message: "You are a helpful AI assistant powered by AG2."
+        llm_config:
+            api_type: "openai"
+            model: "gpt-4o"
+            temperature: 0.7
+            cache_seed: 42
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/basic/multi/fast_timeout_example.yaml

@@ -0,0 +1,37 @@
+# MassGen Fast Configuration with Conservative Timeouts
+# For cost-conscious users who want to prevent runaway usage
+
+# Conservative timeout settings to prevent excessive token usage
+timeout_settings:
+  orchestrator_timeout_seconds: 30   # 30 seconds max coordination
+
+agents:
+  - id: "fast-gpt-5-1"
+    backend:
+      type: "openai"
+      model: "gpt-5"
+      text: 
+        verbosity: "medium"
+      reasoning:
+        effort: "high"
+        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. For any question involving current events, recent information, or real-time data, ALWAYS use web search first."
+  - id: "fast-gpt-5-2"
+    backend:
+      type: "openai"
+      model: "gpt-5"
+      text: 
+        verbosity: "medium"
+      reasoning:
+        effort: "high"
+        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. 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/basic/multi/gemini_4o_claude.yaml

@@ -0,0 +1,31 @@
+# MassGen Three Agent Configuration
+# Gemini-2.5-flash, GPT-4o, and Claude-3.5-Haiku with builtin tools enabled
+
+agents:
+  - id: "gemini2.5flash"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      enable_web_search: true
+      # enable_code_execution: true  # Disabled by default - can preempt web search, resulting in weaker search capability
+    # 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
+      enable_code_execution: 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/basic/multi/gemini_gpt5nano_claude.yaml

@@ -0,0 +1,36 @@
+# MassGen Three Agent Configuration
+# Gemini-2.5-flash, GPT-5-nano, and Claude-3.5-Haiku with builtin tools enabled
+
+agents:
+  - id: "gemini2.5flash"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      enable_web_search: true
+      # enable_code_execution: true  # Disabled by default - can preempt web search, resulting in weaker search capability
+    # 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-5-nano"
+    backend:
+      type: "openai"
+      model: "gpt-5-nano"
+      text:
+        verbosity: "medium"
+      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. 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
+      enable_code_execution: 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/{gemini_4o_claude.yaml → basic/multi/geminicode_4o_claude.yaml}

@@ -1,5 +1,5 @@
 # MassGen Three Agent Configuration
-# Gemini-2.5-flash, GPT-4o-mini, and Grok-3-mini with builtin tools enabled
+# Gemini-2.5-flash, GPT-4o, and claude-3-5-haiku-20241022 with builtin tools enabled
 
 agents:
   - id: "gemini2.5flash"
@@ -7,7 +7,7 @@ agents:
       type: "gemini"
       model: "gemini-2.5-flash"
       enable_web_search: true
-      # enable_code_execution: 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"
@@ -23,7 +23,7 @@ agents:
       type: "claude"
       model: "claude-3-5-haiku-20241022"
       enable_web_search: true
-      return_citations: true
+      enable_code_execution: 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:

massgen/configs/basic/multi/geminicode_gpt5nano_claude.yaml

@@ -0,0 +1,36 @@
+# MassGen Three Agent Configuration
+# Gemini-2.5-flash, GPT-5-nano, and Claude-3.5-Haiku 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-5-nano"
+    backend:
+      type: "openai"
+      model: "gpt-5-nano"
+      text:
+        verbosity: "medium"
+      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. 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
+      enable_code_execution: 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/basic/multi/glm_gemini_claude.yaml

@@ -0,0 +1,25 @@
+agents:
+  - id: "glm-4.5"
+    backend:
+      type: "zai"
+      model: "glm-4.5"
+      base_url: "https://api.z.ai/api/paas/v4/"
+
+  - id: "gemini-2.5-flash"
+    backend:
+      type: "gemini"
+      model: "gemini-2.5-flash"
+      enable_web_search: true
+
+  - id: "claude-3.5-haiku"
+    backend:
+      type: "claude"
+      model: "claude-3-5-haiku-20241022"
+      enable_web_search: true
+      enable_code_execution: true
+
+
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+

massgen/configs/basic/multi/gpt4o_audio_generation.yaml

@@ -0,0 +1,30 @@
+# Example configuration demonstrating context_paths filesystem access control
+# This config shows how agents can access user-specified files with permission control
+# uv run python -m massgen.cli --config massgen/configs/basic/multi/gpt4o_audio_generation.yaml "I want to you tell me a very short introduction about Sherlock Homes in one sentence, and I want you to use emotion voice to read it out loud."
+agents:
+  - id: "gpt4o_1"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      text:
+        verbosity: "medium"
+      cwd: "workspace1"
+      enable_audio_generation: true  # Enable audio generation (currently not implemented)
+
+  - id: "gpt4o_2"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      text:
+        verbosity: "medium"
+      cwd: "workspace2"
+      enable_audio_generation: true  # Enable audio generation (currently not implemented)
+
+orchestrator:
+    snapshot_storage: "snapshots"
+    agent_temporary_workspace: "temp_workspaces"
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/basic/multi/gpt4o_image_generation.yaml

@@ -0,0 +1,31 @@
+# Example configuration demonstrating context_paths filesystem access control
+# This config shows how agents can access user-specified files with permission control
+# massgen --config @examples/basic/multi/gpt4o_image_generation "Generate an image of gray tabby cat hugging an otter with an orange scarf."
+
+agents:
+  - id: "gpt4o_1"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      text:
+        verbosity: "medium"
+      cwd: "workspace1"
+      enable_image_generation: true  # Enable image generation tools
+
+  - id: "gpt4o_2"
+    backend:
+      type: "openai"
+      model: "gpt-4o"
+      text:
+        verbosity: "medium"
+      cwd: "workspace2"
+      enable_image_generation: true  # Enable image generation tools
+
+orchestrator:
+    snapshot_storage: "snapshots"
+    agent_temporary_workspace: "temp_workspaces"
+
+# UI Configuration
+ui:
+  type: "rich_terminal"
+  logging_enabled: true

massgen/configs/basic/multi/gpt5nano_glm_qwen.yaml

@@ -0,0 +1,26 @@
+agents:
+  - id: "gpt-oss-1" # Cerebras AI  
+    backend:
+      type: "chatcompletion"
+      model: "gpt-oss-120b"
+      base_url: "https://api.cerebras.ai/v1"
+      # api_key: "cerebras_api_key"
+
+  - id: "zai_glm45_agent"
+    backend:
+      type: "zai"
+      model: "glm-4.5-air"
+      base_url: "https://api.z.ai/api/paas/v4/"
+      temperature: 0.7
+      top_p: 0.7
+
+  - id: "Qwen3-4b"
+    backend:
+      type: "lmstudio"
+      model: "qwen/qwen3-4b-2507"
+      # api_key: "lm-studio"
+  
+ui:
+  display_type: "rich_terminal"
+  logging_enabled: true
+