@core-ai/mistral 0.3.0 → 0.5.0

package/dist/index.js

@@ -31,11 +31,15 @@ function convertMessage(message) {
     };
   }
   if (message.role === "assistant") {
+    const text = message.parts.flatMap((part) => part.type === "text" ? [part.text] : []).join("");
+    const toolCalls = message.parts.flatMap(
+      (part) => part.type === "tool-call" ? [part.toolCall] : []
+    );
     return {
       role: "assistant",
-      content: message.content,
-      ...message.toolCalls && message.toolCalls.length > 0 ? {
-        toolCalls: message.toolCalls.map((toolCall) => ({
+      content: text.length > 0 ? text : null,
+      ...toolCalls.length > 0 ? {
+        toolCalls: toolCalls.map((toolCall) => ({
           id: toolCall.id,
           type: "function",
           function: {
@@ -117,6 +121,7 @@ function createStructuredOutputOptions(options) {
       type: "tool",
       toolName
     },
+    reasoning: options.reasoning,
     config: options.config,
     providerOptions: options.providerOptions,
     signal: options.signal
@@ -164,16 +169,25 @@ function mapGenerateResponse(response) {
   const firstChoice = response.choices[0];
   if (!firstChoice) {
     return {
+      parts: [],
       content: null,
+      reasoning: null,
       toolCalls: [],
       finishReason: "unknown",
       usage: mapUsage(response.usage)
     };
   }
-  const toolCalls = parseToolCalls(firstChoice.message.toolCalls);
+  const parts = extractAssistantParts(firstChoice.message);
+  const toolCalls = parts.flatMap(
+    (part) => part.type === "tool-call" ? [part.toolCall] : []
+  );
+  const content = parts.flatMap((part) => part.type === "text" ? [part.text] : []).join("");
+  const reasoning = parts.flatMap((part) => part.type === "reasoning" ? [part.text] : []).join("");
   const mappedFinishReason = mapFinishReason(firstChoice.finishReason);
   return {
-    content: extractTextContent(firstChoice.message.content),
+    parts,
+    content: content.length > 0 ? content : null,
+    reasoning: reasoning.length > 0 ? reasoning : null,
     toolCalls,
     finishReason: toolCalls.length > 0 && mappedFinishReason !== "content-filter" ? "tool-calls" : mappedFinishReason,
     usage: mapUsage(response.usage)
@@ -183,11 +197,15 @@ async function* transformStream(stream) {
   const bufferedToolCalls = /* @__PURE__ */ new Map();
   const emittedToolCalls = /* @__PURE__ */ new Set();
   let finishReason = "unknown";
+  let reasoningOpen = false;
   let usage = {
     inputTokens: 0,
     outputTokens: 0,
-    reasoningTokens: 0,
-    totalTokens: 0
+    inputTokenDetails: {
+      cacheReadTokens: 0,
+      cacheWriteTokens: 0
+    },
+    outputTokenDetails: {}
   };
   for await (const event of stream) {
     const chunk = event.data;
@@ -198,13 +216,40 @@ async function* transformStream(stream) {
     if (!choice) {
       continue;
     }
+    const thinkingDeltas = extractThinkingDeltas(choice.delta.content);
+    if (thinkingDeltas.length > 0) {
+      if (!reasoningOpen) {
+        reasoningOpen = true;
+        yield {
+          type: "reasoning-start"
+        };
+      }
+      for (const thinkingDelta of thinkingDeltas) {
+        yield {
+          type: "reasoning-delta",
+          text: thinkingDelta
+        };
+      }
+    }
     for (const textDelta of extractTextDeltas(choice.delta.content)) {
+      if (reasoningOpen) {
+        reasoningOpen = false;
+        yield {
+          type: "reasoning-end"
+        };
+      }
       yield {
-        type: "content-delta",
+        type: "text-delta",
         text: textDelta
       };
     }
     if (choice.delta.toolCalls) {
+      if (reasoningOpen) {
+        reasoningOpen = false;
+        yield {
+          type: "reasoning-end"
+        };
+      }
       for (const [
         position,
         partialToolCall
@@ -258,6 +303,11 @@ async function* transformStream(stream) {
       yield* emitBufferedToolCalls(bufferedToolCalls, emittedToolCalls);
     }
   }
+  if (reasoningOpen) {
+    yield {
+      type: "reasoning-end"
+    };
+  }
   yield* emitBufferedToolCalls(bufferedToolCalls, emittedToolCalls);
   yield {
     type: "finish",
@@ -307,19 +357,47 @@ function mapUsage(usage) {
   return {
     inputTokens: usage?.promptTokens ?? 0,
     outputTokens: usage?.completionTokens ?? 0,
-    reasoningTokens: 0,
-    totalTokens: usage?.totalTokens ?? 0
+    inputTokenDetails: {
+      cacheReadTokens: 0,
+      cacheWriteTokens: 0
+    },
+    outputTokenDetails: {}
   };
 }
-function extractTextContent(content) {
-  if (typeof content === "string") {
-    return content;
+function extractAssistantParts(message) {
+  const parts = [];
+  if (typeof message.content === "string") {
+    if (message.content.length > 0) {
+      parts.push({
+        type: "text",
+        text: message.content
+      });
+    }
+  } else if (Array.isArray(message.content)) {
+    for (const chunk of message.content) {
+      if (chunk.type === "text") {
+        parts.push({
+          type: "text",
+          text: chunk.text
+        });
+        continue;
+      }
+      if (chunk.type === "thinking") {
+        const thinkingText = extractThinkingText(chunk.thinking);
+        parts.push({
+          type: "reasoning",
+          text: thinkingText
+        });
+      }
+    }
   }
-  if (!content || content.length === 0) {
-    return null;
+  for (const toolCall of parseToolCalls(message.toolCalls)) {
+    parts.push({
+      type: "tool-call",
+      toolCall
+    });
   }
-  const text = content.flatMap((chunk) => chunk.type === "text" ? [chunk.text] : []).join("");
-  return text.length > 0 ? text : null;
+  return parts;
 }
 function extractTextDeltas(content) {
   if (typeof content === "string") {
@@ -332,6 +410,36 @@ function extractTextDeltas(content) {
     (chunk) => chunk.type === "text" ? [chunk.text] : []
   );
 }
+function extractThinkingDeltas(content) {
+  if (!content || typeof content === "string") {
+    return [];
+  }
+  return content.flatMap((chunk) => {
+    if (chunk.type !== "thinking") {
+      return [];
+    }
+    const thinkingText = extractThinkingText(chunk.thinking);
+    if (thinkingText.length === 0) {
+      return [];
+    }
+    return [thinkingText];
+  });
+}
+function extractThinkingText(value) {
+  if (typeof value === "string") {
+    return value;
+  }
+  if (!Array.isArray(value)) {
+    return "";
+  }
+  return value.flatMap((part) => {
+    if (!part || typeof part !== "object") {
+      return [];
+    }
+    const text = part.text;
+    return typeof text === "string" ? [text] : [];
+  }).join("");
+}
 function serializeJsonObject(value) {
   const objectValue = asObject(value);
   return Object.keys(objectValue).length > 0 ? JSON.stringify(objectValue) : "";
@@ -462,7 +570,7 @@ async function* transformStructuredOutputStream(stream, schema, provider, toolNa
   let contentBuffer = "";
   const toolArgumentDeltas = /* @__PURE__ */ new Map();
   for await (const event of stream) {
-    if (event.type === "content-delta") {
+    if (event.type === "text-delta") {
       contentBuffer += event.text;
       yield {
         type: "object-delta",

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@core-ai/mistral",
-    "version": "0.3.0",
+    "version": "0.5.0",
     "description": "Mistral provider package for @core-ai/core-ai",
     "license": "MIT",
     "author": "Omnifact (https://omnifact.ai)",
@@ -42,7 +42,7 @@
         "test:watch": "vitest"
     },
     "dependencies": {
-        "@core-ai/core-ai": "^0.3.0",
+        "@core-ai/core-ai": "^0.5.0",
         "@mistralai/mistralai": "^1.14.0",
         "zod-to-json-schema": "^3.25.1"
     },