omnibase_infra 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

omnibase_infra/nodes/contract_registry_reducer/registry/__init__.py

@@ -0,0 +1,9 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+"""Registry module for NodeContractRegistryReducer."""
+
+from omnibase_infra.nodes.contract_registry_reducer.registry.registry_infra_contract_registry_reducer import (
+    RegistryInfraContractRegistryReducer,
+)
+
+__all__ = ["RegistryInfraContractRegistryReducer"]

omnibase_infra/nodes/contract_registry_reducer/registry/registry_infra_contract_registry_reducer.py

@@ -0,0 +1,101 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+"""Registry for NodeContractRegistryReducer dependencies.
+
+This registry provides dependency injection configuration for the
+NodeContractRegistryReducer node, following ONEX container-based DI pattern.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from omnibase_core.models.container.model_onex_container import ModelONEXContainer
+    from omnibase_infra.nodes.contract_registry_reducer.node import (
+        NodeContractRegistryReducer,
+    )
+    from omnibase_infra.nodes.contract_registry_reducer.reducer import (
+        ContractRegistryReducer,
+    )
+
+
+class RegistryInfraContractRegistryReducer:
+    """Registry for NodeContractRegistryReducer dependency injection.
+
+    Why a class instead of a function?
+        ONEX registry pattern (CLAUDE.md) requires registry classes with
+        the naming convention ``RegistryInfra<NodeName>``. This enables:
+
+        - **Future extension**: Additional factory methods can be added
+          (e.g., ``create_reducer_with_cache()``, ``create_test_reducer()``)
+        - **Service registry resolution**: Classes can be registered in the
+          ONEX service registry for container-based resolution
+        - **Consistent pattern**: All node registries follow the same class-based
+          structure, making the codebase predictable and navigable
+        - **Container lifecycle management**: The registry can implement caching,
+          scoping, or other lifecycle behaviors in the future
+
+    Provides factory methods for creating NodeContractRegistryReducer instances
+    with properly configured dependencies from the ONEX container.
+
+    Usage:
+        ```python
+        from omnibase_core.models.container import ModelONEXContainer
+        from omnibase_infra.nodes.contract_registry_reducer.registry import (
+            RegistryInfraContractRegistryReducer,
+        )
+
+        # Create container and registry
+        container = ModelONEXContainer()
+        registry = RegistryInfraContractRegistryReducer(container)
+
+        # Create node instance
+        node = registry.create_node()
+
+        # Or create pure function reducer directly
+        reducer = registry.create_reducer()
+
+        # Use reducer
+        result = reducer.reduce(state, event, metadata)
+        ```
+    """
+
+    def __init__(self, container: ModelONEXContainer) -> None:
+        """Initialize the registry with ONEX container.
+
+        Args:
+            container: ONEX dependency injection container
+        """
+        self._container = container
+
+    def create_node(self) -> NodeContractRegistryReducer:
+        """Create a NodeContractRegistryReducer instance.
+
+        Returns:
+            Configured NodeContractRegistryReducer instance.
+        """
+        from omnibase_infra.nodes.contract_registry_reducer.node import (
+            NodeContractRegistryReducer,
+        )
+
+        return NodeContractRegistryReducer(self._container)
+
+    def create_reducer(self) -> ContractRegistryReducer:
+        """Create a pure function ContractRegistryReducer instance.
+
+        The pure reducer implements the reduce(state, event, metadata) pattern
+        without ONEX container dependencies, suitable for unit testing and
+        standalone use.
+
+        Returns:
+            ContractRegistryReducer instance (pure function reducer).
+        """
+        from omnibase_infra.nodes.contract_registry_reducer.reducer import (
+            ContractRegistryReducer,
+        )
+
+        return ContractRegistryReducer()
+
+
+__all__ = ["RegistryInfraContractRegistryReducer"]

omnibase_infra/nodes/handlers/consul/contract.yaml

@@ -0,0 +1,85 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+# Contract for Consul Handler - Service discovery and KV store operations
+name: "handler_consul"
+node_type: "EFFECT_GENERIC"
+description: "HashiCorp Consul handler for service discovery and KV store operations"
+contract_version:
+  major: 1
+  minor: 0
+  patch: 0
+node_version:
+  major: 0
+  minor: 1
+  patch: 0
+input_model:
+  name: "ModelEventEnvelope"
+  module: "omnibase_core.models.events.model_event_envelope"
+output_model:
+  name: "ModelHandlerOutput"
+  module: "omnibase_core.models.dispatch.model_handler_output"
+handler_routing:
+  routing_strategy: "operation_match"
+  handlers:
+    - handler_type: "consul"
+      handler:
+        name: "HandlerConsul"
+        module: "omnibase_infra.handlers.handler_consul"
+      supported_operations:
+        - "consul.kv_get"
+        - "consul.kv_put"
+        - "consul.register"
+        - "consul.deregister"
+operation_bindings:
+  version:
+    major: 1
+    minor: 0
+    patch: 0
+  global_bindings:
+    - parameter_name: "correlation_id"
+      expression: "${envelope.correlation_id}"
+      required: true
+  bindings:
+    "consul.kv_get":
+      - parameter_name: "key"
+        expression: "${payload.key}"
+        required: true
+    "consul.kv_put":
+      - parameter_name: "key"
+        expression: "${payload.key}"
+        required: true
+      - parameter_name: "value"
+        expression: "${payload.value}"
+        required: true
+    "consul.register":
+      - parameter_name: "service_id"
+        expression: "${payload.service_id}"
+        required: true
+      - parameter_name: "service_name"
+        expression: "${payload.service_name}"
+        required: true
+      - parameter_name: "address"
+        expression: "${payload.address}"
+        required: false
+      - parameter_name: "port"
+        expression: "${payload.port}"
+        required: false
+      - parameter_name: "tags"
+        expression: "${payload.tags}"
+        required: false
+    "consul.deregister":
+      - parameter_name: "service_id"
+        expression: "${payload.service_id}"
+        required: true
+metadata:
+  handler_id: "consul-handler"
+  handler_type: "infra_handler"
+  handler_category: "effect"
+  transport_type: "consul"
+  security_features:
+    - "SecretStr protection for ACL tokens"
+    - "Sanitized error messages"
+  resilience_features:
+    - "Circuit breaker pattern"
+    - "Exponential backoff retry"
+    - "Thread pool management"

omnibase_infra/nodes/handlers/db/contract.yaml

@@ -0,0 +1,72 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+# Contract for Database Handler - PostgreSQL database operations
+name: "handler_db"
+node_type: "EFFECT_GENERIC"
+description: "PostgreSQL database handler for query and execute operations using asyncpg"
+contract_version:
+  major: 1
+  minor: 0
+  patch: 0
+node_version:
+  major: 0
+  minor: 1
+  patch: 0
+input_model:
+  name: "ModelEventEnvelope"
+  module: "omnibase_core.models.events.model_event_envelope"
+output_model:
+  name: "ModelHandlerOutput"
+  module: "omnibase_core.models.dispatch.model_handler_output"
+handler_routing:
+  routing_strategy: "operation_match"
+  handlers:
+    - handler_type: "db"
+      handler:
+        name: "HandlerDb"
+        module: "omnibase_infra.handlers.handler_db"
+      supported_operations:
+        - "db.query"
+        - "db.execute"
+operation_bindings:
+  version:
+    major: 1
+    minor: 0
+    patch: 0
+  # Guardrail overrides (optional) - customize limits for this handler
+  # Defaults: max_expression_length=256, max_path_segments=20
+  # Bounds: expression_length=[32,1024], path_segments=[3,50]
+  max_expression_length: 512 # Allow longer expressions for complex SQL bindings
+  max_path_segments: 25 # Allow deeper nested paths for complex payloads
+  global_bindings:
+    - parameter_name: "correlation_id"
+      expression: "${envelope.correlation_id}"
+      required: true
+  bindings:
+    "db.query":
+      - parameter_name: "sql"
+        expression: "${payload.sql}"
+        required: true
+      - parameter_name: "parameters"
+        expression: "${payload.parameters}"
+        required: false
+    "db.execute":
+      - parameter_name: "sql"
+        expression: "${payload.sql}"
+        required: true
+      - parameter_name: "parameters"
+        expression: "${payload.parameters}"
+        required: false
+metadata:
+  handler_id: "db-handler"
+  handler_type: "infra_handler"
+  handler_category: "effect"
+  transport_type: "database"
+  security_features:
+    - "Parameterized queries for SQL injection protection"
+    - "DSN credential sanitization"
+    - "Single-statement SQL enforcement"
+  resilience_features:
+    - "Circuit breaker pattern"
+    - "Connection pooling (fixed size: 5)"
+    - "Intelligent error classification (transient vs permanent)"

omnibase_infra/nodes/handlers/graph/contract.yaml

@@ -0,0 +1,127 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+# Contract for Graph Handler - Neo4j/Memgraph database operations
+name: "handler_graph"
+node_type: "EFFECT_GENERIC"
+description: "Graph database handler for Cypher queries via neo4j async driver (Memgraph/Neo4j)"
+contract_version:
+  major: 1
+  minor: 0
+  patch: 0
+node_version:
+  major: 0
+  minor: 1
+  patch: 0
+input_model:
+  name: "ModelEventEnvelope"
+  module: "omnibase_core.models.events.model_event_envelope"
+output_model:
+  name: "ModelHandlerOutput"
+  module: "omnibase_core.models.dispatch.model_handler_output"
+handler_routing:
+  routing_strategy: "operation_match"
+  handlers:
+    - handler_type: "graph"
+      handler:
+        name: "HandlerGraph"
+        module: "omnibase_infra.handlers.handler_graph"
+      supported_operations:
+        - "graph.execute_query"
+        - "graph.execute_query_batch"
+        - "graph.create_node"
+        - "graph.create_relationship"
+        - "graph.delete_node"
+        - "graph.delete_relationship"
+        - "graph.traverse"
+operation_bindings:
+  version:
+    major: 1
+    minor: 0
+    patch: 0
+  global_bindings:
+    - parameter_name: "correlation_id"
+      expression: "${envelope.correlation_id}"
+      required: true
+  bindings:
+    "graph.execute_query":
+      - parameter_name: "query"
+        expression: "${payload.query}"
+        required: true
+      - parameter_name: "parameters"
+        expression: "${payload.parameters}"
+        required: false
+    "graph.execute_query_batch":
+      - parameter_name: "queries"
+        expression: "${payload.queries}"
+        required: true
+      - parameter_name: "transaction"
+        expression: "${payload.transaction}"
+        required: false
+    "graph.create_node":
+      - parameter_name: "labels"
+        expression: "${payload.labels}"
+        required: false
+      - parameter_name: "properties"
+        expression: "${payload.properties}"
+        required: false
+    "graph.create_relationship":
+      - parameter_name: "from_node_id"
+        expression: "${payload.from_node_id}"
+        required: true
+      - parameter_name: "to_node_id"
+        expression: "${payload.to_node_id}"
+        required: true
+      - parameter_name: "relationship_type"
+        expression: "${payload.relationship_type}"
+        required: true
+      - parameter_name: "properties"
+        expression: "${payload.properties}"
+        required: false
+    "graph.delete_node":
+      - parameter_name: "node_id"
+        expression: "${payload.node_id}"
+        required: true
+      - parameter_name: "detach"
+        expression: "${payload.detach}"
+        required: false
+    "graph.delete_relationship":
+      - parameter_name: "relationship_id"
+        expression: "${payload.relationship_id}"
+        required: true
+    "graph.traverse":
+      - parameter_name: "start_node_id"
+        expression: "${payload.start_node_id}"
+        required: true
+      - parameter_name: "relationship_types"
+        expression: "${payload.relationship_types}"
+        required: false
+      - parameter_name: "direction"
+        expression: "${payload.direction}"
+        required: false
+      - parameter_name: "max_depth"
+        expression: "${payload.max_depth}"
+        required: false
+      - parameter_name: "filters"
+        expression: "${payload.filters}"
+        required: false
+metadata:
+  handler_id: "graph-handler"
+  handler_type: "infra_handler"
+  handler_category: "effect"
+  transport_type: "graph"
+  security_features:
+    - "Parameterized Cypher queries"
+    - "Label validation for injection prevention"
+    - "Credential sanitization"
+  resilience_features:
+    - "Circuit breaker pattern"
+    - "Connection pool management"
+    - "Transaction support"
+  capabilities:
+    - "cypher"
+    - "parameterized_queries"
+    - "transactions"
+    - "node_crud"
+    - "relationship_crud"
+    - "traversal"
+    - "batch_operations"

omnibase_infra/nodes/handlers/http/contract.yaml

@@ -0,0 +1,74 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+# Contract for HTTP Handler - REST API operations
+name: "handler_http"
+node_type: "EFFECT_GENERIC"
+description: "HTTP REST handler for GET and POST operations using httpx async client"
+contract_version:
+  major: 1
+  minor: 0
+  patch: 0
+node_version:
+  major: 0
+  minor: 1
+  patch: 0
+input_model:
+  name: "ModelEventEnvelope"
+  module: "omnibase_core.models.events.model_event_envelope"
+output_model:
+  name: "ModelHandlerOutput"
+  module: "omnibase_core.models.dispatch.model_handler_output"
+handler_routing:
+  routing_strategy: "operation_match"
+  handlers:
+    - handler_type: "http"
+      handler:
+        name: "HandlerHttpRest"
+        module: "omnibase_infra.handlers.handler_http"
+      supported_operations:
+        - "http.get"
+        - "http.post"
+operation_bindings:
+  version:
+    major: 1
+    minor: 0
+    patch: 0
+  global_bindings:
+    - parameter_name: "correlation_id"
+      expression: "${envelope.correlation_id}"
+      required: true
+  bindings:
+    "http.get":
+      - parameter_name: "url"
+        expression: "${payload.url}"
+        required: true
+      - parameter_name: "headers"
+        expression: "${payload.headers}"
+        required: false
+    "http.post":
+      - parameter_name: "url"
+        expression: "${payload.url}"
+        required: true
+      - parameter_name: "headers"
+        expression: "${payload.headers}"
+        required: false
+      - parameter_name: "body"
+        expression: "${payload.body}"
+        required: false
+metadata:
+  handler_id: "http-handler"
+  handler_type: "infra_handler"
+  handler_category: "effect"
+  transport_type: "http"
+  security_features:
+    - "Configurable request/response size limits"
+    - "Pre-read Content-Length validation"
+    - "Streaming body validation"
+    - "DoS protection"
+  resilience_features:
+    - "Configurable timeout"
+    - "Follow redirects"
+  notes:
+    - "PUT, DELETE, PATCH deferred to Beta"
+    - "Retry logic deferred to Beta"
+    - "Rate limiting deferred to Beta"

omnibase_infra/nodes/handlers/intent/contract.yaml

@@ -0,0 +1,66 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+# Contract for Intent Handler - Graph-based intent storage operations
+name: "handler_intent"
+node_type: "EFFECT_GENERIC"
+description: "Intent handler wrapping HandlerGraph for intent-specific graph operations (DEMO)"
+contract_version:
+  major: 1
+  minor: 0
+  patch: 0
+node_version:
+  major: 0
+  minor: 1
+  patch: 0
+input_model:
+  name: "ModelEventEnvelope"
+  module: "omnibase_core.models.events.model_event_envelope"
+output_model:
+  name: "ModelHandlerOutput"
+  module: "omnibase_core.models.dispatch.model_handler_output"
+handler_routing:
+  routing_strategy: "operation_match"
+  handlers:
+    - handler_type: "intent"
+      handler:
+        name: "HandlerIntent"
+        module: "omnibase_infra.handlers.handler_intent"
+      supported_operations:
+        - "intent.store"
+        - "intent.query_session"
+        - "intent.query_distribution"
+operation_bindings:
+  version:
+    major: 1
+    minor: 0
+    patch: 0
+  global_bindings:
+    - parameter_name: "correlation_id"
+      expression: "${envelope.correlation_id}"
+      required: true
+  bindings:
+    "intent.store":
+      - parameter_name: "intent_type"
+        expression: "${payload.intent_type}"
+        required: true
+      - parameter_name: "session_id"
+        expression: "${payload.session_id}"
+        required: false
+    "intent.query_session":
+      - parameter_name: "session_id"
+        expression: "${payload.session_id}"
+        required: true
+    # No payload bindings required - aggregates all intents by type (read-only)
+    "intent.query_distribution": []
+metadata:
+  handler_id: "intent-handler"
+  handler_type: "intent_handler"
+  handler_category: "effect"
+  transport_type: "graph"
+  notes:
+    - "TEMPORARY demo wiring - will be replaced by contract-driven routing"
+    - "intent.store is NOT idempotent (creates new node each call)"
+    - "intent.query_session is idempotent (read-only)"
+    - "intent.query_distribution is idempotent (read-only aggregation)"
+  dependencies:
+    - "Requires pre-initialized HandlerGraph instance"

omnibase_infra/nodes/handlers/mcp/contract.yaml

@@ -0,0 +1,69 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+# Contract for MCP Handler - Model Context Protocol integration
+name: "handler_mcp"
+node_type: "EFFECT_GENERIC"
+description: "MCP protocol handler for exposing ONEX nodes as AI agent tools"
+contract_version:
+  major: 1
+  minor: 0
+  patch: 0
+node_version:
+  major: 0
+  minor: 1
+  patch: 0
+input_model:
+  name: "ModelEventEnvelope"
+  module: "omnibase_core.models.events.model_event_envelope"
+output_model:
+  name: "ModelHandlerOutput"
+  module: "omnibase_core.models.dispatch.model_handler_output"
+handler_routing:
+  routing_strategy: "operation_match"
+  handlers:
+    - handler_type: "mcp"
+      handler:
+        name: "HandlerMCP"
+        module: "omnibase_infra.handlers.handler_mcp"
+      supported_operations:
+        - "mcp.list_tools"
+        - "mcp.call_tool"
+        - "mcp.describe"
+operation_bindings:
+  version:
+    major: 1
+    minor: 0
+    patch: 0
+  global_bindings:
+    - parameter_name: "correlation_id"
+      expression: "${envelope.correlation_id}"
+      required: true
+  bindings:
+    # No payload bindings required - returns all registered tools
+    "mcp.list_tools": []
+    "mcp.call_tool":
+      - parameter_name: "tool_name"
+        expression: "${payload.tool_name}"
+        required: true
+      - parameter_name: "arguments"
+        expression: "${payload.arguments}"
+        required: false
+    # No payload bindings required - returns MCP server capabilities
+    "mcp.describe": []
+metadata:
+  handler_id: "mcp-handler"
+  handler_type: "infra_handler"
+  handler_category: "effect"
+  transport_type: "mcp"
+  security_features:
+    - "Tool execution timeout enforcement"
+    - "Request size limits"
+    - "Correlation ID propagation"
+    - "Circuit breaker protection"
+  resilience_features:
+    - "Circuit breaker pattern (threshold: 5, reset: 60s)"
+    - "Internal uvicorn server lifecycle management"
+  notes:
+    - "Authentication NOT yet implemented in MVP"
+    - "Streamable HTTP transport for production scalability"
+    - "Dynamic tool discovery from ONEX node registry"

omnibase_infra/nodes/handlers/vault/contract.yaml

@@ -0,0 +1,91 @@
+# SPDX-License-Identifier: MIT
+# Copyright (c) 2025 OmniNode Team
+# Contract for Vault Handler - HashiCorp Vault secret management
+name: "handler_vault"
+node_type: "EFFECT_GENERIC"
+description: "HashiCorp Vault handler for secret management operations using hvac client"
+contract_version:
+  major: 1
+  minor: 0
+  patch: 0
+node_version:
+  major: 0
+  minor: 1
+  patch: 0
+input_model:
+  name: "ModelEventEnvelope"
+  module: "omnibase_core.models.events.model_event_envelope"
+output_model:
+  name: "ModelHandlerOutput"
+  module: "omnibase_core.models.dispatch.model_handler_output"
+handler_routing:
+  routing_strategy: "operation_match"
+  handlers:
+    - handler_type: "vault"
+      handler:
+        name: "HandlerVault"
+        module: "omnibase_infra.handlers.handler_vault"
+      supported_operations:
+        - "vault.read_secret"
+        - "vault.write_secret"
+        - "vault.delete_secret"
+        - "vault.list_secrets"
+        - "vault.renew_token"
+operation_bindings:
+  version:
+    major: 1
+    minor: 0
+    patch: 0
+  global_bindings:
+    - parameter_name: "correlation_id"
+      expression: "${envelope.correlation_id}"
+      required: true
+  bindings:
+    "vault.read_secret":
+      - parameter_name: "path"
+        expression: "${payload.path}"
+        required: true
+      - parameter_name: "mount_point"
+        expression: "${payload.mount_point}"
+        required: false
+    "vault.write_secret":
+      - parameter_name: "path"
+        expression: "${payload.path}"
+        required: true
+      - parameter_name: "secret"
+        expression: "${payload.secret}"
+        required: true
+      - parameter_name: "mount_point"
+        expression: "${payload.mount_point}"
+        required: false
+    "vault.delete_secret":
+      - parameter_name: "path"
+        expression: "${payload.path}"
+        required: true
+      - parameter_name: "mount_point"
+        expression: "${payload.mount_point}"
+        required: false
+    "vault.list_secrets":
+      - parameter_name: "path"
+        expression: "${payload.path}"
+        required: true
+      - parameter_name: "mount_point"
+        expression: "${payload.mount_point}"
+        required: false
+    # No payload bindings required - renews the current auth token from connection config
+    "vault.renew_token": []
+metadata:
+  handler_id: "vault-handler"
+  handler_type: "infra_handler"
+  handler_category: "effect"
+  transport_type: "vault"
+  security_features:
+    - "SecretStr protection for tokens"
+    - "Sanitized error messages"
+    - "SSL verification enabled by default"
+    - "Automatic token renewal management"
+  resilience_features:
+    - "Circuit breaker pattern"
+    - "Exponential backoff retry"
+    - "Thread pool management"
+    - "Token expiration tracking"