@relayfx/sdk 0.0.48 → 0.0.50

package/dist/migrations/sqlite/0002_durable_tool_placement.sql

@@ -0,0 +1,71 @@
+CREATE TABLE "relay_tool_attempts" (
+	"tenant_id" TEXT NOT NULL DEFAULT 'system',
+	"id" TEXT NOT NULL,
+	"tool_call_id" TEXT NOT NULL,
+	"attempt_number" INTEGER NOT NULL,
+	"state" TEXT NOT NULL,
+	"worker_id" TEXT,
+	"claim_expires_at" INTEGER,
+	"error" TEXT,
+	"created_at" INTEGER NOT NULL DEFAULT (CAST(unixepoch('subsec') * 1000 AS INTEGER)),
+	"completed_at" INTEGER,
+	CONSTRAINT "pk_relay_tool_attempts" PRIMARY KEY ("tenant_id", "tool_call_id", "id")
+);
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "definition_json" TEXT;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "placement_kind" TEXT NOT NULL DEFAULT 'local';
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "placement_key" TEXT;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "state" TEXT NOT NULL DEFAULT 'requested';
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "wait_id" TEXT;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "idempotency_key" TEXT NOT NULL DEFAULT '';
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "available_at" INTEGER NOT NULL DEFAULT 0;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "active_attempt_id" TEXT;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "claim_owner" TEXT;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "claimed_at" INTEGER;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "claim_expires_at" INTEGER;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "external_outcome_json" TEXT;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "external_outcome_at" INTEGER;
+--> statement-breakpoint
+ALTER TABLE "relay_tool_calls" ADD COLUMN "updated_at" INTEGER NOT NULL DEFAULT 0;
+--> statement-breakpoint
+UPDATE "relay_tool_calls"
+SET "idempotency_key" = 'tool:' || length("tenant_id") || ':' || "tenant_id" || ':' ||
+	length("execution_id") || ':' || "execution_id" || ':' || length("id") || ':' || "id",
+	"available_at" = "created_at",
+	"updated_at" = "created_at",
+	"state" = CASE
+		WHEN EXISTS (
+			SELECT 1 FROM "relay_tool_results" result
+			WHERE result."tenant_id" = "relay_tool_calls"."tenant_id"
+				AND result."tool_call_id" = "relay_tool_calls"."id"
+				AND result."error" IS NULL
+		) THEN 'completed'
+		WHEN EXISTS (
+			SELECT 1 FROM "relay_tool_results" result
+			WHERE result."tenant_id" = "relay_tool_calls"."tenant_id"
+				AND result."tool_call_id" = "relay_tool_calls"."id"
+		) THEN 'failed'
+		ELSE 'requested'
+	END;
+--> statement-breakpoint
+CREATE UNIQUE INDEX "uq_relay_tool_attempts_number" ON "relay_tool_attempts" ("tenant_id", "tool_call_id", "attempt_number");
+--> statement-breakpoint
+CREATE INDEX "idx_relay_tool_attempts_call" ON "relay_tool_attempts" ("tenant_id", "tool_call_id", "attempt_number");
+--> statement-breakpoint
+CREATE INDEX "idx_relay_tool_calls_remote_claim" ON "relay_tool_calls" ("tenant_id", "placement_kind", "placement_key", "state", "available_at", "created_at", "id");
+--> statement-breakpoint
+CREATE INDEX "idx_relay_tool_calls_stale_claim" ON "relay_tool_calls" ("tenant_id", "state", "claim_expires_at");
+--> statement-breakpoint
+CREATE INDEX "idx_relay_tool_calls_client_pending" ON "relay_tool_calls" ("tenant_id", "placement_kind", "state", "created_at", "id");

package/dist/types/relay/client.d.ts

@@ -1,15 +1,22 @@
-import { AddressBook, AgentRegistry, EventLog, ExecutionService, SkillRegistry, WaitService } from "../runtime/index";
+import { AddressBook, AgentRegistry, EventLog, ExecutionService, SkillRegistry, ToolTransitionCoordinator, WaitService } from "../runtime/index";
 import type { Agent as BatonAgent } from "@batonfx/core";
 import { Address, Agent, Content, Envelope, Execution, Ids, Shared, Skill, Tool } from "../schema/index";
