open-swarm 0.1.1744942884__tar.gz → 0.1.1744943015__tar.gz

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: open-swarm
-Version: 0.1.1744942884
+Version: 0.1.1744943015
 Summary: Open Swarm: Orchestrating AI Agent Swarms with Django
 Project-URL: Homepage, https://github.com/yourusername/open-swarm
 Project-URL: Documentation, https://github.com/yourusername/open-swarm/blob/main/README.md

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "open-swarm"
-version = "0.1.1744942884"
+version = "0.1.1744943015"
 description = "Open Swarm: Orchestrating AI Agent Swarms with Django"
 readme = "README.md"
 requires-python = ">=3.10"

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/digitalbutlers/blueprint_digitalbutlers.py

@@ -16,7 +16,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in DigitalButlersBlueprint: {e}. Check 'openai-agents' install and project structure.")
     print(f"Attempted import from directory: {os.path.dirname(__file__)}")

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/divine_code/blueprint_divine_code.py

@@ -16,7 +16,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in DivineOpsBlueprint: {e}. Check 'openai-agents' install and project structure.")
     print(f"sys.path: {sys.path}")

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/django_chat/blueprint_django_chat.py

@@ -44,7 +44,7 @@ from django.contrib.auth.decorators import login_required
 from django.views.decorators.csrf import csrf_exempt
 from django.contrib.auth.models import User
 from swarm.models import ChatConversation, ChatMessage
-from swarm.extensions.blueprint.blueprint_base import BlueprintBase as Blueprint
+from swarm.core.blueprint_base import BlueprintBase as Blueprint
 from swarm.utils.logger_setup import setup_logger
 
 logger = setup_logger(__name__)

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/echocraft/blueprint_echocraft.py

@@ -6,7 +6,7 @@ from typing import List, Dict, Any, AsyncGenerator
 import uuid # Import uuid to generate IDs
 import time # Import time for timestamp
 
-from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+from swarm.core.blueprint_base import BlueprintBase
 
 logger = logging.getLogger(__name__)
 

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/family_ties/blueprint_family_ties.py

@@ -16,7 +16,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in FamilyTiesBlueprint: {e}. Check dependencies.")
     print(f"sys.path: {sys.path}")

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/mcp_demo/blueprint_mcp_demo.py

@@ -14,7 +14,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in MCPDemoBlueprint: {e}. Check dependencies.")
     print(f"sys.path: {sys.path}")
@@ -128,6 +128,35 @@ class MCPDemoBlueprint(BlueprintBase):
         logger.debug("Sage agent created.")
         return sage_agent
 
+    async def run(self, messages: List[dict], **kwargs):
+        """Main execution entry point for the MCP Demo blueprint."""
+        logger.info("MCPDemoBlueprint run method called.")
+        instruction = messages[-1].get("content", "") if messages else ""
+        try:
+            mcp_servers = kwargs.get("mcp_servers", [])
+            starting_agent = self.create_starting_agent(mcp_servers=mcp_servers)
+            from agents import Runner
+            if not starting_agent.model:
+                yield {"messages": [{"role": "assistant", "content": f"Error: No model instance available for MCPDemo agent. Check your OPENAI_API_KEY, or LITELLM_MODEL/LITELLM_BASE_URL config."}]}
+                return
+            if not starting_agent.tools:
+                yield {"messages": [{"role": "assistant", "content": f"Warning: No tools registered for MCPDemo agent. Only direct LLM output is possible."}]}
+            required_mcps = self.metadata.get('required_mcp_servers', [])
+            missing_mcps = [m for m in required_mcps if m not in [s.name for s in mcp_servers]]
+            if missing_mcps:
+                yield {"messages": [{"role": "assistant", "content": f"Warning: Missing required MCP servers: {', '.join(missing_mcps)}. Some features may not work."}]}
+            from rich.console import Console
+            console = Console()
+            with console.status("Generating...", spinner="dots") as status:
+                async for chunk in Runner.run(starting_agent, instruction):
+                    content = chunk.get("content")
+                    if content and ("function call" in content or "args" in content):
+                        continue
+                    yield chunk
+            logger.info("MCPDemoBlueprint run method finished.")
+        except Exception as e:
+            yield {"messages": [{"role": "assistant", "content": f"Error: {e}"}]}
+
 # Standard Python entry point
 if __name__ == "__main__":
     MCPDemoBlueprint.main()

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/mission_improbable/blueprint_mission_improbable.py

