@bowenqt/qiniu-ai-sdk 0.5.0 → 0.9.0

package/README.md

@@ -21,6 +21,14 @@ TypeScript SDK for Qiniu Cloud AI Token API.
 npm install @bowenqt/qiniu-ai-sdk
 ```
 
+### Installation for Adapter Users
+
+The Vercel AI SDK adapter is optional. Install the peer dependencies when you use it:
+
+```bash
+npm install @bowenqt/qiniu-ai-sdk @ai-sdk/provider ai
+```
+
 ## Quick Start
 
 ```typescript
@@ -38,12 +46,12 @@ const chat = await client.chat.create({
 console.log(chat.choices[0].message.content);
 
 // Image generation (async)
-const imageTask = await client.image.create({
+const imageResult = await client.image.generate({
   model: 'kling-v1',
   prompt: 'A futuristic city',
 });
-const imageResult = await client.image.waitForCompletion(imageTask.task_id);
-console.log(imageResult.data?.[0].url);
+const imageFinal = await client.image.waitForResult(imageResult);
+console.log(imageFinal.data?.[0].url || imageFinal.data?.[0].b64_json);
 
 // Video generation (async)
 const videoTask = await client.video.create({
@@ -72,6 +80,68 @@ for await (const chunk of stream) {
 }
 ```
 
+## Advanced Usage
+
+### StreamAccumulator (Manual Streaming)
+
+```typescript
+import { QiniuAI, createStreamAccumulator, accumulateDelta } from '@bowenqt/qiniu-ai-sdk';
+
+const client = new QiniuAI({ apiKey: 'Sk-xxx' });
+const stream = client.chat.createStream({
+  model: 'gemini-2.5-flash',
+  messages: [{ role: 'user', content: 'Explain streaming in one sentence.' }],
+});
+
+const acc = createStreamAccumulator();
+for await (const chunk of stream) {
+  const delta = chunk.choices[0]?.delta;
+  if (delta) {
+    accumulateDelta(acc, delta);
+  }
+}
+console.log(acc.content);
+```
+
+### parseSSEStream (Custom SSE Parsing)
+
+```typescript
+import { parseSSEStream } from '@bowenqt/qiniu-ai-sdk';
+
+const response = await fetch('https://api.qnaigc.com/v1/chat/completions', {
+  method: 'POST',
+  headers: {
+    'Content-Type': 'application/json',
+    'Authorization': `Bearer ${process.env.QINIU_API_KEY}`,
+  },
+  body: JSON.stringify({
+    model: 'gemini-2.5-flash',
+    messages: [{ role: 'user', content: 'Hello!' }],
+    stream: true,
+  }),
+});
+
+for await (const chunk of parseSSEStream(response)) {
+  console.log(chunk);
+}
+```
+
+### createPoller (Custom Async Polling)
+
+```typescript
+import { createPoller } from '@bowenqt/qiniu-ai-sdk';
+
+const poller = createPoller({
+  intervalMs: 2000,
+  timeoutMs: 60000,
+  isTerminal: (result) => result.status === 'succeed' || result.status === 'failed',
+  getStatus: (id) => client.video.get(id),
+});
+
+const result = await poller.poll('task-id');
+console.log(result.result);
+```
+
 ## Vercel AI SDK Adapter
 
 Use the adapter to integrate with the Vercel AI SDK (`streamText`, `generateText`).
@@ -248,10 +318,14 @@ for await (const chunk of stream) {
 
 #### `client.image`
 
-- `create(params: ImageGenerationRequest): Promise<{ task_id: string }>`
+- `generate(params: ImageGenerationRequest): Promise<ImageCreateResult>`
+- `waitForResult(result: ImageCreateResult, options?: WaitOptions): Promise<ImageGenerateResult>`
+- `create(params: ImageGenerationRequest): Promise<{ task_id: string }>` (deprecated)
 - `edit(params: ImageEditRequest): Promise<ImageEditResponse>`
 - `get(taskId: string): Promise<ImageTaskResponse>`
-- `waitForCompletion(taskId: string, options?: WaitOptions): Promise<ImageTaskResponse>`
+- `waitForCompletion(taskId: string, options?: WaitOptions): Promise<ImageTaskResponse>` (deprecated)
+
+Note: For sync models (Gemini), `waitForResult` returns `status: 'succeed'` set by the SDK when data is returned.
 
 **Image Edit (Kling/Gemini):**
 
@@ -458,7 +532,7 @@ const res = await client.post<any>('/voice/tts', {
 
 ### Wait Options
 
-For `waitForCompletion` methods:
+For `waitForCompletion` and `waitForResult` methods:
 
 ```typescript
 interface WaitOptions {
@@ -494,7 +568,11 @@ const controller = new AbortController();
 setTimeout(() => controller.abort(), 10000);
 
 try {
-  const result = await client.image.waitForCompletion(taskId, {
+  const createResult = await client.image.generate({
+    model: 'kling-v2',
+    prompt: 'A cute cat',
+  });
+  const result = await client.image.waitForResult(createResult, {
     signal: controller.signal,
   });
 } catch (error) {

package/dist/ai/generate-text.d.ts

@@ -0,0 +1,48 @@
+import type { QiniuAI } from '../client';
+import type { ChatMessage, ToolCall } from '../lib/types';
+export interface ToolExecutionContext {
+    toolCallId: string;
+    messages: ChatMessage[];
+    abortSignal?: AbortSignal;
+}
+export interface Tool {
+    description?: string;
+    parameters?: Record<string, unknown>;
+    execute?: (args: unknown, context: ToolExecutionContext) => Promise<unknown> | unknown;
+}
+export interface ToolResult {
+    toolCallId: string;
+    result: string;
+}
+export interface StepResult {
+    type: 'text' | 'tool_call' | 'tool_result';
+    content: string;
+    reasoning?: string;
+    toolCalls?: ToolCall[];
+    toolResults?: ToolResult[];
+}
+export interface GenerateTextOptions {
+    client: QiniuAI;
+    model: string;
+    prompt?: string;
+    messages?: ChatMessage[];
+    system?: string;
+    tools?: Record<string, Tool>;
+    maxSteps?: number;
+    onStepFinish?: (step: StepResult) => void;
+    abortSignal?: AbortSignal;
+}
+export interface GenerateTextResult {
+    text: string;
+    reasoning?: string;
+    steps: StepResult[];
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+    finishReason: string | null;
+}
+export declare function generateText(options: GenerateTextOptions): Promise<GenerateTextResult>;
+export declare function serializeToolResult(result: unknown): string;
+//# sourceMappingURL=generate-text.d.ts.map

package/dist/ai/generate-text.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-text.d.ts","sourceRoot":"","sources":["../../src/ai/generate-text.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,KAAK,EAAyB,WAAW,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAIjF,MAAM,WAAW,oBAAoB;IACjC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,WAAW,CAAC;CAC7B;AAED,MAAM,WAAW,IAAI;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,oBAAoB,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC;CAC1F;AAED,MAAM,WAAW,UAAU;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,UAAU;IACvB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,aAAa,CAAC;IAC3C,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;IACvB,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC;CAC9B;AAED,MAAM,WAAW,mBAAmB;IAChC,MAAM,EAAE,OAAO,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,WAAW,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC7B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,IAAI,CAAC;IAC1C,WAAW,CAAC,EAAE,WAAW,CAAC;CAC7B;AAED,MAAM,WAAW,kBAAkB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,UAAU,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE;QACJ,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACxB,CAAC;IACF,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;CAC/B;AAED,wBAAsB,YAAY,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CA4E5F;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,CAc3D"}

package/dist/ai/generate-text.js

@@ -0,0 +1,169 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateText = generateText;
+exports.serializeToolResult = serializeToolResult;
+const errors_1 = require("../lib/errors");
+async function generateText(options) {
+    const { client, model, tools, maxSteps = 1, onStepFinish, abortSignal, } = options;
+    const messages = normalizeMessages(options);
+    const steps = [];
+    let accumulatedText = '';
+    let accumulatedReasoning = '';
+    let usage;
+    let finishReason = null;
+    for (let stepIndex = 0; stepIndex < maxSteps; stepIndex += 1) {
+        const request = buildChatRequest({ model, messages, tools });
+        const streamResult = await consumeStream(client, request, abortSignal);
+        if (streamResult.usage) {
+            usage = streamResult.usage;
+        }
+        if (streamResult.finishReason !== undefined) {
+            finishReason = streamResult.finishReason;
+        }
+        if (streamResult.content) {
+            accumulatedText += streamResult.content;
+        }
+        if (streamResult.reasoningContent) {
+            accumulatedReasoning += streamResult.reasoningContent;
+        }
+        const textStep = {
+            type: 'text',
+            content: streamResult.content,
+            reasoning: streamResult.reasoningContent || undefined,
+            toolCalls: streamResult.toolCalls.length ? streamResult.toolCalls : undefined,
+        };
+        steps.push(textStep);
+        if (onStepFinish) {
+            onStepFinish(textStep);
+        }
+        if (!streamResult.toolCalls.length || !tools) {
+            return {
+                text: accumulatedText,
+                reasoning: accumulatedReasoning || undefined,
+                steps,
+                usage,
+                finishReason,
+            };
+        }
+        const toolCallSteps = streamResult.toolCalls.map((toolCall) => ({
+            type: 'tool_call',
+            content: toolCall.function.arguments,
+            toolCalls: [toolCall],
+        }));
+        steps.push(...toolCallSteps);
+        const toolResults = await executeTools(streamResult.toolCalls, tools, messages, abortSignal);
+        const toolResultSteps = toolResults.map((toolResult) => ({
+            type: 'tool_result',
+            content: toolResult.result,
+            toolResults: [toolResult],
+        }));
+        steps.push(...toolResultSteps);
+        messages.push(...toolResultsToMessages(streamResult.toolCalls, toolResults));
+    }
+    throw new errors_1.MaxStepsExceededError(maxSteps);
+}
+function serializeToolResult(result) {
+    if (result === undefined) {
+        return '';
+    }
+    if (typeof result === 'string') {
+        return result;
+    }
+    try {
+        return JSON.stringify(result);
+    }
+    catch {
+        return String(result);
+    }
+}
+function normalizeMessages(options) {
+    if (options.messages && options.messages.length > 0) {
+        return [...options.messages];
+    }
+    if (!options.prompt && !options.system) {
+        throw new Error('Either prompt or messages must be provided.');
+    }
+    const messages = [];
+    if (options.system) {
+        messages.push({ role: 'system', content: options.system });
+    }
+    if (options.prompt) {
+        messages.push({ role: 'user', content: options.prompt });
+    }
+    return messages;
+}
+function buildChatRequest(params) {
+    const { model, messages, tools } = params;
+    return {
+        model,
+        messages,
+        tools: tools ? Object.entries(tools).map(([name, tool]) => ({
+            type: 'function',
+            function: {
+                name,
+                description: tool.description,
+                parameters: tool.parameters ?? {},
+            },
+        })) : undefined,
+    };
+}
+async function consumeStream(client, request, abortSignal) {
+    const stream = client.chat.createStream(request, { signal: abortSignal });
+    let finalResult;
+    while (true) {
+        const { value, done } = await stream.next();
+        if (done) {
+            finalResult = value;
+            break;
+        }
+    }
+    return finalResult || {
+        content: '',
+        reasoningContent: '',
+        toolCalls: [],
+        finishReason: null,
+        usage: undefined,
+    };
+}
+function toolResultsToMessages(toolCalls, results) {
+    return toolCalls.map((toolCall) => {
+        const result = results.find((entry) => entry.toolCallId === toolCall.id);
+        return {
+            role: 'tool',
+            name: toolCall.function.name,
+            content: result?.result ?? '',
+        };
+    });
+}
+async function executeTools(toolCalls, tools, messages, abortSignal) {
+    const results = [];
+    for (const toolCall of toolCalls) {
+        const tool = tools[toolCall.function.name];
+        if (!tool || !tool.execute) {
+            throw new errors_1.ToolExecutionError(toolCall.function.name, 'Tool is not implemented.');
+        }
+        const args = parseToolArguments(toolCall.function.arguments);
+        const value = await tool.execute(args, {
+            toolCallId: toolCall.id,
+            messages,
+            abortSignal,
+        });
+        results.push({
+            toolCallId: toolCall.id,
+            result: serializeToolResult(value),
+        });
+    }
+    return results;
+}
+function parseToolArguments(payload) {
+    if (!payload) {
+        return {};
+    }
+    try {
+        return JSON.parse(payload);
+    }
+    catch {
+        return payload;
+    }
+}
+//# sourceMappingURL=generate-text.js.map

package/dist/ai/generate-text.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-text.js","sourceRoot":"","sources":["../../src/ai/generate-text.ts"],"names":[],"mappings":";;AAsDA,oCA4EC;AAED,kDAcC;AA/ID,0CAA0E;AAmDnE,KAAK,UAAU,YAAY,CAAC,OAA4B;IAC3D,MAAM,EACF,MAAM,EACN,KAAK,EACL,KAAK,EACL,QAAQ,GAAG,CAAC,EACZ,YAAY,EACZ,WAAW,GACd,GAAG,OAAO,CAAC;IAEZ,MAAM,QAAQ,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAiB,EAAE,CAAC;IAC/B,IAAI,eAAe,GAAG,EAAE,CAAC;IACzB,IAAI,oBAAoB,GAAG,EAAE,CAAC;IAC9B,IAAI,KAAkC,CAAC;IACvC,IAAI,YAAY,GAAuC,IAAI,CAAC;IAE5D,KAAK,IAAI,SAAS,GAAG,CAAC,EAAE,SAAS,GAAG,QAAQ,EAAE,SAAS,IAAI,CAAC,EAAE,CAAC;QAC3D,MAAM,OAAO,GAAG,gBAAgB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,MAAM,YAAY,GAAG,MAAM,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;QAEvE,IAAI,YAAY,CAAC,KAAK,EAAE,CAAC;YACrB,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC;QAC/B,CAAC;QACD,IAAI,YAAY,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YAC1C,YAAY,GAAG,YAAY,CAAC,YAAY,CAAC;QAC7C,CAAC;QAED,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;YACvB,eAAe,IAAI,YAAY,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,IAAI,YAAY,CAAC,gBAAgB,EAAE,CAAC;YAChC,oBAAoB,IAAI,YAAY,CAAC,gBAAgB,CAAC;QAC1D,CAAC;QAED,MAAM,QAAQ,GAAe;YACzB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,YAAY,CAAC,OAAO;YAC7B,SAAS,EAAE,YAAY,CAAC,gBAAgB,IAAI,SAAS;YACrD,SAAS,EAAE,YAAY,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;SAChF,CAAC;QAEF,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrB,IAAI,YAAY,EAAE,CAAC;YACf,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YAC3C,OAAO;gBACH,IAAI,EAAE,eAAe;gBACrB,SAAS,EAAE,oBAAoB,IAAI,SAAS;gBAC5C,KAAK;gBACL,KAAK;gBACL,YAAY;aACf,CAAC;QACN,CAAC;QAED,MAAM,aAAa,GAAG,YAAY,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAC5D,IAAI,EAAE,WAAoB;YAC1B,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;YACpC,SAAS,EAAE,CAAC,QAAQ,CAAC;SACxB,CAAC,CAAC,CAAC;QACJ,KAAK,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAE7B,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,YAAY,CAAC,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC7F,MAAM,eAAe,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACrD,IAAI,EAAE,aAAsB;YAC5B,OAAO,EAAE,UAAU,CAAC,MAAM;YAC1B,WAAW,EAAE,CAAC,UAAU,CAAC;SAC5B,CAAC,CAAC,CAAC;QACJ,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;QAE/B,QAAQ,CAAC,IAAI,CAAC,GAAG,qBAAqB,CAAC,YAAY,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC;IACjF,CAAC;IAED,MAAM,IAAI,8BAAqB,CAAC,QAAQ,CAAC,CAAC;AAC9C,CAAC;AAED,SAAgB,mBAAmB,CAAC,MAAe;IAC/C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACvB,OAAO,EAAE,CAAC;IACd,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,IAAI,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;AACL,CAAC;AAED,SAAS,iBAAiB,CAAC,OAA4B;IACnD,IAAI,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAClD,OAAO,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IACjC,CAAC;IAED,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,QAAQ,GAAkB,EAAE,CAAC;IACnC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/D,CAAC;IACD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,gBAAgB,CAAC,MAIzB;IACG,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;IAE1C,OAAO;QACH,KAAK;QACL,QAAQ;QACR,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YACxD,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACN,IAAI;gBACJ,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;aACpC;SACJ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;KAClB,CAAC;AACN,CAAC;AAED,KAAK,UAAU,aAAa,CACxB,MAAe,EACf,OAA8B,EAC9B,WAAyB;IAEzB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IAC1E,IAAI,WAAqC,CAAC;IAE1C,OAAO,IAAI,EAAE,CAAC;QACV,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5C,IAAI,IAAI,EAAE,CAAC;YACP,WAAW,GAAG,KAAK,CAAC;YACpB,MAAM;QACV,CAAC;IACL,CAAC;IAED,OAAO,WAAW,IAAI;QAClB,OAAO,EAAE,EAAE;QACX,gBAAgB,EAAE,EAAE;QACpB,SAAS,EAAE,EAAE;QACb,YAAY,EAAE,IAAI;QAClB,KAAK,EAAE,SAAS;KACnB,CAAC;AACN,CAAC;AAED,SAAS,qBAAqB,CAAC,SAAqB,EAAE,OAAqB;IACvE,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,KAAK,QAAQ,CAAC,EAAE,CAAC,CAAC;QACzE,OAAO;YACH,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;YAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE;SAChC,CAAC;IACN,CAAC,CAAC,CAAC;AACP,CAAC;AAED,KAAK,UAAU,YAAY,CACvB,SAAqB,EACrB,KAA2B,EAC3B,QAAuB,EACvB,WAAyB;IAEzB,MAAM,OAAO,GAAiB,EAAE,CAAC;IAEjC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACzB,MAAM,IAAI,2BAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,0BAA0B,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,IAAI,GAAG,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE;YACnC,UAAU,EAAE,QAAQ,CAAC,EAAE;YACvB,QAAQ;YACR,WAAW;SACd,CAAC,CAAC;QAEH,OAAO,CAAC,IAAI,CAAC;YACT,UAAU,EAAE,QAAQ,CAAC,EAAE;YACvB,MAAM,EAAE,mBAAmB,CAAC,KAAK,CAAC;SACrC,CAAC,CAAC;IACP,CAAC;IAED,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAe;IACvC,IAAI,CAAC,OAAO,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,OAAO,CAAC;IACnB,CAAC;AACL,CAAC"}

package/dist/ai/generate-text.mjs

@@ -0,0 +1,165 @@
+import { MaxStepsExceededError, ToolExecutionError } from '../lib/errors.mjs';
+export async function generateText(options) {
+    const { client, model, tools, maxSteps = 1, onStepFinish, abortSignal, } = options;
+    const messages = normalizeMessages(options);
+    const steps = [];
+    let accumulatedText = '';
+    let accumulatedReasoning = '';
+    let usage;
+    let finishReason = null;
+    for (let stepIndex = 0; stepIndex < maxSteps; stepIndex += 1) {
+        const request = buildChatRequest({ model, messages, tools });
+        const streamResult = await consumeStream(client, request, abortSignal);
+        if (streamResult.usage) {
+            usage = streamResult.usage;
+        }
+        if (streamResult.finishReason !== undefined) {
+            finishReason = streamResult.finishReason;
+        }
+        if (streamResult.content) {
+            accumulatedText += streamResult.content;
+        }
+        if (streamResult.reasoningContent) {
+            accumulatedReasoning += streamResult.reasoningContent;
+        }
+        const textStep = {
+            type: 'text',
+            content: streamResult.content,
+            reasoning: streamResult.reasoningContent || undefined,
+            toolCalls: streamResult.toolCalls.length ? streamResult.toolCalls : undefined,
+        };
+        steps.push(textStep);
+        if (onStepFinish) {
+            onStepFinish(textStep);
+        }
+        if (!streamResult.toolCalls.length || !tools) {
+            return {
+                text: accumulatedText,
+                reasoning: accumulatedReasoning || undefined,
+                steps,
+                usage,
+                finishReason,
+            };
+        }
+        const toolCallSteps = streamResult.toolCalls.map((toolCall) => ({
+            type: 'tool_call',
+            content: toolCall.function.arguments,
+            toolCalls: [toolCall],
+        }));
+        steps.push(...toolCallSteps);
+        const toolResults = await executeTools(streamResult.toolCalls, tools, messages, abortSignal);
+        const toolResultSteps = toolResults.map((toolResult) => ({
+            type: 'tool_result',
+            content: toolResult.result,
+            toolResults: [toolResult],
+        }));
+        steps.push(...toolResultSteps);
+        messages.push(...toolResultsToMessages(streamResult.toolCalls, toolResults));
+    }
+    throw new MaxStepsExceededError(maxSteps);
+}
+export function serializeToolResult(result) {
+    if (result === undefined) {
+        return '';
+    }
+    if (typeof result === 'string') {
+        return result;
+    }
+    try {
+        return JSON.stringify(result);
+    }
+    catch {
+        return String(result);
+    }
+}
+function normalizeMessages(options) {
+    if (options.messages && options.messages.length > 0) {
+        return [...options.messages];
+    }
+    if (!options.prompt && !options.system) {
+        throw new Error('Either prompt or messages must be provided.');
+    }
+    const messages = [];
+    if (options.system) {
+        messages.push({ role: 'system', content: options.system });
+    }
+    if (options.prompt) {
+        messages.push({ role: 'user', content: options.prompt });
+    }
+    return messages;
+}
+function buildChatRequest(params) {
+    const { model, messages, tools } = params;
+    return {
+        model,
+        messages,
+        tools: tools ? Object.entries(tools).map(([name, tool]) => ({
+            type: 'function',
+            function: {
+                name,
+                description: tool.description,
+                parameters: tool.parameters ?? {},
+            },
+        })) : undefined,
+    };
+}
+async function consumeStream(client, request, abortSignal) {
+    const stream = client.chat.createStream(request, { signal: abortSignal });
+    let finalResult;
+    while (true) {
+        const { value, done } = await stream.next();
+        if (done) {
+            finalResult = value;
+            break;
+        }
+    }
+    return finalResult || {
+        content: '',
+        reasoningContent: '',
+        toolCalls: [],
+        finishReason: null,
+        usage: undefined,
+    };
+}
+function toolResultsToMessages(toolCalls, results) {
+    return toolCalls.map((toolCall) => {
+        const result = results.find((entry) => entry.toolCallId === toolCall.id);
+        return {
+            role: 'tool',
+            name: toolCall.function.name,
+            content: result?.result ?? '',
+        };
+    });
+}
+async function executeTools(toolCalls, tools, messages, abortSignal) {
+    const results = [];
+    for (const toolCall of toolCalls) {
+        const tool = tools[toolCall.function.name];
+        if (!tool || !tool.execute) {
+            throw new ToolExecutionError(toolCall.function.name, 'Tool is not implemented.');
+        }
+        const args = parseToolArguments(toolCall.function.arguments);
+        const value = await tool.execute(args, {
+            toolCallId: toolCall.id,
+            messages,
+            abortSignal,
+        });
+        results.push({
+            toolCallId: toolCall.id,
+            result: serializeToolResult(value),
+        });
+    }
+    return results;
+}
+function parseToolArguments(payload) {
+    if (!payload) {
+        return {};
+    }
+    try {
+        return JSON.parse(payload);
+    }
+    catch {
+        return payload;
+    }
+}
+//# sourceMappingURL=generate-text.js.map

package/dist/ai-tools/index.d.ts

@@ -0,0 +1,17 @@
+import type { ZodTypeAny } from 'zod';
+export type JsonSchema = {
+    type?: string;
+    properties?: Record<string, JsonSchema>;
+    required?: string[];
+    items?: JsonSchema;
+    enum?: unknown[];
+    oneOf?: JsonSchema[];
+};
+export interface ToolDefinition<PARAMETERS = unknown, RESULT = unknown> {
+    description?: string;
+    parameters: JsonSchema | ZodTypeAny;
+    execute?: (args: PARAMETERS) => Promise<RESULT> | RESULT;
+}
+export declare function tool<PARAMETERS, RESULT>(definition: ToolDefinition<PARAMETERS, RESULT>): ToolDefinition<PARAMETERS, RESULT>;
+export declare function zodToJsonSchema(schema: ZodTypeAny): JsonSchema;
+//# sourceMappingURL=index.d.ts.map

package/dist/ai-tools/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ai-tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,KAAK,CAAC;AAEtC,MAAM,MAAM,UAAU,GAAG;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACxC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC;IACjB,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;CACxB,CAAC;AAEF,MAAM,WAAW,cAAc,CAAC,UAAU,GAAG,OAAO,EAAE,MAAM,GAAG,OAAO;IAClE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,UAAU,GAAG,UAAU,CAAC;IACpC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,UAAU,KAAK,OAAO,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;CAC5D;AAED,wBAAgB,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,GAAG,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,CAE3H;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,UAAU,GAAG,UAAU,CAE9D"}

package/dist/ai-tools/index.js

@@ -0,0 +1,77 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tool = tool;
+exports.zodToJsonSchema = zodToJsonSchema;
+function tool(definition) {
+    return definition;
+}
+function zodToJsonSchema(schema) {
+    return convertSchema(schema);
+}
+function convertSchema(schema) {
+    const def = schema._def;
+    const typeName = def?.typeName;
+    switch (typeName) {
+        case 'ZodString':
+            return { type: 'string' };
+        case 'ZodNumber':
+            return { type: 'number' };
+        case 'ZodBoolean':
+            return { type: 'boolean' };
+        case 'ZodArray':
+            return { type: 'array', items: convertSchema(def.type) };
+        case 'ZodEnum':
+            return { type: 'string', enum: def.values };
+        case 'ZodLiteral':
+            return literalSchema(def.value);
+        case 'ZodObject':
+            return objectSchema(schema);
+        case 'ZodUnion':
+            return { oneOf: def.options.map(convertSchema) };
+        case 'ZodOptional':
+            return convertSchema(def.innerType);
+        case 'ZodNullable':
+            return convertSchema(def.innerType);
+        case 'ZodDefault':
+            return convertSchema(def.innerType);
+        default:
+            return {};
+    }
+}
+function objectSchema(schema) {
+    const def = schema._def;
+    const shapeSource = def?.shape;
+    const shape = typeof shapeSource === 'function' ? shapeSource() : shapeSource || {};
+    const properties = {};
+    const required = [];
+    for (const [key, value] of Object.entries(shape)) {
+        const { inner, optional } = unwrapOptional(value);
+        properties[key] = convertSchema(inner);
+        if (!optional) {
+            required.push(key);
+        }
+    }
+    const schemaResult = { type: 'object', properties };
+    if (required.length) {
+        schemaResult.required = required;
+    }
+    return schemaResult;
+}
+function unwrapOptional(schema) {
+    const def = schema._def;
+    if (!def?.typeName) {
+        return { inner: schema, optional: false };
+    }
+    if (def.typeName === 'ZodOptional' || def.typeName === 'ZodDefault') {
+        return { inner: def.innerType, optional: true };
+    }
+    return { inner: schema, optional: false };
+}
+function literalSchema(value) {
+    const valueType = typeof value;
+    if (valueType === 'string' || valueType === 'number' || valueType === 'boolean') {
+        return { type: valueType, enum: [value] };
+    }
+    return { enum: [value] };
+}
+//# sourceMappingURL=index.js.map

package/dist/ai-tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/ai-tools/index.ts"],"names":[],"mappings":";;AAiBA,oBAEC;AAED,0CAEC;AAND,SAAgB,IAAI,CAAqB,UAA8C;IACnF,OAAO,UAAU,CAAC;AACtB,CAAC;AAED,SAAgB,eAAe,CAAC,MAAkB;IAC9C,OAAO,aAAa,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,aAAa,CAAC,MAAkB;IACrC,MAAM,GAAG,GAAI,MAAmE,CAAC,IAAI,CAAC;IACtF,MAAM,QAAQ,GAAG,GAAG,EAAE,QAAQ,CAAC;IAE/B,QAAQ,QAAQ,EAAE,CAAC;QACf,KAAK,WAAW;YACZ,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAC9B,KAAK,WAAW;YACZ,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAC9B,KAAK,YAAY;YACb,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QAC/B,KAAK,UAAU;YACX,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,CAAE,GAA4B,CAAC,IAAI,CAAC,EAAE,CAAC;QACvF,KAAK,SAAS;YACV,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAG,GAA6B,CAAC,MAAM,EAAE,CAAC;QAC3E,KAAK,YAAY;YACb,OAAO,aAAa,CAAE,GAA0B,CAAC,KAAK,CAAC,CAAC;QAC5D,KAAK,WAAW;YACZ,OAAO,YAAY,CAAC,MAAM,CAAC,CAAC;QAChC,KAAK,UAAU;YACX,OAAO,EAAE,KAAK,EAAG,GAAiC,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QACpF,KAAK,aAAa;YACd,OAAO,aAAa,CAAE,GAAiC,CAAC,SAAS,CAAC,CAAC;QACvE,KAAK,aAAa;YACd,OAAO,aAAa,CAAE,GAAiC,CAAC,SAAS,CAAC,CAAC;QACvE,KAAK,YAAY;YACb,OAAO,aAAa,CAAE,GAAiC,CAAC,SAAS,CAAC,CAAC;QACvE;YACI,OAAO,EAAE,CAAC;IAClB,CAAC;AACL,CAAC;AAED,SAAS,YAAY,CAAC,MAAkB;IACpC,MAAM,GAAG,GAAI,MAAiG,CAAC,IAAI,CAAC;IACpH,MAAM,WAAW,GAAG,GAAG,EAAE,KAAK,CAAC;IAC/B,MAAM,KAAK,GAAG,OAAO,WAAW,KAAK,UAAU,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC;IAEpF,MAAM,UAAU,GAA+B,EAAE,CAAC;IAClD,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC/C,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,cAAc,CAAC,KAAmB,CAAC,CAAC;QAChE,UAAU,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;IACL,CAAC;IAED,MAAM,YAAY,GAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;IAChE,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QAClB,YAAY,CAAC,QAAQ,GAAG,QAAQ,CAAC;IACrC,CAAC;IAED,OAAO,YAAY,CAAC;AACxB,CAAC;AAED,SAAS,cAAc,CAAC,MAAkB;IACtC,MAAM,GAAG,GAAI,MAAmE,CAAC,IAAI,CAAC;IACtF,IAAI,CAAC,GAAG,EAAE,QAAQ,EAAE,CAAC;QACjB,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC9C,CAAC;IAED,IAAI,GAAG,CAAC,QAAQ,KAAK,aAAa,IAAI,GAAG,CAAC,QAAQ,KAAK,YAAY,EAAE,CAAC;QAClE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,SAAuB,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAClE,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;AAC9C,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACjC,MAAM,SAAS,GAAG,OAAO,KAAK,CAAC;IAC/B,IAAI,SAAS,KAAK,QAAQ,IAAI,SAAS,KAAK,QAAQ,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC9E,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;IAC9C,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;AAC7B,CAAC"}