-import { ClusterRegistryRepository, EnvelopeRepository, ExecutionRepository, ScheduleRepository, SessionRepository, SteeringRepository } from "../store-sql/index";
+import { ChildExecutionRepository, ClusterRegistryRepository, EnvelopeRepository, ExecutionRepository, ScheduleRepository, SessionRepository, SteeringRepository, ToolCallRepository } from "../store-sql/index";
 import { Context, Effect, Layer, Schema, Stream } from "effect";
 import { Sharding, ShardingConfig } from "effect/unstable/cluster";
-import type { AckEnvelopeReadyInput, CancelExecutionAccepted, CancelExecutionInput, CancelScheduleInput, CancelScheduleResult, ClaimEnvelopeReadyInput, CreateScheduleInput, CreateScheduleResult, EnvelopeReadyAcked, EnvelopeReadyLease, EnvelopeReadyReleased, GetSessionInput, GetSessionResult, ListExecutionsInput, ListExecutionsResult, ListPendingApprovalsInput, ListRunnersResult, ListSchedulesInput, ListSchedulesResult, ListSessionsInput, ListSessionsResult, ListWaitsInput, PendingToolApprovalList, ReleaseEnvelopeReadyInput, ReplayExecutionInput, ReplayExecutionResult, ResolvePermissionInput, ResolveToolApprovalInput, RouteExecutionResult, StartExecutionInput, StartExecutionResult, SteerAccepted, SteerInput, StreamExecutionInput, SubmitInboundEnvelopeAccepted, SubmitInboundEnvelopeInput, WaitView, WakeAccepted, WakeInput } from "./operation";
+import type { AckEnvelopeReadyInput, CancelExecutionAccepted, CancelExecutionInput, CancelScheduleInput, CancelScheduleResult, ClaimEnvelopeReadyInput, CreateScheduleInput, CreateScheduleResult, EnvelopeReadyAcked, EnvelopeReadyLease, EnvelopeReadyReleased, ExecutionInspection, GetSessionInput, GetSessionResult, ListExecutionsInput, ListExecutionsResult, ListPendingApprovalsInput, ListRunnersResult, ListSchedulesInput, ListSchedulesResult, ListSessionsInput, ListSessionsResult, ListWaitsInput, PendingToolApprovalList, PendingToolCallList, ListPendingToolCallsInput, FulfillToolCallInput, ClaimToolWorkInput, CompleteToolWorkInput, ReleaseToolWorkInput, ListToolAttemptsInput, ToolAttemptList, ToolOutcomeAccepted, ToolWorkLease, ToolWorkReleased, ReleaseEnvelopeReadyInput, ReplayExecutionInput, ReplayExecutionResult, ResolvePermissionInput, ResolveToolApprovalInput, RouteExecutionResult, StartExecutionInput, StartExecutionResult, SpawnChildRunInput, SteerAccepted, SteerInput, StreamExecutionInput, SubmitInboundEnvelopeAccepted, SubmitInboundEnvelopeInput, WaitView, WakeAccepted, WakeInput } from "./operation";
 declare const ClientError_base: Schema.Class<ClientError, Schema.TaggedStruct<"ClientError", {
     readonly message: Schema.String;
 }>, import("effect/Cause").YieldableError>;
 export declare class ClientError extends ClientError_base {
 }