@@ -17,7 +17,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in MissionImprobableBlueprint: {e}. Check dependencies.")
     print(f"sys.path: {sys.path}")

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/monkai_magic/blueprint_monkai_magic.py

@@ -29,7 +29,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in MonkaiMagicBlueprint: {e}. Check dependencies.")
     print(f"sys.path: {sys.path}")

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/nebula_shellz/blueprint_nebula_shellz.py

@@ -13,7 +13,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
     from rich.panel import Panel # Import Panel for splash screen
 except ImportError as e:
     print(f"ERROR: Import failed in nebula_shellz: {e}. Ensure 'openai-agents' install and structure.")
@@ -170,5 +170,33 @@ Initializing NebulaShellzzar Crew...
         logger.debug("NebulaShellzzar agent team created. Morpheus is the starting agent.") # Changed to DEBUG
         return morpheus
 
+    async def run(self, messages: List[dict], **kwargs):
+        logger.info("NebuchaShellzzarBlueprint run method called.")
+        instruction = messages[-1].get("content", "") if messages else ""
+        try:
+            mcp_servers = kwargs.get("mcp_servers", [])
+            starting_agent = self.create_starting_agent(mcp_servers=mcp_servers)
+            from agents import Runner
+            if not starting_agent.model:
+                yield {"messages": [{"role": "assistant", "content": f"Error: No model instance available for NebuchaShellzzar agent. Check your OPENAI_API_KEY, or LITELLM_MODEL/LITELLM_BASE_URL config."}]}
+                return
+            if not starting_agent.tools:
+                yield {"messages": [{"role": "assistant", "content": f"Warning: No tools registered for NebuchaShellzzar agent. Only direct LLM output is possible."}]}
+            required_mcps = self.metadata.get('required_mcp_servers', [])
+            missing_mcps = [m for m in required_mcps if m not in [s.name for s in mcp_servers]]
+            if missing_mcps:
+                yield {"messages": [{"role": "assistant", "content": f"Warning: Missing required MCP servers: {', '.join(missing_mcps)}. Some features may not work."}]}
+            from rich.console import Console
+            console = Console()
+            with console.status("Generating...", spinner="dots") as status:
+                async for chunk in Runner.run(starting_agent, instruction):
+                    content = chunk.get("content")
+                    if content and ("function call" in content or "args" in content):
+                        continue
+                    yield chunk
+            logger.info("NebuchaShellzzarBlueprint run method finished.")
+        except Exception as e:
+            yield {"messages": [{"role": "assistant", "content": f"Error: {e}"}]}
+
 if __name__ == "__main__":
     NebuchaShellzzarBlueprint.main()

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/omniplex/blueprint_omniplex.py

@@ -14,7 +14,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in OmniplexBlueprint: {e}. Check dependencies.")
     print(f"sys.path: {sys.path}")
@@ -216,26 +216,36 @@ class OmniplexBlueprint(BlueprintBase):
         logger.info(f"Omniplex Coordinator created with tools for: {[t.name for t in team_tools]}")
         return coordinator_agent
 
-    async def run(self, messages: List[Dict[str, Any]], **kwargs) -> Any:
-        """Main execution entry point for the Omniplex blueprint."""
-        logger.info("OmniplexBlueprint run method called.")
+    async def run(self, messages: list, **kwargs):
+        import logging
+        if not hasattr(self, "logger"):
+            self.logger = logging.getLogger(__name__)
+        self.logger.info("OmniplexBlueprint run method called.")
         instruction = messages[-1].get("content", "") if messages else ""
-        async for chunk in self._run_non_interactive(instruction, **kwargs):
-            yield chunk
-        logger.info("OmniplexBlueprint run method finished.")
-
-    async def _run_non_interactive(self, instruction: str, **kwargs) -> Any:
-        logger.info(f"Running OmniplexBlueprint non-interactively with instruction: '{instruction[:100]}...'")
-        mcp_servers = kwargs.get("mcp_servers", [])
-        agent = self.create_starting_agent(mcp_servers=mcp_servers)
-        from agents import Runner
-        model_name = os.getenv("LITELLM_MODEL") or os.getenv("DEFAULT_LLM") or "gpt-3.5-turbo"
         try:
