base-pmad-te 0.1.0__tar.gz

base_pmad_te-0.1.0/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: base-pmad-te
+Version: 0.1.0
+Summary: Base Thought Engine for pMADs — full Imperator with admin tools and TEContext coupling
+Requires-Python: >=3.12
+Requires-Dist: langgraph>=0.2.60
+Requires-Dist: langchain-core>=0.3.28
+Requires-Dist: langchain-openai>=0.2.14
+Requires-Dist: httpx>=0.28.1

base_pmad_te-0.1.0/pyproject.toml

@@ -0,0 +1,18 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "base-pmad-te"
+version = "0.1.0"
+description = "Base Thought Engine for pMADs — full Imperator with admin tools and TEContext coupling"
+requires-python = ">=3.12"
+dependencies = [
+    "langgraph>=0.2.60",
+    "langchain-core>=0.3.28",
+    "langchain-openai>=0.2.14",
+    "httpx>=0.28.1",
+]
+
+[tool.setuptools.packages.find]
+where = ["src"]

base_pmad_te-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

base_pmad_te-0.1.0/src/base_pmad_te/__init__.py

@@ -0,0 +1 @@
+"""Context Broker TE — Imperator cognitive agent package."""

base_pmad_te-0.1.0/src/base_pmad_te/_ctx.py

@@ -0,0 +1,103 @@
+"""
+TE/AE decoupling — TEContext protocol and singleton.
+
+Defines the interface the TE needs from the AE kernel, without importing
+any ``app.*`` modules.  The AE bootstrap creates a concrete implementation
+and calls ``initialize(ctx)`` before the Imperator flow is compiled.
+
+This is the ONLY module TE files import for kernel services.
+"""
+
+from __future__ import annotations
+
+from typing import Any, Awaitable, Callable, Optional, Protocol, runtime_checkable
+
+
+@runtime_checkable
+class TEContext(Protocol):
+    """Surface the TE needs from the AE kernel."""
+
+    # ── Properties / fields ──────────────────────────────────────────────
+    config_path: str
+    te_config_path: str
+    effective_utilization_default: float
+
+    # ── Sync helpers ─────────────────────────────────────────────────────
+    def get_pool(self) -> Any:
+        """Return the asyncpg connection pool."""
+        ...
+
+    def load_config(self) -> dict:
+        """Return the current AE config (sync, cached/memoized read)."""
+        ...
+
+    def load_merged_config(self) -> dict:
+        """Return config.yml merged with te.yml overrides."""
+        ...
+
+    def get_chat_model(self, config: dict, role: str = "imperator") -> Any:
+        """Return a LangChain chat model for *role*."""
+        ...
+
+    def get_embeddings_model(self, config: dict, config_key: str = "embeddings") -> Any:
+        """Return a LangChain embeddings model."""
+        ...
+
+    def get_tuning(self, config: dict, key: str, default: Any = None) -> Any:
+        """Read a tuning parameter from *config*."""
+        ...
+
+    def get_api_key(self, provider_config: dict) -> str:
+        """Resolve an API key from provider config + env vars."""
+        ...
+
+    # ── Async helpers ────────────────────────────────────────────────────
+    def async_load_config(self) -> Awaitable[dict]:
+        """Return the current AE config (async variant)."""
+        ...
+
+    def async_load_prompt(self, name: str) -> Awaitable[str]:
+        """Load a prompt template by *name*."""
+        ...
+
+    def dispatch_tool(
+        self,
+        tool_name: str,
+        args: dict,
+        config: dict,
+        state: Any,
+    ) -> Awaitable[dict]:
+        """Dispatch an AE core tool by name."""
+        ...
+
+    # ── Flow builders ────────────────────────────────────────────────────
+    def get_flow_builder(self, name: str) -> Optional[Callable]:
+        """Return a registered AE flow builder by *name*, or ``None``."""
+        ...
+
+
+# ── Module-level singleton ───────────────────────────────────────────────
+
+_ctx: Optional[TEContext] = None
+
+
+def initialize(ctx: TEContext) -> None:
+    """Set the module-level TEContext singleton.
+
+    Called once by the AE bootstrap before the Imperator flow is compiled.
+    """
+    global _ctx
+    _ctx = ctx
+
+
+def get_ctx() -> TEContext:
+    """Return the TEContext singleton.
+
+    Raises ``RuntimeError`` if ``initialize()`` has not been called yet.
+    """
+    if _ctx is None:
+        raise RuntimeError(
+            "TEContext not initialized — call base_pmad_te._ctx.initialize() "
+            "before using TE components."
+        )
+    return _ctx

