massgen 0.1.0a2__py3-none-any.whl → 0.1.0a3__py3-none-any.whl

massgen/__init__.py

@@ -68,7 +68,7 @@ from .chat_agent import (
 from .message_templates import MessageTemplates, get_templates
 from .orchestrator import Orchestrator, create_orchestrator
 
-__version__ = "0.1.0a2"
+__version__ = "0.1.0a3"
 __author__ = "MassGen Contributors"
 
 

massgen/cli.py

@@ -291,29 +291,33 @@ def create_backend(backend_type: str, **kwargs) -> Any:
     if backend_type == "openai":
         api_key = kwargs.get("api_key") or os.getenv("OPENAI_API_KEY")
         if not api_key:
-            print("⚠️  Warning: OpenAI API key not found. Set OPENAI_API_KEY environment variable or add to .env file.", flush=True)
-            print("   .env file locations: current directory, or ~/.massgen/.env", flush=True)
+            raise ConfigurationError(
+                "OpenAI API key not found. Set OPENAI_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+            )
         return ResponseBackend(api_key=api_key, **kwargs)
 
     elif backend_type == "grok":
         api_key = kwargs.get("api_key") or os.getenv("XAI_API_KEY")
         if not api_key:
-            print("⚠️  Warning: Grok API key not found. Set XAI_API_KEY environment variable or add to .env file.", flush=True)
-            print("   .env file locations: current directory, or ~/.massgen/.env", flush=True)
+            raise ConfigurationError(
+                "Grok API key not found. Set XAI_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+            )
         return GrokBackend(api_key=api_key, **kwargs)
 
     elif backend_type == "claude":
         api_key = kwargs.get("api_key") or os.getenv("ANTHROPIC_API_KEY")
         if not api_key:
-            print("⚠️  Warning: Claude API key not found. Set ANTHROPIC_API_KEY environment variable or add to .env file.", flush=True)
-            print("   .env file locations: current directory, or ~/.massgen/.env", flush=True)
+            raise ConfigurationError(
+                "Claude API key not found. Set ANTHROPIC_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+            )
         return ClaudeBackend(api_key=api_key, **kwargs)
 
     elif backend_type == "gemini":
         api_key = kwargs.get("api_key") or os.getenv("GOOGLE_API_KEY") or os.getenv("GEMINI_API_KEY")
         if not api_key:
-            print("⚠️  Warning: Gemini API key not found. Set GOOGLE_API_KEY environment variable or add to .env file.", flush=True)
-            print("   .env file locations: current directory, or ~/.massgen/.env", flush=True)
+            raise ConfigurationError(
+                "Gemini API key not found. Set GOOGLE_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+            )
         return GeminiBackend(api_key=api_key, **kwargs)
 
     elif backend_type == "chatcompletion":
@@ -325,43 +329,81 @@ def create_backend(backend_type: str, **kwargs) -> Any:
             if base_url and "cerebras.ai" in base_url:
                 api_key = os.getenv("CEREBRAS_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("Cerebras AI API key not found. Set CEREBRAS_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "Cerebras AI API key not found. Set CEREBRAS_API_KEY environment variable.\n"
+                        "You can add it to a .env file in:\n"
+                        "  - Current directory: .env\n"
+                        "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and "together.xyz" in base_url:
                 api_key = os.getenv("TOGETHER_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("Together AI API key not found. Set TOGETHER_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "Together AI API key not found. Set TOGETHER_API_KEY environment variable.\n"
+                        "You can add it to a .env file in:\n"
+                        "  - Current directory: .env\n"
+                        "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and "fireworks.ai" in base_url:
                 api_key = os.getenv("FIREWORKS_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("Fireworks AI API key not found. Set FIREWORKS_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "Fireworks AI API key not found. Set FIREWORKS_API_KEY environment variable.\n"
+                        "You can add it to a .env file in:\n"
+                        "  - Current directory: .env\n"
+                        "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and "groq.com" in base_url:
                 api_key = os.getenv("GROQ_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("Groq API key not found. Set GROQ_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "Groq API key not found. Set GROQ_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and "nebius.com" in base_url:
                 api_key = os.getenv("NEBIUS_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("Nebius AI Studio API key not found. Set NEBIUS_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "Nebius AI Studio API key not found. Set NEBIUS_API_KEY environment variable.\n"
+                        "You can add it to a .env file in:\n"
+                        "  - Current directory: .env\n"
+                        "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and "openrouter.ai" in base_url:
                 api_key = os.getenv("OPENROUTER_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("OpenRouter API key not found. Set OPENROUTER_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "OpenRouter API key not found. Set OPENROUTER_API_KEY environment variable.\n"
+                        "You can add it to a .env file in:\n"
+                        "  - Current directory: .env\n"
+                        "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and ("z.ai" in base_url or "bigmodel.cn" in base_url):
                 api_key = os.getenv("ZAI_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("ZAI API key not found. Set ZAI_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "ZAI API key not found. Set ZAI_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and ("moonshot.ai" in base_url or "moonshot.cn" in base_url):
                 api_key = os.getenv("MOONSHOT_API_KEY") or os.getenv("KIMI_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("Kimi/Moonshot API key not found. Set MOONSHOT_API_KEY or KIMI_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "Kimi/Moonshot API key not found. Set MOONSHOT_API_KEY or KIMI_API_KEY environment variable.\n"
+                        "You can add it to a .env file in:\n"
+                        "  - Current directory: .env\n"
+                        "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and "poe.com" in base_url:
                 api_key = os.getenv("POE_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("POE API key not found. Set POE_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "POE API key not found. Set POE_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+                    )
             elif base_url and "aliyuncs.com" in base_url:
                 api_key = os.getenv("QWEN_API_KEY")
                 if not api_key:
-                    raise ConfigurationError("Qwen API key not found. Set QWEN_API_KEY or provide in config.")
+                    raise ConfigurationError(
+                        "Qwen API key not found. Set QWEN_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+                    )
 
         return ChatCompletionsBackend(api_key=api_key, **kwargs)
 
@@ -370,7 +412,9 @@ def create_backend(backend_type: str, **kwargs) -> Any:
         # Supports both global (z.ai) and China (bigmodel.cn) endpoints
         api_key = kwargs.get("api_key") or os.getenv("ZAI_API_KEY")
         if not api_key:
-            raise ConfigurationError("ZAI API key not found. Set ZAI_API_KEY or provide in config.")
+            raise ConfigurationError(
+                "ZAI API key not found. Set ZAI_API_KEY environment variable.\n" "You can add it to a .env file in:\n" "  - Current directory: .env\n" "  - Global config: ~/.massgen/.env",
+            )
         return ChatCompletionsBackend(api_key=api_key, **kwargs)
 
     elif backend_type == "lmstudio":
@@ -1219,13 +1263,13 @@ async def run_interactive_mode(
     rich_console.clear()
 
     # ASCII art for interactive multi-agent mode
-    ascii_art = """[bold cyan]
+    ascii_art = """[bold #4A90E2]
      ███╗   ███╗ █████╗ ███████╗███████╗ ██████╗ ███████╗███╗   ██╗
      ████╗ ████║██╔══██╗██╔════╝██╔════╝██╔════╝ ██╔════╝████╗  ██║
      ██╔████╔██║███████║███████╗███████╗██║  ███╗█████╗  ██╔██╗ ██║
      ██║╚██╔╝██║██╔══██║╚════██║╚════██║██║   ██║██╔══╝  ██║╚██╗██║
      ██║ ╚═╝ ██║██║  ██║███████║███████║╚██████╔╝███████╗██║ ╚████║
-     ╚═╝     ╚═╝╚═╝  ╚═╝╚══════╝╚══════╝ ╚═════╝ ╚══════╝╚═╝  ╚═══╝[/bold cyan]
+     ╚═╝     ╚═╝╚═╝  ╚═╝╚══════╝╚══════╝ ╚═════╝ ╚══════╝╚═╝  ╚═══╝[/bold #4A90E2]
 
      [dim]     🤖 🤖 🤖  →  💬 collaborate  →  🎯 winner  →  📢 final[/dim]
 """
@@ -1233,7 +1277,7 @@ async def run_interactive_mode(
     # Wrap ASCII art in a panel
     ascii_panel = Panel(
         ascii_art,
-        border_style="bold cyan",
+        border_style="bold #4A90E2",
         padding=(0, 2),
         width=80,
     )

massgen/config_builder.py

@@ -33,11 +33,11 @@ load_dotenv()
 # Custom theme for the CLI - using colors that work on both light and dark backgrounds
 custom_theme = Theme(
     {
-        "info": "bright_blue",
-        "warning": "bright_yellow",
-        "error": "bright_red bold",
-        "success": "bright_green bold",
-        "prompt": "bright_magenta bold",
+        "info": "#4A90E2",  # Medium blue - matches system status colors
+        "warning": "#CC6600",  # Orange-brown - works on both light and dark
+        "error": "#CC0000 bold",  # Deep red - strong contrast
+        "success": "#00AA44 bold",  # Deep green - visible on both
+        "prompt": "#6633CC bold",  # Purple - good on both backgrounds
     },
 )
 

{massgen-0.1.0a2.dist-info → massgen-0.1.0a3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: massgen
-Version: 0.1.0a2
+Version: 0.1.0a3
 Summary: Multi-Agent Scaling System - A powerful framework for collaborative AI
 Author-email: MassGen Team <contact@massgen.dev>
 License: Apache-2.0

{massgen-0.1.0a2.dist-info → massgen-0.1.0a3.dist-info}/RECORD

@@ -1,8 +1,8 @@
-massgen/__init__.py,sha256=IeHEyx_AoV8W59BzWjs-kv67adlhGssN5b9Ng1Rn1Y0,7976
+massgen/__init__.py,sha256=QOvL8mdc_ZJsXZYUL9Vz-QnXa0zflAPFBcBQNR59_Yo,7976
 massgen/agent_config.py,sha256=FQZhOvEJq-h9oI2YEDKXX5o6sAWxBg2yu_g9NrI1sP0,30661
 massgen/chat_agent.py,sha256=doJSk6yhaqE1-Qy1AYpsSvUiJwHB92xXCBeS8hGh8wQ,18567
-massgen/cli.py,sha256=K-QfUSvgl9vQKIKzyZLCA6Kl5JAozL_1LcbRWTL1tzc,82016
-massgen/config_builder.py,sha256=oQmFtQMp84SC0UDU9jE8OFOcl7mhFG8og2reodqsTRM,112730
+massgen/cli.py,sha256=5SZ2K1eU13H6kZZERycWSJ5PHmFdF8l_s4QocY2x7iQ,84217
+massgen/config_builder.py,sha256=wR94ptSnnGH8LMw6J0lEZh-X5qHcKj8AWoUM2mIN3Pw,112897
 massgen/coordination_tracker.py,sha256=_r8Man3uomovpXKHrlc4YBI030tkuavYQ-6fEk2JsfM,28201
 massgen/logger_config.py,sha256=wga7v8yIASxDaO7vwjbw_gyCxe0ZxIggV6kGyyMGYlM,22686
 massgen/message_templates.py,sha256=toBE7Kdn0I6SsM6-DxC1El0nPBFWgakFZcAPdQVfl5E,37945
@@ -265,9 +265,9 @@ massgen/v1/examples/fast-4o-mini-config.yaml,sha256=t7zmN8sWQrrWivjaBB9w_WHFDnU-
 massgen/v1/examples/fast_config.yaml,sha256=BDBCpSYJPzTZi9vAEGsCZ3lt3JV25RN-4Nf_17JAhUU,846
 massgen/v1/examples/production.yaml,sha256=hrNqzOqlQG1sL-5U8ZlA8QpCwGSoiCtqMu7Mi2gOd9s,2002
 massgen/v1/examples/single_agent.yaml,sha256=h8ysGIwPjvgnpRxTcBqw5nBz5D2MTXI_Ygdn4D9RdIA,1256
-massgen-0.1.0a2.dist-info/licenses/LICENSE,sha256=_r2TwhSv0GcpGeIKfwbz9z2qzOI62FHCfgSfnZ1z2fc,11386
-massgen-0.1.0a2.dist-info/METADATA,sha256=hr3s2ooXNIjLs9yjQjF3d5OfZFdT8cbC04e3jM3nuYU,60212
-massgen-0.1.0a2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-massgen-0.1.0a2.dist-info/entry_points.txt,sha256=On6OV7swWO0StsAXeOri3MzmuxhO9C-PUvKW-5pJ6gA,49
-massgen-0.1.0a2.dist-info/top_level.txt,sha256=jm8ZxEGdkjsh8aPWXvCpnexL_S2ZRkFeT9vl6vIm_3I,8
-massgen-0.1.0a2.dist-info/RECORD,,
+massgen-0.1.0a3.dist-info/licenses/LICENSE,sha256=_r2TwhSv0GcpGeIKfwbz9z2qzOI62FHCfgSfnZ1z2fc,11386
+massgen-0.1.0a3.dist-info/METADATA,sha256=kvSVW9g9NhVwuVaPZx0WuDqRBypOEvVz8EQlo5SZC3w,60212
+massgen-0.1.0a3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+massgen-0.1.0a3.dist-info/entry_points.txt,sha256=On6OV7swWO0StsAXeOri3MzmuxhO9C-PUvKW-5pJ6gA,49
+massgen-0.1.0a3.dist-info/top_level.txt,sha256=jm8ZxEGdkjsh8aPWXvCpnexL_S2ZRkFeT9vl6vIm_3I,8
+massgen-0.1.0a3.dist-info/RECORD,,