-            for chunk in Runner.run(agent, instruction):
-                yield chunk
+            mcp_servers = kwargs.get("mcp_servers", [])
+            agent = self.create_starting_agent(mcp_servers=mcp_servers)
+            from agents import Runner
+            if not agent.model:
+                yield {"messages": [{"role": "assistant", "content": f"Error: No model instance available for Omniplex agent. Check your OPENAI_API_KEY, or LITELLM_MODEL/LITELLM_BASE_URL config."}]}
+                return
+            if not agent.tools:
+                yield {"messages": [{"role": "assistant", "content": f"Warning: No tools registered for Omniplex agent. Only direct LLM output is possible."}]}
+            required_mcps = self.metadata.get('required_mcp_servers', [])
+            missing_mcps = [m for m in required_mcps if m not in [s.name for s in mcp_servers]]
+            if missing_mcps:
+                yield {"messages": [{"role": "assistant", "content": f"Warning: Missing required MCP servers: {', '.join(missing_mcps)}. Some features may not work."}]}
+            from rich.console import Console
+            console = Console()
+            with console.status("Generating...", spinner="dots") as status:
+                async for chunk in Runner.run(agent, instruction):
+                    content = chunk.get("content")
+                    if content and ("function call" in content or "args" in content):
+                        continue
+                    yield chunk
+            self.logger.info("OmniplexBlueprint run method finished.")
         except Exception as e:
-            logger.error(f"Error during non-interactive run: {e}", exc_info=True)
-            yield {"messages": [{"role": "assistant", "content": f"An error occurred: {e}"}]}
+            yield {"messages": [{"role": "assistant", "content": f"Error: {e}"}]}
 
 # Standard Python entry point
 if __name__ == "__main__":

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/rue_code/blueprint_rue_code.py

@@ -13,7 +13,7 @@ logger = logging.getLogger(__name__)
 
 # Attempt to import BlueprintBase, handle potential ImportError during early setup/testing
 try:
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     logger.error(f"Import failed: {e}. Check 'openai-agents' install and project structure.")
     # *** REMOVED sys.exit(1) ***

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/suggestion/blueprint_suggestion.py

@@ -14,7 +14,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
      print(f"ERROR: Failed to import 'agents' or 'BlueprintBase'. Is 'openai-agents' installed and src in PYTHONPATH? Details: {e}")
      sys.exit(1)

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/blueprints/unapologetic_press/blueprint_unapologetic_press.py

@@ -18,7 +18,7 @@ try:
     from agents.models.interface import Model
     from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
     from openai import AsyncOpenAI
-    from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+    from swarm.core.blueprint_base import BlueprintBase
 except ImportError as e:
     print(f"ERROR: Import failed in UnapologeticPressBlueprint: {e}. Check dependencies.")
     print(f"sys.path: {sys.path}")

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/extensions/cli/commands/blueprint_management.py

@@ -1,7 +1,7 @@
 # Handles blueprint discovery and validation for the CLI
 
-from swarm.extensions.blueprint.discovery import discover_blueprints
-from swarm.extensions.config.config_loader import load_server_config
+from swarm.core.blueprint_discovery import discover_blueprints
+from swarm.core.config_loader import load_server_config
 
 def list_blueprints():
     """List available blueprints and their metadata."""
@@ -16,7 +16,7 @@ def list_blueprints():
 cat > src/swarm/extensions/cli/commands/config_management.py << 'EOF'
 # Handles configuration management workflows (e.g., LLM, MCP servers)
 