base_pmad_te-0.1.0/src/base_pmad_te/_kernel_ctx.py

@@ -0,0 +1,106 @@
+"""
+KernelTEContext — concrete TEContext backed by ``app.*`` modules.
+
+This is the ONLY file in the TE package that imports from ``app.*``.
+It is the adapter boundary between the TE and the AE kernel.
+
+NOTE: This file lives in the TE package for now but conceptually belongs
+to the AE/kernel.  It will be moved when the template is created.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any, Callable, Optional
+
+
+@dataclass
+class KernelTEContext:
+    """Delegate every TEContext method to the corresponding ``app.*`` call."""
+
+    # ── Properties ───────────────────────────────────────────────────────
+
+    @property
+    def config_path(self) -> str:
+        from app.config import CONFIG_PATH
+
+        return CONFIG_PATH
+
+    @property
+    def te_config_path(self) -> str:
+        from app.config import TE_CONFIG_PATH
+
+        return TE_CONFIG_PATH
+
+    @property
+    def effective_utilization_default(self) -> float:
+        from app.budget import EFFECTIVE_UTILIZATION_DEFAULT
+
+        return EFFECTIVE_UTILIZATION_DEFAULT
+
+    # ── Sync helpers ─────────────────────────────────────────────────────
+
+    def get_pool(self) -> Any:
+        from app.database import get_pg_pool
+
+        return get_pg_pool()
+
+    def load_config(self) -> dict:
+        from app.config import load_config
+
+        return load_config()
+
+    def load_merged_config(self) -> dict:
+        from app.config import load_merged_config
+
+        return load_merged_config()
+
+    def get_chat_model(self, config: dict, role: str = "imperator") -> Any:
+        from app.config import get_chat_model
+
+        return get_chat_model(config, role=role)
+
+    def get_embeddings_model(self, config: dict, config_key: str = "embeddings") -> Any:
+        from app.config import get_embeddings_model
+
+        return get_embeddings_model(config, config_key=config_key)
+
+    def get_tuning(self, config: dict, key: str, default: Any = None) -> Any:
+        from app.config import get_tuning
+
+        return get_tuning(config, key, default)
+
+    def get_api_key(self, provider_config: dict) -> str:
+        from app.config import get_api_key
+
+        return get_api_key(provider_config)
+
+    # ── Async helpers ────────────────────────────────────────────────────
+
+    async def async_load_config(self) -> dict:
+        from app.config import async_load_config
+
+        return await async_load_config()
+
+    async def async_load_prompt(self, name: str) -> str:
+        from app.prompt_loader import async_load_prompt
+
+        return await async_load_prompt(name)
+
+    async def dispatch_tool(
+        self,
+        tool_name: str,
+        args: dict,
+        config: dict,
+        state: Any,
+    ) -> dict:
+        from app.flows.tool_dispatch import dispatch_tool
+
+        return await dispatch_tool(tool_name, args, config, state)
+
+    # ── Flow builders ────────────────────────────────────────────────────
+
+    def get_flow_builder(self, name: str) -> Optional[Callable]:
+        from app.stategraph_registry import get_flow_builder
+
+        return get_flow_builder(name)