effect-assistant 0.1.0 → 0.2.0

package/dist/internal/loop.js

@@ -1,9 +1,7 @@
-import { fromCallId } from "../DecisionId.js";
-import { CallOrigin, DecisionRequested, ModelOutput, ModelPart, ToolResult, ToolStarted } from "../AssistantEvent.js";
-import { TurnDecisions } from "./decisions.js";
+import { ModelOutput, ModelPart, ToolResult, ToolStarted } from "../AssistantEvent.js";
 import { unansweredCalls } from "./history.js";
 import { chatSpan, toolOutput, toolSpan } from "./telemetry.js";
-import { empty, failed, needsDecisionOf } from "./toolkit.js";
+import { empty, needingApproval, needsApprovalMessage } from "./toolkit.js";
 import { isTurnEnding } from "./tools.js";
 import * as Predicate from "effect/Predicate";
 import * as Context from "effect/Context";
@@ -27,7 +25,7 @@ const readHistory = (read) => "messages" in read ? {
 	throughSeq: void 0
 };
 const requestToolkit = (definitions) => ({
-	tools: Object.fromEntries(definitions.map((tool) => [tool.name, tool.needsApproval === void 0 ? tool : tool.setNeedsApproval(false)])),
+	tools: Object.fromEntries(definitions.map((tool) => [tool.name, tool])),
 	handle: () => Effect.succeed(Stream.empty)
 });
 const generate = (prompt, definitions, toolChoice) => LanguageModel.generateText({
@@ -50,35 +48,14 @@ const runCall = Effect.fnUntraced(function* (tools, call) {
 		...result
 	});
 });
-const park = Effect.fnUntraced(function* (call) {
-	if (Option.isNone(yield* TurnDecisions)) return ToolResult.make({
-		callId: call.id,
-		tool: call.name,
-		...failed(`${call.name} needs the user's decision, and only the assistant can ask them. Don't run it: say in your answer what needs deciding, with the details.`, "NeedsDecision")
-	});
-	return DecisionRequested.make({
-		id: fromCallId(call.id),
-		origin: CallOrigin.make({
-			callId: call.id,
-			tool: call.name,
-			params: call.params
-		})
-	});
-}, (effect, call) => effect.pipe(toolSpan(call)));
-const runCalls = (tools, pending, history, then) => Stream.unwrap(Effect.gen(function* () {
-	const parks = yield* Effect.forEach(pending, ({ call, approved }) => approved ? Effect.succeed(false) : needsDecisionOf(tools, call, history));
-	const started = pending.flatMap(({ call }, i) => parks[i] ? [] : [value(ToolStarted.make({
-		callId: call.id,
-		tool: call.name
-	}))]);
-	const results = Effect.forEach(pending, ({ call }, i) => parks[i] ? park(call) : runCall(tools, call), { concurrency: "unbounded" });
-	const reread = (done) => done.some((event) => Predicate.isTagged(event, "DecisionRequested") || isFailure(event));
-	return Stream.fromIterable(started).pipe(Stream.concat(Stream.unwrap(Effect.gen(function* () {
-		const done = yield* results;
-		return Stream.fromIterable(done.map(value)).pipe(Stream.concat(then(reread(done))));
-	}))));
-}));
-const isFailure = (event) => Predicate.isTagged(event, "ToolResult") && event.outcome === "failure";
+const runCalls = (tools, calls, then) => Stream.fromIterable(calls.map((call) => value(ToolStarted.make({
+	callId: call.id,
+	tool: call.name
+})))).pipe(Stream.concat(Stream.unwrap(Effect.gen(function* () {
+	const done = yield* Effect.forEach(calls, (call) => runCall(tools, call), { concurrency: "unbounded" });
+	return Stream.fromIterable(done.map(value)).pipe(Stream.concat(then(done.some(isFailure))));
+}))));
+const isFailure = (result) => result.outcome === "failure";
 const Steps = Data.taggedEnum();
 /** @internal */
 const value = (value) => Steps.Value({ value });
