agenthooksprotocol 0.1.0__py3-none-any.whl

agenthooksprotocol/__init__.py

@@ -0,0 +1,64 @@
+"""Asynchronous Agent Hooks Protocol harness and schema-generated models."""
+
+from . import (
+    capability,
+    candidate,
+    state,
+    diagnostics,
+    auth,
+    effect,
+    event,
+    generated,
+    path,
+    registration,
+    subscription,
+    tool,
+    transport,
+)
+from ._content import ContentContext as ContentContext
+from ._models import *  # noqa: F403
+from ._models import __all__ as _model_names
+from ._hooks import (
+    HookResult as HookResult,
+    Hooks as Hooks,
+    PendingInvocation as PendingInvocation,
+    HooksClosedError as HooksClosedError,
+)
+from .permission import Permission as Permission
+from .content import (
+    OwnedContentSource as OwnedContentSource,
+    ContentSources as ContentSources,
+)
+from .codec import Codec as Codec, IdentityCodec as IdentityCodec
+from .runtime import (
+    ProtocolError as ProtocolError,
+    OperationCancelledError as OperationCancelledError,
+)
+
+__all__ = list(_model_names) + [
+    "ContentContext",
+    "OwnedContentSource",
+    "ContentSources",
+    "Permission",
+    "state",
+    "candidate",
+    "diagnostics",
+    "auth",
+    "HooksClosedError",
+    "OperationCancelledError",
+    "capability",
+    "effect",
+    "event",
+    "path",
+    "registration",
+    "subscription",
+    "tool",
+    "transport",
+    "Hooks",
+    "HookResult",
+    "PendingInvocation",
+    "ProtocolError",
+    "Codec",
+    "IdentityCodec",
+    "generated",
+]

agenthooksprotocol/_boundaries.py

