agent-runtime-core 0.5.2__py3-none-any.whl → 0.6.0__py3-none-any.whl

agent_runtime_core/__init__.py

@@ -34,7 +34,7 @@ Example usage:
             return RunResult(final_output={"message": "Hello!"})
 """
 
-__version__ = "0.5.2"
+__version__ = "0.6.0"
 
 # Core interfaces
 from agent_runtime_core.interfaces import (

agent_runtime_core/registry.py

@@ -3,72 +3,151 @@ Agent runtime registry.
 
 Provides a global registry for agent runtimes, allowing them to be
 looked up by key.
+
+Supports:
+- Manual registration via register_runtime()
+- Factory functions for lazy instantiation
+- Class registration (auto-instantiation)
 """
 
-from typing import Optional
+import logging
+from typing import Callable, Optional, Type, Union
 
 from agent_runtime_core.interfaces import AgentRuntime
 
+logger = logging.getLogger(__name__)
 
 # Global registry
 _runtimes: dict[str, AgentRuntime] = {}
+_runtime_factories: dict[str, Callable[[], AgentRuntime]] = {}
+
+
+def _is_agent_runtime(obj) -> bool:
+    """Check if an object is an AgentRuntime instance."""
+    return isinstance(obj, AgentRuntime)
+
 
+def _is_agent_runtime_class(cls) -> bool:
+    """Check if a class is an AgentRuntime subclass."""
+    if not isinstance(cls, type):
+        return False
+    try:
+        return issubclass(cls, AgentRuntime)
+    except TypeError:
+        return False
 
-def register_runtime(runtime: AgentRuntime) -> None:
+
+def register_runtime(
+    runtime: Union[AgentRuntime, Type[AgentRuntime], Callable[[], AgentRuntime]],
+    key: Optional[str] = None,
+) -> None:
     """
     Register an agent runtime.
-    
+
     Args:
-        runtime: The runtime to register
-        
+        runtime: Runtime instance, class, or factory function
+        key: Optional key override (uses runtime.key if not provided)
+
+    Examples:
+        # Register an instance
+        register_runtime(MyRuntime())
+
+        # Register a class (will be instantiated)
+        register_runtime(MyRuntime)
+
+        # Register with custom key
+        register_runtime(MyRuntime(), key="custom-key")
+
+        # Register a factory
+        register_runtime(lambda: MyRuntime(config=get_config()), key="my-runtime")
+
     Raises:
-        ValueError: If a runtime with the same key is already registered
+        ValueError: If key is required but not provided
+        TypeError: If runtime is not a valid type
     """
-    key = runtime.key
-    if key in _runtimes:
-        raise ValueError(f"Runtime already registered: {key}")
-    _runtimes[key] = runtime
+    if _is_agent_runtime(runtime):
+        # Instance provided
+        runtime_key = key or runtime.key
+        _runtimes[runtime_key] = runtime
+        logger.info(f"Registered agent runtime: {runtime_key}")
+
+    elif _is_agent_runtime_class(runtime):
+        # Class provided - instantiate it
+        instance = runtime()
+        runtime_key = key or instance.key
+        _runtimes[runtime_key] = instance
+        logger.info(f"Registered agent runtime: {runtime_key}")
 
+    elif callable(runtime):
+        # Factory function provided
+        if not key:
+            raise ValueError("key is required when registering a factory function")
+        _runtime_factories[key] = runtime
+        logger.info(f"Registered agent runtime factory: {key}")
 
-def get_runtime(key: str) -> Optional[AgentRuntime]:
+    else:
+        raise TypeError(
+            f"runtime must be AgentRuntime instance, class, or callable, got {type(runtime)}"
+        )
+
+
+def get_runtime(key: str) -> AgentRuntime:
     """
     Get a registered runtime by key.
-    
+
     Args:
         key: The runtime key
-        
+
     Returns:
-        The runtime, or None if not found
+        The runtime instance
+
+    Raises:
+        KeyError: If runtime not found
     """
-    return _runtimes.get(key)
+    # Check instances first
+    if key in _runtimes:
+        return _runtimes[key]
+
+    # Check factories
+    if key in _runtime_factories:
+        instance = _runtime_factories[key]()
+        _runtimes[key] = instance
+        return instance
+
+    raise KeyError(f"Agent runtime not found: {key}. Available: {list_runtimes()}")
 
 
 def list_runtimes() -> list[str]:
     """
     List all registered runtime keys.
-    
+
     Returns:
         List of runtime keys
     """
-    return list(_runtimes.keys())
+    return list(set(_runtimes.keys()) | set(_runtime_factories.keys()))
 
 
 def unregister_runtime(key: str) -> bool:
     """
     Unregister a runtime.
-    
+
     Args:
         key: The runtime key
-        
+
     Returns:
         True if unregistered, False if not found
     """
