soe-ai 0.2.0b3__tar.gz → 0.2.0b4__tar.gz

{soe_ai-0.2.0b3 → soe_ai-0.2.0b4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: soe-ai
-Version: 0.2.0b3
+Version: 0.2.0b4
 Summary: Signal-driven Orchestration Engine - Agent orchestration with event-driven workflow engine
 Author-email: Pedro Garcia <pgarcia14180@gmail.com>
 License-Expression: MIT

{soe_ai-0.2.0b3 → soe_ai-0.2.0b4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "soe-ai"
-version = "0.2.0b3"
+version = "0.2.0b4"
 description = "Signal-driven Orchestration Engine - Agent orchestration with event-driven workflow engine"
 readme = "README.md"
 requires-python = ">=3.8"

{soe_ai-0.2.0b3 → soe_ai-0.2.0b4}/soe/init.py

@@ -62,7 +62,13 @@ def create_all_nodes(
 
         tools_list = []
         if tools_registry:
-            tools_list = [{"function": func, "max_retries": 0} for func in tools_registry.values()]
+            for entry in tools_registry.values():
+                if callable(entry):
+                    # Simple function format: {"name": function}
+                    tools_list.append({"function": entry, "max_retries": 0})
+                elif isinstance(entry, dict) and "function" in entry:
+                    # Full config format: {"name": {"function": fn, ...}}
+                    tools_list.append(entry)
         nodes["agent"] = create_agent_node_caller(backends, tools_list, call_llm, broadcast_signals_caller)
 
     if tools_registry is not None:

{soe_ai-0.2.0b3 → soe_ai-0.2.0b4}/soe_ai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: soe-ai
-Version: 0.2.0b3
+Version: 0.2.0b4
 Summary: Signal-driven Orchestration Engine - Agent orchestration with event-driven workflow engine
 Author-email: Pedro Garcia <pgarcia14180@gmail.com>
 License-Expression: MIT

{soe_ai-0.2.0b3 → soe_ai-0.2.0b4}/soe_ai.egg-info/SOURCES.txt

@@ -147,4 +147,5 @@ soe_ai.egg-info/dependency_links.txt
 soe_ai.egg-info/requires.txt
 soe_ai.egg-info/top_level.txt
 tests/test_local_storage_backends.py
+tests/test_setup_orchestration.py
 tests/test_validation_errors.py

soe_ai-0.2.0b4/tests/test_setup_orchestration.py

@@ -0,0 +1,146 @@
+"""
+Tests for setup_orchestration and create_all_nodes.
+
+These test the high-level setup API that external users call.
+Ensures both simple function format and full config dict format work.
+"""
+
+import pytest
+from soe import setup_orchestration, create_all_nodes, orchestrate
+from tests.test_cases.lib import create_test_backends, create_call_llm
+
+
+def sample_tool(x: int) -> int:
+    """A simple test tool."""
+    return x * 2
+
+
+def another_tool(name: str) -> str:
+    """Another test tool."""
+    return f"Hello, {name}!"
+
+
+class TestCreateAllNodesToolsRegistry:
+    """Test create_all_nodes with different tools_registry formats."""
+
+    def test_tools_registry_simple_functions(self):
+        """
+        tools_registry with simple functions: {"name": function}
+        This is the format used in documentation examples.
+        """
+        backends = create_test_backends("simple_funcs")
+        call_llm = create_call_llm(stub=lambda p, c: '{"output": "test"}')
+
+        tools_registry = {
+            "sample_tool": sample_tool,
+            "another_tool": another_tool,
+        }
+
+        nodes, broadcast = create_all_nodes(
+            backends=backends,
+            call_llm=call_llm,
+            tools_registry=tools_registry,
+        )
+
+        assert "router" in nodes
+        assert "llm" in nodes
+        assert "agent" in nodes
+        assert "tool" in nodes
+
+        backends.cleanup_all()
+
+    def test_tools_registry_full_config_dicts(self):
+        """
+        tools_registry with full config dicts: {"name": {"function": fn, "max_retries": 3}}
+        This is the format used by soe_evolve and agentic_playground.
+        """
+        backends = create_test_backends("config_dicts")
+        call_llm = create_call_llm(stub=lambda p, c: '{"output": "test"}')
+
+        tools_registry = {
+            "sample_tool": {"function": sample_tool, "max_retries": 3},
+            "another_tool": {"function": another_tool, "failure_signal": "TOOL_FAILED"},
+        }
+
+        nodes, broadcast = create_all_nodes(
+            backends=backends,
+            call_llm=call_llm,
+            tools_registry=tools_registry,
+        )
+
+        assert "router" in nodes
+        assert "llm" in nodes
+        assert "agent" in nodes
+        assert "tool" in nodes
+
+        backends.cleanup_all()
+
+    def test_tools_registry_mixed_formats(self):
+        """
+        tools_registry with mixed formats: some functions, some dicts.
+        Should handle both gracefully.
+        """
+        backends = create_test_backends("mixed_formats")
+        call_llm = create_call_llm(stub=lambda p, c: '{"output": "test"}')
+
+        tools_registry = {
+            "sample_tool": sample_tool,  # Simple function
+            "another_tool": {"function": another_tool, "max_retries": 2},  # Full config
+        }
+
+        nodes, broadcast = create_all_nodes(
+            backends=backends,
+            call_llm=call_llm,
+            tools_registry=tools_registry,
+        )
+
+        assert "router" in nodes
+        assert "llm" in nodes
+        assert "agent" in nodes
+        assert "tool" in nodes
+
+        backends.cleanup_all()
+
+
+class TestSetupOrchestration:
+    """Test setup_orchestration convenience function."""
+
+    def test_setup_with_simple_functions(self):
+        """setup_orchestration works with simple function format."""
+        call_llm = create_call_llm(stub=lambda p, c: '{"output": "test"}')
+
+        tools_registry = {
+            "sample_tool": sample_tool,
+        }
+
+        backends, broadcast = setup_orchestration(
+            storage_dir="./test_data/setup_simple",
+            call_llm=call_llm,
+            tools_registry=tools_registry,
+        )
+
+        assert backends is not None
+        assert broadcast is not None
+        assert callable(broadcast)
+
+        backends.cleanup_all()
+
+    def test_setup_with_config_dicts(self):
+        """setup_orchestration works with full config dict format."""
+        call_llm = create_call_llm(stub=lambda p, c: '{"output": "test"}')
+
+        tools_registry = {
+            "sample_tool": {"function": sample_tool, "max_retries": 3},
+        }
+
+        backends, broadcast = setup_orchestration(
+            storage_dir="./test_data/setup_dicts",
+            call_llm=call_llm,
+            tools_registry=tools_registry,
+        )
+
+        assert backends is not None
+        assert broadcast is not None
+        assert callable(broadcast)
+
+        backends.cleanup_all()