Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 25 additions & 1 deletion coding-agent/src/agentkit/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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,
Expand Down Expand Up @@ -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",
Expand Down
2 changes: 2 additions & 0 deletions coding-agent/src/agentkit/storage/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from agentkit.storage.checkpoint_fs import FSCheckpointStore
from agentkit.storage.protocols import (
ArtifactStore,
CheckpointStore,
DocIndex,
SessionStore,
Expand All @@ -9,6 +10,7 @@

__all__ = [
"CheckpointStore",
"ArtifactStore",
"DocIndex",
"FileSessionStore",
"FSCheckpointStore",
Expand Down
19 changes: 19 additions & 0 deletions coding-agent/tests/agentkit/test_public_exports.py
Original file line number Diff line number Diff line change
@@ -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"