+    removed = False
     if key in _runtimes:
         del _runtimes[key]
-        return True
-    return False
+        removed = True
+    if key in _runtime_factories:
+        del _runtime_factories[key]
+        removed = True
+    return removed
 
 
 def clear_registry() -> None:
     """Clear all registered runtimes. Useful for testing."""
     _runtimes.clear()
+    _runtime_factories.clear()

{agent_runtime_core-0.5.2.dist-info → agent_runtime_core-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-runtime-core
-Version: 0.5.2
+Version: 0.6.0
 Summary: Framework-agnostic Python library for executing AI agents with consistent patterns
 Project-URL: Homepage, https://github.com/makemore/agent-runtime-core
 Project-URL: Repository, https://github.com/makemore/agent-runtime-core
@@ -53,6 +53,7 @@ A lightweight, framework-agnostic Python library for building AI agent systems.
 
 | Version | Date | Changes |
 |---------|------|---------|
+| **0.6.0** | 2025-01-23 | Enhanced registry with factory functions and class registration |
 | **0.5.2** | 2025-01-14 | Add ToolCallingAgent base class, execute_with_events helper |
 | **0.5.1** | 2025-01-13 | Bug fixes and improvements |
 | **0.5.0** | 2025-01-12 | Initial stable release |

{agent_runtime_core-0.5.2.dist-info → agent_runtime_core-0.6.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
-agent_runtime_core/__init__.py,sha256=pn2I6UI_nbnho9K8KViC05RsWTRm0ftXyi7TF0dbyLQ,4139
+agent_runtime_core/__init__.py,sha256=qPJb-oJ424TMygK4OxUpLqoOu8e4I68MWok1ufxOAzQ,4139
 agent_runtime_core/config.py,sha256=e3_uB5brAuQcWU36sOhWF9R6RoJrngtCS-xEB3n2fas,4986
 agent_runtime_core/interfaces.py,sha256=T74pgS229tvarQD-_o25oflylUR7jq_jbgUjnvVs6IA,12191
-agent_runtime_core/registry.py,sha256=hrbEdNNdqEz7-uN-82qofsXFTZBRDxZ2Ht9qwmp1qkw,1476
+agent_runtime_core/registry.py,sha256=QmazCAcHTsPt236Z_xEBJjdppm6jUuufE-gfvcGMUCk,3959
 agent_runtime_core/runner.py,sha256=M3It72UhfmLt17jVnSvObiSfQ1_RN4JVUIJsjnRd2Ps,12771
 agent_runtime_core/steps.py,sha256=XpVFK7P-ZOpr7NwaP7XFygduIpjrKld-OIig7dHNMKE,11994
 agent_runtime_core/testing.py,sha256=ordECGprBappLBMWxlETvuf2AoIPNomJFeSedXaY30E,11131
@@ -32,7 +32,7 @@ agent_runtime_core/state/sqlite.py,sha256=HKZwDiC_7F1W8Z_Pz8roEs91XhQ9rUHfGpuQ7W
 agent_runtime_core/tracing/__init__.py,sha256=u1QicGc39e30gWyQD4cQWxGGjITnkwoOPUhNrG6aNyI,1266
 agent_runtime_core/tracing/langfuse.py,sha256=Rj2sUlatk5sFro0y68tw5X6fQcSwWxcBOSOjB0F7JTU,3660
 agent_runtime_core/tracing/noop.py,sha256=SpsbpsUcNG6C3xZG3uyiNPUHY8etloISx3w56Q8D3KE,751
-agent_runtime_core-0.5.2.dist-info/METADATA,sha256=Lu_Z_hwn2hWC8czhLGqaUFcT7dEOQ2GSXRlx12RR9KE,23765
-agent_runtime_core-0.5.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-agent_runtime_core-0.5.2.dist-info/licenses/LICENSE,sha256=fDlWep3_mUrj8KHV_jk275tHVEW7_9sJRhkNuGCZ_TA,1068
-agent_runtime_core-0.5.2.dist-info/RECORD,,
+agent_runtime_core-0.6.0.dist-info/METADATA,sha256=Q9egtbhBSMcgmeA9TM4gy_8P5pUcN0WFMCNaAOii12w,23858
+agent_runtime_core-0.6.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+agent_runtime_core-0.6.0.dist-info/licenses/LICENSE,sha256=fDlWep3_mUrj8KHV_jk275tHVEW7_9sJRhkNuGCZ_TA,1068
+agent_runtime_core-0.6.0.dist-info/RECORD,,