@effect-agent/testing 0.1.0-beta.19 → 0.1.0-beta.21

package/dist/index.d.mts

@@ -993,6 +993,42 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
     readonly createdAt: string;
     readonly deploymentId: string;
     readonly payload: {
+      readonly _tag: "AbortRequested";
+      readonly submissionId: string;
+      readonly author: string;
+      readonly reason: string;
+    } | {
+      readonly _tag: "ToolCallResolved";
+      readonly runId: string;
+      readonly toolCallId: string;
+      readonly resolution: "completed-with-result" | "failed-with-error" | "never-started" | "safe-retry";
+      readonly author: string;
+      readonly reason: string;
+    } | {
+      readonly _tag: "ToolCallPrepared";
+      readonly runId: string;
+      readonly turnId: string;
+      readonly turn: number;
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly parameters: Schema.Json;
+      readonly parametersDigest: string;
+    } | {
+      readonly _tag: "ToolCallUnknown";
+      readonly runId: string;
+      readonly turn: number;
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly reason: string;
+    } | {
+      readonly _tag: "ToolApprovalDecided";
+      readonly runId: string;
+      readonly turn: number;
+      readonly toolCallId: string;
+      readonly decision: "approved" | "denied";
+      readonly resolver: string;
+      readonly reason: string;
+    } | {
       readonly _tag: "ConversationCreated";
       readonly agentId: string;
       readonly definitions: {
@@ -1020,15 +1056,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly inputTokens?: number | undefined;
       readonly outputTokens?: number | undefined;
       readonly costMicrousd?: number | undefined;
-    } | {
-      readonly _tag: "ToolCallPrepared";
-      readonly runId: string;
-      readonly turnId: string;
-      readonly turn: number;
-      readonly toolCallId: string;
-      readonly toolName: string;
-      readonly parameters: Schema.Json;
-      readonly parametersDigest: string;
     } | {
       readonly _tag: "ToolCallSettled";
       readonly runId: string;
@@ -1036,20 +1063,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly toolName: string;
       readonly result: Schema.Json;
       readonly isFailure: boolean;
-    } | {
-      readonly _tag: "ToolCallUnknown";
-      readonly runId: string;
-      readonly turn: number;
-      readonly toolCallId: string;
-      readonly toolName: string;
-      readonly reason: string;
-    } | {
-      readonly _tag: "ToolCallResolved";
-      readonly runId: string;
-      readonly toolCallId: string;
-      readonly resolution: "completed-with-result" | "failed-with-error" | "never-started" | "safe-retry";
-      readonly author: string;
-      readonly reason: string;
     } | {
       readonly _tag: "ToolStepSettled";
       readonly runId: string;
@@ -1065,14 +1078,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly toolCallId: string;
       readonly toolName: string;
       readonly parametersDigest: string;
-    } | {
-      readonly _tag: "ToolApprovalDecided";
-      readonly runId: string;
-      readonly turn: number;
-      readonly toolCallId: string;
-      readonly decision: "approved" | "denied";
-      readonly resolver: string;
-      readonly reason: string;
     } | {
       readonly _tag: "ModelResponseInterrupted";
       readonly runId: string;
@@ -1094,11 +1099,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly _tag: "RunCompleted";
       readonly runId: string;
       readonly output: Schema.Json;
-    } | {
-      readonly _tag: "AbortRequested";
-      readonly submissionId: string;
-      readonly author: string;
-      readonly reason: string;
     } | {
       readonly _tag: "SubmissionSettled";
       readonly submissionId: string;
@@ -1182,6 +1182,42 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
     readonly createdAt: string;
     readonly deploymentId: string;
     readonly payload: {
+      readonly _tag: "AbortRequested";
+      readonly submissionId: string;
+      readonly author: string;
+      readonly reason: string;
+    } | {
+      readonly _tag: "ToolCallResolved";
+      readonly runId: string;
+      readonly toolCallId: string;
+      readonly resolution: "completed-with-result" | "failed-with-error" | "never-started" | "safe-retry";
+      readonly author: string;
+      readonly reason: string;
+    } | {
+      readonly _tag: "ToolCallPrepared";
+      readonly runId: string;
+      readonly turnId: string;
+      readonly turn: number;
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly parameters: Schema.Json;
+      readonly parametersDigest: string;
+    } | {
+      readonly _tag: "ToolCallUnknown";
+      readonly runId: string;
+      readonly turn: number;
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly reason: string;
+    } | {
+      readonly _tag: "ToolApprovalDecided";
+      readonly runId: string;
+      readonly turn: number;
+      readonly toolCallId: string;
+      readonly decision: "approved" | "denied";
+      readonly resolver: string;
+      readonly reason: string;
+    } | {
       readonly _tag: "ConversationCreated";
       readonly agentId: string;
       readonly definitions: {
@@ -1209,15 +1245,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly inputTokens?: number | undefined;
       readonly outputTokens?: number | undefined;
       readonly costMicrousd?: number | undefined;
-    } | {
-      readonly _tag: "ToolCallPrepared";
-      readonly runId: string;
-      readonly turnId: string;
-      readonly turn: number;
-      readonly toolCallId: string;
-      readonly toolName: string;
-      readonly parameters: Schema.Json;
-      readonly parametersDigest: string;
     } | {
       readonly _tag: "ToolCallSettled";
       readonly runId: string;
@@ -1225,20 +1252,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly toolName: string;
       readonly result: Schema.Json;
       readonly isFailure: boolean;
-    } | {
-      readonly _tag: "ToolCallUnknown";
-      readonly runId: string;
-      readonly turn: number;
-      readonly toolCallId: string;
-      readonly toolName: string;
-      readonly reason: string;
-    } | {
-      readonly _tag: "ToolCallResolved";
-      readonly runId: string;
-      readonly toolCallId: string;
-      readonly resolution: "completed-with-result" | "failed-with-error" | "never-started" | "safe-retry";
-      readonly author: string;
-      readonly reason: string;
     } | {
       readonly _tag: "ToolStepSettled";
       readonly runId: string;
@@ -1254,14 +1267,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly toolCallId: string;
       readonly toolName: string;
       readonly parametersDigest: string;
-    } | {
-      readonly _tag: "ToolApprovalDecided";
-      readonly runId: string;
-      readonly turn: number;
-      readonly toolCallId: string;
-      readonly decision: "approved" | "denied";
-      readonly resolver: string;
-      readonly reason: string;
     } | {
       readonly _tag: "ModelResponseInterrupted";
       readonly runId: string;
@@ -1283,11 +1288,6 @@ declare const phase3TravelPlannerEncodedFixture: readonly {
       readonly _tag: "RunCompleted";
       readonly runId: string;
       readonly output: Schema.Json;
-    } | {
-      readonly _tag: "AbortRequested";
-      readonly submissionId: string;
-      readonly author: string;
-      readonly reason: string;
     } | {
       readonly _tag: "SubmissionSettled";
       readonly submissionId: string;

package/dist/index.mjs

@@ -2,7 +2,7 @@ import { McpConnectionRequest, McpConnector, McpServerIdentity, McpToolkitMismat
 import { Agent, AgentId, AgentPolicy, ConversationId, IdGenerator, RunId, SubmissionId, ToolCallId, TurnId } from "@effect-agent/core";
 import { DurableStep, DurableStepError, ToolExecutionClass } from "@effect-agent/engine";
 import { AbortCommand, AgentBindingResolver, ApprovalDecisionCommand, BatchId, CanonicalBatch, CanonicalRecordEnvelope, CertificationReport, CertificationSweepResult, CertificationTierThreeReport, CertifiedAdapterIdentity, ConversationCheckpoint, ConversationExportRequest, ConversationProjection, ConversationStore, DEFAULT_OWNERSHIP_LEASE_DURATION, DefinitionDigests, DeploymentId, Digest, DurableAgentRuntime, DurableRuntimeConfig, DurableRuntimeFailpoint, DurableRuntimeFailpointError, DurableRuntimeFailpointLocation, DurableRuntimeFailpointTestControl, DurableWorkerBinding, IdempotencyKey, LoadCheckpointRequest, ObligationThresholds, PersistedJson, Principal, ProducerId, ReconciliationCompleted, ReconciliationSafeToRetry, ReconciliationUncertain, RecordEnvelope, RecordId, ResolutionAbortSubmission, ResolutionCompletedWithResult, ResolutionNeverHappened, ResolutionSafeToRetry, SubmissionLedger, SubmissionLookupById, ToolReconciler, ToolReconcilerError, UnknownResolutionCommand, WakeScheduler, certifyPorts, childConversationIdFor, verifyConversationInvariants } from "@effect-agent/session";
-import { Cause, Clock, Context, DateTime, Deferred, Duration, Effect, Exit, Fiber, Layer, Option, Queue, Ref, Schema, Stream } from "effect";
+import { Cause, Clock, Context, DateTime, Deferred, Duration, Effect, Exit, Fiber, JsonSchema, Layer, Option, Queue, Ref, Schema, Stream } from "effect";
 import { FastCheck, TestClock } from "effect/testing";
 import { AiError, LanguageModel, Model, Response, Tool, Toolkit } from "effect/unstable/ai";
 import { CodeExecutionHost, CodeExecutionLimits, CodeExecutionNamespace, CodeExecutionProtocolError, CodeExecutionRequest, CodeExecutionResourceUse, CodeExecutionResult, CodeExecutionTimeoutError, CodeExecutor, CodeExecutorUnsupportedError, CodeHostCall, CodeHostCallFailure, CodeHostCallLimitError, CodeHostCallResult, CodeHostCallSuccess, CodeOutputLimitError, CodeProgramFailedError, CodeSourceError, NetworkAllowlist, NetworkDisabled, SandboxImplementation } from "@effect-agent/sandbox";
@@ -2387,10 +2387,25 @@ const docsMcpIdentity = McpServerIdentity.make({
 		version: "1.0.0"
 	})
 });
+/**
+* `Tool.getJsonSchema` produces a `$ref`/`$defs`-shaped schema for
+* `FetchDocument`'s named, refined parameters type, but `McpSchema.Tool`'s
+* `inputSchema` requires a flat `{ type: "object", ... }` root — the shape a
+* real MCP server advertises on the wire. This inlines the single top-level
+* `$ref` so the derivation described above still holds byte-for-byte.
+*/
+const flattenTopLevelRef = (schema) => {
+	const ref = schema["$ref"];
+	const defs = schema["$defs"];
+	if (typeof ref !== "string" || defs === void 0) return schema;
+	return JsonSchema.resolve$ref(ref, defs) ?? schema;
+};
+const fetchDocumentOutputSchema = flattenTopLevelRef(Tool.getJsonSchemaFromSchema(FetchDocument.successSchema));
 const discoveredFetchDocument = McpSchema.Tool.make({
 	name: FetchDocument.name,
 	description: "Fetch one bounded research document by its identifier.",
-	inputSchema: Tool.getJsonSchema(FetchDocument)
+	inputSchema: flattenTopLevelRef(Tool.getJsonSchema(FetchDocument)),
+	...fetchDocumentOutputSchema.type === "object" ? { outputSchema: fetchDocumentOutputSchema } : {}
 });
 const scriptedConnector = (tools) => Layer.succeed(McpConnector)({ connect: () => Effect.acquireRelease(Effect.succeed({
 	identity: docsMcpIdentity,