+declare const ExecutionNotFound_base: Schema.Class<ExecutionNotFound, Schema.TaggedStruct<"ExecutionNotFound", {
+    readonly execution_id: Schema.brand<Schema.String, "Relay.ExecutionId"> & {
+        make: (input: unknown, options?: import("effect/SchemaAST").ParseOptions) => string & import("effect/Brand").Brand<"Relay.ExecutionId">;
+    };
+}>, import("effect/Cause").YieldableError>;
+export declare class ExecutionNotFound extends ExecutionNotFound_base {
+}
 export interface StartExecutionByAddressInput {
     readonly address_id: Ids.AddressId;
     readonly session_id?: Ids.SessionId;
@@ -73,6 +80,7 @@ export interface Interface {
     readonly cancelExecution: (input: CancelExecutionInput) => Effect.Effect<CancelExecutionAccepted, ClientError>;
     readonly steer: (input: SteerInput) => Effect.Effect<SteerAccepted, ClientError>;
     readonly getExecution: (id: Ids.ExecutionId) => Effect.Effect<Execution.Execution | undefined, ClientError>;
+    readonly inspectExecution: (executionId: Ids.ExecutionId) => Effect.Effect<ExecutionInspection, ExecutionNotFound | ClientError>;
     readonly listExecutions: (input: ListExecutionsInput) => Effect.Effect<ListExecutionsResult, ClientError>;
     readonly listSessions: (input: ListSessionsInput) => Effect.Effect<ListSessionsResult, ClientError>;
     readonly getSession: (input: GetSessionInput) => Effect.Effect<GetSessionResult, ClientError>;
@@ -86,8 +94,14 @@ export interface Interface {
     readonly listPendingApprovals: (input: ListPendingApprovalsInput) => Effect.Effect<PendingToolApprovalList, ClientError>;
     readonly resolveToolApproval: (input: ResolveToolApprovalInput) => Effect.Effect<WakeAccepted, ClientError>;
     readonly resolvePermission: (input: ResolvePermissionInput) => Effect.Effect<WakeAccepted, ClientError>;
+    readonly listPendingToolCalls: (input: ListPendingToolCallsInput) => Effect.Effect<PendingToolCallList, ClientError>;
+    readonly fulfillToolCall: (input: FulfillToolCallInput) => Effect.Effect<ToolOutcomeAccepted, ClientError>;
+    readonly claimToolWork: (input: ClaimToolWorkInput) => Effect.Effect<ToolWorkLease | null, ClientError>;
+    readonly completeToolWork: (input: CompleteToolWorkInput) => Effect.Effect<ToolOutcomeAccepted, ClientError>;
+    readonly releaseToolWork: (input: ReleaseToolWorkInput) => Effect.Effect<ToolWorkReleased, ClientError>;
+    readonly listToolAttempts: (input: ListToolAttemptsInput) => Effect.Effect<ToolAttemptList, ClientError>;
     readonly submitInboundEnvelope: (input: SubmitInboundEnvelopeInput) => Effect.Effect<SubmitInboundEnvelopeAccepted, ClientError>;
-    readonly spawnChildRun: (input: Execution.SpawnChildRunInput) => Effect.Effect<Execution.ChildRunAccepted, ClientError>;
+    readonly spawnChildRun: (input: SpawnChildRunInput) => Effect.Effect<Execution.ChildRunAccepted, ClientError>;
     readonly claimEnvelopeReady: (input: ClaimEnvelopeReadyInput) => Effect.Effect<EnvelopeReadyLease | null, ClientError>;
     readonly ackEnvelopeReady: (input: AckEnvelopeReadyInput) => Effect.Effect<EnvelopeReadyAcked, ClientError>;
     readonly releaseEnvelopeReady: (input: ReleaseEnvelopeReadyInput) => Effect.Effect<EnvelopeReadyReleased, ClientError>;
@@ -98,7 +112,7 @@ export interface Interface {
 declare const Service_base: Context.ServiceClass<Service, "@relayfx/sdk/Client", Interface>;
 export declare class Service extends Service_base {
 }
-export declare const layerFromRuntime: Layer.Layer<Service, never, AgentRegistry.Service | AddressBook.Service | ExecutionService.Service | WaitService.Service | EventLog.Service | EnvelopeRepository.Service | ExecutionRepository.Service | SessionRepository.Service | SteeringRepository.Service | ClusterRegistryRepository.Service | ScheduleRepository.Service | SkillRegistry.Service | Sharding.Sharding | ShardingConfig.ShardingConfig>;
+export declare const layerFromRuntime: Layer.Layer<Service, never, AgentRegistry.Service | AddressBook.Service | ExecutionService.Service | WaitService.Service | EventLog.Service | EnvelopeRepository.Service | ExecutionRepository.Service | ToolCallRepository.Service | ChildExecutionRepository.Service | SessionRepository.Service | SteeringRepository.Service | ClusterRegistryRepository.Service | ScheduleRepository.Service | SkillRegistry.Service | ToolTransitionCoordinator.Service | Sharding.Sharding | ShardingConfig.ShardingConfig>;
 export declare const testLayer: (implementation: Interface) => Layer.Layer<Service, never, never>;
 export declare const registerAgentDefinition: (input: Agent.RegisterDefinitionPayload) => Effect.Effect<Agent.DefinitionRegistered, ClientError, Service>;
 export declare const registerAgent: (input: RegisterAgentInput) => Effect.Effect<Agent.DefinitionRegistered, ClientError, Service>;
@@ -124,14 +138,21 @@ export declare const listWaits: (input: ListWaitsInput) => Effect.Effect<readonl
 export declare const replayExecution: (input: ReplayExecutionInput) => Effect.Effect<ReplayExecutionResult, ClientError, Service>;
 export declare const listRunners: () => Effect.Effect<ListRunnersResult, ClientError, Service>;
 export declare const routeExecution: (executionId: string & import("effect/Brand").Brand<"Relay.ExecutionId">) => Effect.Effect<RouteExecutionResult, ClientError, Service>;
+export declare const inspectExecution: (executionId: string & import("effect/Brand").Brand<"Relay.ExecutionId">) => Effect.Effect<ExecutionInspection, ClientError | ExecutionNotFound, Service>;
 export declare const send: (input: Envelope.SendInput) => Effect.Effect<Envelope.EnvelopeAccepted, ClientError, Service>;
 export declare const streamExecution: (input: StreamExecutionInput) => Stream.Stream<Execution.ExecutionEvent, ClientError, Service>;
 export declare const wake: (input: WakeInput) => Effect.Effect<WakeAccepted, ClientError, Service>;
 export declare const listPendingApprovals: (input: ListPendingApprovalsInput) => Effect.Effect<PendingToolApprovalList, ClientError, Service>;
 export declare const resolveToolApproval: (input: ResolveToolApprovalInput) => Effect.Effect<WakeAccepted, ClientError, Service>;
 export declare const resolvePermission: (input: ResolvePermissionInput) => Effect.Effect<WakeAccepted, ClientError, Service>;
+export declare const listPendingToolCalls: (input: ListPendingToolCallsInput) => Effect.Effect<PendingToolCallList, ClientError, Service>;
+export declare const fulfillToolCall: (input: FulfillToolCallInput) => Effect.Effect<ToolOutcomeAccepted, ClientError, Service>;
+export declare const claimToolWork: (input: ClaimToolWorkInput) => Effect.Effect<ToolWorkLease | null, ClientError, Service>;
+export declare const completeToolWork: (input: CompleteToolWorkInput) => Effect.Effect<ToolOutcomeAccepted, ClientError, Service>;
+export declare const releaseToolWork: (input: ReleaseToolWorkInput) => Effect.Effect<ToolWorkReleased, ClientError, Service>;
+export declare const listToolAttempts: (input: ListToolAttemptsInput) => Effect.Effect<ToolAttemptList, ClientError, Service>;
 export declare const submitInboundEnvelope: (input: SubmitInboundEnvelopeInput) => Effect.Effect<SubmitInboundEnvelopeAccepted, ClientError, Service>;
-export declare const spawnChildRun: (input: Execution.SpawnChildRunInput) => Effect.Effect<Execution.ChildRunAccepted, ClientError, Service>;
+export declare const spawnChildRun: (input: SpawnChildRunInput) => Effect.Effect<Execution.ChildRunAccepted, ClientError, Service>;
 export declare const claimEnvelopeReady: (input: ClaimEnvelopeReadyInput) => Effect.Effect<EnvelopeReadyLease | null, ClientError, Service>;
 export declare const ackEnvelopeReady: (input: AckEnvelopeReadyInput) => Effect.Effect<EnvelopeReadyAcked, ClientError, Service>;
 export declare const releaseEnvelopeReady: (input: ReleaseEnvelopeReadyInput) => Effect.Effect<EnvelopeReadyReleased, ClientError, Service>;

package/dist/types/relay/index.d.ts

@@ -2,6 +2,7 @@ export * as AdapterOutbox from "./adapter-outbox";
 export * as Client from "./client";
 export * as Database from "./database";
 export * as Operation from "./operation";
+export * as ToolWorker from "./tool-worker";
 export { Address, Agent, Content, Envelope, Execution, Ids, Schedule, Shared, Skill, Tool } from "../schema/index";
 export { AddressBook, AddressResolution, AgentLoop, AgentRegistry, ActivityVersionRegistry, ArtifactStore, BlobStore, ChildRunService, EnvelopeService, EventLog, ExecutionEntity, ExecutionService, ExecutionWorkflow, LanguageModelService, ModelCallPolicy, PromptAssembler, SchedulerService, SchemaRegistry, SkillRegistry, ToolRuntime, WaitService, } from "../runtime/index";
 export { RunnerRuntime } from "../runtime/index";