@revojs/ai 0.3.1 → 0.3.3

package/dist/protocols/openai-chat-completions.js

@@ -150,6 +150,24 @@ function toFinishReason(raw) {
 		raw: raw ?? void 0
 	};
 }
+function toStreamError(chunk) {
+	if (typeof chunk.error === "string") return chunk.error;
+	if (chunk.error !== void 0) return chunk.error.message ?? "Unknown AI provider error";
+	if (chunk.object === "error") return chunk.message ?? "Unknown AI provider error";
+}
+function hasCompleteToolCalls(tools) {
+	if (tools.size === 0) return false;
+	for (const accumulator of tools.values()) {
+		if (accumulator.name.length === 0) return false;
+		try {
+			const input = JSON.parse(toToolInput(accumulator.arguments));
+			if (typeof input !== "object" || input === null || Array.isArray(input)) return false;
+		} catch {
+			return false;
+		}
+	}
+	return true;
+}
 async function* readParts(response, warnings) {
 	yield {
 		type: "stream-start",
@@ -160,6 +178,7 @@ async function* readParts(response, warnings) {
 	let textStarted = false;
 	let reasoningStarted = false;
 	let finishRaw;
+	let failure;
 	let usage;
 	for await (const data of readDataFrames(response)) {
 		if (data === "[DONE]") break;
@@ -169,6 +188,8 @@ async function* readParts(response, warnings) {
 		} catch {
 			continue;
 		}
+		failure = toStreamError(chunk);
+		if (failure !== void 0) break;
 		if (!metadataSent) {
 			metadataSent = true;
 			yield {
@@ -213,11 +234,14 @@ async function* readParts(response, warnings) {
 		}
 		for (const partial of delta.tool_calls ?? []) {
 			const index = partial.index ?? 0;
+			const id = typeof partial.id === "string" && partial.id.length > 0 ? partial.id : void 0;
+			const name = typeof partial.function?.name === "string" ? partial.function.name : "";
+			const input = typeof partial.function?.arguments === "string" ? partial.function.arguments : "";
 			let accumulator = tools.get(index);
 			if (accumulator === void 0) {
 				accumulator = {
-					streamId: partial.id ?? `openai-tool-${index}`,
-					callId: partial.id ?? "",
+					streamId: id ?? `openai-tool-${index}`,
+					callId: id ?? "",
 					name: "",
 					arguments: ""
 				};
@@ -225,22 +249,25 @@ async function* readParts(response, warnings) {
 				yield {
 					type: "tool-input-start",
 					id: accumulator.streamId,
-					toolName: partial.function?.name ?? ""
+					toolName: name
 				};
 			}
-			if (partial.id !== void 0 && partial.id !== accumulator.callId) accumulator.callId = partial.id;
-			if (partial.function?.name !== void 0) accumulator.name += partial.function.name;
-			if (partial.function?.arguments !== void 0) {
-				accumulator.arguments += partial.function.arguments;
+			if (id !== void 0) accumulator.callId = id;
+			accumulator.name += name;
+			if (input.length > 0) {
+				accumulator.arguments += input;
 				yield {
 					type: "tool-input-delta",
 					id: accumulator.streamId,
-					delta: partial.function.arguments
+					delta: input
 				};
 			}
 		}
 	}
-	if (finishRaw === void 0) throw new Error("Incomplete chat completions stream.");
+	if (finishRaw === void 0) {
+		if (!hasCompleteToolCalls(tools)) throw new Error(failure ?? "Incomplete chat completions stream.");
+		finishRaw = "tool_calls";
+	}
 	if (reasoningStarted) yield {
 		type: "reasoning-end",
 		id: "openai-reasoning-0"
@@ -289,9 +316,9 @@ function chatCompletionsGenerateProtocol(config) {
 		});
 		for (const call of choice.message.tool_calls ?? []) content.push({
 			type: "tool-call",
-			toolCallId: call.id ?? "",
-			toolName: call.function?.name ?? "",
-			input: toToolInput(call.function?.arguments)
+			toolCallId: typeof call.id === "string" ? call.id : "",
+			toolName: typeof call.function?.name === "string" ? call.function.name : "",
+			input: toToolInput(typeof call.function?.arguments === "string" ? call.function.arguments : void 0)
 		});
 		return {
 			content,

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@revojs/ai",
-    "version": "0.3.1",
+    "version": "0.3.3",
     "license": "MIT",
     "repository": {
         "type": "git",
@@ -30,10 +30,10 @@
     },
     "dependencies": {
         "@cfworker/json-schema": "^4.1.1",
-        "@revojs/server": "0.3.1",
-        "revojs": "0.3.1"
+        "@revojs/server": "0.3.3",
+        "revojs": "0.3.3"
     },
     "devDependencies": {
-        "@revojs/tsconfig": "0.3.1"
+        "@revojs/tsconfig": "0.3.3"
     }
 }