@@ -0,0 +1,498 @@
+# Generated by ahp-codegen. DO NOT EDIT.
+from __future__ import annotations
+from typing import TYPE_CHECKING, Any, cast
+
+if TYPE_CHECKING:
+    from ._hooks import HookResult, Hooks
+    from . import _models as models
+
+
+class BoundaryMixin:
+    """Schema-derived named boundaries; implemented by Hooks.dispatch."""
+
+    async def config_change_after(
+        self, input: models.ConfigChangeAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "config.change.after", input, **kwargs
+        )
+
+    async def config_change_before(
+        self, input: models.ConfigChangeBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "config.change.before", input, **kwargs
+        )
+
+    async def context_compact_after(
+        self, input: models.ContextCompactAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "context.compact.after", input, **kwargs
+        )
+
+    async def context_compact_before(
+        self, input: models.ContextCompactBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "context.compact.before", input, **kwargs
+        )
+
+    async def file_changed(
+        self, input: models.FileChangedInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("file.changed", input, **kwargs)
+
+    async def hook_failure(
+        self, input: models.HookFailureInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("hook.failure", input, **kwargs)
+
+    async def model_error(
+        self, input: models.ModelErrorInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("model.error", input, **kwargs)
+
+    async def model_request_before(
+        self, input: models.ModelRequestBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "model.request.before", input, **kwargs
+        )
+
+    async def model_response_after(
+        self, input: models.ModelResponseAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "model.response.after", input, **kwargs
+        )
+
+    async def model_switch_after(
+        self, input: models.ModelSwitchAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("model.switch.after", input, **kwargs)
+
+    async def model_switch_before(
+        self, input: models.ModelSwitchBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "model.switch.before", input, **kwargs
+        )
+
+    async def session_end(
+        self, input: models.SessionEndInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("session.end", input, **kwargs)
+
+    async def session_start(
+        self, input: models.SessionStartInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("session.start", input, **kwargs)
+
+    async def task_change_after(
+        self, input: models.TaskChangeAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("task.change.after", input, **kwargs)
+
+    async def task_change_before(
+        self, input: models.TaskChangeBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("task.change.before", input, **kwargs)
+
+    async def tool_after(
+        self, input: models.ToolAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("tool.after", input, **kwargs)
+
+    async def tool_batch_after(
+        self, input: models.ToolBatchAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("tool.batch.after", input, **kwargs)
+
+    async def tool_before(
+        self, input: models.ToolBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("tool.before", input, **kwargs)
+
+    async def tool_permission_request(
+        self, input: models.ToolPermissionRequestInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "tool.permission.request", input, **kwargs
+        )
+
+    async def tool_permission_resolved(
+        self, input: models.ToolPermissionResolvedInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "tool.permission.resolved", input, **kwargs
+        )
+
+    async def tool_progress(
+        self, input: models.ToolProgressInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("tool.progress", input, **kwargs)
+
+    async def turn_end(
+        self, input: models.TurnEndInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("turn.end", input, **kwargs)
+
+    async def turn_finish_before(
+        self, input: models.TurnFinishBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("turn.finish.before", input, **kwargs)
+
+    async def turn_progress(
+        self, input: models.TurnProgressInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("turn.progress", input, **kwargs)
+
+    async def turn_start(
+        self, input: models.TurnStartInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("turn.start", input, **kwargs)
+
+    async def user_attention(
+        self, input: models.UserAttentionInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch("user.attention", input, **kwargs)
+
+    async def user_elicitation_request(
+        self, input: models.UserElicitationRequestInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "user.elicitation.request", input, **kwargs
+        )
+
+    async def user_elicitation_result(
+        self, input: models.UserElicitationResultInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "user.elicitation.result", input, **kwargs
+        )
+
+    async def user_message_inbound(
+        self, input: models.UserMessageInboundInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "user.message.inbound", input, **kwargs
+        )
+
+    async def user_message_outbound(
+        self, input: models.UserMessageOutboundInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "user.message.outbound", input, **kwargs
+        )
+
+    async def workspace_change_after(
+        self, input: models.WorkspaceChangeAfterInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "workspace.change.after", input, **kwargs
+        )
+
+    async def workspace_change_before(
+        self, input: models.WorkspaceChangeBeforeInput | dict[str, Any], **kwargs: Any
+    ) -> HookResult:
+        return await cast("Hooks", self).dispatch(
+            "workspace.change.before", input, **kwargs
+        )
+
+
+CONTENT_SOURCE_SLOTS: dict[str, tuple[str, tuple[str, ...]]] = {
+    "config.change.after.items": (
+        "config.change.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "config.change.before.items": (
+        "config.change.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "context.compact.after.items": (
+        "context.compact.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "context.compact.after.summary": ("context.compact.after", ("summary",)),
+    "context.compact.before.instructions": (
+        "context.compact.before",
+        ("instructions",),
+    ),
+    "context.compact.before.items": (
+        "context.compact.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "file.changed.items": (
+        "file.changed",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "hook.failure.items": (
+        "hook.failure",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "model.error.items": (
+        "model.error",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "model.request.before.items": (
+        "model.request.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "model.response.after.items": (
+        "model.response.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "model.switch.after.items": (
+        "model.switch.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "model.switch.before.items": (
+        "model.switch.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "session.end.items": (
+        "session.end",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "session.start.items": (
+        "session.start",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "task.change.after.items": (
+        "task.change.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "task.change.before.items": (
+        "task.change.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "tool.after.file_changes_after": (
+        "tool.after",
+        (
+            "fileChanges",
+            "*",
+            "after",
+        ),
+    ),
+    "tool.after.file_changes_before": (
+        "tool.after",
+        (
+            "fileChanges",
+            "*",
+            "before",
+        ),
+    ),
+    "tool.after.items": (
+        "tool.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "tool.batch.after.items": (
+        "tool.batch.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "tool.before.items": (
+        "tool.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "tool.permission.request.items": (
+        "tool.permission.request",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "tool.permission.resolved.items": (
+        "tool.permission.resolved",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "tool.progress.items": (
+        "tool.progress",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "tool.progress.partial_output": ("tool.progress", ("partialOutput",)),
+    "turn.end.items": (
+        "turn.end",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "turn.finish.before.items": (
+        "turn.finish.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "turn.progress.delta": ("turn.progress", ("delta",)),
+    "turn.progress.items": (
+        "turn.progress",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "turn.start.items": (
+        "turn.start",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "user.attention.items": (
+        "user.attention",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "user.attention.attention_message": (
+        "user.attention",
+        (
+            "attention",
+            "message",
+            "*",
+        ),
+    ),
+    "user.attention.attention_title": (
+        "user.attention",
+        (
+            "attention",
+            "title",
+            "*",
+        ),
+    ),
+    "user.elicitation.request.items": (
+        "user.elicitation.request",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "user.elicitation.request.elicitation_request": (
+        "user.elicitation.request",
+        (
+            "elicitation",
+            "request",
+        ),
+    ),
+    "user.elicitation.result.items": (
+        "user.elicitation.result",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "user.elicitation.result.elicitation_result": (
+        "user.elicitation.result",
+        (
+            "elicitation",
+            "result",
+        ),
+    ),
+    "user.message.inbound.items": (
+        "user.message.inbound",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "user.message.inbound.message_text": (
+        "user.message.inbound",
+        (
+            "message",
+            "text",
+            "*",
+        ),
+    ),
+    "user.message.outbound.items": (
+        "user.message.outbound",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "user.message.outbound.message_payload": (
+        "user.message.outbound",
+        (
+            "message",
+            "payload",
+            "*",
+        ),
+    ),
+    "workspace.change.after.items": (
+        "workspace.change.after",
+        (
+            "items",
+            "*",
+        ),
+    ),
+    "workspace.change.before.items": (
+        "workspace.change.before",
+        (
+            "items",
+            "*",
+        ),
+    ),
+}