aigp-langchain 1.0.0__tar.gz

aigp_langchain-1.0.0/.gitignore

@@ -0,0 +1,40 @@
+# Python
+__pycache__/
+*.py[cod]
+*.egg-info/
+dist/
+build/
+.venv/
+.eggs/
+
+# Node
+node_modules/
+dist/
+
+# Go
+bin/
+
+# Rust
+target/
+
+# .NET
+bin/
+obj/
+
+# Java
+*.class
+out/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Secrets (never commit)
+.env
+*.pem
+*.key

aigp_langchain-1.0.0/PKG-INFO

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: aigp-langchain
+Version: 1.0.0
+Summary: AIGP governance adapter for LangChain/LangGraph
+Requires-Python: >=3.11
+Requires-Dist: aigp-agent-core>=1.0.0

aigp_langchain-1.0.0/aigp_langchain/__init__.py

@@ -0,0 +1,7 @@
+"""AIGP-LangChain — Governance adapter for LangChain/LangGraph."""
+
+__version__ = "1.0.0"
+
+from .handler import AigpCallbackHandler
+
+__all__ = ["AigpCallbackHandler"]

aigp_langchain-1.0.0/aigp_langchain/handler.py

@@ -0,0 +1,100 @@
+"""LangChain Callback Handler — maps LangChain lifecycle to AIGP governance."""
+
+import asyncio
+import logging
+import time
+from typing import Any
+from uuid import UUID
+
+from aigp_agent_core import AgentGovernance
+from aigp_agent_core.governance import GovernanceBlockedError
+
+logger = logging.getLogger(__name__)
+
+
+class AigpCallbackHandler:
+    """LangChain BaseCallbackHandler implementing AIGP governance.
+
+    Usage:
+        handler = AigpCallbackHandler(gov_url=..., app_id=..., hmac_secret=...)
+        agent = AgentExecutor(agent=..., tools=[...], callbacks=[handler])
+    """
+
+    def __init__(self, gov_url: str, app_id: str, hmac_secret: str, **kwargs):
+        self._gov = AgentGovernance(gov_url, app_id, hmac_secret, **kwargs)
+        self._root_run_id: UUID | None = None
+        self._tool_starts: dict[UUID, float] = {}
+
+    # --- Chain events ---
+
+    def on_chain_start(self, serialized: dict, inputs: dict, *, run_id: UUID, parent_run_id: UUID | None = None, **kwargs):
+        """Root chain start → AIGP CHECK."""
+        if parent_run_id is None:
+            self._root_run_id = run_id
+            agent_name = serialized.get("name", serialized.get("id", ["agent"])[-1])
+            model_id = kwargs.get("model_id", "unknown")
+            self._gov.pre_invoke(agent_name, model_id, kwargs.get("user_id", ""))
+
+    def on_chain_end(self, outputs: dict, *, run_id: UUID, parent_run_id: UUID | None = None, **kwargs):
+        """Root chain end → RECORD + TRACE + evidence."""
+        if parent_run_id is None and run_id == self._root_run_id:
+            self._run_async(self._gov.post_invoke())
+
+    def on_chain_error(self, error: BaseException, *, run_id: UUID, parent_run_id: UUID | None = None, **kwargs):
+        """Root chain error → RECORD with ERROR."""
+        if parent_run_id is None:
+            self._run_async(self._gov.on_error(error))
+
+    # --- LLM events ---
+
+    def on_llm_start(self, serialized: dict, prompts: list[str], *, run_id: UUID, **kwargs):
+        """LLM invocation starting."""
+        pass
+
+    def on_llm_end(self, response: Any, *, run_id: UUID, **kwargs):
+        """LLM invocation complete — accumulate tokens."""
+        usage = {}
+        if hasattr(response, "llm_output") and response.llm_output:
+            usage = response.llm_output.get("usage", response.llm_output.get("token_usage", {}))
+        elif hasattr(response, "usage_metadata") and response.usage_metadata:
+            usage = response.usage_metadata
+        self._gov.on_model_call(usage=usage)
+
+    def on_llm_error(self, error: BaseException, *, run_id: UUID, **kwargs):
+        pass
+
+    # --- Tool events ---
+
+    def on_tool_start(self, serialized: dict, input_str: str, *, run_id: UUID, **kwargs):
+        """Tool execution starting."""
+        self._tool_starts[run_id] = time.time()
+
+    def on_tool_end(self, output: str, *, run_id: UUID, **kwargs):
+        """Tool execution complete."""
+        start = self._tool_starts.pop(run_id, time.time())
+        duration_ms = int((time.time() - start) * 1000)
+        tool_name = kwargs.get("name", "unknown_tool")
+        self._run_async(self._gov.on_tool_call(tool_name, duration_ms=duration_ms))
+
+    def on_tool_error(self, error: BaseException, *, run_id: UUID, **kwargs):
+        self._tool_starts.pop(run_id, None)
+
+    # --- Retriever events ---
+
+    def on_retriever_start(self, serialized: dict, query: str, *, run_id: UUID, **kwargs):
+        pass
+
+    def on_retriever_end(self, documents: list, *, run_id: UUID, **kwargs):
+        pass
+
+    # --- Helpers ---
+
+    def _run_async(self, coro):
+        try:
+            loop = asyncio.get_event_loop()
+            if loop.is_running():
+                asyncio.ensure_future(coro)
+            else:
+                loop.run_until_complete(coro)
+        except RuntimeError:
+            asyncio.run(coro)

aigp_langchain-1.0.0/pyproject.toml

@@ -0,0 +1,10 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "aigp-langchain"
+version = "1.0.0"
+description = "AIGP governance adapter for LangChain/LangGraph"
+requires-python = ">=3.11"
+dependencies = ["aigp-agent-core>=1.0.0"]