@core-ai/core-ai 0.4.0 → 0.5.1

package/dist/index.d.ts

@@ -31,10 +31,27 @@ type FilePart = {
     mimeType: string;
     filename?: string;
 };
+type ReasoningEffort = 'minimal' | 'low' | 'medium' | 'high' | 'max';
+type ReasoningConfig = {
+    effort: ReasoningEffort;
+};
+type AssistantTextPart = {
+    type: 'text';
+    text: string;
+};
+type ReasoningPart = {
+    type: 'reasoning';
+    text: string;
+    providerMetadata?: Record<string, unknown>;
+};
+type ToolCallPart = {
+    type: 'tool-call';
+    toolCall: ToolCall;
+};
+type AssistantContentPart = AssistantTextPart | ReasoningPart | ToolCallPart;
 type AssistantMessage = {
     role: 'assistant';
-    content: string | null;
-    toolCalls?: ToolCall[];
+    parts: AssistantContentPart[];
 };
 type ToolCall = {
     id: string;
@@ -75,6 +92,7 @@ type ModelConfig = {
 };
 type GenerateOptions = {
     messages: Message[];
+    reasoning?: ReasoningConfig;
     tools?: ToolSet;
     toolChoice?: ToolChoice;
     config?: ModelConfig;
@@ -82,7 +100,9 @@ type GenerateOptions = {
     signal?: AbortSignal;
 };
 type GenerateResult = {
+    parts: AssistantContentPart[];
     content: string | null;
+    reasoning: string | null;
     toolCalls: ToolCall[];
     finishReason: FinishReason;
     usage: ChatUsage;
@@ -92,6 +112,7 @@ type GenerateObjectOptions<TSchema extends z.ZodType> = {
     schema: TSchema;
     schemaName?: string;
     schemaDescription?: string;
+    reasoning?: ReasoningConfig;
     config?: ModelConfig;
     providerOptions?: Record<string, unknown>;
     signal?: AbortSignal;
@@ -138,12 +159,19 @@ type ChatInputTokenDetails = {
 type ChatOutputTokenDetails = {
     /**
      * Tokens consumed by internal reasoning/thinking. Subset of `outputTokens`.
-     * For non-reasoning models (or providers that don't report it), this is `0`.
+     * Omitted when the provider does not report a breakdown.
      */
-    reasoningTokens: number;
+    reasoningTokens?: number;
 };
 type StreamEvent = {
-    type: 'content-delta';
+    type: 'reasoning-start';
+} | {
+    type: 'reasoning-delta';
+    text: string;
+} | {
+    type: 'reasoning-end';
+} | {
+    type: 'text-delta';
     text: string;
 } | {
     type: 'tool-call-start';
@@ -251,6 +279,12 @@ declare class StructuredOutputValidationError extends StructuredOutputError {
 
 declare function defineTool(options: ToolDefinition): ToolDefinition;
 
+type ResultToMessageOptions = {
+    includeReasoning?: boolean;
+};
+declare function resultToMessage(result: GenerateResult, options?: ResultToMessageOptions): AssistantMessage;
+declare function assistantMessage(content: string): AssistantMessage;
+
 type GenerateParams = GenerateOptions & {
     model: ChatModel;
 };
@@ -284,4 +318,4 @@ type GenerateImageParams = ImageGenerateOptions & {
 };
 declare function generateImage(params: GenerateImageParams): Promise<ImageGenerateResult>;
 
-export { type AssistantMessage, type ChatInputTokenDetails, type ChatModel, type ChatOutputTokenDetails, type ChatUsage, type EmbedOptions, type EmbedResult, type EmbeddingModel, type EmbeddingUsage, type FilePart, type FinishReason, type GenerateObjectOptions, type GenerateObjectResult, type GenerateOptions, type GenerateResult, type GeneratedImage, type ImageGenerateOptions, type ImageGenerateResult, type ImageModel, type ImagePart, LLMError, type Message, type ModelConfig, type ObjectStreamEvent, ProviderError, type StreamEvent, type StreamObjectOptions, type StreamObjectResult, type StreamResult, StructuredOutputError, StructuredOutputNoObjectGeneratedError, StructuredOutputParseError, StructuredOutputValidationError, type SystemMessage, type TextPart, type ToolCall, type ToolChoice, type ToolDefinition, type ToolResultMessage, type ToolSet, type UserContentPart, type UserMessage, createObjectStreamResult, createStreamResult, defineTool, embed, generate, generateImage, generateObject, stream, streamObject };
+export { type AssistantContentPart, type AssistantMessage, type AssistantTextPart, type ChatInputTokenDetails, type ChatModel, type ChatOutputTokenDetails, type ChatUsage, type EmbedOptions, type EmbedResult, type EmbeddingModel, type EmbeddingUsage, type FilePart, type FinishReason, type GenerateObjectOptions, type GenerateObjectResult, type GenerateOptions, type GenerateResult, type GeneratedImage, type ImageGenerateOptions, type ImageGenerateResult, type ImageModel, type ImagePart, LLMError, type Message, type ModelConfig, type ObjectStreamEvent, ProviderError, type ReasoningConfig, type ReasoningEffort, type ReasoningPart, type StreamEvent, type StreamObjectOptions, type StreamObjectResult, type StreamResult, StructuredOutputError, StructuredOutputNoObjectGeneratedError, StructuredOutputParseError, StructuredOutputValidationError, type SystemMessage, type TextPart, type ToolCall, type ToolCallPart, type ToolChoice, type ToolDefinition, type ToolResultMessage, type ToolSet, type UserContentPart, type UserMessage, assistantMessage, createObjectStreamResult, createStreamResult, defineTool, embed, generate, generateImage, generateObject, resultToMessage, stream, streamObject };

package/dist/index.js

@@ -52,6 +52,22 @@ function defineTool(options) {
   return options;
 }
 
+// src/result-to-message.ts
+function resultToMessage(result, options) {
+  const includeReasoning = options?.includeReasoning ?? true;
+  const parts = includeReasoning ? [...result.parts] : result.parts.filter((part) => part.type !== "reasoning");
+  return {
+    role: "assistant",
+    parts
+  };
+}
+function assistantMessage(content) {
+  return {
+    role: "assistant",
+    parts: [{ type: "text", text: content }]
+  };
+}
+
 // src/generate.ts
 async function generate(params) {
   if (params.messages.length === 0) {
@@ -107,9 +123,7 @@ function createObjectStreamResult(source) {
         cacheReadTokens: 0,
         cacheWriteTokens: 0
       },
-      outputTokenDetails: {
-        reasoningTokens: 0
-      }
+      outputTokenDetails: {}
     };
     try {
       for await (const event of source) {
@@ -169,8 +183,10 @@ function createStreamResult(source) {
   });
   let iteratorCreated = false;
   async function* iterate() {
-    let content = "";
-    const toolCalls = [];
+    const parts = [];
+    let textBuffer = "";
+    let reasoningBuffer = "";
+    let insideReasoning = false;
     let finishReason = "unknown";
     let usage = {
       inputTokens: 0,
@@ -179,23 +195,73 @@ function createStreamResult(source) {
         cacheReadTokens: 0,
         cacheWriteTokens: 0
       },
-      outputTokenDetails: {
-        reasoningTokens: 0
+      outputTokenDetails: {}
+    };
+    const flushText = () => {
+      if (textBuffer.length === 0) {
+        return;
       }
+      parts.push({
+        type: "text",
+        text: textBuffer
+      });
+      textBuffer = "";
+    };
+    const flushReasoning = () => {
+      if (reasoningBuffer.length === 0) {
+        return;
+      }
+      parts.push({
+        type: "reasoning",
+        text: reasoningBuffer
+      });
+      reasoningBuffer = "";
     };
     for await (const event of source) {
-      if (event.type === "content-delta") {
-        content += event.text;
+      if (event.type === "reasoning-start") {
+        flushText();
+        flushReasoning();
+        insideReasoning = true;
+      } else if (event.type === "reasoning-delta") {
+        if (!insideReasoning) {
+          flushText();
+          insideReasoning = true;
+        }
+        reasoningBuffer += event.text;
+      } else if (event.type === "reasoning-end") {
+        flushReasoning();
+        insideReasoning = false;
+      } else if (event.type === "text-delta") {
+        if (insideReasoning) {
+          flushReasoning();
+          insideReasoning = false;
+        }
+        textBuffer += event.text;
       } else if (event.type === "tool-call-end") {
-        toolCalls.push(event.toolCall);
+        flushText();
+        flushReasoning();
+        insideReasoning = false;
+        parts.push({
+          type: "tool-call",
+          toolCall: event.toolCall
+        });
       } else if (event.type === "finish") {
         finishReason = event.finishReason;
         usage = event.usage;
       }
       yield event;
     }
+    flushText();
+    flushReasoning();
+    const content = parts.flatMap((part) => part.type === "text" ? [part.text] : []).join("");
+    const reasoning = parts.flatMap((part) => part.type === "reasoning" ? [part.text] : []).join("");
+    const toolCalls = parts.flatMap(
+      (part) => part.type === "tool-call" ? [part.toolCall] : []
+    );
     resolveResponse?.({
+      parts,
       content: content.length > 0 ? content : null,
+      reasoning: reasoning.length > 0 ? reasoning : null,
       toolCalls,
       finishReason,
       usage
@@ -251,6 +317,7 @@ export {
   StructuredOutputNoObjectGeneratedError,
   StructuredOutputParseError,
   StructuredOutputValidationError,
+  assistantMessage,
   createObjectStreamResult,
   createStreamResult,
   defineTool,
@@ -258,6 +325,7 @@ export {
   generate,
   generateImage,
   generateObject,
+  resultToMessage,
   stream,
   streamObject
 };

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@core-ai/core-ai",
-    "version": "0.4.0",
+    "version": "0.5.1",
     "description": "Type-safe LLM abstraction layer over native provider SDKs",
     "license": "MIT",
     "author": "Omnifact (https://omnifact.ai)",
@@ -39,7 +39,6 @@
         "build": "tsup",
         "lint": "eslint src/ --max-warnings 0",
         "check-types": "tsc --noEmit",
-        "prepublishOnly": "npm run build",
         "test": "vitest run",
         "test:watch": "vitest"
     },