@iowarp/clio-coder 0.5.1 → 0.5.2

package/dist/{chunk-7KI4ZSEB.js → chunk-ELDJWJKM.js}

@@ -3,30 +3,30 @@ import {
   GATEWAY_SCHEMA_RUNTIME_ID,
   RESPONSE_SCHEMA_RUNTIME_ID,
   responseSchemaDialectFor
-} from "./chunk-VWKR2NN5.js";
+} from "./chunk-PF36KOGI.js";
 import {
   hashToolCall
 } from "./chunk-7QXCO3AF.js";
 import {
   isDispositionedToolResultError
-} from "./chunk-OPD7GL6F.js";
+} from "./chunk-H6FQVNZC.js";
 import {
   engineStreamSimple,
   resolvedRequestContext
 } from "./chunk-L7G7K4I2.js";
 import {
   classify
-} from "./chunk-VYRQHORQ.js";
+} from "./chunk-NMQSO6Z6.js";
 import {
   protectArtifact,
   protectedArtifactMutationBlockReason
-} from "./chunk-P4GNZPXO.js";
+} from "./chunk-OYUS3UZP.js";
 import {
   resolveReadPath
 } from "./chunk-C2CPOIH2.js";
 import {
   ToolNames
-} from "./chunk-V4XJWDHR.js";
+} from "./chunk-6UKFRTXI.js";
 import {
   GUARDRAIL_DEFAULTS,
   resolveGuardrail,
@@ -828,8 +828,50 @@ function resolveDeliveryTools(allowedTools, product) {
 
 // src/engine/agent.ts
 init_esm_shims();
-import { createInitialSystemMessage, toToolDeclaration } from "@earendil-works/pi-ai";
-import { Agent } from "@earendil-works/pi-agent-core";
+import {
+  createAssistantMessageEventStream,
+  createInitialSystemMessage,
+  toToolDeclaration
+} from "@earendil-works/pi-ai";
+import {
+  Agent
+} from "@earendil-works/pi-agent-core";
+function batchAwareBeforeToolCall(batch, delegate) {
+  const decisions = /* @__PURE__ */ new WeakMap();
+  return async (context, signal) => {
+    let decision = decisions.get(context.assistantMessage);
+    if (!decision) {
+      decision = Promise.resolve().then(() => batch({ assistantMessage: context.assistantMessage, context: context.context }, signal)).catch(() => ({ reason: "Tool batch admission failed" }));
+      decisions.set(context.assistantMessage, decision);
+    }
+    const rejection = await decision;
+    if (rejection) return { block: true, reason: rejection.reason.slice(0, 1024) || "Tool batch rejected" };
+    return delegate?.(context, signal);
+  };
+}
+function refusedStream(request, reason, aborted = false) {
+  const stream = createAssistantMessageEventStream();
+  const message = {
+    role: "assistant",
+    content: [],
+    api: request.model.api,
+    provider: request.model.provider,
+    model: request.model.id,
+    usage: {
+      input: 0,
+      output: 0,
+      cacheRead: 0,
+      cacheWrite: 0,
+      totalTokens: 0,
+      cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
+    },
+    stopReason: aborted ? "aborted" : "error",
+    errorMessage: reason.slice(0, 1024) || "Request admission refused",
+    timestamp: Date.now()
+  };
+  stream.push({ type: "error", reason: aborted ? "aborted" : "error", error: message });
+  return stream;
+}
 function dispositionAwareAfterToolCall(delegate) {
   return async (context, signal) => {
     const override = await delegate?.(context, signal);
@@ -839,15 +881,51 @@ function dispositionAwareAfterToolCall(delegate) {
   };
 }
 function createEngineAgent(options = {}) {
-  const { streamFn, transcriptStreamFn = engineStreamSimple, onStreamInvocation, ...agentOptions } = options;
+  const {
+    streamFn,
+    transcriptStreamFn = engineStreamSimple,
+    onStreamInvocation,
+    beforeToolBatch,
+    beforeStreamRequest,
+    ...agentOptions
+  } = options;
+  const correlations = /* @__PURE__ */ new WeakMap();
+  let activeCorrelationId;
+  const invoke = (model, context, streamOptions) => {
+    onStreamInvocation?.();
+    return streamFn ? streamFn(model, resolvedRequestContext(context), streamOptions) : transcriptStreamFn(model, context, streamOptions);
+  };
+  const admit = (model, context, streamOptions) => {
+    if (!beforeStreamRequest) return invoke(model, context, streamOptions);
+    activeCorrelationId = void 0;
+    const request = { model, context, options: streamOptions };
+    let admission;
+    try {
+      admission = beforeStreamRequest(request);
+    } catch {
+      return refusedStream(request, "Request admission failed", streamOptions?.signal?.aborted);
+    }
+    if (streamOptions?.signal?.aborted) return refusedStream(request, "Request aborted before invocation", true);
+    if (admission.block) return refusedStream(request, admission.reason);
+    activeCorrelationId = admission.correlationId;
+    return invoke(model, context, streamOptions);
+  };
   const agent = new Agent({
     ...agentOptions,
-    streamFn: (model, context, streamOptions) => {
-      onStreamInvocation?.();
-      return streamFn ? streamFn(model, resolvedRequestContext(context), streamOptions) : transcriptStreamFn(model, context, streamOptions);
-    },
+    streamFn: beforeStreamRequest ? admit : invoke,
+    ...beforeToolBatch ? { beforeToolCall: batchAwareBeforeToolCall(beforeToolBatch, options.beforeToolCall) } : {},
     afterToolCall: dispositionAwareAfterToolCall(options.afterToolCall)
   });
+  if (beforeStreamRequest) {
+    agent.subscribe((event) => {
+      if (event.type === "message_end" && event.message.role === "assistant" && activeCorrelationId !== void 0) {
+        correlations.set(event.message, activeCorrelationId);
+      }
+      if (event.type === "agent_start" || event.type === "turn_end" || event.type === "agent_end") {
+        activeCorrelationId = void 0;
+      }
+    });
+  }
   const convertToLlm = agent.convertToLlm.bind(agent);
   agent.convertToLlm = async (messages) => (await convertToLlm(messages)).map((message) => {
     if (message.role !== "assistant" || message.stopReason !== "aborted") return message;
@@ -861,7 +939,8 @@ ${partial}` : "No visible assistant text was retained.")
   });
   return {
     agent,
-    state: () => agent.state
+    state: () => agent.state,
+    requestCorrelationId: (message) => correlations.get(message)
   };
 }
 function replaceEngineMessages(agent, messages, systemPrompt = agent.state.systemPrompt) {
@@ -871,6 +950,9 @@ function replaceEngineMessages(agent, messages, systemPrompt = agent.state.syste
 function setEngineSystemPrompt(agent, systemPrompt) {
   replaceEngineMessages(agent, agent.state.messages, systemPrompt);
 }
+function continueEngineWithoutInput(agent) {
+  return agent.prompt([]);
+}
 
 // src/engine/provider-payload.ts
 init_esm_shims();
@@ -1094,6 +1176,7 @@ export {
   createEngineAgent,
   replaceEngineMessages,
   setEngineSystemPrompt,
+  continueEngineWithoutInput,
   supportsNamedToolChoice,
   patchProviderThinkingPayload,
   patchToolChoiceNonePayload,
@@ -1101,4 +1184,4 @@ export {
   patchResponseSchemaPayloadForDialect,
   patchWorkerRequestPayload
 };
-//# sourceMappingURL=chunk-7KI4ZSEB.js.map
+//# sourceMappingURL=chunk-ELDJWJKM.js.map

package/dist/{chunk-GCQFN7YI.js → chunk-EMYEQLWO.js}

@@ -4,7 +4,7 @@ import {
 } from "./chunk-L7G7K4I2.js";
 import {
   ToolNames
-} from "./chunk-V4XJWDHR.js";
+} from "./chunk-6UKFRTXI.js";
 import {
   typebox_exports
 } from "./chunk-GH5ZR6JP.js";
@@ -29,4 +29,4 @@ var steerToolSurface = {
 export {
   steerToolSurface
 };
-//# sourceMappingURL=chunk-GCQFN7YI.js.map
+//# sourceMappingURL=chunk-EMYEQLWO.js.map