feat: add streaming state storage

crates/crank-registry/src/model.rs

@@ -1,8 +1,10 @@
 use crank_core::{
-    Agent, AgentId, AgentOperationBinding, AgentStatus, AgentVersion, AuthProfile, DescriptorId,
-    ExportMode, InvitationToken, InvocationLevel, InvocationLog, InvocationSource, MembershipRole,
-    Operation, OperationId, OperationStatus, PlatformApiKey, Protocol, SampleId, Secret, SecretId,
-    SecretVersion, UsagePeriod, UsageRollup, User, UserSessionId, Workspace, WorkspaceId,
+    Agent, AgentId, AgentOperationBinding, AgentStatus, AgentVersion, AsyncJobHandle, AsyncJobId,
+    AuthProfile, DescriptorId, ExecutionMode, ExportMode, InvitationToken, InvocationLevel,
+    InvocationLog, InvocationSource, JobStatus, MembershipRole, Operation, OperationId,
+    OperationStatus, PlatformApiKey, Protocol, SampleId, Secret, SecretId, SecretVersion,
+    StreamSession, StreamSessionId, StreamStatus, UsagePeriod, UsageRollup, User, UserSessionId,
+    Workspace, WorkspaceId,
 };
 use crank_mapping::MappingSet;
 use crank_schema::Schema;
@@ -42,6 +44,12 @@ define_registry_id!(YamlImportJobId);
 
 pub type RegistryOperation = Operation<Schema, MappingSet>;
 
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+pub struct Page<T> {
+    pub items: Vec<T>,
+    pub total: u64,
+}
+
 #[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
 pub struct WorkspaceRecord {
     pub workspace: Workspace,
@@ -95,6 +103,16 @@ pub struct SecretVersionRecord {
     pub secret_version: SecretVersion,
 }
 
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+pub struct StreamSessionRecord {
+    pub session: StreamSession,
+}
+
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+pub struct AsyncJobRecord {
+    pub job: AsyncJobHandle,
+}
+
 #[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
 pub struct InvocationLogRecord {
     pub log: InvocationLog,
@@ -432,6 +450,60 @@ pub struct RotateSecretRequest<'a> {
     pub created_by: Option<&'a crank_core::UserId>,
 }
 
+#[derive(Clone, Debug, PartialEq)]
+pub struct CreateStreamSessionRequest<'a> {
+    pub session: &'a StreamSession,
+}
+
+#[derive(Clone, Debug, PartialEq)]
+pub struct UpdateStreamSessionStateRequest<'a> {
+    pub session_id: &'a StreamSessionId,
+    pub current_status: StreamStatus,
+    pub next_status: StreamStatus,
+    pub cursor: Option<&'a Value>,
+    pub state: &'a Value,
+    pub expires_at: Option<&'a str>,
+    pub last_poll_at: Option<&'a str>,
+    pub closed_at: Option<&'a str>,
+}
+
+#[derive(Clone, Debug, PartialEq, Eq)]
+pub struct StreamSessionFilter<'a> {
+    pub workspace_id: &'a WorkspaceId,
+    pub agent_id: Option<&'a AgentId>,
+    pub operation_id: Option<&'a OperationId>,
+    pub status: Option<StreamStatus>,
+    pub mode: Option<ExecutionMode>,
+    pub limit: u32,
+}
+
+#[derive(Clone, Debug, PartialEq)]
+pub struct CreateAsyncJobRequest<'a> {
+    pub job: &'a AsyncJobHandle,
+}
+
+#[derive(Clone, Debug, PartialEq)]
+pub struct UpdateAsyncJobStatusRequest<'a> {
+    pub job_id: &'a AsyncJobId,
+    pub current_status: JobStatus,
+    pub next_status: JobStatus,
+    pub progress: &'a Value,
+    pub result: Option<&'a Value>,
+    pub error: Option<&'a Value>,
+    pub expires_at: Option<&'a str>,
+    pub updated_at: &'a str,
+    pub finished_at: Option<&'a str>,
+}
+
+#[derive(Clone, Debug, PartialEq, Eq)]
+pub struct AsyncJobFilter<'a> {
+    pub workspace_id: &'a WorkspaceId,
+    pub agent_id: Option<&'a AgentId>,
+    pub operation_id: Option<&'a OperationId>,
+    pub status: Option<JobStatus>,
+    pub limit: u32,
+}
+
 #[derive(Clone, Debug, PartialEq, Eq)]
 pub struct SaveSampleMetadataRequest<'a> {
     pub sample: &'a OperationSampleMetadata,