@absolutejs/ai 0.3.4 → 0.4.1

package/dist/ai/providers/openrouter.js

@@ -567,7 +567,7 @@ async function* tapStream(source, params, providerName) {
   let lastUsage;
   try {
     for await (const chunk of source) {
-      if (chunk.type === "done" && chunk.usage) {
+      if ((chunk.type === "done" || chunk.type === "usage_update" && chunk.final) && chunk.usage) {
         lastUsage = chunk.usage;
       }
       yield chunk;
@@ -1180,6 +1180,8 @@ var processCompleted = function* (parsed) {
   const { response } = parsed;
   const usage = extractUsage(response);
   const metadata = extractResponseMetadata(response);
+  if (usage)
+    yield { type: "usage_update", usage, metadata, final: true };
   if (isRecordArray(response.output)) {
     yield* extractCitationsFromOutput(response.output);
     yield* extractImageFromOutput(response.output);
@@ -1218,6 +1220,15 @@ var processSSEEvent = function* (eventType, parsed, pendingCalls, providerName)
       break;
     case "response.output_item.added":
       processOutputItemAdded(parsed, pendingCalls);
+      if (isRecord(parsed.item) && parsed.item.type === "function_call" && typeof parsed.item.name === "string") {
+        const id = typeof parsed.item.call_id === "string" ? parsed.item.call_id : parsed.item.id;
+        if (typeof id === "string")
+          yield {
+            type: "tool_input_start",
+            id,
+            name: parsed.item.name
+          };
+      }
       break;
     case "response.output_item.done":
       yield* processOutputItemDone(parsed);
@@ -1234,8 +1245,19 @@ var processSSEEvent = function* (eventType, parsed, pendingCalls, providerName)
     case "response.failed":
     case "response.incomplete":
     case "response.error":
-    case "error":
+    case "error": {
+      if (isRecord(parsed.response)) {
+        const usage = extractUsage(parsed.response);
+        if (usage)
+          yield {
+            type: "usage_update",
+            usage,
+            metadata: extractResponseMetadata(parsed.response),
+            final: false
+          };
+      }
       throw responseFailure(eventType, parsed, providerName);
+    }
   }
 };
 var flushSSEBuffer = function* (state) {
@@ -1657,6 +1679,14 @@ var processDelta = function* (delta, pendingToolCalls) {
   }
   if (isRecordArray2(delta.tool_calls)) {
     processToolCallDeltas(delta.tool_calls, pendingToolCalls);
+    for (const call of delta.tool_calls) {
+      if (isRecord2(call.function) && typeof call.function.name === "string" && call.function.name)
+        yield {
+          type: "tool_input_start",
+          id: typeof call.id === "string" ? call.id : String(call.index),
+          name: call.function.name
+        };
+    }
   }
   if (Array.isArray(delta.annotations)) {
     for (const annotation of delta.annotations) {
@@ -1755,7 +1785,12 @@ var processSSELine2 = function* (line, pendingToolCalls, providerName) {
   }
   const usageUpdate = narrowUsageRecord(parsed);
   if (usageUpdate) {
-    yield { type: "usage_update", usage: usageUpdate };
+    yield {
+      type: "usage_update",
+      usage: usageUpdate,
+      final: Array.isArray(parsed.choices) && parsed.choices.length === 0,
+      metadata: narrowResponseMetadata(parsed)
+    };
   }
   const metadata = narrowResponseMetadata(parsed);
   if (metadata) {
@@ -1784,10 +1819,9 @@ var processSSELine2 = function* (line, pendingToolCalls, providerName) {
   }
   yield* processChoice(firstChoice, pendingToolCalls);
 };
-var isUsageUpdate = (chunk) => chunk.type === "usage_update";
 var collectYieldableChunks = (line, pendingToolCalls, usageRef, metadataRef, providerName) => {
   const allChunks = Array.from(processSSELine2(line, pendingToolCalls, providerName));
-  const usageChunks = allChunks.filter(isUsageUpdate);
+  const usageChunks = allChunks.filter((chunk) => chunk.type === "usage_update");
   const lastUsage = usageChunks.at(NOT_FOUND);
   if (lastUsage) {
     usageRef.current = lastUsage.usage;
@@ -1804,7 +1838,7 @@ var collectYieldableChunks = (line, pendingToolCalls, usageRef, metadataRef, pro
       }
     };
   }
-  return allChunks.filter((chunk) => !isUsageUpdate(chunk) && chunk.type !== "response_metadata");
+  return allChunks.filter((chunk) => chunk.type !== "response_metadata").map((chunk) => chunk.type === "usage_update" ? { ...chunk, metadata: metadataRef.current } : chunk);
 };
 var processSSELines2 = function* (lines, pendingToolCalls, usageRef, metadataRef, providerName) {
   for (const line of lines) {
@@ -1922,7 +1956,6 @@ var API_VERSION = "2023-06-01";
 var DEFAULT_MAX_TOKENS = 32000;
 var EVENT_PREFIX_LENGTH2 = 7;
 var DATA_PREFIX_LENGTH2 = 6;
-var EMPTY_CHUNKS = [];
 var isRecord3 = (val) => typeof val === "object" && val !== null;
 var mapContentBlock = (block) => {
   if (block.type === "thinking") {
@@ -2201,14 +2234,79 @@ var getNumber = (obj, key) => {
   }
   return 0;
 };
+var TOOL_INPUT_PROGRESS_CHARS = 2048;
+var TOOL_INPUT_FIELD_MAX_CHARS = 200;
+var readJsonString = (json, start) => {
+  for (let index = start + 1;index < json.length; index += 1) {
+    if (json[index] === "\\")
+      index += 1;
+    else if (json[index] === '"')
+      return { end: index + 1, raw: json.slice(start + 1, index) };
+  }
+  return;
+};
+var decodeJsonString = (raw) => {
+  try {
+    const value = JSON.parse(`"${raw}"`);
+    return typeof value === "string" ? value : undefined;
+  } catch {
+    return;
+  }
+};
+var completeStringFields = (partialJson) => {
+  const fields = {};
+  let depth = 0;
+  let key;
+  let index = 0;
+  while (index < partialJson.length) {
+    const character = partialJson[index];
+    if (character === '"') {
+      const text = readJsonString(partialJson, index);
+      if (!text)
+        break;
+      if (depth === 1 && key === undefined) {
+        const after = partialJson.slice(text.end).trimStart();
+        if (after.startsWith(":"))
+          key = text.raw;
+      } else if (depth === 1 && key !== undefined) {
+        const value = decodeJsonString(text.raw);
+        if (value !== undefined && value.length <= TOOL_INPUT_FIELD_MAX_CHARS)
+          fields[key] = value;
+        key = undefined;
+      }
+      index = text.end;
+      continue;
+    }
+    if (character === "{" || character === "[")
+      depth += 1;
+    else if (character === "}" || character === "]")
+      depth -= 1;
+    else if (character === "," && depth === 1)
+      key = undefined;
+    index += 1;
+  }
+  return fields;
+};
+var toolInputProgress = (state) => {
+  state.toolInputReported = state.toolInputJson.length;
+  return {
+    fields: completeStringFields(state.toolInputJson),
+    id: state.currentToolId,
+    name: state.currentToolName,
+    receivedChars: state.toolInputJson.length,
+    type: "tool_input"
+  };
+};
 var handleContentBlockStart = (parsed, state) => {
   const block = getRecord(parsed, "content_block");
   if (block && block.type === "tool_use") {
     state.currentToolId = getString(block, "id");
     state.currentToolName = getString(block, "name");
     state.toolInputJson = "";
+    state.toolInputReported = 0;
     state.isThinkingBlock = false;
     state.currentProviderBlock = undefined;
+    return toolInputProgress(state);
   } else if (block && block.type === "thinking") {
     state.isThinkingBlock = true;
     state.thinkingSignature = "";
@@ -2218,6 +2316,7 @@ var handleContentBlockStart = (parsed, state) => {
     state.currentProviderBlock = block && block.type !== "text" ? { ...block } : undefined;
     state.providerBlockInputJson = "";
   }
+  return;
 };
 var handleContentBlockDelta = (parsed, state) => {
   const delta = getRecord(parsed, "delta");
@@ -2252,6 +2351,8 @@ var handleContentBlockDelta = (parsed, state) => {
       state.providerBlockInputJson += getString(delta, "partial_json");
     } else {
       state.toolInputJson += getString(delta, "partial_json");
+      if (state.toolInputJson.length - state.toolInputReported >= TOOL_INPUT_PROGRESS_CHARS)
+        return toolInputProgress(state);
     }
   }
   if (delta.type === "signature_delta") {
@@ -2371,8 +2472,7 @@ var handleError = (parsed, state) => {
 var processEvent = (eventType, parsed, state) => {
   switch (eventType) {
     case "content_block_start": {
-      handleContentBlockStart(parsed, state);
-      return;
+      return handleContentBlockStart(parsed, state);
     }
     case "content_block_delta": {
       return handleContentBlockDelta(parsed, state);
@@ -2383,11 +2483,21 @@ var processEvent = (eventType, parsed, state) => {
     case "message_delta": {
       handleMessageDelta(parsed, state);
       mergeMetadata(parsed, state);
-      return;
+      return state.usage ? {
+        type: "usage_update",
+        usage: state.usage,
+        metadata: state.metadata,
+        final: false
+      } : undefined;
     }
     case "message_start": {
       handleMessageStart(parsed, state);
-      return;
+      return state.usage ? {
+        type: "usage_update",
+        usage: state.usage,
+        metadata: state.metadata,
+        final: false
+      } : undefined;
     }
     case "message_stop": {
       mergeMetadata(parsed, state);
@@ -2421,49 +2531,6 @@ var processSingleEvent = (event, state) => {
   }
   return processEvent(eventType, parsed, state);
 };
-var collectChunk = (event, state) => {
-  const chunk = processSingleEvent(event, state);
-  return chunk ? [chunk] : [];
-};
-var processBufferedEvents = (eventsText, state) => {
-  const events = eventsText.split(`
-
-`);
-  state.buffer = events.pop() ?? "";
-  return events.flatMap((event) => collectChunk(event, state));
-};
-var readNextChunks = async (reader, decoder, state, signal) => {
-  if (signal?.aborted) {
-    return { chunks: EMPTY_CHUNKS, done: true };
-  }
-  const { done, value } = await reader.read();
-  if (done) {
-    return { chunks: EMPTY_CHUNKS, done: true };
-  }
-  const rawText = state.buffer + decoder.decode(value, { stream: true });
-  const chunks = processBufferedEvents(rawText, state);
-  return { chunks, done: false };
-};
-var findDoneChunk = (chunks) => chunks.findIndex((c) => c.type === "done");
-var sseStreamLoop = async (reader, decoder, state, signal) => {
-  const result = await readNextChunks(reader, decoder, state, signal);
-  if (result.done) {
-    return { chunks: result.chunks, finished: true };
-  }
-  const doneIdx = findDoneChunk(result.chunks);
-  if (doneIdx >= 0) {
-    return { chunks: result.chunks.slice(0, doneIdx + 1), finished: true };
-  }
-  return { chunks: result.chunks, finished: false };
-};
-async function* streamChunks(reader, decoder, state, signal) {
-  let finished = false;
-  while (!finished) {
-    const result = await sseStreamLoop(reader, decoder, state, signal);
-    ({ finished } = result);
-    yield* result.chunks;
-  }
-}
 async function* parseSSEStream3(body, providerName, signal) {
   const reader = body.getReader();
   const decoder = new TextDecoder;
@@ -2475,13 +2542,40 @@ async function* parseSSEStream3(body, providerName, signal) {
     stopReason: "",
     thinkingSignature: "",
     toolInputJson: "",
+    toolInputReported: 0,
     usage: undefined,
     providerName,
     providerBlockInputJson: ""
   };
   try {
-    yield* streamChunks(reader, decoder, state, signal);
+    while (true) {
+      signal?.throwIfAborted();
+      const { done, value } = await reader.read();
+      if (done)
+        return;
+      const events = (state.buffer + decoder.decode(value, { stream: true })).split(`
+
+`);
+      state.buffer = events.pop() ?? "";
+      for (const event of events) {
+        const chunk = processSingleEvent(event, state);
+        if (!chunk)
+          continue;
+        if (chunk.type === "tool_input" && chunk.receivedChars === 0)
+          yield {
+            type: "tool_input_start",
+            id: chunk.id,
+            name: chunk.name
+          };
+        yield chunk;
+        if (chunk.type === "done")
+          return;
+      }
+    }
   } finally {
+    await reader.cancel().catch(() => {
+      return;
+    });
     reader.releaseLock();
   }
 }
@@ -3518,5 +3612,5 @@ export {
   verifyOpenRouterWebhookSignature
 };
 
-//# debugId=47F788AD8F60934664756E2164756E21
+//# debugId=197F4FFE0DB1613B64756E2164756E21
 //# sourceMappingURL=openrouter.js.map