@@ -103,25 +80,16 @@ const step = (options, then) => {
 	const toolkit = options.toolkit ?? empty;
 	const instructions = Predicate.isString(options.instructions) ? { text: options.instructions } : options.instructions;
 	const after = (result) => Stream.suspend(() => then(result));
+	const needingApproval$1 = needingApproval(toolkit.definitions);
+	if (needingApproval$1.length > 0) return Stream.die(needsApprovalMessage(needingApproval$1));
 	return Stream.unwrap(Effect.gen(function* () {
 		const { messages: history, throughSeq } = readHistory(yield* options.history);
 		const cutOff = unansweredCalls(history);
-		const decisions = yield* TurnDecisions;
-		const approved = Option.isNone(decisions) ? [] : (yield* decisions.value.approved).filter((call) => toolkit.definitions.some((tool) => tool.name === call.name));
-		if (cutOff.length > 0 || approved.length > 0) {
-			const pending = [...cutOff.map((call) => ({
-				call,
-				approved: false
-			})), ...approved.map((call) => ({
-				call,
-				approved: true
-			}))];
-			return runCalls(toolkit, pending, history, () => after({
-				output: void 0,
-				done: false,
-				historyLength: history.length + pending.length
-			}));
-		}
+		if (cutOff.length > 0) return runCalls(toolkit, cutOff, () => after({
+			output: void 0,
+			done: false,
+			historyLength: history.length + cutOff.length
+		}));
 		const prompt = Prompt.fromMessages([Prompt.systemMessage({ content: instructions.text }), ...history]);
 		const name = yield* Effect.serviceOption(Model.ModelName);
 		const started = yield* Clock.currentTimeMillis;
@@ -137,12 +105,9 @@ const step = (options, then) => {
 		});
 		const calls = toolCalls(output.parts);
 		const historyLength = history.length + Prompt.fromResponseParts(output.parts).content.length + calls.length;
-		return Stream.make(value(output)).pipe(Stream.concat(runCalls(toolkit, calls.map((call) => ({
-			call,
-			approved: false
-		})), history, (reread) => after({
+		return Stream.make(value(output)).pipe(Stream.concat(runCalls(toolkit, calls, (failed) => after({
 			output,
-			done: calls.length === 0 || !reread && calls.some(isTurnEnding),
+			done: calls.length === 0 || !failed && calls.some(isTurnEnding),
 			historyLength
 		}))));
 	}));

package/dist/internal/loop.js.map

@@ -1 +1 @@
-{"version":3,"file":"loop.js","names":["AssistantEvent.ModelPart","Telemetry.toolOutput","Telemetry.toolSpan","Decisions.TurnDecisions","Toolkits.failed","DecisionId.fromCallId","Toolkits.needsDecisionOf","Toolkits.empty","History.unansweredCalls","Telemetry.chatSpan"],"sources":["../../src/internal/loop.ts"],"sourcesContent":["import type * as AiError from \"effect/ai/AiError\"\nimport * as LanguageModel from \"effect/ai/LanguageModel\"\nimport * as Model from \"effect/ai/Model\"\nimport * as Prompt from \"effect/ai/Prompt\"\nimport type * as Response from \"effect/ai/Response\"\nimport type * as Tool from \"effect/ai/Tool\"\nimport type * as Toolkit from \"effect/ai/Toolkit\"\nimport * as Clock from \"effect/Clock\"\nimport * as Context from \"effect/Context\"\nimport * as Data from \"effect/Data\"\nimport * as Effect from \"effect/Effect\"\nimport * as Option from \"effect/Option\"\nimport * as Predicate from \"effect/Predicate\"\nimport * as Schema from \"effect/Schema\"\nimport * as Stream from \"effect/Stream\"\nimport * as AssistantEvent from \"../AssistantEvent.ts\"\nimport * as DecisionId from \"../DecisionId.ts\"\nimport type * as Instructions from \"../Instructions.ts\"\nimport * as Decisions from \"./decisions.ts\"\nimport * as History from \"./history.ts\"\nimport * as Telemetry from \"./telemetry.ts\"\nimport * as Toolkits from \"./toolkit.ts\"\nimport { isTurnEnding } from \"./tools.ts\"\n\n/** @internal */\nexport type Event =\n  | AssistantEvent.ModelOutput\n  | AssistantEvent.ToolStarted\n  | AssistantEvent.ToolResult\n  | AssistantEvent.DecisionRequested\n\n// The assistant's own tools with the turn's services, so an app's own loop in a turn can message the user. A\n// specialist's run has none.\n/** @internal */\nexport const TurnTools = Context.Reference<Option.Option<Toolkits.Tools>>(\"effect-assistant/TurnTools\", {\n  defaultValue: Option.none\n})\n\n/** @internal */\nexport interface ChatHistory {\n  readonly messages: ReadonlyArray<Prompt.Message>\n  readonly throughSeq: number\n}\n\nconst readHistory = (read: ReadonlyArray<Prompt.Message> | ChatHistory) =>\n  \"messages\" in read\n    ? { messages: read.messages, throughSeq: read.throughSeq }\n    : { messages: read, throughSeq: undefined }\n\n// `AssistantLoop` takes native toolkits; turns and specialist runs come here with their tools already merged.\n/** @internal */\nexport interface StepOptions<R = never> {\n  readonly instructions: string | Instructions.Resolved\n  readonly toolkit?: Toolkits.Tools<R> | undefined\n  readonly history: Effect.Effect<ReadonlyArray<Prompt.Message> | ChatHistory>\n  readonly toolChoice?: LanguageModel.ToolChoice<string> | undefined\n}\n\n/** @internal */\nexport interface Options<R = never> extends StepOptions<R> {\n  readonly maxSteps?: number | undefined\n}\n\n// Its member docs show through `AssistantLoop.StepResult`.\n/** @internal */\nexport interface StepResult {\n  /** The model's output; unset when the step only ran calls a crash cut off. */\n  readonly output: AssistantEvent.ModelOutput | undefined\n  /**\n   * Whether the model is done: it made no tool calls, or ended its turn (`wait`, or a message\n   * with `wait: true`) and none of the step's calls failed.\n   */\n  readonly done: boolean\n  /**\n   * The history's length with this step's output and results: a longer one got something new meanwhile.\n   */\n  readonly historyLength: number\n}\n\n// The loop runs the calls itself, after storing each as started, so the model call runs none.\nconst requestToolkit = (definitions: ReadonlyArray<Tool.Any>): Toolkit.WithHandler<Record<string, Tool.Any>> => ({\n  // The loop decides which calls wait for the user's decision; the model call would ask for approvals of its own.\n  tools: Object.fromEntries(\n    definitions.map((tool) => [tool.name, tool.needsApproval === undefined ? tool : tool.setNeedsApproval(false)])\n  ),\n  handle: () => Effect.succeed(Stream.empty)\n})\n\n// Tool call resolution stays on with a toolkit that runs nothing: with it off, a call whose params don't fit its\n// tool's schema fails the whole model call, and here the model reads why instead.\nconst generate = (\n  prompt: Prompt.Prompt,\n  definitions: ReadonlyArray<Tool.Any>,\n  toolChoice: LanguageModel.ToolChoice<string> | undefined\n): Effect.Effect<\n  LanguageModel.GenerateTextResponse<Record<string, Tool.Any>, \"opaque\">,\n  AiError.AiError,\n  LanguageModel.LanguageModel\n> => LanguageModel.generateText({ prompt, toolkit: requestToolkit(definitions), toolChoice })\n\nconst decodePart = Schema.decodeUnknownOption(AssistantEvent.ModelPart)\n\nconst modelParts = (content: ReadonlyArray<Response.AllParts<Record<string, Tool.Any>>>) =>\n  content.flatMap((part) =>\n    Option.toArray(decodePart(part)).filter((decoded) => !(decoded.type === \"tool-call\" && decoded.providerExecuted))\n  )\n\nconst toolCalls = (parts: ReadonlyArray<AssistantEvent.ModelPart>): ReadonlyArray<AssistantEvent.ToolCallPart> =>\n  parts.flatMap((part) => (part.type === \"tool-call\" ? [part] : []))\n\nconst runCall = Effect.fnUntraced(function* <R>(tools: Toolkits.Tools<R>, call: AssistantEvent.ToolCallPart) {\n  const result = yield* Effect.gen(function* () {\n    const result = yield* tools.run(call)\n\n    yield* Telemetry.toolOutput(result.result)\n\n    return result\n  }).pipe(Telemetry.toolSpan(call))\n\n  return AssistantEvent.ToolResult.make({ callId: call.id, tool: call.name, ...result })\n})\n\ninterface Pending {\n  readonly call: AssistantEvent.ToolCallPart\n  readonly approved: boolean\n}\n\n// Only a turn parks: a specialist's call fails instead.\nconst park = Effect.fnUntraced(\n  function* (call: AssistantEvent.ToolCallPart) {\n    if (Option.isNone(yield* Decisions.TurnDecisions)) {\n      return AssistantEvent.ToolResult.make({\n        callId: call.id,\n        tool: call.name,\n        ...Toolkits.failed(\n          `${call.name} needs the user's decision, and only the assistant can ask them. Don't run it: say in your answer what needs deciding, with the details.`,\n          \"NeedsDecision\"\n        )\n      })\n    }\n\n    return AssistantEvent.DecisionRequested.make({\n      id: DecisionId.fromCallId(call.id),\n      origin: AssistantEvent.CallOrigin.make({ callId: call.id, tool: call.name, params: call.params })\n    })\n  },\n  (effect, call) => effect.pipe(Telemetry.toolSpan(call))\n)\n\n// Every `ToolStarted` is stored before any call runs. `then` is told whether the model has to read the results: one\n// failed, or waits for the user's decision.\nconst runCalls = <R, S, E, R2>(\n  tools: Toolkits.Tools<R>,\n  pending: ReadonlyArray<Pending>,\n  history: ReadonlyArray<Prompt.Message>,\n  then: (reread: boolean) => Stream.Stream<Step<never, S>, E, R2>\n): Stream.Stream<Step<Event, S>, E, R | R2> =>\n  Stream.unwrap(\n    Effect.gen(function* () {\n      const parks = yield* Effect.forEach(pending, ({ call, approved }) =>\n        approved ? Effect.succeed(false) : Toolkits.needsDecisionOf(tools, call, history)\n      )\n\n      const started = pending.flatMap(({ call }, i) =>\n        parks[i] ? [] : [value(AssistantEvent.ToolStarted.make({ callId: call.id, tool: call.name }))]\n      )\n\n      const results = Effect.forEach(pending, ({ call }, i) => (parks[i] ? park(call) : runCall(tools, call)), {\n        concurrency: \"unbounded\"\n      })\n\n      const reread = (done: ReadonlyArray<Event>) =>\n        done.some((event) => Predicate.isTagged(event, \"DecisionRequested\") || isFailure(event))\n\n      return Stream.fromIterable(started).pipe(\n        Stream.concat(\n          Stream.unwrap(\n            Effect.gen(function* () {\n              const done = yield* results\n\n              return Stream.fromIterable(done.map(value)).pipe(Stream.concat(then(reread(done))))\n            })\n          )\n        )\n      )\n    })\n  )\n\nconst isFailure = (event: Event) => Predicate.isTagged(event, \"ToolResult\") && event.outcome === \"failure\"\n\n/** @internal */\nexport type Step<A, S> = Data.TaggedEnum<{\n  Value: { readonly value: A }\n  Next: { readonly state: S }\n  Stop: Record<never, never>\n}>\n\ninterface StepDefinition extends Data.TaggedEnum.WithGenerics<2> {\n  readonly taggedEnum: Step<this[\"A\"], this[\"B\"]>\n}\n\nconst Steps = Data.taggedEnum<StepDefinition>()\n\n/** @internal */\nexport const value = <A>(value: A): Step<A, never> => Steps.Value({ value })\n\n/** @internal */\nexport const next = <S>(state: S): Stream.Stream<Step<never, S>> => Stream.succeed(Steps.Next({ state }))\n\n/** @internal */\n// oxlint-disable-next-line no-unnecessary-type-parameters -- `stop<number>()` types a loop's state where inference can't.\nexport const stop = <S = never>(): Stream.Stream<Step<never, S>> => Stream.succeed(Steps.Stop())\n\nconst isValue = <A, S>(step: Step<A, S>): step is Data.TaggedEnum.Value<Step<A, S>, \"Value\"> =>\n  Predicate.isTagged(step, \"Value\")\n\n// A body that ends without `next` or `stop` stops too.\n/** @internal */\nexport const loop = <S, A, E, R>(\n  initial: S,\n  body: (state: S) => Stream.Stream<Step<A, S>, E, R>\n): Stream.Stream<A, E, R> => {\n  const go = (state: S): Stream.Stream<A, E, R> =>\n    Stream.suspend(() => {\n      let after: Option.Option<S> = Option.none()\n\n      return body(state).pipe(\n        Stream.tap((step) =>\n          Effect.sync(() => (after = Predicate.isTagged(step, \"Next\") ? Option.some(step.state) : after))\n        ),\n        Stream.filter(isValue),\n        Stream.map((step) => step.value),\n        Stream.concat(Stream.suspend(() => Option.match(after, { onNone: () => Stream.empty, onSome: go })))\n      )\n    })\n\n  return go(initial)\n}\n\n/** @internal */\nexport const step = <R, S, E2 = never, R2 = never>(\n  options: StepOptions<R>,\n  then: (result: StepResult) => Stream.Stream<Step<never, S>, E2, R2>\n): Stream.Stream<Step<Event, S>, AiError.AiError | E2, LanguageModel.LanguageModel | R | R2> => {\n  const toolkit = options.toolkit ?? Toolkits.empty\n  const instructions = Predicate.isString(options.instructions) ? { text: options.instructions } : options.instructions\n\n  const after = (result: StepResult) => Stream.suspend(() => then(result))\n\n  return Stream.unwrap(\n    Effect.gen(function* () {\n      const { messages: history, throughSeq } = readHistory(yield* options.history)\n      // Calls a crash cut off, with no result: they run again, with the same ids, before the model goes on.\n      const cutOff = History.unansweredCalls(history)\n\n      // Calls the user approved: they run as the model made them, before the model goes on. Not in a loop without them.\n      const decisions = yield* Decisions.TurnDecisions\n\n      const approved = Option.isNone(decisions)\n        ? []\n        : (yield* decisions.value.approved).filter((call) =>\n            toolkit.definitions.some((tool) => tool.name === call.name)\n          )\n\n      if (cutOff.length > 0 || approved.length > 0) {\n        const pending = [\n          ...cutOff.map((call) => ({ call, approved: false })),\n          ...approved.map((call) => ({ call, approved: true }))\n        ]\n\n        return runCalls(toolkit, pending, history, () =>\n          after({ output: undefined, done: false, historyLength: history.length + pending.length })\n        )\n      }\n\n      const prompt = Prompt.fromMessages([Prompt.systemMessage({ content: instructions.text }), ...history])\n      const name = yield* Effect.serviceOption(Model.ModelName)\n      const started = yield* Clock.currentTimeMillis\n\n      const response = yield* generate(prompt, toolkit.definitions, options.toolChoice).pipe(\n        Telemetry.chatSpan(prompt, instructions.attributes)\n      )\n\n      const metadata = response.content.find((part) => part.type === \"response-metadata\")\n\n      const output = AssistantEvent.ModelOutput.make({\n        parts: modelParts(response.content),\n        seenThroughSeq: throughSeq,\n        model: metadata?.modelId ?? Option.getOrUndefined(name),\n        responseId: metadata?.id,\n        usage: response.usage,\n        durationMillis: (yield* Clock.currentTimeMillis) - started\n      })\n\n      const calls = toolCalls(output.parts)\n      const historyLength = history.length + Prompt.fromResponseParts(output.parts).content.length + calls.length\n\n      // A failed call doesn't end the turn: a message that failed never reached the user, so the model reads why. Nor\n      // does a parked one: the model asks the user for the decision it waits for.\n      return Stream.make(value(output)).pipe(\n        Stream.concat(\n          runCalls(\n            toolkit,\n            calls.map((call) => ({ call, approved: false })),\n            history,\n            (reread) =>\n              after({\n                output,\n                done: calls.length === 0 || (!reread && calls.some(isTurnEnding)),\n                historyLength\n              })\n          )\n        )\n      )\n    })\n  )\n}\n\n/** @internal */\nexport const run = <R = never>(\n  options: Options<R>\n): Stream.Stream<Event, AiError.AiError, LanguageModel.LanguageModel | R> => {\n  const maxSteps = options.maxSteps ?? 16\n\n  // Done: stop, unless something arrived while the model was working; then it reads that first.\n  const afterDone = (steps: number, historyLength: number) =>\n    Stream.unwrap(\n      Effect.gen(function* () {\n        const read = yield* options.history\n\n        return readHistory(read).messages.length > historyLength ? next(steps) : stop<number>()\n      })\n    )\n\n  return loop(0, (steps: number) =>\n    step(options, (result) => {\n      const taken = result.output === undefined ? steps : steps + 1\n\n      if (taken >= maxSteps) return stop<number>()\n\n      return result.done ? afterDone(taken, result.historyLength) : next(taken)\n    })\n  )\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAkCA,MAAa,YAAY,QAAQ,UAAyC,8BAA8B,EACtG,cAAc,OAAO,KACvB,CAAC;AAQD,MAAM,eAAe,SACnB,cAAc,OACV;CAAE,UAAU,KAAK;CAAU,YAAY,KAAK;AAAW,IACvD;CAAE,UAAU;CAAM,YAAY,KAAA;AAAU;AAiC9C,MAAM,kBAAkB,iBAAyF;CAE/G,OAAO,OAAO,YACZ,YAAY,KAAK,SAAS,CAAC,KAAK,MAAM,KAAK,kBAAkB,KAAA,IAAY,OAAO,KAAK,iBAAiB,KAAK,CAAC,CAAC,CAC/G;CACA,cAAc,OAAO,QAAQ,OAAO,KAAK;AAC3C;AAIA,MAAM,YACJ,QACA,aACA,eAKG,cAAc,aAAa;CAAE;CAAQ,SAAS,eAAe,WAAW;CAAG;AAAW,CAAC;AAE5F,MAAM,aAAa,OAAO,oBAAoBA,SAAwB;AAEtE,MAAM,cAAc,YAClB,QAAQ,SAAS,SACf,OAAO,QAAQ,WAAW,IAAI,CAAC,CAAC,CAAC,QAAQ,YAAY,EAAE,QAAQ,SAAS,eAAe,QAAQ,iBAAiB,CAClH;AAEF,MAAM,aAAa,UACjB,MAAM,SAAS,SAAU,KAAK,SAAS,cAAc,CAAC,IAAI,IAAI,CAAC,CAAE;AAEnE,MAAM,UAAU,OAAO,WAAW,WAAc,OAA0B,MAAmC;CAC3G,MAAM,SAAS,OAAO,OAAO,IAAI,aAAa;EAC5C,MAAM,SAAS,OAAO,MAAM,IAAI,IAAI;EAEpC,OAAOC,WAAqB,OAAO,MAAM;EAEzC,OAAO;CACT,CAAC,CAAC,CAAC,KAAKC,SAAmB,IAAI,CAAC;CAEhC,OAAA,WAAiC,KAAK;EAAE,QAAQ,KAAK;EAAI,MAAM,KAAK;EAAM,GAAG;CAAO,CAAC;AACvF,CAAC;AAQD,MAAM,OAAO,OAAO,WAClB,WAAW,MAAmC;CAC5C,IAAI,OAAO,OAAO,OAAOC,aAAuB,GAC9C,OAAA,WAAiC,KAAK;EACpC,QAAQ,KAAK;EACb,MAAM,KAAK;EACX,GAAGC,OACD,GAAG,KAAK,KAAK,2IACb,eACF;CACF,CAAC;CAGH,OAAA,kBAAwC,KAAK;EAC3C,IAAIC,WAAsB,KAAK,EAAE;EACjC,QAAA,WAAkC,KAAK;GAAE,QAAQ,KAAK;GAAI,MAAM,KAAK;GAAM,QAAQ,KAAK;EAAO,CAAC;CAClG,CAAC;AACH,IACC,QAAQ,SAAS,OAAO,KAAKH,SAAmB,IAAI,CAAC,CACxD;AAIA,MAAM,YACJ,OACA,SACA,SACA,SAEA,OAAO,OACL,OAAO,IAAI,aAAa;CACtB,MAAM,QAAQ,OAAO,OAAO,QAAQ,UAAU,EAAE,MAAM,eACpD,WAAW,OAAO,QAAQ,KAAK,IAAII,gBAAyB,OAAO,MAAM,OAAO,CAClF;CAEA,MAAM,UAAU,QAAQ,SAAS,EAAE,QAAQ,MACzC,MAAM,KAAK,CAAC,IAAI,CAAC,MAAA,YAAiC,KAAK;EAAE,QAAQ,KAAK;EAAI,MAAM,KAAK;CAAK,CAAC,CAAC,CAAC,CAC/F;CAEA,MAAM,UAAU,OAAO,QAAQ,UAAU,EAAE,QAAQ,MAAO,MAAM,KAAK,KAAK,IAAI,IAAI,QAAQ,OAAO,IAAI,GAAI,EACvG,aAAa,YACf,CAAC;CAED,MAAM,UAAU,SACd,KAAK,MAAM,UAAU,UAAU,SAAS,OAAO,mBAAmB,KAAK,UAAU,KAAK,CAAC;CAEzF,OAAO,OAAO,aAAa,OAAO,CAAC,CAAC,KAClC,OAAO,OACL,OAAO,OACL,OAAO,IAAI,aAAa;EACtB,MAAM,OAAO,OAAO;EAEpB,OAAO,OAAO,aAAa,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,OAAO,OAAO,KAAK,OAAO,IAAI,CAAC,CAAC,CAAC;CACpF,CAAC,CACH,CACF,CACF;AACF,CAAC,CACH;AAEF,MAAM,aAAa,UAAiB,UAAU,SAAS,OAAO,YAAY,KAAK,MAAM,YAAY;AAajG,MAAM,QAAQ,KAAK,WAA2B;;AAG9C,MAAa,SAAY,UAA6B,MAAM,MAAM,EAAE,MAAM,CAAC;;AAG3E,MAAa,QAAW,UAA4C,OAAO,QAAQ,MAAM,KAAK,EAAE,MAAM,CAAC,CAAC;;AAIxG,MAAa,aAAuD,OAAO,QAAQ,MAAM,KAAK,CAAC;AAE/F,MAAM,WAAiB,SACrB,UAAU,SAAS,MAAM,OAAO;;AAIlC,MAAa,QACX,SACA,SAC2B;CAC3B,MAAM,MAAM,UACV,OAAO,cAAc;EACnB,IAAI,QAA0B,OAAO,KAAK;EAE1C,OAAO,KAAK,KAAK,CAAC,CAAC,KACjB,OAAO,KAAK,SACV,OAAO,WAAY,QAAQ,UAAU,SAAS,MAAM,MAAM,IAAI,OAAO,KAAK,KAAK,KAAK,IAAI,KAAM,CAChG,GACA,OAAO,OAAO,OAAO,GACrB,OAAO,KAAK,SAAS,KAAK,KAAK,GAC/B,OAAO,OAAO,OAAO,cAAc,OAAO,MAAM,OAAO;GAAE,cAAc,OAAO;GAAO,QAAQ;EAAG,CAAC,CAAC,CAAC,CACrG;CACF,CAAC;CAEH,OAAO,GAAG,OAAO;AACnB;;AAGA,MAAa,QACX,SACA,SAC8F;CAC9F,MAAM,UAAU,QAAQ,WAAWC;CACnC,MAAM,eAAe,UAAU,SAAS,QAAQ,YAAY,IAAI,EAAE,MAAM,QAAQ,aAAa,IAAI,QAAQ;CAEzG,MAAM,SAAS,WAAuB,OAAO,cAAc,KAAK,MAAM,CAAC;CAEvE,OAAO,OAAO,OACZ,OAAO,IAAI,aAAa;EACtB,MAAM,EAAE,UAAU,SAAS,eAAe,YAAY,OAAO,QAAQ,OAAO;EAE5E,MAAM,SAASC,gBAAwB,OAAO;EAG9C,MAAM,YAAY,OAAOL;EAEzB,MAAM,WAAW,OAAO,OAAO,SAAS,IACpC,CAAC,KACA,OAAO,UAAU,MAAM,SAAA,CAAU,QAAQ,SACxC,QAAQ,YAAY,MAAM,SAAS,KAAK,SAAS,KAAK,IAAI,CAC5D;EAEJ,IAAI,OAAO,SAAS,KAAK,SAAS,SAAS,GAAG;GAC5C,MAAM,UAAU,CACd,GAAG,OAAO,KAAK,UAAU;IAAE;IAAM,UAAU;GAAM,EAAE,GACnD,GAAG,SAAS,KAAK,UAAU;IAAE;IAAM,UAAU;GAAK,EAAE,CACtD;GAEA,OAAO,SAAS,SAAS,SAAS,eAChC,MAAM;IAAE,QAAQ,KAAA;IAAW,MAAM;IAAO,eAAe,QAAQ,SAAS,QAAQ;GAAO,CAAC,CAC1F;EACF;EAEA,MAAM,SAAS,OAAO,aAAa,CAAC,OAAO,cAAc,EAAE,SAAS,aAAa,KAAK,CAAC,GAAG,GAAG,OAAO,CAAC;EACrG,MAAM,OAAO,OAAO,OAAO,cAAc,MAAM,SAAS;EACxD,MAAM,UAAU,OAAO,MAAM;EAE7B,MAAM,WAAW,OAAO,SAAS,QAAQ,QAAQ,aAAa,QAAQ,UAAU,CAAC,CAAC,KAChFM,SAAmB,QAAQ,aAAa,UAAU,CACpD;EAEA,MAAM,WAAW,SAAS,QAAQ,MAAM,SAAS,KAAK,SAAS,mBAAmB;EAElF,MAAM,SAAA,YAAoC,KAAK;GAC7C,OAAO,WAAW,SAAS,OAAO;GAClC,gBAAgB;GAChB,OAAO,UAAU,WAAW,OAAO,eAAe,IAAI;GACtD,YAAY,UAAU;GACtB,OAAO,SAAS;GAChB,iBAAiB,OAAO,MAAM,qBAAqB;EACrD,CAAC;EAED,MAAM,QAAQ,UAAU,OAAO,KAAK;EACpC,MAAM,gBAAgB,QAAQ,SAAS,OAAO,kBAAkB,OAAO,KAAK,CAAC,CAAC,QAAQ,SAAS,MAAM;EAIrG,OAAO,OAAO,KAAK,MAAM,MAAM,CAAC,CAAC,CAAC,KAChC,OAAO,OACL,SACE,SACA,MAAM,KAAK,UAAU;GAAE;GAAM,UAAU;EAAM,EAAE,GAC/C,UACC,WACC,MAAM;GACJ;GACA,MAAM,MAAM,WAAW,KAAM,CAAC,UAAU,MAAM,KAAK,YAAY;GAC/D;EACF,CAAC,CACL,CACF,CACF;CACF,CAAC,CACH;AACF;;AAGA,MAAa,OACX,YAC2E;CAC3E,MAAM,WAAW,QAAQ,YAAY;CAGrC,MAAM,aAAa,OAAe,kBAChC,OAAO,OACL,OAAO,IAAI,aAAa;EACtB,MAAM,OAAO,OAAO,QAAQ;EAE5B,OAAO,YAAY,IAAI,CAAC,CAAC,SAAS,SAAS,gBAAgB,KAAK,KAAK,IAAI,KAAa;CACxF,CAAC,CACH;CAEF,OAAO,KAAK,IAAI,UACd,KAAK,UAAU,WAAW;EACxB,MAAM,QAAQ,OAAO,WAAW,KAAA,IAAY,QAAQ,QAAQ;EAE5D,IAAI,SAAS,UAAU,OAAO,KAAa;EAE3C,OAAO,OAAO,OAAO,UAAU,OAAO,OAAO,aAAa,IAAI,KAAK,KAAK;CAC1E,CAAC,CACH;AACF"}
+{"version":3,"file":"loop.js","names":["AssistantEvent.ModelPart","Telemetry.toolOutput","Telemetry.toolSpan","Toolkits.empty","needingApproval","Toolkits.needingApproval","Toolkits.needsApprovalMessage","History.unansweredCalls","Telemetry.chatSpan"],"sources":["../../src/internal/loop.ts"],"sourcesContent":["import type * as AiError from \"effect/ai/AiError\"\nimport * as LanguageModel from \"effect/ai/LanguageModel\"\nimport * as Model from \"effect/ai/Model\"\nimport * as Prompt from \"effect/ai/Prompt\"\nimport type * as Response from \"effect/ai/Response\"\nimport type * as Tool from \"effect/ai/Tool\"\nimport type * as Toolkit from \"effect/ai/Toolkit\"\nimport * as Clock from \"effect/Clock\"\nimport * as Context from \"effect/Context\"\nimport * as Data from \"effect/Data\"\nimport * as Effect from \"effect/Effect\"\nimport * as Option from \"effect/Option\"\nimport * as Predicate from \"effect/Predicate\"\nimport * as Schema from \"effect/Schema\"\nimport * as Stream from \"effect/Stream\"\nimport * as AssistantEvent from \"../AssistantEvent.ts\"\nimport type * as Instructions from \"../Instructions.ts\"\nimport * as History from \"./history.ts\"\nimport * as Telemetry from \"./telemetry.ts\"\nimport * as Toolkits from \"./toolkit.ts\"\nimport { isTurnEnding } from \"./tools.ts\"\n\n/** @internal */\nexport type Event = AssistantEvent.ModelOutput | AssistantEvent.ToolStarted | AssistantEvent.ToolResult\n\n// The assistant's own tools with the turn's services, so an app's own loop in a turn can message the user. A\n// specialist's run has none.\n/** @internal */\nexport const TurnTools = Context.Reference<Option.Option<Toolkits.Tools>>(\"effect-assistant/TurnTools\", {\n  defaultValue: Option.none\n})\n\n/** @internal */\nexport interface ChatHistory {\n  readonly messages: ReadonlyArray<Prompt.Message>\n  readonly throughSeq: number\n}\n\nconst readHistory = (read: ReadonlyArray<Prompt.Message> | ChatHistory) =>\n  \"messages\" in read\n    ? { messages: read.messages, throughSeq: read.throughSeq }\n    : { messages: read, throughSeq: undefined }\n\n// `AssistantLoop` takes native toolkits; turns and specialist runs come here with their tools already merged.\n/** @internal */\nexport interface StepOptions<R = never> {\n  readonly instructions: string | Instructions.Resolved\n  readonly toolkit?: Toolkits.Tools<R> | undefined\n  readonly history: Effect.Effect<ReadonlyArray<Prompt.Message> | ChatHistory>\n  readonly toolChoice?: LanguageModel.ToolChoice<string> | undefined\n}\n\n/** @internal */\nexport interface Options<R = never> extends StepOptions<R> {\n  readonly maxSteps?: number | undefined\n}\n\n// Its member docs show through `AssistantLoop.StepResult`.\n/** @internal */\nexport interface StepResult {\n  /** The model's output; unset when the step only ran calls a crash cut off. */\n  readonly output: AssistantEvent.ModelOutput | undefined\n  /**\n   * Whether the model is done: it made no tool calls, or ended its turn (`wait`, or a message\n   * with `wait: true`) and none of the step's calls failed.\n   */\n  readonly done: boolean\n  /**\n   * The history's length with this step's output and results: a longer one got something new meanwhile.\n   */\n  readonly historyLength: number\n}\n\n// The loop runs the calls itself, after storing each as started, so the model call runs none.\nconst requestToolkit = (definitions: ReadonlyArray<Tool.Any>): Toolkit.WithHandler<Record<string, Tool.Any>> => ({\n  tools: Object.fromEntries(definitions.map((tool) => [tool.name, tool])),\n  handle: () => Effect.succeed(Stream.empty)\n})\n\n// Tool call resolution stays on with a toolkit that runs nothing: with it off, a call whose params don't fit its\n// tool's schema fails the whole model call, and here the model reads why instead.\nconst generate = (\n  prompt: Prompt.Prompt,\n  definitions: ReadonlyArray<Tool.Any>,\n  toolChoice: LanguageModel.ToolChoice<string> | undefined\n): Effect.Effect<\n  LanguageModel.GenerateTextResponse<Record<string, Tool.Any>, \"opaque\">,\n  AiError.AiError,\n  LanguageModel.LanguageModel\n> => LanguageModel.generateText({ prompt, toolkit: requestToolkit(definitions), toolChoice })\n\nconst decodePart = Schema.decodeUnknownOption(AssistantEvent.ModelPart)\n\nconst modelParts = (content: ReadonlyArray<Response.AllParts<Record<string, Tool.Any>>>) =>\n  content.flatMap((part) =>\n    Option.toArray(decodePart(part)).filter((decoded) => !(decoded.type === \"tool-call\" && decoded.providerExecuted))\n  )\n\nconst toolCalls = (parts: ReadonlyArray<AssistantEvent.ModelPart>): ReadonlyArray<AssistantEvent.ToolCallPart> =>\n  parts.flatMap((part) => (part.type === \"tool-call\" ? [part] : []))\n\nconst runCall = Effect.fnUntraced(function* <R>(tools: Toolkits.Tools<R>, call: AssistantEvent.ToolCallPart) {\n  const result = yield* Effect.gen(function* () {\n    const result = yield* tools.run(call)\n\n    yield* Telemetry.toolOutput(result.result)\n\n    return result\n  }).pipe(Telemetry.toolSpan(call))\n\n  return AssistantEvent.ToolResult.make({ callId: call.id, tool: call.name, ...result })\n})\n\n// Every `ToolStarted` is stored before any call runs. `then` is told whether one of the calls failed, so the model\n// has to read the results.\nconst runCalls = <R, S, E, R2>(\n  tools: Toolkits.Tools<R>,\n  calls: ReadonlyArray<AssistantEvent.ToolCallPart>,\n  then: (failed: boolean) => Stream.Stream<Step<never, S>, E, R2>\n): Stream.Stream<Step<Event, S>, E, R | R2> =>\n  Stream.fromIterable(\n    calls.map((call) => value(AssistantEvent.ToolStarted.make({ callId: call.id, tool: call.name })))\n  ).pipe(\n    Stream.concat(\n      Stream.unwrap(\n        Effect.gen(function* () {\n          const done = yield* Effect.forEach(calls, (call) => runCall(tools, call), { concurrency: \"unbounded\" })\n\n          return Stream.fromIterable(done.map(value)).pipe(Stream.concat(then(done.some(isFailure))))\n        })\n      )\n    )\n  )\n\nconst isFailure = (result: AssistantEvent.ToolResult) => result.outcome === \"failure\"\n\n/** @internal */\nexport type Step<A, S> = Data.TaggedEnum<{\n  Value: { readonly value: A }\n  Next: { readonly state: S }\n  Stop: Record<never, never>\n}>\n\ninterface StepDefinition extends Data.TaggedEnum.WithGenerics<2> {\n  readonly taggedEnum: Step<this[\"A\"], this[\"B\"]>\n}\n\nconst Steps = Data.taggedEnum<StepDefinition>()\n\n/** @internal */\nexport const value = <A>(value: A): Step<A, never> => Steps.Value({ value })\n\n/** @internal */\nexport const next = <S>(state: S): Stream.Stream<Step<never, S>> => Stream.succeed(Steps.Next({ state }))\n\n/** @internal */\n// oxlint-disable-next-line no-unnecessary-type-parameters -- `stop<number>()` types a loop's state where inference can't.\nexport const stop = <S = never>(): Stream.Stream<Step<never, S>> => Stream.succeed(Steps.Stop())\n\nconst isValue = <A, S>(step: Step<A, S>): step is Data.TaggedEnum.Value<Step<A, S>, \"Value\"> =>\n  Predicate.isTagged(step, \"Value\")\n\n// A body that ends without `next` or `stop` stops too.\n/** @internal */\nexport const loop = <S, A, E, R>(\n  initial: S,\n  body: (state: S) => Stream.Stream<Step<A, S>, E, R>\n): Stream.Stream<A, E, R> => {\n  const go = (state: S): Stream.Stream<A, E, R> =>\n    Stream.suspend(() => {\n      let after: Option.Option<S> = Option.none()\n\n      return body(state).pipe(\n        Stream.tap((step) =>\n          Effect.sync(() => (after = Predicate.isTagged(step, \"Next\") ? Option.some(step.state) : after))\n        ),\n        Stream.filter(isValue),\n        Stream.map((step) => step.value),\n        Stream.concat(Stream.suspend(() => Option.match(after, { onNone: () => Stream.empty, onSome: go })))\n      )\n    })\n\n  return go(initial)\n}\n\n/** @internal */\nexport const step = <R, S, E2 = never, R2 = never>(\n  options: StepOptions<R>,\n  then: (result: StepResult) => Stream.Stream<Step<never, S>, E2, R2>\n): Stream.Stream<Step<Event, S>, AiError.AiError | E2, LanguageModel.LanguageModel | R | R2> => {\n  const toolkit = options.toolkit ?? Toolkits.empty\n  const instructions = Predicate.isString(options.instructions) ? { text: options.instructions } : options.instructions\n\n  const after = (result: StepResult) => Stream.suspend(() => then(result))\n  const needingApproval = Toolkits.needingApproval(toolkit.definitions)\n\n  if (needingApproval.length > 0) return Stream.die(Toolkits.needsApprovalMessage(needingApproval))\n\n  return Stream.unwrap(\n    Effect.gen(function* () {\n      const { messages: history, throughSeq } = readHistory(yield* options.history)\n      // Calls a crash cut off, with no result: they run again, with the same ids, before the model goes on.\n      const cutOff = History.unansweredCalls(history)\n\n      if (cutOff.length > 0) {\n        return runCalls(toolkit, cutOff, () =>\n          after({ output: undefined, done: false, historyLength: history.length + cutOff.length })\n        )\n      }\n\n      const prompt = Prompt.fromMessages([Prompt.systemMessage({ content: instructions.text }), ...history])\n      const name = yield* Effect.serviceOption(Model.ModelName)\n      const started = yield* Clock.currentTimeMillis\n\n      const response = yield* generate(prompt, toolkit.definitions, options.toolChoice).pipe(\n        Telemetry.chatSpan(prompt, instructions.attributes)\n      )\n\n      const metadata = response.content.find((part) => part.type === \"response-metadata\")\n\n      const output = AssistantEvent.ModelOutput.make({\n        parts: modelParts(response.content),\n        seenThroughSeq: throughSeq,\n        model: metadata?.modelId ?? Option.getOrUndefined(name),\n        responseId: metadata?.id,\n        usage: response.usage,\n        durationMillis: (yield* Clock.currentTimeMillis) - started\n      })\n\n      const calls = toolCalls(output.parts)\n      const historyLength = history.length + Prompt.fromResponseParts(output.parts).content.length + calls.length\n\n      // A failed call doesn't end the turn: a message that failed never reached the user, so the model reads why.\n      return Stream.make(value(output)).pipe(\n        Stream.concat(\n          runCalls(toolkit, calls, (failed) =>\n            after({\n              output,\n              done: calls.length === 0 || (!failed && calls.some(isTurnEnding)),\n              historyLength\n            })\n          )\n        )\n      )\n    })\n  )\n}\n\n/** @internal */\nexport const run = <R = never>(\n  options: Options<R>\n): Stream.Stream<Event, AiError.AiError, LanguageModel.LanguageModel | R> => {\n  const maxSteps = options.maxSteps ?? 16\n\n  // Done: stop, unless something arrived while the model was working; then it reads that first.\n  const afterDone = (steps: number, historyLength: number) =>\n    Stream.unwrap(\n      Effect.gen(function* () {\n        const read = yield* options.history\n\n        return readHistory(read).messages.length > historyLength ? next(steps) : stop<number>()\n      })\n    )\n\n  return loop(0, (steps: number) =>\n    step(options, (result) => {\n      const taken = result.output === undefined ? steps : steps + 1\n\n      if (taken >= maxSteps) return stop<number>()\n\n      return result.done ? afterDone(taken, result.historyLength) : next(taken)\n    })\n  )\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AA4BA,MAAa,YAAY,QAAQ,UAAyC,8BAA8B,EACtG,cAAc,OAAO,KACvB,CAAC;AAQD,MAAM,eAAe,SACnB,cAAc,OACV;CAAE,UAAU,KAAK;CAAU,YAAY,KAAK;AAAW,IACvD;CAAE,UAAU;CAAM,YAAY,KAAA;AAAU;AAiC9C,MAAM,kBAAkB,iBAAyF;CAC/G,OAAO,OAAO,YAAY,YAAY,KAAK,SAAS,CAAC,KAAK,MAAM,IAAI,CAAC,CAAC;CACtE,cAAc,OAAO,QAAQ,OAAO,KAAK;AAC3C;AAIA,MAAM,YACJ,QACA,aACA,eAKG,cAAc,aAAa;CAAE;CAAQ,SAAS,eAAe,WAAW;CAAG;AAAW,CAAC;AAE5F,MAAM,aAAa,OAAO,oBAAoBA,SAAwB;AAEtE,MAAM,cAAc,YAClB,QAAQ,SAAS,SACf,OAAO,QAAQ,WAAW,IAAI,CAAC,CAAC,CAAC,QAAQ,YAAY,EAAE,QAAQ,SAAS,eAAe,QAAQ,iBAAiB,CAClH;AAEF,MAAM,aAAa,UACjB,MAAM,SAAS,SAAU,KAAK,SAAS,cAAc,CAAC,IAAI,IAAI,CAAC,CAAE;AAEnE,MAAM,UAAU,OAAO,WAAW,WAAc,OAA0B,MAAmC;CAC3G,MAAM,SAAS,OAAO,OAAO,IAAI,aAAa;EAC5C,MAAM,SAAS,OAAO,MAAM,IAAI,IAAI;EAEpC,OAAOC,WAAqB,OAAO,MAAM;EAEzC,OAAO;CACT,CAAC,CAAC,CAAC,KAAKC,SAAmB,IAAI,CAAC;CAEhC,OAAA,WAAiC,KAAK;EAAE,QAAQ,KAAK;EAAI,MAAM,KAAK;EAAM,GAAG;CAAO,CAAC;AACvF,CAAC;AAID,MAAM,YACJ,OACA,OACA,SAEA,OAAO,aACL,MAAM,KAAK,SAAS,MAAA,YAAiC,KAAK;CAAE,QAAQ,KAAK;CAAI,MAAM,KAAK;AAAK,CAAC,CAAC,CAAC,CAClG,CAAC,CAAC,KACA,OAAO,OACL,OAAO,OACL,OAAO,IAAI,aAAa;CACtB,MAAM,OAAO,OAAO,OAAO,QAAQ,QAAQ,SAAS,QAAQ,OAAO,IAAI,GAAG,EAAE,aAAa,YAAY,CAAC;CAEtG,OAAO,OAAO,aAAa,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,OAAO,OAAO,KAAK,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC;AAC5F,CAAC,CACH,CACF,CACF;AAEF,MAAM,aAAa,WAAsC,OAAO,YAAY;AAa5E,MAAM,QAAQ,KAAK,WAA2B;;AAG9C,MAAa,SAAY,UAA6B,MAAM,MAAM,EAAE,MAAM,CAAC;;AAG3E,MAAa,QAAW,UAA4C,OAAO,QAAQ,MAAM,KAAK,EAAE,MAAM,CAAC,CAAC;;AAIxG,MAAa,aAAuD,OAAO,QAAQ,MAAM,KAAK,CAAC;AAE/F,MAAM,WAAiB,SACrB,UAAU,SAAS,MAAM,OAAO;;AAIlC,MAAa,QACX,SACA,SAC2B;CAC3B,MAAM,MAAM,UACV,OAAO,cAAc;EACnB,IAAI,QAA0B,OAAO,KAAK;EAE1C,OAAO,KAAK,KAAK,CAAC,CAAC,KACjB,OAAO,KAAK,SACV,OAAO,WAAY,QAAQ,UAAU,SAAS,MAAM,MAAM,IAAI,OAAO,KAAK,KAAK,KAAK,IAAI,KAAM,CAChG,GACA,OAAO,OAAO,OAAO,GACrB,OAAO,KAAK,SAAS,KAAK,KAAK,GAC/B,OAAO,OAAO,OAAO,cAAc,OAAO,MAAM,OAAO;GAAE,cAAc,OAAO;GAAO,QAAQ;EAAG,CAAC,CAAC,CAAC,CACrG;CACF,CAAC;CAEH,OAAO,GAAG,OAAO;AACnB;;AAGA,MAAa,QACX,SACA,SAC8F;CAC9F,MAAM,UAAU,QAAQ,WAAWC;CACnC,MAAM,eAAe,UAAU,SAAS,QAAQ,YAAY,IAAI,EAAE,MAAM,QAAQ,aAAa,IAAI,QAAQ;CAEzG,MAAM,SAAS,WAAuB,OAAO,cAAc,KAAK,MAAM,CAAC;CACvE,MAAMC,oBAAkBC,gBAAyB,QAAQ,WAAW;CAEpE,IAAID,kBAAgB,SAAS,GAAG,OAAO,OAAO,IAAIE,qBAA8BF,iBAAe,CAAC;CAEhG,OAAO,OAAO,OACZ,OAAO,IAAI,aAAa;EACtB,MAAM,EAAE,UAAU,SAAS,eAAe,YAAY,OAAO,QAAQ,OAAO;EAE5E,MAAM,SAASG,gBAAwB,OAAO;EAE9C,IAAI,OAAO,SAAS,GAClB,OAAO,SAAS,SAAS,cACvB,MAAM;GAAE,QAAQ,KAAA;GAAW,MAAM;GAAO,eAAe,QAAQ,SAAS,OAAO;EAAO,CAAC,CACzF;EAGF,MAAM,SAAS,OAAO,aAAa,CAAC,OAAO,cAAc,EAAE,SAAS,aAAa,KAAK,CAAC,GAAG,GAAG,OAAO,CAAC;EACrG,MAAM,OAAO,OAAO,OAAO,cAAc,MAAM,SAAS;EACxD,MAAM,UAAU,OAAO,MAAM;EAE7B,MAAM,WAAW,OAAO,SAAS,QAAQ,QAAQ,aAAa,QAAQ,UAAU,CAAC,CAAC,KAChFC,SAAmB,QAAQ,aAAa,UAAU,CACpD;EAEA,MAAM,WAAW,SAAS,QAAQ,MAAM,SAAS,KAAK,SAAS,mBAAmB;EAElF,MAAM,SAAA,YAAoC,KAAK;GAC7C,OAAO,WAAW,SAAS,OAAO;GAClC,gBAAgB;GAChB,OAAO,UAAU,WAAW,OAAO,eAAe,IAAI;GACtD,YAAY,UAAU;GACtB,OAAO,SAAS;GAChB,iBAAiB,OAAO,MAAM,qBAAqB;EACrD,CAAC;EAED,MAAM,QAAQ,UAAU,OAAO,KAAK;EACpC,MAAM,gBAAgB,QAAQ,SAAS,OAAO,kBAAkB,OAAO,KAAK,CAAC,CAAC,QAAQ,SAAS,MAAM;EAGrG,OAAO,OAAO,KAAK,MAAM,MAAM,CAAC,CAAC,CAAC,KAChC,OAAO,OACL,SAAS,SAAS,QAAQ,WACxB,MAAM;GACJ;GACA,MAAM,MAAM,WAAW,KAAM,CAAC,UAAU,MAAM,KAAK,YAAY;GAC/D;EACF,CAAC,CACH,CACF,CACF;CACF,CAAC,CACH;AACF;;AAGA,MAAa,OACX,YAC2E;CAC3E,MAAM,WAAW,QAAQ,YAAY;CAGrC,MAAM,aAAa,OAAe,kBAChC,OAAO,OACL,OAAO,IAAI,aAAa;EACtB,MAAM,OAAO,OAAO,QAAQ;EAE5B,OAAO,YAAY,IAAI,CAAC,CAAC,SAAS,SAAS,gBAAgB,KAAK,KAAK,IAAI,KAAa;CACxF,CAAC,CACH;CAEF,OAAO,KAAK,IAAI,UACd,KAAK,UAAU,WAAW;EACxB,MAAM,QAAQ,OAAO,WAAW,KAAA,IAAY,QAAQ,QAAQ;EAE5D,IAAI,SAAS,UAAU,OAAO,KAAa;EAE3C,OAAO,OAAO,OAAO,UAAU,OAAO,OAAO,aAAa,IAAI,KAAK,KAAK;CAC1E,CAAC,CACH;AACF"}

package/dist/internal/specialists.js

@@ -3,7 +3,6 @@ import { RunId, make } from "../RunId.js";
 import { SpecialistFinished, SpecialistStarted } from "../AssistantEvent.js";
 import { CurrentChat } from "../CurrentChat.js";
 import { Brief, Runner, SpecialistError } from "../Specialist.js";
-import { TurnDecisions } from "./decisions.js";
 import { toolMessage, userText } from "./history.js";
 import { agentOutput, agentSpan } from "./telemetry.js";
 import { ToolFailure, fail } from "../ToolFailure.js";
@@ -64,7 +63,7 @@ const runBrief = Effect.fnUntraced(function* (handler, options) {
 		}
 	};
 	const record = Effect.fnUntraced(function* (event) {
-		if (Predicate.isTagged(event, "ToolStarted") || Predicate.isTagged(event, "DecisionRequested")) return;
+		if (Predicate.isTagged(event, "ToolStarted")) return;
 		if (Predicate.isTagged(event, "ToolResult")) return yield* add([toolMessage(event)]);
 		yield* add(Prompt.fromResponseParts(event.parts).content);
 		const text = finalText(event);
@@ -176,7 +175,7 @@ const makeRunner = Effect.fnUntraced(function* (options) {
 		history: run.history,
 		inbox: run.inbox,
 		chatId: chat.chatId
-	}).pipe(Effect.provideContext(Context.merge(options.model, services)), Effect.provideService(TurnDecisions, Option.none()), Effect.provideService(TurnTools, Option.none()), Effect.matchCause({
+	}).pipe(Effect.provideContext(Context.merge(options.model, services)), Effect.provideService(TurnTools, Option.none()), Effect.matchCause({
 		onSuccess: (answer) => ({
 			status: "answered",
 			report: answer

package/dist/internal/specialists.js.map

@@ -1 +1 @@
-{"version":3,"file":"specialists.js","names":["run","Loop.run","Skill.withCatalog","Toolkits.merge","Skills.toTools","Toolkits.fromToolkitOrEmpty","Telemetry.agentOutput","Telemetry.agentSpan","InputId.fromKey","RunId.RunId","ToolFailure.ToolFailure","Specialist.Runner","ToolFailure.fail","Toolkits.empty","Toolkits.fromWithHandler","Decisions.TurnDecisions","Loop.TurnTools","Specialist.SpecialistError","RunId.make"],"sources":["../../src/internal/specialists.ts"],"sourcesContent":["import type * as LanguageModel from \"effect/ai/LanguageModel\"\nimport * as Prompt from \"effect/ai/Prompt\"\nimport * as Tool from \"effect/ai/Tool\"\nimport * as Toolkit from \"effect/ai/Toolkit\"\nimport * as Cause from \"effect/Cause\"\nimport * as Context from \"effect/Context\"\nimport * as Effect from \"effect/Effect\"\nimport * as Exit from \"effect/Exit\"\nimport * as Fiber from \"effect/Fiber\"\nimport * as FiberMap from \"effect/FiberMap\"\nimport * as FiberSet from \"effect/FiberSet\"\nimport * as Option from \"effect/Option\"\nimport * as Predicate from \"effect/Predicate\"\nimport * as Queue from \"effect/Queue\"\nimport * as Ref from \"effect/Ref\"\nimport * as Schema from \"effect/Schema\"\nimport * as Semaphore from \"effect/Semaphore\"\nimport * as Stream from \"effect/Stream\"\nimport * as AssistantEvent from \"../AssistantEvent.ts\"\nimport type * as AssistantLoop from \"../AssistantLoop.ts\"\nimport * as Loop from \"./loop.ts\"\nimport { CurrentChat } from \"../CurrentChat.ts\"\nimport * as InputId from \"../InputId.ts\"\nimport * as RunId from \"../RunId.ts\"\nimport * as Skill from \"../Skill.ts\"\nimport * as Skills from \"./skills.ts\"\nimport * as Specialist from \"../Specialist.ts\"\nimport * as ToolFailure from \"../ToolFailure.ts\"\nimport * as Decisions from \"./decisions.ts\"\nimport { toolMessage, userText } from \"./history.ts\"\nimport { mapText } from \"./instructions.ts\"\nimport * as Telemetry from \"./telemetry.ts\"\nimport * as Toolkits from \"./toolkit.ts\"\nimport type { Writer } from \"./writer.ts\"\n\n/** @internal */\nexport type RunHandler = (run: Specialist.Run) => Stream.Stream<AssistantLoop.Event, unknown, Specialist.RunServices>\n\n/** @internal */\nexport const briefText = (brief: Specialist.Brief): string =>\n  [\n    `Goal: ${brief.goal}`,\n    brief.context && `Context: ${brief.context}`,\n    brief.successCriteria && `Done when: ${brief.successCriteria}`,\n    brief.report && `Report back: ${brief.report}`,\n    brief.limits && `Limits: ${brief.limits}`\n  ]\n    .filter(Boolean)\n    .join(\"\\n\")\n\nconst finalText = (output: AssistantEvent.ModelOutput): string =>\n  output.parts\n    .flatMap((part) => (part.type === \"text\" ? [part.text] : []))\n    .join(\"\")\n    .trim()\n\n// The answer is the last text the model wrote.\n/** @internal */\nexport const runBrief = Effect.fnUntraced(\n  function* (\n    handler: RunHandler,\n    options: {\n      readonly specialist: Specialist.Any\n      readonly runId: RunId.RunId\n      readonly brief: Specialist.Brief\n      readonly history: Ref.Ref<ReadonlyArray<Prompt.Message>>\n      readonly inbox: Queue.Queue<Prompt.Message>\n      readonly chatId: Writer[\"chatId\"]\n    }\n  ) {\n    const { history, inbox } = options\n    const answer = yield* Ref.make(\"\")\n    const add = (messages: ReadonlyArray<Prompt.Message>) => Ref.update(history, (all) => [...all, ...messages])\n\n    // Read before every model call, when no tool calls are open: messages sent meanwhile join it now.\n    const current = Effect.gen(function* () {\n      yield* add(yield* Queue.clear(inbox))\n\n      return yield* Ref.get(history)\n    })\n\n    const run: Specialist.Run = {\n      runId: options.runId,\n      chatId: options.chatId,\n      brief: options.brief,\n      history: current,\n      loop: (loop) => {\n        const skills = loop.skills ?? []\n\n        return Loop.run({\n          instructions: mapText(loop.instructions, (text) => Skill.withCatalog(text, skills)),\n          toolkit: Toolkits.merge(Skills.toTools(skills), Toolkits.fromToolkitOrEmpty(loop.toolkit)),\n          history: current,\n          maxSteps: loop.maxSteps ?? 8\n        })\n      }\n    }\n\n    // A run lives in memory, so there's nothing to run again after a crash: `ToolStarted` isn't kept. A run parks no\n    // call (it has no user to ask), so there's no `DecisionRequested` either.\n    const record = Effect.fnUntraced(function* (event: AssistantLoop.Event) {\n      if (Predicate.isTagged(event, \"ToolStarted\") || Predicate.isTagged(event, \"DecisionRequested\")) return\n\n      if (Predicate.isTagged(event, \"ToolResult\")) return yield* add([toolMessage(event)])\n\n      yield* add(Prompt.fromResponseParts(event.parts).content)\n      const text = finalText(event)\n\n      if (text.length > 0) yield* Ref.set(answer, text)\n    })\n\n    yield* Stream.runForEach(handler(run), record)\n\n    const text = yield* Ref.get(answer)\n\n    yield* Telemetry.agentOutput(text)\n\n    return text\n  },\n  (effect, _handler, options) => effect.pipe(Telemetry.agentSpan(options.specialist.name, { chatId: options.chatId }))\n)\n\ntype Outcome =\n  | { readonly status: \"answered\"; readonly report: string }\n  | { readonly status: \"failed\"; readonly error: string }\n  | { readonly status: \"stopped\" | \"cut_off\" }\n\n// Only `stop` interrupts a brief's loop alone; a failure or a defect failed it.\nconst unanswered = (cause: Cause.Cause<unknown>): Outcome =>\n  Cause.hasInterruptsOnly(cause) ? { status: \"stopped\" } : { status: \"failed\", error: String(Cause.squash(cause)) }\n\n// The id is fixed per run and brief, so a replay is stored once.\n/** @internal */\nexport const finished = (\n  specialist: string,\n  run: { readonly runId: RunId.RunId; readonly brief: number; readonly goal: string },\n  outcome: Outcome\n): AssistantEvent.SpecialistFinished =>\n  AssistantEvent.SpecialistFinished.make({\n    id: InputId.fromKey(`${run.runId}:${run.brief}`),\n    runId: run.runId,\n    specialist,\n    brief: run.brief,\n    goal: run.goal,\n    ...outcome\n  })\n\nconst RunIdInput = RunId.RunId.annotate({ description: \"The runId spawn_specialist returned.\" })\n\nconst RunStatus = <const Status extends string>(status: ReadonlyArray<Status>) =>\n  Schema.Struct({ runId: RunId.RunId, status: Schema.Literals(status) })\n\nconst MessageSpecialist = Tool.make(\"message_specialist\", {\n  description:\n    \"Send more to one of your specialists: new instructions for a running one (it reads them at its next step), or a follow-up for a finished one (it continues with what it already found).\",\n  parameters: Schema.Struct({ runId: RunIdInput, message: Schema.String }),\n  success: RunStatus([\"steered\", \"continued\"]),\n  failure: ToolFailure.ToolFailure,\n  dependencies: [Specialist.Runner]\n})\n\nconst StopSpecialist = Tool.make(\"stop_specialist\", {\n  description:\n    \"Stop one of your running specialists, for example when the user changed their mind. It posts no answer.\",\n  parameters: Schema.Struct({ runId: RunIdInput }),\n  success: RunStatus([\"stopped\"]),\n  failure: ToolFailure.ToolFailure,\n  dependencies: [Specialist.Runner]\n})\n\nconst refused = <A, R>(effect: Effect.Effect<A, Specialist.SpecialistError, R>) =>\n  Effect.catch(effect, (error) => ToolFailure.fail(error.message, { reason: error.reason }))\n\n/** @internal */\nexport const toolkit = Effect.fnUntraced(function* (\n  specialists: ReadonlyArray<Specialist.Any>\n): Effect.fn.Return<Toolkits.Tools<Specialist.Runner>> {\n  if (specialists.length === 0) return Toolkits.empty\n\n  const catalog = specialists.map((specialist) => `- ${specialist.name}: ${specialist.description}`).join(\"\\n\")\n\n  const SpawnSpecialist = Tool.make(\"spawn_specialist\", {\n    description: `Start a specialist on a job. It works in the background and its report arrives as a <specialist> input, often within seconds. Write a self-contained brief: it can't see this chat. For a quick question, just call wait; for slow work, tell the user in one line first.\\nPass one of these as specialist:\\n${catalog}`,\n    parameters: Schema.Struct({\n      specialist: Schema.Literals(specialists.map((specialist) => specialist.name)).annotate({\n        description: \"The specialist to start, by name.\"\n      }),\n      ...Specialist.Brief.fields\n    }),\n    success: RunStatus([\"started\"]),\n    failure: ToolFailure.ToolFailure,\n    dependencies: [Specialist.Runner]\n  })\n\n  const SpecialistTools = Toolkit.make(SpawnSpecialist, MessageSpecialist, StopSpecialist)\n\n  const handlers = yield* SpecialistTools.toHandlers({\n    spawn_specialist: Effect.fnUntraced(function* ({ specialist, ...brief }) {\n      const runId = yield* refused(Specialist.Runner.use((runner) => runner.spawn({ specialist, brief })))\n\n      return { runId, status: \"started\" }\n    }),\n    message_specialist: Effect.fnUntraced(function* ({ runId, message }) {\n      const status = yield* refused(Specialist.Runner.use((runner) => runner.message(runId, message)))\n\n      return { runId, status }\n    }),\n    stop_specialist: Effect.fnUntraced(function* ({ runId }) {\n      yield* refused(Specialist.Runner.use((runner) => runner.stop(runId)))\n\n      return { runId, status: \"stopped\" }\n    })\n  })\n\n  return Toolkits.fromWithHandler(yield* Effect.provideContext(SpecialistTools, handlers))\n})\n\n// How long a finished run stays available for follow-ups.\nconst warm = \"15 minutes\"\n\ninterface Run {\n  readonly runId: RunId.RunId\n  readonly specialist: Specialist.Any\n  readonly handler: RunHandler\n  readonly goal: string\n  readonly brief: Specialist.Brief\n  readonly history: Ref.Ref<ReadonlyArray<Prompt.Message>>\n  // Messages sent while it works; the loop reads them at its next step.\n  readonly inbox: Queue.Queue<Prompt.Message>\n  // Counts follow-ups too; numbers the reports.\n  readonly briefs: Ref.Ref<number>\n  // Read and changed under `lock`.\n  readonly busy: Ref.Ref<boolean>\n  // Orders a message against the run finishing, so none is dropped or starts a second loop.\n  readonly lock: Semaphore.Semaphore\n}\n\n/** @internal */\nexport const makeRunner = Effect.fnUntraced(function* (options: {\n  readonly chat: Writer\n  readonly handle: CurrentChat[\"Service\"]\n  readonly specialists: ReadonlyArray<{\n    readonly specialist: Specialist.Any\n    readonly handler: RunHandler\n  }>\n  readonly model: Context.Context<LanguageModel.LanguageModel>\n  // An interruption keeps the hold.\n  readonly holding: (effect: Effect.Effect<void>) => Effect.Effect<void>\n  readonly report: (finished: AssistantEvent.SpecialistFinished) => Effect.Effect<void>\n}) {\n  const { chat } = options\n  const runs = new Map<string, Run>()\n  const byName = new Map(options.specialists.map((entry) => [entry.specialist.name, entry]))\n  // The brief loops working now, and the finished runs' countdowns to being forgotten.\n  const working = yield* FiberMap.make<string, Outcome>()\n  const forgetting = yield* FiberMap.make<string>()\n  const background = yield* FiberSet.make()\n\n  const report = Effect.fnUntraced(function* (run: Run, outcome: Outcome) {\n    const brief = yield* Ref.get(run.briefs)\n\n    yield* options.report(finished(run.specialist.name, { runId: run.runId, brief, goal: run.goal }, outcome))\n  })\n\n  const once = (run: Run) =>\n    runBrief(run.handler, {\n      specialist: run.specialist,\n      runId: run.runId,\n      brief: run.brief,\n      history: run.history,\n      inbox: run.inbox,\n      chatId: chat.chatId\n    }).pipe(\n      Effect.provideContext(Context.merge(options.model, services)),\n      // Spawned from a turn, a run would inherit the turn's decisions: it can't park a call or run an approved one.\n      Effect.provideService(Decisions.TurnDecisions, Option.none()),\n      // Nor its own tools: a run can't message the user.\n      Effect.provideService(Loop.TurnTools, Option.none()),\n      Effect.matchCause({\n        onSuccess: (answer): Outcome => ({ status: \"answered\", report: answer }),\n        onFailure: unanswered\n      })\n    )\n\n  // A message that arrived after the last read: keep working before reporting.\n  const untilCaughtUp = Effect.fnUntraced(function* (run: Run) {\n    let outcome = yield* once(run)\n\n    while ((yield* Queue.size(run.inbox)) > 0) outcome = yield* once(run)\n\n    return outcome\n  })\n\n  const work = Effect.fnUntraced(\n    function* (run: Run) {\n      yield* chat.append([\n        AssistantEvent.SpecialistStarted.make({\n          runId: run.runId,\n          specialist: run.specialist.name,\n          goal: run.goal\n        })\n      ])\n\n      let finished = false\n\n      while (!finished) {\n        const exit = yield* Fiber.await(yield* FiberMap.run(working, run.runId, untilCaughtUp(run)))\n        const outcome = Exit.isSuccess(exit) ? exit.value : unanswered(exit.cause)\n\n        finished = yield* run.lock.withPermit(\n          Effect.gen(function* () {\n            // A message steered in after the loop's last read: work on it before reporting.\n            if (outcome.status !== \"stopped\" && (yield* Queue.size(run.inbox)) > 0) return false\n\n            yield* report(run, outcome)\n            yield* Ref.set(run.busy, false)\n            yield* FiberMap.run(\n              forgetting,\n              run.runId,\n              Effect.gen(function* () {\n                yield* Effect.sleep(warm)\n                runs.delete(run.runId)\n              })\n            )\n\n            return true\n          })\n        )\n      }\n    },\n    // A shutdown interrupts the whole run: no turn can start now, so the report waits for the chat's next start.\n    (effect, run) => Effect.onInterrupt(effect, () => report(run, { status: \"cut_off\" }))\n  )\n\n  const launch = Effect.fnUntraced(function* (run: Run) {\n    yield* FiberMap.remove(forgetting, run.runId)\n    yield* FiberSet.run(background, options.holding(work(run)))\n  })\n\n  // Under the run's lock.\n  const send = Effect.fnUntraced(function* (run: Run, text: string): Effect.fn.Return<\"steered\" | \"continued\"> {\n    yield* Ref.update(run.briefs, (n) => n + 1)\n\n    if (yield* Ref.get(run.busy)) {\n      yield* Queue.offer(run.inbox, userText(text))\n\n      return \"steered\"\n    }\n\n    yield* Ref.update(run.history, (all) => [...all, userText(text)])\n    yield* Ref.set(run.busy, true)\n    yield* launch(run)\n\n    return \"continued\"\n  })\n\n  const runner = Specialist.Runner.of({\n    spawn: Effect.fnUntraced(function* (request: Specialist.SpawnRequest) {\n      const found = byName.get(request.specialist)\n\n      if (found === undefined) {\n        return yield* new Specialist.SpecialistError({\n          reason: \"UnknownSpecialist\",\n          message: `No specialist named ${request.specialist}.`\n        })\n      }\n\n      const run: Run = {\n        runId: yield* RunId.make(),\n        specialist: found.specialist,\n        handler: found.handler,\n        goal: request.brief.goal,\n        brief: request.brief,\n        history: yield* Ref.make<ReadonlyArray<Prompt.Message>>([userText(briefText(request.brief))]),\n        inbox: yield* Queue.unbounded<Prompt.Message>(),\n        briefs: yield* Ref.make(1),\n        busy: yield* Ref.make(true),\n        lock: yield* Semaphore.make(1)\n      }\n\n      runs.set(run.runId, run)\n      yield* launch(run)\n\n      return run.runId\n    }),\n    message: Effect.fnUntraced(function* (\n      runId: RunId.RunId,\n      text: string\n    ): Effect.fn.Return<\"steered\" | \"continued\", Specialist.SpecialistError> {\n      const run = runs.get(runId)\n\n      if (run === undefined) {\n        return yield* new Specialist.SpecialistError({\n          reason: \"RunNotFound\",\n          message: `Run ${runId} is no longer available. Spawn a new one with the full context.`\n        })\n      }\n\n      return yield* run.lock.withPermit(send(run, text))\n    }),\n    stop: Effect.fnUntraced(function* (runId: RunId.RunId) {\n      if (!(yield* FiberMap.has(working, runId))) {\n        return yield* new Specialist.SpecialistError({ reason: \"NotRunning\", message: `Run ${runId} isn't running.` })\n      }\n\n      yield* FiberMap.remove(working, runId)\n    })\n  })\n\n  // What a run's handler gets: its chat, this runner, and (merged in `once`) the model.\n  const services = Context.make(CurrentChat, options.handle).pipe(Context.add(Specialist.Runner, runner))\n\n  return runner\n})\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAuCA,MAAa,aAAa,UACxB;CACE,SAAS,MAAM;CACf,MAAM,WAAW,YAAY,MAAM;CACnC,MAAM,mBAAmB,cAAc,MAAM;CAC7C,MAAM,UAAU,gBAAgB,MAAM;CACtC,MAAM,UAAU,WAAW,MAAM;AACnC,CAAC,CACE,OAAO,OAAO,CAAC,CACf,KAAK,IAAI;AAEd,MAAM,aAAa,WACjB,OAAO,MACJ,SAAS,SAAU,KAAK,SAAS,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAE,CAAC,CAC5D,KAAK,EAAE,CAAC,CACR,KAAK;;AAIV,MAAa,WAAW,OAAO,WAC7B,WACE,SACA,SAQA;CACA,MAAM,EAAE,SAAS,UAAU;CAC3B,MAAM,SAAS,OAAO,IAAI,KAAK,EAAE;CACjC,MAAM,OAAO,aAA4C,IAAI,OAAO,UAAU,QAAQ,CAAC,GAAG,KAAK,GAAG,QAAQ,CAAC;CAG3G,MAAM,UAAU,OAAO,IAAI,aAAa;EACtC,OAAO,IAAI,OAAO,MAAM,MAAM,KAAK,CAAC;EAEpC,OAAO,OAAO,IAAI,IAAI,OAAO;CAC/B,CAAC;CAED,MAAMA,QAAsB;EAC1B,OAAO,QAAQ;EACf,QAAQ,QAAQ;EAChB,OAAO,QAAQ;EACf,SAAS;EACT,OAAO,SAAS;GACd,MAAM,SAAS,KAAK,UAAU,CAAC;GAE/B,OAAOC,IAAS;IACd,cAAc,QAAQ,KAAK,eAAe,SAASC,YAAkB,MAAM,MAAM,CAAC;IAClF,SAASC,MAAeC,QAAe,MAAM,GAAGC,mBAA4B,KAAK,OAAO,CAAC;IACzF,SAAS;IACT,UAAU,KAAK,YAAY;GAC7B,CAAC;EACH;CACF;CAIA,MAAM,SAAS,OAAO,WAAW,WAAW,OAA4B;EACtE,IAAI,UAAU,SAAS,OAAO,aAAa,KAAK,UAAU,SAAS,OAAO,mBAAmB,GAAG;EAEhG,IAAI,UAAU,SAAS,OAAO,YAAY,GAAG,OAAO,OAAO,IAAI,CAAC,YAAY,KAAK,CAAC,CAAC;EAEnF,OAAO,IAAI,OAAO,kBAAkB,MAAM,KAAK,CAAC,CAAC,OAAO;EACxD,MAAM,OAAO,UAAU,KAAK;EAE5B,IAAI,KAAK,SAAS,GAAG,OAAO,IAAI,IAAI,QAAQ,IAAI;CAClD,CAAC;CAED,OAAO,OAAO,WAAW,QAAQL,KAAG,GAAG,MAAM;CAE7C,MAAM,OAAO,OAAO,IAAI,IAAI,MAAM;CAElC,OAAOM,YAAsB,IAAI;CAEjC,OAAO;AACT,IACC,QAAQ,UAAU,YAAY,OAAO,KAAKC,UAAoB,QAAQ,WAAW,MAAM,EAAE,QAAQ,QAAQ,OAAO,CAAC,CAAC,CACrH;AAQA,MAAM,cAAc,UAClB,MAAM,kBAAkB,KAAK,IAAI,EAAE,QAAQ,UAAU,IAAI;CAAE,QAAQ;CAAU,OAAO,OAAO,MAAM,OAAO,KAAK,CAAC;AAAE;;AAIlH,MAAa,YACX,YACA,KACA,YAAA,mBAEkC,KAAK;CACrC,IAAIC,QAAgB,GAAG,IAAI,MAAM,GAAG,IAAI,OAAO;CAC/C,OAAO,IAAI;CACX;CACA,OAAO,IAAI;CACX,MAAM,IAAI;CACV,GAAG;AACL,CAAC;AAEH,MAAM,aAAA,MAAyB,SAAS,EAAE,aAAa,uCAAuC,CAAC;AAE/F,MAAM,aAA0C,WAC9C,OAAO,OAAO;CAAE,OAAOC;CAAa,QAAQ,OAAO,SAAS,MAAM;AAAE,CAAC;AAEvE,MAAM,oBAAoB,KAAK,KAAK,sBAAsB;CACxD,aACE;CACF,YAAY,OAAO,OAAO;EAAE,OAAO;EAAY,SAAS,OAAO;CAAO,CAAC;CACvE,SAAS,UAAU,CAAC,WAAW,WAAW,CAAC;CAC3C,SAASC;CACT,cAAc,CAACC,MAAiB;AAClC,CAAC;AAED,MAAM,iBAAiB,KAAK,KAAK,mBAAmB;CAClD,aACE;CACF,YAAY,OAAO,OAAO,EAAE,OAAO,WAAW,CAAC;CAC/C,SAAS,UAAU,CAAC,SAAS,CAAC;CAC9B,SAASD;CACT,cAAc,CAACC,MAAiB;AAClC,CAAC;AAED,MAAM,WAAiB,WACrB,OAAO,MAAM,SAAS,UAAUC,KAAiB,MAAM,SAAS,EAAE,QAAQ,MAAM,OAAO,CAAC,CAAC;;AAG3F,MAAa,UAAU,OAAO,WAAW,WACvC,aACqD;CACrD,IAAI,YAAY,WAAW,GAAG,OAAOC;CAErC,MAAM,UAAU,YAAY,KAAK,eAAe,KAAK,WAAW,KAAK,IAAI,WAAW,aAAa,CAAC,CAAC,KAAK,IAAI;CAE5G,MAAM,kBAAkB,KAAK,KAAK,oBAAoB;EACpD,aAAa,gTAAgT;EAC7T,YAAY,OAAO,OAAO;GACxB,YAAY,OAAO,SAAS,YAAY,KAAK,eAAe,WAAW,IAAI,CAAC,CAAC,CAAC,SAAS,EACrF,aAAa,oCACf,CAAC;GACD,GAAA,MAAoB;EACtB,CAAC;EACD,SAAS,UAAU,CAAC,SAAS,CAAC;EAC9B,SAASH;EACT,cAAc,CAACC,MAAiB;CAClC,CAAC;CAED,MAAM,kBAAkB,QAAQ,KAAK,iBAAiB,mBAAmB,cAAc;CAEvF,MAAM,WAAW,OAAO,gBAAgB,WAAW;EACjD,kBAAkB,OAAO,WAAW,WAAW,EAAE,YAAY,GAAG,SAAS;GAGvE,OAAO;IAAE,OAAA,OAFY,QAAA,OAA0B,KAAK,WAAW,OAAO,MAAM;KAAE;KAAY;IAAM,CAAC,CAAC,CAAC;IAEnF,QAAQ;GAAU;EACpC,CAAC;EACD,oBAAoB,OAAO,WAAW,WAAW,EAAE,OAAO,WAAW;GAGnE,OAAO;IAAE;IAAO,QAAA,OAFM,QAAA,OAA0B,KAAK,WAAW,OAAO,QAAQ,OAAO,OAAO,CAAC,CAAC;GAExE;EACzB,CAAC;EACD,iBAAiB,OAAO,WAAW,WAAW,EAAE,SAAS;GACvD,OAAO,QAAA,OAA0B,KAAK,WAAW,OAAO,KAAK,KAAK,CAAC,CAAC;GAEpE,OAAO;IAAE;IAAO,QAAQ;GAAU;EACpC,CAAC;CACH,CAAC;CAED,OAAOG,gBAAyB,OAAO,OAAO,eAAe,iBAAiB,QAAQ,CAAC;AACzF,CAAC;AAGD,MAAM,OAAO;;AAoBb,MAAa,aAAa,OAAO,WAAW,WAAW,SAWpD;CACD,MAAM,EAAE,SAAS;CACjB,MAAM,uBAAO,IAAI,IAAiB;CAClC,MAAM,SAAS,IAAI,IAAI,QAAQ,YAAY,KAAK,UAAU,CAAC,MAAM,WAAW,MAAM,KAAK,CAAC,CAAC;CAEzF,MAAM,UAAU,OAAO,SAAS,KAAsB;CACtD,MAAM,aAAa,OAAO,SAAS,KAAa;CAChD,MAAM,aAAa,OAAO,SAAS,KAAK;CAExC,MAAM,SAAS,OAAO,WAAW,WAAW,KAAU,SAAkB;EACtE,MAAM,QAAQ,OAAO,IAAI,IAAI,IAAI,MAAM;EAEvC,OAAO,QAAQ,OAAO,SAAS,IAAI,WAAW,MAAM;GAAE,OAAO,IAAI;GAAO;GAAO,MAAM,IAAI;EAAK,GAAG,OAAO,CAAC;CAC3G,CAAC;CAED,MAAM,QAAQ,QACZ,SAAS,IAAI,SAAS;EACpB,YAAY,IAAI;EAChB,OAAO,IAAI;EACX,OAAO,IAAI;EACX,SAAS,IAAI;EACb,OAAO,IAAI;EACX,QAAQ,KAAK;CACf,CAAC,CAAC,CAAC,KACD,OAAO,eAAe,QAAQ,MAAM,QAAQ,OAAO,QAAQ,CAAC,GAE5D,OAAO,eAAeC,eAAyB,OAAO,KAAK,CAAC,GAE5D,OAAO,eAAeC,WAAgB,OAAO,KAAK,CAAC,GACnD,OAAO,WAAW;EAChB,YAAY,YAAqB;GAAE,QAAQ;GAAY,QAAQ;EAAO;EACtE,WAAW;CACb,CAAC,CACH;CAGF,MAAM,gBAAgB,OAAO,WAAW,WAAW,KAAU;EAC3D,IAAI,UAAU,OAAO,KAAK,GAAG;EAE7B,QAAQ,OAAO,MAAM,KAAK,IAAI,KAAK,KAAK,GAAG,UAAU,OAAO,KAAK,GAAG;EAEpE,OAAO;CACT,CAAC;CAED,MAAM,OAAO,OAAO,WAClB,WAAW,KAAU;EACnB,OAAO,KAAK,OAAO,CAAA,kBACgB,KAAK;GACpC,OAAO,IAAI;GACX,YAAY,IAAI,WAAW;GAC3B,MAAM,IAAI;EACZ,CAAC,CACH,CAAC;EAED,IAAI,WAAW;EAEf,OAAO,CAAC,UAAU;GAChB,MAAM,OAAO,OAAO,MAAM,MAAM,OAAO,SAAS,IAAI,SAAS,IAAI,OAAO,cAAc,GAAG,CAAC,CAAC;GAC3F,MAAM,UAAU,KAAK,UAAU,IAAI,IAAI,KAAK,QAAQ,WAAW,KAAK,KAAK;GAEzE,WAAW,OAAO,IAAI,KAAK,WACzB,OAAO,IAAI,aAAa;IAEtB,IAAI,QAAQ,WAAW,cAAc,OAAO,MAAM,KAAK,IAAI,KAAK,KAAK,GAAG,OAAO;IAE/E,OAAO,OAAO,KAAK,OAAO;IAC1B,OAAO,IAAI,IAAI,IAAI,MAAM,KAAK;IAC9B,OAAO,SAAS,IACd,YACA,IAAI,OACJ,OAAO,IAAI,aAAa;KACtB,OAAO,OAAO,MAAM,IAAI;KACxB,KAAK,OAAO,IAAI,KAAK;IACvB,CAAC,CACH;IAEA,OAAO;GACT,CAAC,CACH;EACF;CACF,IAEC,QAAQ,QAAQ,OAAO,YAAY,cAAc,OAAO,KAAK,EAAE,QAAQ,UAAU,CAAC,CAAC,CACtF;CAEA,MAAM,SAAS,OAAO,WAAW,WAAW,KAAU;EACpD,OAAO,SAAS,OAAO,YAAY,IAAI,KAAK;EAC5C,OAAO,SAAS,IAAI,YAAY,QAAQ,QAAQ,KAAK,GAAG,CAAC,CAAC;CAC5D,CAAC;CAGD,MAAM,OAAO,OAAO,WAAW,WAAW,KAAU,MAAyD;EAC3G,OAAO,IAAI,OAAO,IAAI,SAAS,MAAM,IAAI,CAAC;EAE1C,IAAI,OAAO,IAAI,IAAI,IAAI,IAAI,GAAG;GAC5B,OAAO,MAAM,MAAM,IAAI,OAAO,SAAS,IAAI,CAAC;GAE5C,OAAO;EACT;EAEA,OAAO,IAAI,OAAO,IAAI,UAAU,QAAQ,CAAC,GAAG,KAAK,SAAS,IAAI,CAAC,CAAC;EAChE,OAAO,IAAI,IAAI,IAAI,MAAM,IAAI;EAC7B,OAAO,OAAO,GAAG;EAEjB,OAAO;CACT,CAAC;CAED,MAAM,SAAA,OAA2B,GAAG;EAClC,OAAO,OAAO,WAAW,WAAW,SAAkC;GACpE,MAAM,QAAQ,OAAO,IAAI,QAAQ,UAAU;GAE3C,IAAI,UAAU,KAAA,GACZ,OAAO,OAAO,IAAIC,gBAA2B;IAC3C,QAAQ;IACR,SAAS,uBAAuB,QAAQ,WAAW;GACrD,CAAC;GAGH,MAAM,MAAW;IACf,OAAO,OAAOC,KAAW;IACzB,YAAY,MAAM;IAClB,SAAS,MAAM;IACf,MAAM,QAAQ,MAAM;IACpB,OAAO,QAAQ;IACf,SAAS,OAAO,IAAI,KAAoC,CAAC,SAAS,UAAU,QAAQ,KAAK,CAAC,CAAC,CAAC;IAC5F,OAAO,OAAO,MAAM,UAA0B;IAC9C,QAAQ,OAAO,IAAI,KAAK,CAAC;IACzB,MAAM,OAAO,IAAI,KAAK,IAAI;IAC1B,MAAM,OAAO,UAAU,KAAK,CAAC;GAC/B;GAEA,KAAK,IAAI,IAAI,OAAO,GAAG;GACvB,OAAO,OAAO,GAAG;GAEjB,OAAO,IAAI;EACb,CAAC;EACD,SAAS,OAAO,WAAW,WACzB,OACA,MACuE;GACvE,MAAM,MAAM,KAAK,IAAI,KAAK;GAE1B,IAAI,QAAQ,KAAA,GACV,OAAO,OAAO,IAAID,gBAA2B;IAC3C,QAAQ;IACR,SAAS,OAAO,MAAM;GACxB,CAAC;GAGH,OAAO,OAAO,IAAI,KAAK,WAAW,KAAK,KAAK,IAAI,CAAC;EACnD,CAAC;EACD,MAAM,OAAO,WAAW,WAAW,OAAoB;GACrD,IAAI,EAAE,OAAO,SAAS,IAAI,SAAS,KAAK,IACtC,OAAO,OAAO,IAAIA,gBAA2B;IAAE,QAAQ;IAAc,SAAS,OAAO,MAAM;GAAiB,CAAC;GAG/G,OAAO,SAAS,OAAO,SAAS,KAAK;EACvC,CAAC;CACH,CAAC;CAGD,MAAM,WAAW,QAAQ,KAAK,aAAa,QAAQ,MAAM,CAAC,CAAC,KAAK,QAAQ,IAAIN,QAAmB,MAAM,CAAC;CAEtG,OAAO;AACT,CAAC"}
+{"version":3,"file":"specialists.js","names":["run","Loop.run","Skill.withCatalog","Toolkits.merge","Skills.toTools","Toolkits.fromToolkitOrEmpty","Telemetry.agentOutput","Telemetry.agentSpan","InputId.fromKey","RunId.RunId","ToolFailure.ToolFailure","Specialist.Runner","ToolFailure.fail","Toolkits.empty","Toolkits.fromWithHandler","Loop.TurnTools","Specialist.SpecialistError","RunId.make"],"sources":["../../src/internal/specialists.ts"],"sourcesContent":["import type * as LanguageModel from \"effect/ai/LanguageModel\"\nimport * as Prompt from \"effect/ai/Prompt\"\nimport * as Tool from \"effect/ai/Tool\"\nimport * as Toolkit from \"effect/ai/Toolkit\"\nimport * as Cause from \"effect/Cause\"\nimport * as Context from \"effect/Context\"\nimport * as Effect from \"effect/Effect\"\nimport * as Exit from \"effect/Exit\"\nimport * as Fiber from \"effect/Fiber\"\nimport * as FiberMap from \"effect/FiberMap\"\nimport * as FiberSet from \"effect/FiberSet\"\nimport * as Option from \"effect/Option\"\nimport * as Predicate from \"effect/Predicate\"\nimport * as Queue from \"effect/Queue\"\nimport * as Ref from \"effect/Ref\"\nimport * as Schema from \"effect/Schema\"\nimport * as Semaphore from \"effect/Semaphore\"\nimport * as Stream from \"effect/Stream\"\nimport * as AssistantEvent from \"../AssistantEvent.ts\"\nimport type * as AssistantLoop from \"../AssistantLoop.ts\"\nimport * as Loop from \"./loop.ts\"\nimport { CurrentChat } from \"../CurrentChat.ts\"\nimport * as InputId from \"../InputId.ts\"\nimport * as RunId from \"../RunId.ts\"\nimport * as Skill from \"../Skill.ts\"\nimport * as Skills from \"./skills.ts\"\nimport * as Specialist from \"../Specialist.ts\"\nimport * as ToolFailure from \"../ToolFailure.ts\"\nimport { toolMessage, userText } from \"./history.ts\"\nimport { mapText } from \"./instructions.ts\"\nimport * as Telemetry from \"./telemetry.ts\"\nimport * as Toolkits from \"./toolkit.ts\"\nimport type { Writer } from \"./writer.ts\"\n\n/** @internal */\nexport type RunHandler = (run: Specialist.Run) => Stream.Stream<AssistantLoop.Event, unknown, Specialist.RunServices>\n\n/** @internal */\nexport const briefText = (brief: Specialist.Brief): string =>\n  [\n    `Goal: ${brief.goal}`,\n    brief.context && `Context: ${brief.context}`,\n    brief.successCriteria && `Done when: ${brief.successCriteria}`,\n    brief.report && `Report back: ${brief.report}`,\n    brief.limits && `Limits: ${brief.limits}`\n  ]\n    .filter(Boolean)\n    .join(\"\\n\")\n\nconst finalText = (output: AssistantEvent.ModelOutput): string =>\n  output.parts\n    .flatMap((part) => (part.type === \"text\" ? [part.text] : []))\n    .join(\"\")\n    .trim()\n\n// The answer is the last text the model wrote.\n/** @internal */\nexport const runBrief = Effect.fnUntraced(\n  function* (\n    handler: RunHandler,\n    options: {\n      readonly specialist: Specialist.Any\n      readonly runId: RunId.RunId\n      readonly brief: Specialist.Brief\n      readonly history: Ref.Ref<ReadonlyArray<Prompt.Message>>\n      readonly inbox: Queue.Queue<Prompt.Message>\n      readonly chatId: Writer[\"chatId\"]\n    }\n  ) {\n    const { history, inbox } = options\n    const answer = yield* Ref.make(\"\")\n    const add = (messages: ReadonlyArray<Prompt.Message>) => Ref.update(history, (all) => [...all, ...messages])\n\n    // Read before every model call, when no tool calls are open: messages sent meanwhile join it now.\n    const current = Effect.gen(function* () {\n      yield* add(yield* Queue.clear(inbox))\n\n      return yield* Ref.get(history)\n    })\n\n    const run: Specialist.Run = {\n      runId: options.runId,\n      chatId: options.chatId,\n      brief: options.brief,\n      history: current,\n      loop: (loop) => {\n        const skills = loop.skills ?? []\n\n        return Loop.run({\n          instructions: mapText(loop.instructions, (text) => Skill.withCatalog(text, skills)),\n          toolkit: Toolkits.merge(Skills.toTools(skills), Toolkits.fromToolkitOrEmpty(loop.toolkit)),\n          history: current,\n          maxSteps: loop.maxSteps ?? 8\n        })\n      }\n    }\n\n    // A run lives in memory, so there's nothing to run again after a crash: `ToolStarted` isn't kept.\n    const record = Effect.fnUntraced(function* (event: AssistantLoop.Event) {\n      if (Predicate.isTagged(event, \"ToolStarted\")) return\n\n      if (Predicate.isTagged(event, \"ToolResult\")) return yield* add([toolMessage(event)])\n\n      yield* add(Prompt.fromResponseParts(event.parts).content)\n      const text = finalText(event)\n\n      if (text.length > 0) yield* Ref.set(answer, text)\n    })\n\n    yield* Stream.runForEach(handler(run), record)\n\n    const text = yield* Ref.get(answer)\n\n    yield* Telemetry.agentOutput(text)\n\n    return text\n  },\n  (effect, _handler, options) => effect.pipe(Telemetry.agentSpan(options.specialist.name, { chatId: options.chatId }))\n)\n\ntype Outcome =\n  | { readonly status: \"answered\"; readonly report: string }\n  | { readonly status: \"failed\"; readonly error: string }\n  | { readonly status: \"stopped\" | \"cut_off\" }\n\n// Only `stop` interrupts a brief's loop alone; a failure or a defect failed it.\nconst unanswered = (cause: Cause.Cause<unknown>): Outcome =>\n  Cause.hasInterruptsOnly(cause) ? { status: \"stopped\" } : { status: \"failed\", error: String(Cause.squash(cause)) }\n\n// The id is fixed per run and brief, so a replay is stored once.\n/** @internal */\nexport const finished = (\n  specialist: string,\n  run: { readonly runId: RunId.RunId; readonly brief: number; readonly goal: string },\n  outcome: Outcome\n): AssistantEvent.SpecialistFinished =>\n  AssistantEvent.SpecialistFinished.make({\n    id: InputId.fromKey(`${run.runId}:${run.brief}`),\n    runId: run.runId,\n    specialist,\n    brief: run.brief,\n    goal: run.goal,\n    ...outcome\n  })\n\nconst RunIdInput = RunId.RunId.annotate({ description: \"The runId spawn_specialist returned.\" })\n\nconst RunStatus = <const Status extends string>(status: ReadonlyArray<Status>) =>\n  Schema.Struct({ runId: RunId.RunId, status: Schema.Literals(status) })\n\nconst MessageSpecialist = Tool.make(\"message_specialist\", {\n  description:\n    \"Send more to one of your specialists: new instructions for a running one (it reads them at its next step), or a follow-up for a finished one (it continues with what it already found).\",\n  parameters: Schema.Struct({ runId: RunIdInput, message: Schema.String }),\n  success: RunStatus([\"steered\", \"continued\"]),\n  failure: ToolFailure.ToolFailure,\n  dependencies: [Specialist.Runner]\n})\n\nconst StopSpecialist = Tool.make(\"stop_specialist\", {\n  description:\n    \"Stop one of your running specialists, for example when the user changed their mind. It posts no answer.\",\n  parameters: Schema.Struct({ runId: RunIdInput }),\n  success: RunStatus([\"stopped\"]),\n  failure: ToolFailure.ToolFailure,\n  dependencies: [Specialist.Runner]\n})\n\nconst refused = <A, R>(effect: Effect.Effect<A, Specialist.SpecialistError, R>) =>\n  Effect.catch(effect, (error) => ToolFailure.fail(error.message, { reason: error.reason }))\n\n/** @internal */\nexport const toolkit = Effect.fnUntraced(function* (\n  specialists: ReadonlyArray<Specialist.Any>\n): Effect.fn.Return<Toolkits.Tools<Specialist.Runner>> {\n  if (specialists.length === 0) return Toolkits.empty\n\n  const catalog = specialists.map((specialist) => `- ${specialist.name}: ${specialist.description}`).join(\"\\n\")\n\n  const SpawnSpecialist = Tool.make(\"spawn_specialist\", {\n    description: `Start a specialist on a job. It works in the background and its report arrives as a <specialist> input, often within seconds. Write a self-contained brief: it can't see this chat. For a quick question, just call wait; for slow work, tell the user in one line first.\\nPass one of these as specialist:\\n${catalog}`,\n    parameters: Schema.Struct({\n      specialist: Schema.Literals(specialists.map((specialist) => specialist.name)).annotate({\n        description: \"The specialist to start, by name.\"\n      }),\n      ...Specialist.Brief.fields\n    }),\n    success: RunStatus([\"started\"]),\n    failure: ToolFailure.ToolFailure,\n    dependencies: [Specialist.Runner]\n  })\n\n  const SpecialistTools = Toolkit.make(SpawnSpecialist, MessageSpecialist, StopSpecialist)\n\n  const handlers = yield* SpecialistTools.toHandlers({\n    spawn_specialist: Effect.fnUntraced(function* ({ specialist, ...brief }) {\n      const runId = yield* refused(Specialist.Runner.use((runner) => runner.spawn({ specialist, brief })))\n\n      return { runId, status: \"started\" }\n    }),\n    message_specialist: Effect.fnUntraced(function* ({ runId, message }) {\n      const status = yield* refused(Specialist.Runner.use((runner) => runner.message(runId, message)))\n\n      return { runId, status }\n    }),\n    stop_specialist: Effect.fnUntraced(function* ({ runId }) {\n      yield* refused(Specialist.Runner.use((runner) => runner.stop(runId)))\n\n      return { runId, status: \"stopped\" }\n    })\n  })\n\n  return Toolkits.fromWithHandler(yield* Effect.provideContext(SpecialistTools, handlers))\n})\n\n// How long a finished run stays available for follow-ups.\nconst warm = \"15 minutes\"\n\ninterface Run {\n  readonly runId: RunId.RunId\n  readonly specialist: Specialist.Any\n  readonly handler: RunHandler\n  readonly goal: string\n  readonly brief: Specialist.Brief\n  readonly history: Ref.Ref<ReadonlyArray<Prompt.Message>>\n  // Messages sent while it works; the loop reads them at its next step.\n  readonly inbox: Queue.Queue<Prompt.Message>\n  // Counts follow-ups too; numbers the reports.\n  readonly briefs: Ref.Ref<number>\n  // Read and changed under `lock`.\n  readonly busy: Ref.Ref<boolean>\n  // Orders a message against the run finishing, so none is dropped or starts a second loop.\n  readonly lock: Semaphore.Semaphore\n}\n\n/** @internal */\nexport const makeRunner = Effect.fnUntraced(function* (options: {\n  readonly chat: Writer\n  readonly handle: CurrentChat[\"Service\"]\n  readonly specialists: ReadonlyArray<{\n    readonly specialist: Specialist.Any\n    readonly handler: RunHandler\n  }>\n  readonly model: Context.Context<LanguageModel.LanguageModel>\n  // An interruption keeps the hold.\n  readonly holding: (effect: Effect.Effect<void>) => Effect.Effect<void>\n  readonly report: (finished: AssistantEvent.SpecialistFinished) => Effect.Effect<void>\n}) {\n  const { chat } = options\n  const runs = new Map<string, Run>()\n  const byName = new Map(options.specialists.map((entry) => [entry.specialist.name, entry]))\n  // The brief loops working now, and the finished runs' countdowns to being forgotten.\n  const working = yield* FiberMap.make<string, Outcome>()\n  const forgetting = yield* FiberMap.make<string>()\n  const background = yield* FiberSet.make()\n\n  const report = Effect.fnUntraced(function* (run: Run, outcome: Outcome) {\n    const brief = yield* Ref.get(run.briefs)\n\n    yield* options.report(finished(run.specialist.name, { runId: run.runId, brief, goal: run.goal }, outcome))\n  })\n\n  const once = (run: Run) =>\n    runBrief(run.handler, {\n      specialist: run.specialist,\n      runId: run.runId,\n      brief: run.brief,\n      history: run.history,\n      inbox: run.inbox,\n      chatId: chat.chatId\n    }).pipe(\n      Effect.provideContext(Context.merge(options.model, services)),\n      // Spawned from a turn, a run would inherit the turn's own tools: a run can't message the user.\n      Effect.provideService(Loop.TurnTools, Option.none()),\n      Effect.matchCause({\n        onSuccess: (answer): Outcome => ({ status: \"answered\", report: answer }),\n        onFailure: unanswered\n      })\n    )\n\n  // A message that arrived after the last read: keep working before reporting.\n  const untilCaughtUp = Effect.fnUntraced(function* (run: Run) {\n    let outcome = yield* once(run)\n\n    while ((yield* Queue.size(run.inbox)) > 0) outcome = yield* once(run)\n\n    return outcome\n  })\n\n  const work = Effect.fnUntraced(\n    function* (run: Run) {\n      yield* chat.append([\n        AssistantEvent.SpecialistStarted.make({\n          runId: run.runId,\n          specialist: run.specialist.name,\n          goal: run.goal\n        })\n      ])\n\n      let finished = false\n\n      while (!finished) {\n        const exit = yield* Fiber.await(yield* FiberMap.run(working, run.runId, untilCaughtUp(run)))\n        const outcome = Exit.isSuccess(exit) ? exit.value : unanswered(exit.cause)\n\n        finished = yield* run.lock.withPermit(\n          Effect.gen(function* () {\n            // A message steered in after the loop's last read: work on it before reporting.\n            if (outcome.status !== \"stopped\" && (yield* Queue.size(run.inbox)) > 0) return false\n\n            yield* report(run, outcome)\n            yield* Ref.set(run.busy, false)\n            yield* FiberMap.run(\n              forgetting,\n              run.runId,\n              Effect.gen(function* () {\n                yield* Effect.sleep(warm)\n                runs.delete(run.runId)\n              })\n            )\n\n            return true\n          })\n        )\n      }\n    },\n    // A shutdown interrupts the whole run: no turn can start now, so the report waits for the chat's next start.\n    (effect, run) => Effect.onInterrupt(effect, () => report(run, { status: \"cut_off\" }))\n  )\n\n  const launch = Effect.fnUntraced(function* (run: Run) {\n    yield* FiberMap.remove(forgetting, run.runId)\n    yield* FiberSet.run(background, options.holding(work(run)))\n  })\n\n  // Under the run's lock.\n  const send = Effect.fnUntraced(function* (run: Run, text: string): Effect.fn.Return<\"steered\" | \"continued\"> {\n    yield* Ref.update(run.briefs, (n) => n + 1)\n\n    if (yield* Ref.get(run.busy)) {\n      yield* Queue.offer(run.inbox, userText(text))\n\n      return \"steered\"\n    }\n\n    yield* Ref.update(run.history, (all) => [...all, userText(text)])\n    yield* Ref.set(run.busy, true)\n    yield* launch(run)\n\n    return \"continued\"\n  })\n\n  const runner = Specialist.Runner.of({\n    spawn: Effect.fnUntraced(function* (request: Specialist.SpawnRequest) {\n      const found = byName.get(request.specialist)\n\n      if (found === undefined) {\n        return yield* new Specialist.SpecialistError({\n          reason: \"UnknownSpecialist\",\n          message: `No specialist named ${request.specialist}.`\n        })\n      }\n\n      const run: Run = {\n        runId: yield* RunId.make(),\n        specialist: found.specialist,\n        handler: found.handler,\n        goal: request.brief.goal,\n        brief: request.brief,\n        history: yield* Ref.make<ReadonlyArray<Prompt.Message>>([userText(briefText(request.brief))]),\n        inbox: yield* Queue.unbounded<Prompt.Message>(),\n        briefs: yield* Ref.make(1),\n        busy: yield* Ref.make(true),\n        lock: yield* Semaphore.make(1)\n      }\n\n      runs.set(run.runId, run)\n      yield* launch(run)\n\n      return run.runId\n    }),\n    message: Effect.fnUntraced(function* (\n      runId: RunId.RunId,\n      text: string\n    ): Effect.fn.Return<\"steered\" | \"continued\", Specialist.SpecialistError> {\n      const run = runs.get(runId)\n\n      if (run === undefined) {\n        return yield* new Specialist.SpecialistError({\n          reason: \"RunNotFound\",\n          message: `Run ${runId} is no longer available. Spawn a new one with the full context.`\n        })\n      }\n\n      return yield* run.lock.withPermit(send(run, text))\n    }),\n    stop: Effect.fnUntraced(function* (runId: RunId.RunId) {\n      if (!(yield* FiberMap.has(working, runId))) {\n        return yield* new Specialist.SpecialistError({ reason: \"NotRunning\", message: `Run ${runId} isn't running.` })\n      }\n\n      yield* FiberMap.remove(working, runId)\n    })\n  })\n\n  // What a run's handler gets: its chat, this runner, and (merged in `once`) the model.\n  const services = Context.make(CurrentChat, options.handle).pipe(Context.add(Specialist.Runner, runner))\n\n  return runner\n})\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsCA,MAAa,aAAa,UACxB;CACE,SAAS,MAAM;CACf,MAAM,WAAW,YAAY,MAAM;CACnC,MAAM,mBAAmB,cAAc,MAAM;CAC7C,MAAM,UAAU,gBAAgB,MAAM;CACtC,MAAM,UAAU,WAAW,MAAM;AACnC,CAAC,CACE,OAAO,OAAO,CAAC,CACf,KAAK,IAAI;AAEd,MAAM,aAAa,WACjB,OAAO,MACJ,SAAS,SAAU,KAAK,SAAS,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAE,CAAC,CAC5D,KAAK,EAAE,CAAC,CACR,KAAK;;AAIV,MAAa,WAAW,OAAO,WAC7B,WACE,SACA,SAQA;CACA,MAAM,EAAE,SAAS,UAAU;CAC3B,MAAM,SAAS,OAAO,IAAI,KAAK,EAAE;CACjC,MAAM,OAAO,aAA4C,IAAI,OAAO,UAAU,QAAQ,CAAC,GAAG,KAAK,GAAG,QAAQ,CAAC;CAG3G,MAAM,UAAU,OAAO,IAAI,aAAa;EACtC,OAAO,IAAI,OAAO,MAAM,MAAM,KAAK,CAAC;EAEpC,OAAO,OAAO,IAAI,IAAI,OAAO;CAC/B,CAAC;CAED,MAAMA,QAAsB;EAC1B,OAAO,QAAQ;EACf,QAAQ,QAAQ;EAChB,OAAO,QAAQ;EACf,SAAS;EACT,OAAO,SAAS;GACd,MAAM,SAAS,KAAK,UAAU,CAAC;GAE/B,OAAOC,IAAS;IACd,cAAc,QAAQ,KAAK,eAAe,SAASC,YAAkB,MAAM,MAAM,CAAC;IAClF,SAASC,MAAeC,QAAe,MAAM,GAAGC,mBAA4B,KAAK,OAAO,CAAC;IACzF,SAAS;IACT,UAAU,KAAK,YAAY;GAC7B,CAAC;EACH;CACF;CAGA,MAAM,SAAS,OAAO,WAAW,WAAW,OAA4B;EACtE,IAAI,UAAU,SAAS,OAAO,aAAa,GAAG;EAE9C,IAAI,UAAU,SAAS,OAAO,YAAY,GAAG,OAAO,OAAO,IAAI,CAAC,YAAY,KAAK,CAAC,CAAC;EAEnF,OAAO,IAAI,OAAO,kBAAkB,MAAM,KAAK,CAAC,CAAC,OAAO;EACxD,MAAM,OAAO,UAAU,KAAK;EAE5B,IAAI,KAAK,SAAS,GAAG,OAAO,IAAI,IAAI,QAAQ,IAAI;CAClD,CAAC;CAED,OAAO,OAAO,WAAW,QAAQL,KAAG,GAAG,MAAM;CAE7C,MAAM,OAAO,OAAO,IAAI,IAAI,MAAM;CAElC,OAAOM,YAAsB,IAAI;CAEjC,OAAO;AACT,IACC,QAAQ,UAAU,YAAY,OAAO,KAAKC,UAAoB,QAAQ,WAAW,MAAM,EAAE,QAAQ,QAAQ,OAAO,CAAC,CAAC,CACrH;AAQA,MAAM,cAAc,UAClB,MAAM,kBAAkB,KAAK,IAAI,EAAE,QAAQ,UAAU,IAAI;CAAE,QAAQ;CAAU,OAAO,OAAO,MAAM,OAAO,KAAK,CAAC;AAAE;;AAIlH,MAAa,YACX,YACA,KACA,YAAA,mBAEkC,KAAK;CACrC,IAAIC,QAAgB,GAAG,IAAI,MAAM,GAAG,IAAI,OAAO;CAC/C,OAAO,IAAI;CACX;CACA,OAAO,IAAI;CACX,MAAM,IAAI;CACV,GAAG;AACL,CAAC;AAEH,MAAM,aAAA,MAAyB,SAAS,EAAE,aAAa,uCAAuC,CAAC;AAE/F,MAAM,aAA0C,WAC9C,OAAO,OAAO;CAAE,OAAOC;CAAa,QAAQ,OAAO,SAAS,MAAM;AAAE,CAAC;AAEvE,MAAM,oBAAoB,KAAK,KAAK,sBAAsB;CACxD,aACE;CACF,YAAY,OAAO,OAAO;EAAE,OAAO;EAAY,SAAS,OAAO;CAAO,CAAC;CACvE,SAAS,UAAU,CAAC,WAAW,WAAW,CAAC;CAC3C,SAASC;CACT,cAAc,CAACC,MAAiB;AAClC,CAAC;AAED,MAAM,iBAAiB,KAAK,KAAK,mBAAmB;CAClD,aACE;CACF,YAAY,OAAO,OAAO,EAAE,OAAO,WAAW,CAAC;CAC/C,SAAS,UAAU,CAAC,SAAS,CAAC;CAC9B,SAASD;CACT,cAAc,CAACC,MAAiB;AAClC,CAAC;AAED,MAAM,WAAiB,WACrB,OAAO,MAAM,SAAS,UAAUC,KAAiB,MAAM,SAAS,EAAE,QAAQ,MAAM,OAAO,CAAC,CAAC;;AAG3F,MAAa,UAAU,OAAO,WAAW,WACvC,aACqD;CACrD,IAAI,YAAY,WAAW,GAAG,OAAOC;CAErC,MAAM,UAAU,YAAY,KAAK,eAAe,KAAK,WAAW,KAAK,IAAI,WAAW,aAAa,CAAC,CAAC,KAAK,IAAI;CAE5G,MAAM,kBAAkB,KAAK,KAAK,oBAAoB;EACpD,aAAa,gTAAgT;EAC7T,YAAY,OAAO,OAAO;GACxB,YAAY,OAAO,SAAS,YAAY,KAAK,eAAe,WAAW,IAAI,CAAC,CAAC,CAAC,SAAS,EACrF,aAAa,oCACf,CAAC;GACD,GAAA,MAAoB;EACtB,CAAC;EACD,SAAS,UAAU,CAAC,SAAS,CAAC;EAC9B,SAASH;EACT,cAAc,CAACC,MAAiB;CAClC,CAAC;CAED,MAAM,kBAAkB,QAAQ,KAAK,iBAAiB,mBAAmB,cAAc;CAEvF,MAAM,WAAW,OAAO,gBAAgB,WAAW;EACjD,kBAAkB,OAAO,WAAW,WAAW,EAAE,YAAY,GAAG,SAAS;GAGvE,OAAO;IAAE,OAAA,OAFY,QAAA,OAA0B,KAAK,WAAW,OAAO,MAAM;KAAE;KAAY;IAAM,CAAC,CAAC,CAAC;IAEnF,QAAQ;GAAU;EACpC,CAAC;EACD,oBAAoB,OAAO,WAAW,WAAW,EAAE,OAAO,WAAW;GAGnE,OAAO;IAAE;IAAO,QAAA,OAFM,QAAA,OAA0B,KAAK,WAAW,OAAO,QAAQ,OAAO,OAAO,CAAC,CAAC;GAExE;EACzB,CAAC;EACD,iBAAiB,OAAO,WAAW,WAAW,EAAE,SAAS;GACvD,OAAO,QAAA,OAA0B,KAAK,WAAW,OAAO,KAAK,KAAK,CAAC,CAAC;GAEpE,OAAO;IAAE;IAAO,QAAQ;GAAU;EACpC,CAAC;CACH,CAAC;CAED,OAAOG,gBAAyB,OAAO,OAAO,eAAe,iBAAiB,QAAQ,CAAC;AACzF,CAAC;AAGD,MAAM,OAAO;;AAoBb,MAAa,aAAa,OAAO,WAAW,WAAW,SAWpD;CACD,MAAM,EAAE,SAAS;CACjB,MAAM,uBAAO,IAAI,IAAiB;CAClC,MAAM,SAAS,IAAI,IAAI,QAAQ,YAAY,KAAK,UAAU,CAAC,MAAM,WAAW,MAAM,KAAK,CAAC,CAAC;CAEzF,MAAM,UAAU,OAAO,SAAS,KAAsB;CACtD,MAAM,aAAa,OAAO,SAAS,KAAa;CAChD,MAAM,aAAa,OAAO,SAAS,KAAK;CAExC,MAAM,SAAS,OAAO,WAAW,WAAW,KAAU,SAAkB;EACtE,MAAM,QAAQ,OAAO,IAAI,IAAI,IAAI,MAAM;EAEvC,OAAO,QAAQ,OAAO,SAAS,IAAI,WAAW,MAAM;GAAE,OAAO,IAAI;GAAO;GAAO,MAAM,IAAI;EAAK,GAAG,OAAO,CAAC;CAC3G,CAAC;CAED,MAAM,QAAQ,QACZ,SAAS,IAAI,SAAS;EACpB,YAAY,IAAI;EAChB,OAAO,IAAI;EACX,OAAO,IAAI;EACX,SAAS,IAAI;EACb,OAAO,IAAI;EACX,QAAQ,KAAK;CACf,CAAC,CAAC,CAAC,KACD,OAAO,eAAe,QAAQ,MAAM,QAAQ,OAAO,QAAQ,CAAC,GAE5D,OAAO,eAAeC,WAAgB,OAAO,KAAK,CAAC,GACnD,OAAO,WAAW;EAChB,YAAY,YAAqB;GAAE,QAAQ;GAAY,QAAQ;EAAO;EACtE,WAAW;CACb,CAAC,CACH;CAGF,MAAM,gBAAgB,OAAO,WAAW,WAAW,KAAU;EAC3D,IAAI,UAAU,OAAO,KAAK,GAAG;EAE7B,QAAQ,OAAO,MAAM,KAAK,IAAI,KAAK,KAAK,GAAG,UAAU,OAAO,KAAK,GAAG;EAEpE,OAAO;CACT,CAAC;CAED,MAAM,OAAO,OAAO,WAClB,WAAW,KAAU;EACnB,OAAO,KAAK,OAAO,CAAA,kBACgB,KAAK;GACpC,OAAO,IAAI;GACX,YAAY,IAAI,WAAW;GAC3B,MAAM,IAAI;EACZ,CAAC,CACH,CAAC;EAED,IAAI,WAAW;EAEf,OAAO,CAAC,UAAU;GAChB,MAAM,OAAO,OAAO,MAAM,MAAM,OAAO,SAAS,IAAI,SAAS,IAAI,OAAO,cAAc,GAAG,CAAC,CAAC;GAC3F,MAAM,UAAU,KAAK,UAAU,IAAI,IAAI,KAAK,QAAQ,WAAW,KAAK,KAAK;GAEzE,WAAW,OAAO,IAAI,KAAK,WACzB,OAAO,IAAI,aAAa;IAEtB,IAAI,QAAQ,WAAW,cAAc,OAAO,MAAM,KAAK,IAAI,KAAK,KAAK,GAAG,OAAO;IAE/E,OAAO,OAAO,KAAK,OAAO;IAC1B,OAAO,IAAI,IAAI,IAAI,MAAM,KAAK;IAC9B,OAAO,SAAS,IACd,YACA,IAAI,OACJ,OAAO,IAAI,aAAa;KACtB,OAAO,OAAO,MAAM,IAAI;KACxB,KAAK,OAAO,IAAI,KAAK;IACvB,CAAC,CACH;IAEA,OAAO;GACT,CAAC,CACH;EACF;CACF,IAEC,QAAQ,QAAQ,OAAO,YAAY,cAAc,OAAO,KAAK,EAAE,QAAQ,UAAU,CAAC,CAAC,CACtF;CAEA,MAAM,SAAS,OAAO,WAAW,WAAW,KAAU;EACpD,OAAO,SAAS,OAAO,YAAY,IAAI,KAAK;EAC5C,OAAO,SAAS,IAAI,YAAY,QAAQ,QAAQ,KAAK,GAAG,CAAC,CAAC;CAC5D,CAAC;CAGD,MAAM,OAAO,OAAO,WAAW,WAAW,KAAU,MAAyD;EAC3G,OAAO,IAAI,OAAO,IAAI,SAAS,MAAM,IAAI,CAAC;EAE1C,IAAI,OAAO,IAAI,IAAI,IAAI,IAAI,GAAG;GAC5B,OAAO,MAAM,MAAM,IAAI,OAAO,SAAS,IAAI,CAAC;GAE5C,OAAO;EACT;EAEA,OAAO,IAAI,OAAO,IAAI,UAAU,QAAQ,CAAC,GAAG,KAAK,SAAS,IAAI,CAAC,CAAC;EAChE,OAAO,IAAI,IAAI,IAAI,MAAM,IAAI;EAC7B,OAAO,OAAO,GAAG;EAEjB,OAAO;CACT,CAAC;CAED,MAAM,SAAA,OAA2B,GAAG;EAClC,OAAO,OAAO,WAAW,WAAW,SAAkC;GACpE,MAAM,QAAQ,OAAO,IAAI,QAAQ,UAAU;GAE3C,IAAI,UAAU,KAAA,GACZ,OAAO,OAAO,IAAIC,gBAA2B;IAC3C,QAAQ;IACR,SAAS,uBAAuB,QAAQ,WAAW;GACrD,CAAC;GAGH,MAAM,MAAW;IACf,OAAO,OAAOC,KAAW;IACzB,YAAY,MAAM;IAClB,SAAS,MAAM;IACf,MAAM,QAAQ,MAAM;IACpB,OAAO,QAAQ;IACf,SAAS,OAAO,IAAI,KAAoC,CAAC,SAAS,UAAU,QAAQ,KAAK,CAAC,CAAC,CAAC;IAC5F,OAAO,OAAO,MAAM,UAA0B;IAC9C,QAAQ,OAAO,IAAI,KAAK,CAAC;IACzB,MAAM,OAAO,IAAI,KAAK,IAAI;IAC1B,MAAM,OAAO,UAAU,KAAK,CAAC;GAC/B;GAEA,KAAK,IAAI,IAAI,OAAO,GAAG;GACvB,OAAO,OAAO,GAAG;GAEjB,OAAO,IAAI;EACb,CAAC;EACD,SAAS,OAAO,WAAW,WACzB,OACA,MACuE;GACvE,MAAM,MAAM,KAAK,IAAI,KAAK;GAE1B,IAAI,QAAQ,KAAA,GACV,OAAO,OAAO,IAAID,gBAA2B;IAC3C,QAAQ;IACR,SAAS,OAAO,MAAM;GACxB,CAAC;GAGH,OAAO,OAAO,IAAI,KAAK,WAAW,KAAK,KAAK,IAAI,CAAC;EACnD,CAAC;EACD,MAAM,OAAO,WAAW,WAAW,OAAoB;GACrD,IAAI,EAAE,OAAO,SAAS,IAAI,SAAS,KAAK,IACtC,OAAO,OAAO,IAAIA,gBAA2B;IAAE,QAAQ;IAAc,SAAS,OAAO,MAAM;GAAiB,CAAC;GAG/G,OAAO,SAAS,OAAO,SAAS,KAAK;EACvC,CAAC;CACH,CAAC;CAGD,MAAM,WAAW,QAAQ,KAAK,aAAa,QAAQ,MAAM,CAAC,CAAC,KAAK,QAAQ,IAAIL,QAAmB,MAAM,CAAC;CAEtG,OAAO;AACT,CAAC"}

package/dist/internal/toolkit.d.ts

@@ -3,11 +3,10 @@ import { AssistantToolCall } from "../AssistantToolCall.js";
 import * as Context from "effect/Context";
 import * as Effect from "effect/Effect";
 import * as Schema from "effect/Schema";
-import * as Prompt from "effect/ai/Prompt";
 import * as Tool from "effect/ai/Tool";
 import * as Toolkit from "effect/ai/Toolkit";
 declare namespace toolkit_d_exports {
-  export { CallResult, ToolServices, Tools, duplicateNames, empty, failed, fromToolkit, fromToolkitOrEmpty, fromWithHandler, merge, needsDecision, needsDecisionOf, provide, runNative, runOnly, withContext };
+  export { CallResult, ToolServices, Tools, duplicateNames, empty, failed, fromToolkit, fromToolkitOrEmpty, fromWithHandler, merge, needingApproval, needsApprovalMessage, provide, runNative, runOnly, withContext };
 }
 /** @internal */
 export interface CallResult {
@@ -19,26 +18,15 @@ export interface CallResult {
 export interface Tools<out R = never> {
   readonly definitions: ReadonlyArray<Tool.Any>;
   readonly run: (call: ToolCallPart) => Effect.Effect<CallResult, never, R>;
-  readonly needsDecision?: ((call: ToolCallPart, history: ReadonlyArray<Prompt.Message>) => Effect.Effect<boolean, never, R>) | undefined;
 }
 /** @internal */
-export type ToolServices<T extends Record<string, Tool.Any>> = Exclude<Tool.HandlerServices<T[keyof T]>, AssistantToolCall> | T[keyof T]["parametersSchema"]["DecodingServices"];
+export type ToolServices<T extends Record<string, Tool.Any>> = Exclude<Tool.HandlerServices<T[keyof T]>, AssistantToolCall>;
 /** @internal */
 export declare const failed: (message: string, reason: string) => CallResult;
 /** @internal */
-export declare const needsDecision: <T extends Record<string, Tool.Any>>(toolkit: {
-  readonly tools: T;
-}, call: {
-  readonly type: "tool-call";
-  readonly id: string;
-  readonly params: Schema.Json;
-  readonly providerExecuted: boolean;
-  readonly "~effect/ai/Response/Part": "~effect/ai/Response/Part";
-  readonly metadata: {
-    readonly [x: string]: Schema.Json;
-  };
-  readonly name: string;
-}, history: readonly Prompt.Message[]) => Effect.Effect<boolean, never, ToolServices<T>>;
+export declare const needingApproval: (definitions: ReadonlyArray<Tool.Any>) => ReadonlyArray<string>;
+/** @internal */
+export declare const needsApprovalMessage: (names: ReadonlyArray<string>) => string;
 /** @internal */
 export declare const runNative: <T extends Record<string, Tool.Any>>(toolkit: Toolkit.WithHandler<T>, call: {
   readonly type: "tool-call";
@@ -50,7 +38,7 @@ export declare const runNative: <T extends Record<string, Tool.Any>>(toolkit: To
     readonly [x: string]: Schema.Json;
   };
   readonly name: string;
-}) => Effect.Effect<CallResult, never, ToolServices<T>>;
+}) => Effect.Effect<CallResult, never, Exclude<Tool.HandlerServices<T[keyof T]>, AssistantToolCall>>;
 /** @internal */
 export declare const fromWithHandler: <T extends Record<string, Tool.Any>>(toolkit: Toolkit.WithHandler<T>) => Tools<ToolServices<T>>;
 /** @internal */
@@ -64,8 +52,6 @@ export declare const empty: Tools;
 /** @internal */
 export declare const merge: <A, B>(self: Tools<A>, that: Tools<B>) => Tools<A | B>;
 /** @internal */
-export declare const needsDecisionOf: <R>(tools: Tools<R>, call: ToolCallPart, history: ReadonlyArray<Prompt.Message>) => Effect.Effect<boolean, never, R>;
-/** @internal */
 export declare const provide: <R>(tools: Tools<R>, context: Context.Context<R>) => Tools;
 /** @internal */
 export declare const duplicateNames: (tools: Tools<unknown>) => ReadonlyArray<string>;

package/dist/internal/toolkit.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"toolkit.d.ts","names":[],"sources":["../../src/internal/toolkit.ts"],"mappings":";;;;;;;;;;;;iBAgBiB;WACN,SAAS;WACT,QAAQ,OAAO;WACf;;;iBAKM,UAAU;WAChB,aAAa,cAAc,KAAK;WAChC,MAAM,MAAM,iBAAgC,OAAO,OAAO,mBAAmB;WAE7E,kBACH,MAAM,cAA6B,SAAS,cAAc,OAAO,aAAa,OAAO,uBAAuB;;;YAKxG,aAAa,UAAU,eAAe,KAAK,QACnD,QAAQ,KAAK,gBAAgB,QAAQ,KAAK,qBAE1C,QAAQ;;qBAKC,SAAM,iBAAmB,mBAAmB;;qBAkC5C,gBAA6C,UAAU,eAAe,KAAK,MAAI;WAC/D,OAAA;;;;;;;;;;;GAoB3B,kBAAA,OAAA,cAAA,OAAA,uBAAA,aAAA;;qBASW,YAAyC,UAAU,eAAe,KAAK,MAAI,SAAA,QAAA,YAAA,IAAA;;;;;;;;;;MAiCtF,OAAA,OAAA,mBAAA,aAAA;;qBAGW,kBAAmB,UAAU,eAAe,KAAK,MAAI,SACvD,QAAQ,YAAY,OAC5B,MAAM,aAAa;;qBAOT,cAAe,UAAU,eAAe,KAAK,MAAI,SACnD,QAAQ,QAAQ,OACxB,MAAM,aAAa,KAAK,KAAK,YAAY;;qBAU/B,cAAe,UAAU,eAAe,KAAK,MAAM,GAAC,SACtD,QAAQ,QAAQ,IAAE,WAChB,SAAS,QAAQ,QAAQ,OAAO,QAAQ,aAAa,OAC/D,MAAM,IAAI,aAAa;;qBAYb,qBAAsB,UAAU,eAAe,KAAK,MAAI,SAC1D,QAAQ,QAAQ,mBACxB,MAAM,aAAa,KAAK,KAAK,YAAY;;qBAG/B,OAAO;;qBAMP,QAAS,GAAG,GAAC,MAAQ,MAAM,IAAE,MAAQ,MAAM,OAAK,MAAM,IAAI;;qBAY1D,kBAAmB,GAAC,OACxB,MAAM,IAAE,MACT,cAA2B,SACxB,cAAc,OAAO,aAC7B,OAAO,uBAAuB;;qBAKpB,UAAW,GAAC,OAAS,MAAM,IAAE,SAAW,QAAQ,QAAQ,OAAK;;qBAO7D,iBAAc,OAAW,mBAAiB;;qBAQ1C,UAAW,GAAC,OAAS,MAAM,IAAE,OAAS,uBAAqB,oBAAoB,MAAM"}
+{"version":3,"file":"toolkit.d.ts","names":[],"sources":["../../src/internal/toolkit.ts"],"mappings":";;;;;;;;;;;iBAciB;WACN,SAAS;WACT,QAAQ,OAAO;WACf;;;iBAKM,UAAU;WAChB,aAAa,cAAc,KAAK;WAChC,MAAM,MAAM,iBAAgC,OAAO,OAAO,mBAAmB;;;YAI5E,aAAa,UAAU,eAAe,KAAK,QAAQ,QAC7D,KAAK,gBAAgB,QAAQ,KAC7B;;qBAMW,SAAM,iBAAmB,mBAAmB;;qBA6B5C,kBAAe,aAAiB,cAAc,KAAK,SAAO;;qBAI1D,uBAAoB,OAAW;;qBAU/B,YAAyC,UAAU,eAAe,KAAK,MAAI,SAAA,QAAA,YAAA,IAAA;;;;;;;;;;MAiCtF,OAAA,OAAA,mBAAA,QAAA,KAAA,gBAAA,QAAA,KAAA;;qBAGW,kBAAmB,UAAU,eAAe,KAAK,MAAI,SACvD,QAAQ,YAAY,OAC5B,MAAM,aAAa;;qBAMT,cAAe,UAAU,eAAe,KAAK,MAAI,SACnD,QAAQ,QAAQ,OACxB,MAAM,aAAa,KAAK,KAAK,YAAY;;qBAS/B,cAAe,UAAU,eAAe,KAAK,MAAM,GAAC,SACtD,QAAQ,QAAQ,IAAE,WAChB,SAAS,QAAQ,QAAQ,OAAO,QAAQ,aAAa,OAC/D,MAAM,IAAI,aAAa;;qBAWb,qBAAsB,UAAU,eAAe,KAAK,MAAI,SAC1D,QAAQ,QAAQ,mBACxB,MAAM,aAAa,KAAK,KAAK,YAAY;;qBAG/B,OAAO;;qBAMP,QAAS,GAAG,GAAC,MAAQ,MAAM,IAAE,MAAQ,MAAM,OAAK,MAAM,IAAI;;qBAW1D,UAAW,GAAC,OAAS,MAAM,IAAE,SAAW,QAAQ,QAAQ,OAAK;;qBAM7D,iBAAc,OAAW,mBAAiB;;qBAQ1C,UAAW,GAAC,OAAS,MAAM,IAAE,OAAS,uBAAqB,oBAAoB,MAAM"}

package/dist/internal/toolkit.js

@@ -1,7 +1,6 @@
 import { toJson } from "./schema.js";
 import { AssistantToolCall } from "../AssistantToolCall.js";
 import { ToolFailure } from "../ToolFailure.js";
-import * as Predicate from "effect/Predicate";
 import * as Effect from "effect/Effect";
 import * as Schema from "effect/Schema";
 import * as Stream from "effect/Stream";
@@ -26,22 +25,10 @@ const failure = (result) => ({
 	reason: Option.getOrUndefined(Option.orElse(Option.map(decodeReason(result), ({ reason }) => reason), () => Option.map(decodeTag(result), ({ _tag }) => _tag)))
 });
 const hasTool = (toolkit, name) => Object.hasOwn(toolkit.tools, name);
-const toolNamed = (toolkit, name) => Object.hasOwn(toolkit.tools, name) ? toolkit.tools[name] : void 0;
-const decodeParams = (tool, params) => Schema.decodeUnknownEffect(tool.parametersSchema)(params);
 /** @internal */
-const needsDecision = Effect.fnUntraced(function* (toolkit, call, history) {
-	const tool = toolNamed(toolkit, call.name);
-	if (tool === void 0) return false;
-	const needs = tool.needsApproval;
-	if (needs === void 0 || Predicate.isBoolean(needs)) return needs === true;
-	return yield* Effect.gen(function* () {
-		const decided = needs(yield* decodeParams(tool, call.params), {
-			toolCallId: call.id,
-			messages: history
-		});
-		return Effect.isEffect(decided) ? yield* decided : decided;
-	}).pipe(Effect.orElseSucceed(() => false));
-});
+const needingApproval = (definitions) => definitions.flatMap((tool) => tool.needsApproval === void 0 || tool.needsApproval === false ? [] : [tool.name]);
+/** @internal */
+const needsApprovalMessage = (names) => `${names.join(", ")} ${names.length === 1 ? "sets" : "set"} needsApproval, and effect-assistant doesn't run tools that wait for approval. Remove needsApproval: let the model call the tool with final values, or show a block whose button calls your app's API, which does the work and submits an app event that wakes the chat.`;
 const isParams = (params) => params !== void 0;
 /** @internal */
 const runNative = Effect.fnUntraced(function* (toolkit, call) {
@@ -67,16 +54,14 @@ const runNative = Effect.fnUntraced(function* (toolkit, call) {
 /** @internal */
 const fromWithHandler = (toolkit) => ({
 	definitions: Object.values(toolkit.tools),
-	run: (call) => runNative(toolkit, call),
-	needsDecision: (call, history) => needsDecision(toolkit, call, history)
+	run: (call) => runNative(toolkit, call)
 });
 /** @internal */
 const fromToolkit = (toolkit) => ({
 	definitions: Object.values(toolkit.tools),
 	run: Effect.fnUntraced(function* (call) {
 		return yield* runNative(yield* toolkit, call);
-	}),
-	needsDecision: (call, history) => needsDecision(toolkit, call, history)
+	})
 });
 /** @internal */
 const withContext = (toolkit, handlers) => ({
@@ -85,8 +70,7 @@ const withContext = (toolkit, handlers) => ({
 		const context = yield* Effect.context();
 		const resolved = yield* Effect.provideContext(toolkit, yield* toolkit.toHandlers(handlers(context)));
 		return yield* runNative(resolved, call);
-	}),
-	needsDecision: (call, history) => needsDecision(toolkit, call, history)
+	})
 });
 /** @internal */
 const fromToolkitOrEmpty = (toolkit) => toolkit === void 0 ? empty : fromToolkit(toolkit);
@@ -100,17 +84,13 @@ const merge = (self, that) => {
 	const inSelf = (call) => self.definitions.some((tool) => tool.name === call.name);
 	return {
 		definitions: [...self.definitions, ...that.definitions],
-		run: (call) => inSelf(call) ? self.run(call) : that.run(call),
-		needsDecision: (call, history) => inSelf(call) ? needsDecisionOf(self, call, history) : needsDecisionOf(that, call, history)
+		run: (call) => inSelf(call) ? self.run(call) : that.run(call)
 	};
 };
 /** @internal */
-const needsDecisionOf = (tools, call, history) => tools.needsDecision === void 0 ? Effect.succeed(false) : tools.needsDecision(call, history);
-/** @internal */
 const provide = (tools, context) => ({
 	definitions: tools.definitions,
-	run: (call) => Effect.provideContext(tools.run(call), context),
-	needsDecision: (call, history) => Effect.provideContext(needsDecisionOf(tools, call, history), context)
+	run: (call) => Effect.provideContext(tools.run(call), context)
 });
 /** @internal */
 const duplicateNames = (tools) => {
@@ -120,10 +100,9 @@ const duplicateNames = (tools) => {
 /** @internal */
 const runOnly = (tools, names, message) => ({
 	definitions: tools.definitions,
-	run: (call) => names.includes(call.name) ? tools.run(call) : Effect.succeed(failed(message, "NotNow")),
-	needsDecision: (call, history) => names.includes(call.name) ? needsDecisionOf(tools, call, history) : Effect.succeed(false)
+	run: (call) => names.includes(call.name) ? tools.run(call) : Effect.succeed(failed(message, "NotNow"))
 });
 //#endregion
-export { duplicateNames, empty, failed, fromToolkit, fromToolkitOrEmpty, fromWithHandler, merge, needsDecision, needsDecisionOf, provide, runNative, runOnly, withContext };
+export { duplicateNames, empty, failed, fromToolkit, fromToolkitOrEmpty, fromWithHandler, merge, needingApproval, needsApprovalMessage, provide, runNative, runOnly, withContext };
 
 //# sourceMappingURL=toolkit.js.map

package/dist/internal/toolkit.js.map

@@ -1 +1 @@
-{"version":3,"file":"toolkit.js","names":["ToolFailure.ToolFailure"],"sources":["../../src/internal/toolkit.ts"],"sourcesContent":["import * as AiError from \"effect/ai/AiError\"\nimport type * as Prompt from \"effect/ai/Prompt\"\nimport type * as Tool from \"effect/ai/Tool\"\nimport type * as Toolkit from \"effect/ai/Toolkit\"\nimport type * as Context from \"effect/Context\"\nimport * as Effect from \"effect/Effect\"\nimport * as Option from \"effect/Option\"\nimport * as Predicate from \"effect/Predicate\"\nimport * as Schema from \"effect/Schema\"\nimport * as Stream from \"effect/Stream\"\nimport type * as AssistantEvent from \"../AssistantEvent.ts\"\nimport { AssistantToolCall } from \"../AssistantToolCall.ts\"\nimport * as ToolFailure from \"../ToolFailure.ts\"\nimport { toJson } from \"./schema.ts\"\n\n/** @internal */\nexport interface CallResult {\n  readonly outcome: AssistantEvent.ToolOutcome\n  readonly result: Schema.Json\n  readonly reason?: string | undefined\n}\n\n// Apps give the loop native toolkits; these are what the loop makes of them, merged with its own.\n/** @internal */\nexport interface Tools<out R = never> {\n  readonly definitions: ReadonlyArray<Tool.Any>\n  readonly run: (call: AssistantEvent.ToolCallPart) => Effect.Effect<CallResult, never, R>\n  // Unset: no call waits for the user's decision.\n  readonly needsDecision?:\n    | ((call: AssistantEvent.ToolCallPart, history: ReadonlyArray<Prompt.Message>) => Effect.Effect<boolean, never, R>)\n    | undefined\n}\n\n/** @internal */\nexport type ToolServices<T extends Record<string, Tool.Any>> =\n  | Exclude<Tool.HandlerServices<T[keyof T]>, AssistantToolCall>\n  // Already in the handlers' services; stated for the loop, which decodes params to decide on a call.\n  | T[keyof T][\"parametersSchema\"][\"DecodingServices\"]\n\nconst encodeFailure = Schema.encodeSync(ToolFailure.ToolFailure)\n\n/** @internal */\nexport const failed = (message: string, reason: string): CallResult => ({\n  outcome: \"failure\",\n  result: toJson(encodeFailure(new ToolFailure.ToolFailure({ message, reason }))),\n  reason\n})\n\nconst decodeReason = Schema.decodeUnknownOption(Schema.Struct({ reason: Schema.String }))\n\nconst decodeTag = Schema.decodeUnknownOption(Schema.Struct({ _tag: Schema.String }))\n\nconst failure = (result: Schema.Json): CallResult => ({\n  outcome: \"failure\",\n  result,\n  reason: Option.getOrUndefined(\n    Option.orElse(\n      Option.map(decodeReason(result), ({ reason }) => reason),\n      () => Option.map(decodeTag(result), ({ _tag }) => _tag)\n    )\n  )\n})\n\nconst hasTool = <T extends Record<string, Tool.Any>>(\n  toolkit: { readonly tools: T },\n  name: PropertyKey\n): name is keyof T => Object.hasOwn(toolkit.tools, name)\n\n// With noUncheckedIndexedAccess, an index doesn't narrow on `hasTool`.\nconst toolNamed = (toolkit: { readonly tools: Record<string, Tool.Any> }, name: string): Tool.Any | undefined =>\n  Object.hasOwn(toolkit.tools, name) ? toolkit.tools[name] : undefined\n\nconst decodeParams = (tool: Tool.Any, params: Schema.Json) => Schema.decodeUnknownEffect(tool.parametersSchema)(params)\n\n// Params that don't fit the tool don't park: the call runs, and fails with why.\n/** @internal */\nexport const needsDecision = Effect.fnUntraced(function* <T extends Record<string, Tool.Any>>(\n  toolkit: { readonly tools: T },\n  call: AssistantEvent.ToolCallPart,\n  history: ReadonlyArray<Prompt.Message>\n): Effect.fn.Return<boolean, never, ToolServices<T>> {\n  const tool = toolNamed(toolkit, call.name)\n\n  if (tool === undefined) return false\n\n  const needs = tool.needsApproval\n\n  if (needs === undefined || Predicate.isBoolean(needs)) return needs === true\n\n  return yield* Effect.gen(function* () {\n    const decided = needs(yield* decodeParams(tool, call.params), {\n      toolCallId: call.id,\n      messages: history\n    })\n\n    return Effect.isEffect(decided) ? yield* decided : decided\n  }).pipe(Effect.orElseSucceed(() => false))\n})\n\n// The toolkit decodes the params with the tool's schema before its handler sees them, and a call whose params don't\n// fit fails with a validation error the model reads: the model's JSON is what `handle` takes.\nconst isParams = <T>(params: unknown): params is Tool.ParametersEncoded<T> => params !== undefined\n\n// Every failure is the call's result, for the model to read. With `failureMode: \"return\"` the toolkit encodes the\n// tool's own failure with its schema first.\n/** @internal */\nexport const runNative = Effect.fnUntraced(function* <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.WithHandler<T>,\n  call: AssistantEvent.ToolCallPart\n): Effect.fn.Return<CallResult, never, ToolServices<T>> {\n  const name: PropertyKey = call.name\n\n  if (!hasTool(toolkit, name)) return failed(`No tool named ${call.name}.`, \"ToolNotFoundError\")\n\n  const params: unknown = call.params\n\n  if (!isParams<T[keyof T]>(params)) {\n    return failed(`Invalid parameters for ${call.name}.`, \"ToolParameterValidationError\")\n  }\n\n  return yield* Effect.gen(function* () {\n    const results = yield* toolkit.handle<keyof T>(name, params, call.id)\n    const last = yield* Stream.runLast(Stream.filter(results, (result) => !result.preliminary))\n\n    return Option.match(last, {\n      onNone: () => failed(`${call.name} returned no result.`, \"NoResult\"),\n      onSome: (result): CallResult =>\n        result.isFailure\n          ? failure(toJson(result.encodedResult))\n          : { outcome: \"success\", result: toJson(result.encodedResult) }\n    })\n  }).pipe(\n    Effect.catch((error) =>\n      Effect.succeed(\n        AiError.isAiError(error) ? failed(error.reason.message, error.reason._tag) : failure(toJson(error))\n      )\n    ),\n    Effect.provideService(AssistantToolCall, { callId: call.id, tool: call.name })\n  )\n})\n\n/** @internal */\nexport const fromWithHandler = <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.WithHandler<T>\n): Tools<ToolServices<T>> => ({\n  definitions: Object.values(toolkit.tools),\n  run: (call) => runNative(toolkit, call),\n  needsDecision: (call, history) => needsDecision(toolkit, call, history)\n})\n\n/** @internal */\nexport const fromToolkit = <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.Toolkit<T>\n): Tools<ToolServices<T> | Tool.HandlersFor<T>> => ({\n  definitions: Object.values(toolkit.tools),\n  run: Effect.fnUntraced(function* (call) {\n    return yield* runNative(yield* toolkit, call)\n  }),\n  needsDecision: (call, history) => needsDecision(toolkit, call, history)\n})\n\n// The handlers use services of the loop they run in, so they're made for each call, with its context.\n/** @internal */\nexport const withContext = <T extends Record<string, Tool.Any>, R>(\n  toolkit: Toolkit.Toolkit<T>,\n  handlers: (context: Context.Context<R>) => Toolkit.HandlersFrom<T>\n): Tools<R | ToolServices<T>> => ({\n  definitions: Object.values(toolkit.tools),\n  run: Effect.fnUntraced(function* (call) {\n    const context = yield* Effect.context<R>()\n    const resolved = yield* Effect.provideContext(toolkit, yield* toolkit.toHandlers(handlers(context)))\n\n    return yield* runNative(resolved, call)\n  }),\n  needsDecision: (call, history) => needsDecision(toolkit, call, history)\n})\n\n/** @internal */\nexport const fromToolkitOrEmpty = <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.Toolkit<T> | undefined\n): Tools<ToolServices<T> | Tool.HandlersFor<T>> => (toolkit === undefined ? empty : fromToolkit(toolkit))\n\n/** @internal */\nexport const empty: Tools = {\n  definitions: [],\n  run: (call) => Effect.succeed(failed(`No tool named ${call.name}.`, \"ToolNotFoundError\"))\n}\n\n/** @internal */\nexport const merge = <A, B>(self: Tools<A>, that: Tools<B>): Tools<A | B> => {\n  const inSelf = (call: AssistantEvent.ToolCallPart) => self.definitions.some((tool) => tool.name === call.name)\n\n  return {\n    definitions: [...self.definitions, ...that.definitions],\n    run: (call) => (inSelf(call) ? self.run(call) : that.run(call)),\n    needsDecision: (call, history) =>\n      inSelf(call) ? needsDecisionOf(self, call, history) : needsDecisionOf(that, call, history)\n  }\n}\n\n/** @internal */\nexport const needsDecisionOf = <R>(\n  tools: Tools<R>,\n  call: AssistantEvent.ToolCallPart,\n  history: ReadonlyArray<Prompt.Message>\n): Effect.Effect<boolean, never, R> =>\n  tools.needsDecision === undefined ? Effect.succeed(false) : tools.needsDecision(call, history)\n\n// A turn's own services, for a loop the app writes.\n/** @internal */\nexport const provide = <R>(tools: Tools<R>, context: Context.Context<R>): Tools => ({\n  definitions: tools.definitions,\n  run: (call) => Effect.provideContext(tools.run(call), context),\n  needsDecision: (call, history) => Effect.provideContext(needsDecisionOf(tools, call, history), context)\n})\n\n/** @internal */\nexport const duplicateNames = (tools: Tools<unknown>): ReadonlyArray<string> => {\n  const names = tools.definitions.map((tool) => tool.name)\n\n  return names.filter((name, i) => names.indexOf(name) !== i)\n}\n\n// The same tools are offered, so the prompt doesn't change.\n/** @internal */\nexport const runOnly = <R>(tools: Tools<R>, names: ReadonlyArray<string>, message: string): Tools<R> => ({\n  definitions: tools.definitions,\n  run: (call) => (names.includes(call.name) ? tools.run(call) : Effect.succeed(failed(message, \"NotNow\"))),\n  needsDecision: (call, history) =>\n    names.includes(call.name) ? needsDecisionOf(tools, call, history) : Effect.succeed(false)\n})\n"],"mappings":";;;;;;;;;;AAuCA,MAAM,gBAAgB,OAAO,WAAWA,WAAuB;;AAG/D,MAAa,UAAU,SAAiB,YAAgC;CACtE,SAAS;CACT,QAAQ,OAAO,cAAc,IAAIA,YAAwB;EAAE;EAAS;CAAO,CAAC,CAAC,CAAC;CAC9E;AACF;AAEA,MAAM,eAAe,OAAO,oBAAoB,OAAO,OAAO,EAAE,QAAQ,OAAO,OAAO,CAAC,CAAC;AAExF,MAAM,YAAY,OAAO,oBAAoB,OAAO,OAAO,EAAE,MAAM,OAAO,OAAO,CAAC,CAAC;AAEnF,MAAM,WAAW,YAAqC;CACpD,SAAS;CACT;CACA,QAAQ,OAAO,eACb,OAAO,OACL,OAAO,IAAI,aAAa,MAAM,IAAI,EAAE,aAAa,MAAM,SACjD,OAAO,IAAI,UAAU,MAAM,IAAI,EAAE,WAAW,IAAI,CACxD,CACF;AACF;AAEA,MAAM,WACJ,SACA,SACoB,OAAO,OAAO,QAAQ,OAAO,IAAI;AAGvD,MAAM,aAAa,SAAuD,SACxE,OAAO,OAAO,QAAQ,OAAO,IAAI,IAAI,QAAQ,MAAM,QAAQ,KAAA;AAE7D,MAAM,gBAAgB,MAAgB,WAAwB,OAAO,oBAAoB,KAAK,gBAAgB,CAAC,CAAC,MAAM;;AAItH,MAAa,gBAAgB,OAAO,WAAW,WAC7C,SACA,MACA,SACmD;CACnD,MAAM,OAAO,UAAU,SAAS,KAAK,IAAI;CAEzC,IAAI,SAAS,KAAA,GAAW,OAAO;CAE/B,MAAM,QAAQ,KAAK;CAEnB,IAAI,UAAU,KAAA,KAAa,UAAU,UAAU,KAAK,GAAG,OAAO,UAAU;CAExE,OAAO,OAAO,OAAO,IAAI,aAAa;EACpC,MAAM,UAAU,MAAM,OAAO,aAAa,MAAM,KAAK,MAAM,GAAG;GAC5D,YAAY,KAAK;GACjB,UAAU;EACZ,CAAC;EAED,OAAO,OAAO,SAAS,OAAO,IAAI,OAAO,UAAU;CACrD,CAAC,CAAC,CAAC,KAAK,OAAO,oBAAoB,KAAK,CAAC;AAC3C,CAAC;AAID,MAAM,YAAe,WAAyD,WAAW,KAAA;;AAKzF,MAAa,YAAY,OAAO,WAAW,WACzC,SACA,MACsD;CACtD,MAAM,OAAoB,KAAK;CAE/B,IAAI,CAAC,QAAQ,SAAS,IAAI,GAAG,OAAO,OAAO,iBAAiB,KAAK,KAAK,IAAI,mBAAmB;CAE7F,MAAM,SAAkB,KAAK;CAE7B,IAAI,CAAC,SAAqB,MAAM,GAC9B,OAAO,OAAO,0BAA0B,KAAK,KAAK,IAAI,8BAA8B;CAGtF,OAAO,OAAO,OAAO,IAAI,aAAa;EACpC,MAAM,UAAU,OAAO,QAAQ,OAAgB,MAAM,QAAQ,KAAK,EAAE;EACpE,MAAM,OAAO,OAAO,OAAO,QAAQ,OAAO,OAAO,UAAU,WAAW,CAAC,OAAO,WAAW,CAAC;EAE1F,OAAO,OAAO,MAAM,MAAM;GACxB,cAAc,OAAO,GAAG,KAAK,KAAK,uBAAuB,UAAU;GACnE,SAAS,WACP,OAAO,YACH,QAAQ,OAAO,OAAO,aAAa,CAAC,IACpC;IAAE,SAAS;IAAW,QAAQ,OAAO,OAAO,aAAa;GAAE;EACnE,CAAC;CACH,CAAC,CAAC,CAAC,KACD,OAAO,OAAO,UACZ,OAAO,QACL,QAAQ,UAAU,KAAK,IAAI,OAAO,MAAM,OAAO,SAAS,MAAM,OAAO,IAAI,IAAI,QAAQ,OAAO,KAAK,CAAC,CACpG,CACF,GACA,OAAO,eAAe,mBAAmB;EAAE,QAAQ,KAAK;EAAI,MAAM,KAAK;CAAK,CAAC,CAC/E;AACF,CAAC;;AAGD,MAAa,mBACX,aAC4B;CAC5B,aAAa,OAAO,OAAO,QAAQ,KAAK;CACxC,MAAM,SAAS,UAAU,SAAS,IAAI;CACtC,gBAAgB,MAAM,YAAY,cAAc,SAAS,MAAM,OAAO;AACxE;;AAGA,MAAa,eACX,aACkD;CAClD,aAAa,OAAO,OAAO,QAAQ,KAAK;CACxC,KAAK,OAAO,WAAW,WAAW,MAAM;EACtC,OAAO,OAAO,UAAU,OAAO,SAAS,IAAI;CAC9C,CAAC;CACD,gBAAgB,MAAM,YAAY,cAAc,SAAS,MAAM,OAAO;AACxE;;AAIA,MAAa,eACX,SACA,cACgC;CAChC,aAAa,OAAO,OAAO,QAAQ,KAAK;CACxC,KAAK,OAAO,WAAW,WAAW,MAAM;EACtC,MAAM,UAAU,OAAO,OAAO,QAAW;EACzC,MAAM,WAAW,OAAO,OAAO,eAAe,SAAS,OAAO,QAAQ,WAAW,SAAS,OAAO,CAAC,CAAC;EAEnG,OAAO,OAAO,UAAU,UAAU,IAAI;CACxC,CAAC;CACD,gBAAgB,MAAM,YAAY,cAAc,SAAS,MAAM,OAAO;AACxE;;AAGA,MAAa,sBACX,YACkD,YAAY,KAAA,IAAY,QAAQ,YAAY,OAAO;;AAGvG,MAAa,QAAe;CAC1B,aAAa,CAAC;CACd,MAAM,SAAS,OAAO,QAAQ,OAAO,iBAAiB,KAAK,KAAK,IAAI,mBAAmB,CAAC;AAC1F;;AAGA,MAAa,SAAe,MAAgB,SAAiC;CAC3E,MAAM,UAAU,SAAsC,KAAK,YAAY,MAAM,SAAS,KAAK,SAAS,KAAK,IAAI;CAE7G,OAAO;EACL,aAAa,CAAC,GAAG,KAAK,aAAa,GAAG,KAAK,WAAW;EACtD,MAAM,SAAU,OAAO,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,IAAI,IAAI;EAC7D,gBAAgB,MAAM,YACpB,OAAO,IAAI,IAAI,gBAAgB,MAAM,MAAM,OAAO,IAAI,gBAAgB,MAAM,MAAM,OAAO;CAC7F;AACF;;AAGA,MAAa,mBACX,OACA,MACA,YAEA,MAAM,kBAAkB,KAAA,IAAY,OAAO,QAAQ,KAAK,IAAI,MAAM,cAAc,MAAM,OAAO;;AAI/F,MAAa,WAAc,OAAiB,aAAwC;CAClF,aAAa,MAAM;CACnB,MAAM,SAAS,OAAO,eAAe,MAAM,IAAI,IAAI,GAAG,OAAO;CAC7D,gBAAgB,MAAM,YAAY,OAAO,eAAe,gBAAgB,OAAO,MAAM,OAAO,GAAG,OAAO;AACxG;;AAGA,MAAa,kBAAkB,UAAiD;CAC9E,MAAM,QAAQ,MAAM,YAAY,KAAK,SAAS,KAAK,IAAI;CAEvD,OAAO,MAAM,QAAQ,MAAM,MAAM,MAAM,QAAQ,IAAI,MAAM,CAAC;AAC5D;;AAIA,MAAa,WAAc,OAAiB,OAA8B,aAA+B;CACvG,aAAa,MAAM;CACnB,MAAM,SAAU,MAAM,SAAS,KAAK,IAAI,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,QAAQ,OAAO,SAAS,QAAQ,CAAC;CACtG,gBAAgB,MAAM,YACpB,MAAM,SAAS,KAAK,IAAI,IAAI,gBAAgB,OAAO,MAAM,OAAO,IAAI,OAAO,QAAQ,KAAK;AAC5F"}
+{"version":3,"file":"toolkit.js","names":["ToolFailure.ToolFailure"],"sources":["../../src/internal/toolkit.ts"],"sourcesContent":["import * as AiError from \"effect/ai/AiError\"\nimport type * as Tool from \"effect/ai/Tool\"\nimport type * as Toolkit from \"effect/ai/Toolkit\"\nimport type * as Context from \"effect/Context\"\nimport * as Effect from \"effect/Effect\"\nimport * as Option from \"effect/Option\"\nimport * as Schema from \"effect/Schema\"\nimport * as Stream from \"effect/Stream\"\nimport type * as AssistantEvent from \"../AssistantEvent.ts\"\nimport { AssistantToolCall } from \"../AssistantToolCall.ts\"\nimport * as ToolFailure from \"../ToolFailure.ts\"\nimport { toJson } from \"./schema.ts\"\n\n/** @internal */\nexport interface CallResult {\n  readonly outcome: AssistantEvent.ToolOutcome\n  readonly result: Schema.Json\n  readonly reason?: string | undefined\n}\n\n// Apps give the loop native toolkits; these are what the loop makes of them, merged with its own.\n/** @internal */\nexport interface Tools<out R = never> {\n  readonly definitions: ReadonlyArray<Tool.Any>\n  readonly run: (call: AssistantEvent.ToolCallPart) => Effect.Effect<CallResult, never, R>\n}\n\n/** @internal */\nexport type ToolServices<T extends Record<string, Tool.Any>> = Exclude<\n  Tool.HandlerServices<T[keyof T]>,\n  AssistantToolCall\n>\n\nconst encodeFailure = Schema.encodeSync(ToolFailure.ToolFailure)\n\n/** @internal */\nexport const failed = (message: string, reason: string): CallResult => ({\n  outcome: \"failure\",\n  result: toJson(encodeFailure(new ToolFailure.ToolFailure({ message, reason }))),\n  reason\n})\n\nconst decodeReason = Schema.decodeUnknownOption(Schema.Struct({ reason: Schema.String }))\n\nconst decodeTag = Schema.decodeUnknownOption(Schema.Struct({ _tag: Schema.String }))\n\nconst failure = (result: Schema.Json): CallResult => ({\n  outcome: \"failure\",\n  result,\n  reason: Option.getOrUndefined(\n    Option.orElse(\n      Option.map(decodeReason(result), ({ reason }) => reason),\n      () => Option.map(decodeTag(result), ({ _tag }) => _tag)\n    )\n  )\n})\n\nconst hasTool = <T extends Record<string, Tool.Any>>(\n  toolkit: { readonly tools: T },\n  name: PropertyKey\n): name is keyof T => Object.hasOwn(toolkit.tools, name)\n\n// Approval would park a call until the user decides, which this library doesn't do: the model calls a tool with\n// final values, and a user's own commitment goes through the app's API from a block.\n/** @internal */\nexport const needingApproval = (definitions: ReadonlyArray<Tool.Any>): ReadonlyArray<string> =>\n  definitions.flatMap((tool) => (tool.needsApproval === undefined || tool.needsApproval === false ? [] : [tool.name]))\n\n/** @internal */\nexport const needsApprovalMessage = (names: ReadonlyArray<string>): string =>\n  `${names.join(\", \")} ${names.length === 1 ? \"sets\" : \"set\"} needsApproval, and effect-assistant doesn't run tools that wait for approval. Remove needsApproval: let the model call the tool with final values, or show a block whose button calls your app's API, which does the work and submits an app event that wakes the chat.`\n\n// The toolkit decodes the params with the tool's schema before its handler sees them, and a call whose params don't\n// fit fails with a validation error the model reads: the model's JSON is what `handle` takes.\nconst isParams = <T>(params: unknown): params is Tool.ParametersEncoded<T> => params !== undefined\n\n// Every failure is the call's result, for the model to read. With `failureMode: \"return\"` the toolkit encodes the\n// tool's own failure with its schema first.\n/** @internal */\nexport const runNative = Effect.fnUntraced(function* <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.WithHandler<T>,\n  call: AssistantEvent.ToolCallPart\n): Effect.fn.Return<CallResult, never, ToolServices<T>> {\n  const name: PropertyKey = call.name\n\n  if (!hasTool(toolkit, name)) return failed(`No tool named ${call.name}.`, \"ToolNotFoundError\")\n\n  const params: unknown = call.params\n\n  if (!isParams<T[keyof T]>(params)) {\n    return failed(`Invalid parameters for ${call.name}.`, \"ToolParameterValidationError\")\n  }\n\n  return yield* Effect.gen(function* () {\n    const results = yield* toolkit.handle<keyof T>(name, params, call.id)\n    const last = yield* Stream.runLast(Stream.filter(results, (result) => !result.preliminary))\n\n    return Option.match(last, {\n      onNone: () => failed(`${call.name} returned no result.`, \"NoResult\"),\n      onSome: (result): CallResult =>\n        result.isFailure\n          ? failure(toJson(result.encodedResult))\n          : { outcome: \"success\", result: toJson(result.encodedResult) }\n    })\n  }).pipe(\n    Effect.catch((error) =>\n      Effect.succeed(\n        AiError.isAiError(error) ? failed(error.reason.message, error.reason._tag) : failure(toJson(error))\n      )\n    ),\n    Effect.provideService(AssistantToolCall, { callId: call.id, tool: call.name })\n  )\n})\n\n/** @internal */\nexport const fromWithHandler = <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.WithHandler<T>\n): Tools<ToolServices<T>> => ({\n  definitions: Object.values(toolkit.tools),\n  run: (call) => runNative(toolkit, call)\n})\n\n/** @internal */\nexport const fromToolkit = <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.Toolkit<T>\n): Tools<ToolServices<T> | Tool.HandlersFor<T>> => ({\n  definitions: Object.values(toolkit.tools),\n  run: Effect.fnUntraced(function* (call) {\n    return yield* runNative(yield* toolkit, call)\n  })\n})\n\n// The handlers use services of the loop they run in, so they're made for each call, with its context.\n/** @internal */\nexport const withContext = <T extends Record<string, Tool.Any>, R>(\n  toolkit: Toolkit.Toolkit<T>,\n  handlers: (context: Context.Context<R>) => Toolkit.HandlersFrom<T>\n): Tools<R | ToolServices<T>> => ({\n  definitions: Object.values(toolkit.tools),\n  run: Effect.fnUntraced(function* (call) {\n    const context = yield* Effect.context<R>()\n    const resolved = yield* Effect.provideContext(toolkit, yield* toolkit.toHandlers(handlers(context)))\n\n    return yield* runNative(resolved, call)\n  })\n})\n\n/** @internal */\nexport const fromToolkitOrEmpty = <T extends Record<string, Tool.Any>>(\n  toolkit: Toolkit.Toolkit<T> | undefined\n): Tools<ToolServices<T> | Tool.HandlersFor<T>> => (toolkit === undefined ? empty : fromToolkit(toolkit))\n\n/** @internal */\nexport const empty: Tools = {\n  definitions: [],\n  run: (call) => Effect.succeed(failed(`No tool named ${call.name}.`, \"ToolNotFoundError\"))\n}\n\n/** @internal */\nexport const merge = <A, B>(self: Tools<A>, that: Tools<B>): Tools<A | B> => {\n  const inSelf = (call: AssistantEvent.ToolCallPart) => self.definitions.some((tool) => tool.name === call.name)\n\n  return {\n    definitions: [...self.definitions, ...that.definitions],\n    run: (call) => (inSelf(call) ? self.run(call) : that.run(call))\n  }\n}\n\n// A turn's own services, for a loop the app writes.\n/** @internal */\nexport const provide = <R>(tools: Tools<R>, context: Context.Context<R>): Tools => ({\n  definitions: tools.definitions,\n  run: (call) => Effect.provideContext(tools.run(call), context)\n})\n\n/** @internal */\nexport const duplicateNames = (tools: Tools<unknown>): ReadonlyArray<string> => {\n  const names = tools.definitions.map((tool) => tool.name)\n\n  return names.filter((name, i) => names.indexOf(name) !== i)\n}\n\n// The same tools are offered, so the prompt doesn't change.\n/** @internal */\nexport const runOnly = <R>(tools: Tools<R>, names: ReadonlyArray<string>, message: string): Tools<R> => ({\n  definitions: tools.definitions,\n  run: (call) => (names.includes(call.name) ? tools.run(call) : Effect.succeed(failed(message, \"NotNow\")))\n})\n"],"mappings":";;;;;;;;;AAiCA,MAAM,gBAAgB,OAAO,WAAWA,WAAuB;;AAG/D,MAAa,UAAU,SAAiB,YAAgC;CACtE,SAAS;CACT,QAAQ,OAAO,cAAc,IAAIA,YAAwB;EAAE;EAAS;CAAO,CAAC,CAAC,CAAC;CAC9E;AACF;AAEA,MAAM,eAAe,OAAO,oBAAoB,OAAO,OAAO,EAAE,QAAQ,OAAO,OAAO,CAAC,CAAC;AAExF,MAAM,YAAY,OAAO,oBAAoB,OAAO,OAAO,EAAE,MAAM,OAAO,OAAO,CAAC,CAAC;AAEnF,MAAM,WAAW,YAAqC;CACpD,SAAS;CACT;CACA,QAAQ,OAAO,eACb,OAAO,OACL,OAAO,IAAI,aAAa,MAAM,IAAI,EAAE,aAAa,MAAM,SACjD,OAAO,IAAI,UAAU,MAAM,IAAI,EAAE,WAAW,IAAI,CACxD,CACF;AACF;AAEA,MAAM,WACJ,SACA,SACoB,OAAO,OAAO,QAAQ,OAAO,IAAI;;AAKvD,MAAa,mBAAmB,gBAC9B,YAAY,SAAS,SAAU,KAAK,kBAAkB,KAAA,KAAa,KAAK,kBAAkB,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,CAAE;;AAGrH,MAAa,wBAAwB,UACnC,GAAG,MAAM,KAAK,IAAI,EAAE,GAAG,MAAM,WAAW,IAAI,SAAS,MAAM;AAI7D,MAAM,YAAe,WAAyD,WAAW,KAAA;;AAKzF,MAAa,YAAY,OAAO,WAAW,WACzC,SACA,MACsD;CACtD,MAAM,OAAoB,KAAK;CAE/B,IAAI,CAAC,QAAQ,SAAS,IAAI,GAAG,OAAO,OAAO,iBAAiB,KAAK,KAAK,IAAI,mBAAmB;CAE7F,MAAM,SAAkB,KAAK;CAE7B,IAAI,CAAC,SAAqB,MAAM,GAC9B,OAAO,OAAO,0BAA0B,KAAK,KAAK,IAAI,8BAA8B;CAGtF,OAAO,OAAO,OAAO,IAAI,aAAa;EACpC,MAAM,UAAU,OAAO,QAAQ,OAAgB,MAAM,QAAQ,KAAK,EAAE;EACpE,MAAM,OAAO,OAAO,OAAO,QAAQ,OAAO,OAAO,UAAU,WAAW,CAAC,OAAO,WAAW,CAAC;EAE1F,OAAO,OAAO,MAAM,MAAM;GACxB,cAAc,OAAO,GAAG,KAAK,KAAK,uBAAuB,UAAU;GACnE,SAAS,WACP,OAAO,YACH,QAAQ,OAAO,OAAO,aAAa,CAAC,IACpC;IAAE,SAAS;IAAW,QAAQ,OAAO,OAAO,aAAa;GAAE;EACnE,CAAC;CACH,CAAC,CAAC,CAAC,KACD,OAAO,OAAO,UACZ,OAAO,QACL,QAAQ,UAAU,KAAK,IAAI,OAAO,MAAM,OAAO,SAAS,MAAM,OAAO,IAAI,IAAI,QAAQ,OAAO,KAAK,CAAC,CACpG,CACF,GACA,OAAO,eAAe,mBAAmB;EAAE,QAAQ,KAAK;EAAI,MAAM,KAAK;CAAK,CAAC,CAC/E;AACF,CAAC;;AAGD,MAAa,mBACX,aAC4B;CAC5B,aAAa,OAAO,OAAO,QAAQ,KAAK;CACxC,MAAM,SAAS,UAAU,SAAS,IAAI;AACxC;;AAGA,MAAa,eACX,aACkD;CAClD,aAAa,OAAO,OAAO,QAAQ,KAAK;CACxC,KAAK,OAAO,WAAW,WAAW,MAAM;EACtC,OAAO,OAAO,UAAU,OAAO,SAAS,IAAI;CAC9C,CAAC;AACH;;AAIA,MAAa,eACX,SACA,cACgC;CAChC,aAAa,OAAO,OAAO,QAAQ,KAAK;CACxC,KAAK,OAAO,WAAW,WAAW,MAAM;EACtC,MAAM,UAAU,OAAO,OAAO,QAAW;EACzC,MAAM,WAAW,OAAO,OAAO,eAAe,SAAS,OAAO,QAAQ,WAAW,SAAS,OAAO,CAAC,CAAC;EAEnG,OAAO,OAAO,UAAU,UAAU,IAAI;CACxC,CAAC;AACH;;AAGA,MAAa,sBACX,YACkD,YAAY,KAAA,IAAY,QAAQ,YAAY,OAAO;;AAGvG,MAAa,QAAe;CAC1B,aAAa,CAAC;CACd,MAAM,SAAS,OAAO,QAAQ,OAAO,iBAAiB,KAAK,KAAK,IAAI,mBAAmB,CAAC;AAC1F;;AAGA,MAAa,SAAe,MAAgB,SAAiC;CAC3E,MAAM,UAAU,SAAsC,KAAK,YAAY,MAAM,SAAS,KAAK,SAAS,KAAK,IAAI;CAE7G,OAAO;EACL,aAAa,CAAC,GAAG,KAAK,aAAa,GAAG,KAAK,WAAW;EACtD,MAAM,SAAU,OAAO,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,IAAI,IAAI;CAC/D;AACF;;AAIA,MAAa,WAAc,OAAiB,aAAwC;CAClF,aAAa,MAAM;CACnB,MAAM,SAAS,OAAO,eAAe,MAAM,IAAI,IAAI,GAAG,OAAO;AAC/D;;AAGA,MAAa,kBAAkB,UAAiD;CAC9E,MAAM,QAAQ,MAAM,YAAY,KAAK,SAAS,KAAK,IAAI;CAEvD,OAAO,MAAM,QAAQ,MAAM,MAAM,MAAM,QAAQ,IAAI,MAAM,CAAC;AAC5D;;AAIA,MAAa,WAAc,OAAiB,OAA8B,aAA+B;CACvG,aAAa,MAAM;CACnB,MAAM,SAAU,MAAM,SAAS,KAAK,IAAI,IAAI,MAAM,IAAI,IAAI,IAAI,OAAO,QAAQ,OAAO,SAAS,QAAQ,CAAC;AACxG"}