-from swarm.extensions.config.config_loader import (
+from swarm.core.config_loader import (
     load_server_config,
     save_server_config,
 )

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/extensions/cli/commands/config_management.py

@@ -1,15 +1,14 @@
 # Handles configuration management workflows (e.g., LLM, MCP servers)
 
-from swarm.extensions.config.config_loader import (
-    load_server_config,
-    save_server_config,
-)
+from swarm.core import config_loader
+from swarm.core import config_manager
+from swarm.core import server_config
 
 def add_llm(model_name, api_key):
     """Add a new LLM configuration."""
-    config = load_server_config()
+    config = config_loader.load_server_config()
     if "llms" not in config:
         config["llms"] = {}
     config["llms"][model_name] = {"api_key": api_key}
-    save_server_config(config)
+    config_manager.save_server_config(config)
     print(f"Added LLM '{model_name}' to configuration.")

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/extensions/cli/commands/edit_config.py

@@ -1,8 +1,10 @@
 import argparse
 import json
-from swarm.extensions.config.config_loader import (
-    load_server_config,
-    save_server_config,
+from swarm.core import (
+    config_loader,
+    config_manager,
+    server_config,
+    setup_wizard,
 )
 from pathlib import Path
 
@@ -60,7 +62,7 @@ def main():
     args = parser.parse_args()
 
     try:
-        config = load_server_config(str(CONFIG_PATH))
+        config = config_loader.load_server_config(str(CONFIG_PATH))
     except FileNotFoundError as e:
         print(f"Error: {e}")
         return
@@ -69,9 +71,9 @@ def main():
         list_config(config)
     elif args.interactive:
         edit_config_interactive(config)
-        save_server_config(str(CONFIG_PATH), config)
+        config_loader.save_server_config(str(CONFIG_PATH), config)
     elif args.field and args.value:
         edit_config_field(config, args.field, args.value)
-        save_server_config(str(CONFIG_PATH), config)
+        config_loader.save_server_config(str(CONFIG_PATH), config)
     else:
         parser.print_help()

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/extensions/cli/commands/list_blueprints.py

@@ -4,7 +4,7 @@ Description: Lists all blueprints available in the system.
 """
 
 from pathlib import Path
-from swarm.extensions.blueprint.blueprint_discovery import discover_blueprints
+from swarm.core.blueprint_discovery import discover_blueprints
 
 # Metadata for dynamic registration
 description = "Lists all blueprints available in the system."

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/extensions/cli/commands/validate_env.py

@@ -1,7 +1,7 @@
 import os
 import argparse
-from swarm.extensions.config.config_loader import load_server_config
-from swarm.extensions.blueprint.blueprint_base import BlueprintBase
+from swarm.core import config_loader, config_manager, server_config
+from swarm.core.blueprint_base import BlueprintBase
 
 CONFIG_PATH = "swarm_config.json"
 
@@ -43,7 +43,7 @@ def main():
     args = parser.parse_args()
 
     try:
-        config = load_server_config(CONFIG_PATH)
+        config = server_config.load_server_config(CONFIG_PATH)
     except FileNotFoundError as e:
         print(f"Error: {e}")
         return

{open_swarm-0.1.1744942884 → open_swarm-0.1.1744943015}/src/swarm/extensions/cli/commands/validate_envvars.py

@@ -1,5 +1,5 @@
-from swarm.extensions.blueprint.blueprint_discovery import discover_blueprints
-from swarm.extensions.config.config_loader import load_env_config, validate_env_vars
+from swarm.core.blueprint_discovery import discover_blueprints
+from swarm.core import config_loader, config_manager, server_config
 import argparse
 
 def validate_envvars(blueprint_name=None):
@@ -19,16 +19,16 @@ def validate_envvars(blueprint_name=None):
             print(f"Blueprint '{blueprint_name}' not found.")
             return
         required_vars = blueprint.get("env_vars", [])
-        env_vars = load_env_config()
-        validation = validate_env_vars(env_vars, required_vars)
+        env_vars = config_loader.load_env_config()
+        validation = config_manager.validate_env_vars(env_vars, required_vars)
         print(f"Validation for '{blueprint_name}': {validation}")
     else:
         # Global validation
-        env_vars = load_env_config()
+        env_vars = config_loader.load_env_config()
         print("Global Environment Validation:")
         for blueprint_name, blueprint_data in blueprints.items():
             required_vars = blueprint_data.get("env_vars", [])
-            validation = validate_env_vars(env_vars, required_vars)
+            validation = config_manager.validate_env_vars(env_vars, required_vars)
             print(f"Validation for '{blueprint_name}': {validation}")
 
 def main():