diff --git a/coding-agent/src/agentkit/__init__.py b/coding-agent/src/agentkit/__init__.py index 7f6e6e2cb..e428cd997 100644 --- a/coding-agent/src/agentkit/__init__.py +++ b/coding-agent/src/agentkit/__init__.py @@ -5,7 +5,8 @@ Plugin, PluginRegistry — Plugin system Directive, Approve, Reject, AskUser, Checkpoint, MemoryRecord — Effect descriptions Entry, Tape, ForkTapeStore — Conversation history - TapeStore, DocIndex, SessionStore — Storage protocols + TapeStore, DocIndex, SessionStore, ArtifactStore — Storage protocols + TurnResult, SessionResult, ArtifactRef — Result models ToolSchema, ToolRegistry, tool — Tool system Channel, LocalChannel — Communication AgentConfig, load_config — Configuration @@ -48,7 +49,19 @@ ) from agentkit.runtime import HookRuntime, Lifecycle, Pipeline, PipelineContext from agentkit.checkpoint import CheckpointMeta, CheckpointService, CheckpointSnapshot +from agentkit.result import ( + ArtifactRef, + FailureSummary, + ResultRef, + SessionResult, + TurnResult, + VerificationSummary, + latest_turn_result, + result_from_turn_trace, + verification_summary_from_turn_trace, +) from agentkit.storage import ( + ArtifactStore, CheckpointStore, DocIndex, FSCheckpointStore, @@ -84,10 +97,21 @@ "CheckpointSnapshot", # Storage "CheckpointStore", + "ArtifactStore", "TapeStore", "DocIndex", "FSCheckpointStore", "SessionStore", + # Results + "ArtifactRef", + "FailureSummary", + "ResultRef", + "SessionResult", + "TurnResult", + "VerificationSummary", + "latest_turn_result", + "result_from_turn_trace", + "verification_summary_from_turn_trace", # Tools "ToolSchema", "ToolRegistry", diff --git a/coding-agent/src/agentkit/storage/__init__.py b/coding-agent/src/agentkit/storage/__init__.py index fa9351d14..b18282c52 100644 --- a/coding-agent/src/agentkit/storage/__init__.py +++ b/coding-agent/src/agentkit/storage/__init__.py @@ -1,5 +1,6 @@ from agentkit.storage.checkpoint_fs import FSCheckpointStore from agentkit.storage.protocols import ( + ArtifactStore, CheckpointStore, DocIndex, SessionStore, @@ -9,6 +10,7 @@ __all__ = [ "CheckpointStore", + "ArtifactStore", "DocIndex", "FileSessionStore", "FSCheckpointStore", diff --git a/coding-agent/tests/agentkit/test_public_exports.py b/coding-agent/tests/agentkit/test_public_exports.py new file mode 100644 index 000000000..44d341fd5 --- /dev/null +++ b/coding-agent/tests/agentkit/test_public_exports.py @@ -0,0 +1,19 @@ +"""Tests for stable agentkit package-level exports.""" + +import agentkit +from agentkit.storage import ArtifactStore + + +def test_agentkit_exports_result_models_and_reducers() -> None: + assert agentkit.ArtifactRef.__name__ == "ArtifactRef" + assert agentkit.ResultRef.__name__ == "ResultRef" + assert agentkit.TurnResult.__name__ == "TurnResult" + assert agentkit.SessionResult.__name__ == "SessionResult" + assert agentkit.VerificationSummary.__name__ == "VerificationSummary" + assert agentkit.FailureSummary.__name__ == "FailureSummary" + assert callable(agentkit.result_from_turn_trace) + assert callable(agentkit.verification_summary_from_turn_trace) + + +def test_agentkit_storage_exports_artifact_store_protocol() -> None: + assert ArtifactStore.__name__ == "ArtifactStore"