@opentiny/tiny-robot-kit 0.4.0 → 0.4.1-alpha.1

package/dist/core.d.ts

@@ -0,0 +1,281 @@
+import { ChatCompletionMessageParam, ChatCompletionMessageToolCall, ChatCompletion, ChatCompletionChunk, ChatCompletionTool } from 'openai/resources/index';
+import { M as MaybePromise } from './types-CePh-Jcx.js';
+import { Ref, ComputedRef } from 'vue';
+
+type DeepReadonly<T> = T extends (...args: any[]) => any ? T : T extends Array<infer U> ? ReadonlyArray<DeepReadonly<U>> : T extends object ? {
+    readonly [K in keyof T]: DeepReadonly<T[K]>;
+} : T;
+type RequestState = 'idle' | 'processing' | 'completed' | 'aborted' | 'error';
+type RequestProcessingState = 'requesting' | 'completing' | string;
+type ChatMessage<Metadata extends object = Record<string, unknown>, State extends object = Record<string, unknown>> = ChatCompletionMessageParam & {
+    tool_calls?: Array<ChatCompletionMessageToolCall>;
+    loading?: boolean;
+    metadata?: Metadata;
+    state?: State;
+    [key: string]: any;
+};
+interface MessageRequestBody {
+    messages: Array<ChatMessage>;
+    [key: string]: any;
+}
+type ResponseProvider<T extends ChatCompletion | ChatCompletionChunk = ChatCompletion | ChatCompletionChunk> = (requestBody: MessageRequestBody, abortSignal: AbortSignal) => Promise<T> | AsyncGenerator<T> | Promise<AsyncGenerator<T>>;
+interface PublicMessageState {
+    requestState: RequestState;
+    processingState?: RequestProcessingState;
+    messages: ChatMessage[];
+    isProcessing: boolean;
+}
+interface InternalMessageState {
+    requestState: RequestState;
+    processingState?: RequestProcessingState;
+    messages: ChatMessage[];
+}
+interface MessageRuntime {
+    currentTurn: ChatMessage[];
+    customContext: Record<string, unknown>;
+    abortController: AbortController | null;
+    responseProvider: ResponseProvider;
+}
+interface MessageEngine {
+    getState(): PublicMessageState;
+    subscribe(listener: (state: PublicMessageState) => void): () => void;
+    subscribe(kinds: MessageUpdateKinds, listener: (state: PublicMessageState) => void): () => void;
+    sendMessage(content: string): Promise<void>;
+    send(...msgs: ChatMessage[]): Promise<void>;
+    abort(): Promise<void>;
+    setResponseProvider(provider: ResponseProvider): void;
+}
+type MessageUpdateKind = 'messages' | 'requestState';
+type MessageUpdateKinds = MessageUpdateKind | MessageUpdateKind[];
+type MessageMutationRecipe = (draft: InternalMessageState, skipNotify: () => void) => void;
+interface MutateMessageStateFn {
+    /**
+     * 在受控上下文中修改消息状态，并按声明的更新类型分发状态变更通知。
+     *
+     * 这里的“通知”语义对应 `subscribe` 订阅链路，用于驱动依赖 engine 状态快照的观察者；
+     * 它并不等同于具体框架的响应式更新机制。对于 Vue 等运行时，界面更新可能同时依赖
+     * 框架自身的响应式追踪与 `subscribe` 侧的状态通知，两者职责不同，不应混用。
+     */
+    (kinds: MessageUpdateKinds, recipe: MessageMutationRecipe): void;
+}
+interface MessageStateAdapter {
+    initialize(initialState: InternalMessageState): void;
+    getState(): PublicMessageState;
+    /**
+     * 创建一条适配当前运行时的消息对象。
+     *
+     * engine 和 core 插件里凡是“新建消息”的入口，都应统一走这个方法，
+     * 以便不同平台注入各自的运行时能力。
+     *
+     * 例如：
+     * - native/纯 TS 场景：直接返回原对象即可
+     *   `createMessage(message) { return message }`
+     * - Vue 场景：返回 reactive proxy，使后续对 message.content / message.state 的原地修改能够被追踪
+     *   `createMessage(message) { return reactive(message) }`
+     */
+    createMessage<T extends ChatMessage>(message: T): T;
+    mutate: MutateMessageStateFn;
+    subscribe(listener: (state: PublicMessageState) => void): () => void;
+    subscribe(kinds: MessageUpdateKinds, listener: (state: PublicMessageState) => void): () => void;
+}
+interface BasePluginContext {
+    getState(): PublicMessageState;
+    /**
+     * 创建一条适配当前运行时的消息对象。
+     *
+     * 插件在运行过程中如果需要主动创建并追加消息，应统一通过此接口构造消息实例，
+     * 而不是直接持有普通对象字面量。这样可以确保消息对象具备当前平台所要求的运行时能力。
+     *
+     * 对于采用响应式机制的平台（例如 Vue），该接口通常会返回带有响应式包装的消息实例。
+     * 如果插件绕过此接口直接创建普通对象，再参与后续的状态更新或原地字段修改，
+     * 可能导致消息内容、状态字段等变更无法被正确追踪。
+     */
+    createMessage<T extends ChatMessage>(message: T): T;
+    /**
+     * 在受控上下文中修改状态，并触发与 `subscribe` 对应的状态更新通知。
+     *
+     * 该通知机制面向 engine 的订阅接口，而不是具体框架的响应式系统本身。
+     * 因此，在需要让 `subscribe` 观察者感知到变更时，应通过此接口完成状态写入。
+     */
+    mutate: MutateMessageStateFn;
+    abortSignal: AbortSignal;
+    currentTurn: ChatMessage[];
+    customContext: Record<string, unknown>;
+    setRequestState: (state: RequestState, processingState?: RequestProcessingState) => void;
+    setCustomContext: (data: Record<string, unknown>) => void;
+}
+interface BeforeRequestContext extends BasePluginContext {
+    requestBody: MessageRequestBody;
+}
+interface AfterRequestContext extends BasePluginContext {
+    currentMessage: DeepReadonly<ChatMessage>;
+    lastChoice?: ChatCompletion.Choice | ChatCompletionChunk.Choice;
+    /**
+     * 使用 appendMessage 函数追加消息，可触发消息更新通知。
+     */
+    appendMessage: (message: ChatMessage | ChatMessage[]) => void;
+    requestNext: () => void;
+}
+interface CompletionChunkContext extends BasePluginContext {
+    /**
+     * 当前消息，只读。需要使用 updateCurrentMessage 函数修改当前消息，才能正常触发消息更新通知。
+     */
+    currentMessage: DeepReadonly<ChatMessage>;
+    /**
+     * 使用 updateCurrentMessage 函数修改当前消息，才能正常触发消息更新通知。
+     */
+    updateCurrentMessage: (recipe: (message: ChatMessage) => void) => void;
+    choice: ChatCompletion.Choice | ChatCompletionChunk.Choice;
+    chunk: ChatCompletion | ChatCompletionChunk;
+}
+interface MessageEnginePlugin {
+    /**
+     * 插件名称。
+     */
+    name?: string;
+    /**
+     * 是否禁用插件。
+     */
+    disabled?: boolean | ((context: BasePluginContext) => boolean);
+    /**
+     * 一次对话回合（turn）开始钩子：用户消息入队后、正式发起请求之前触发。
+     * 按插件注册顺序串行执行，便于做有序初始化/校验；出错则中断流程。
+     */
+    onTurnStart?: (context: BasePluginContext) => MaybePromise<void>;
+    /**
+     * 一次对话回合（turn）结束的生命周期钩子。
+     * 触发时机：本轮对话完成（成功、被中止）后。
+     * 执行策略：按插件注册顺序串行执行，有错误则中断流程。
+     */
+    onTurnEnd?: (context: BasePluginContext) => MaybePromise<void>;
+    /**
+     * 请求开始前的生命周期钩子。
+     * 触发时机：已组装 requestBody，正式发起请求之前。
+     * 执行策略：按插件注册顺序串行执行，避免并发修改 requestBody 产生冲突。
+     */
+    onBeforeRequest?: (context: BeforeRequestContext) => MaybePromise<void>;
+    /**
+     * 请求完成后的生命周期钩子。
+     * 触发时机：本次请求（含流式）结束后。
+     * 执行策略：并行执行（Promise.all）。
+     */
+    onAfterRequest?: (context: AfterRequestContext) => MaybePromise<void>;
+    /**
+     * 数据块处理钩子，在接收到每个响应数据块时触发。
+     * 无论是流式响应（多个增量数据块）还是非流式响应（单个完整数据块），都会触发此钩子。
+     */
+    onCompletionChunk?: (context: CompletionChunkContext) => void;
+    onError?: (context: BasePluginContext & {
+        error: unknown;
+    }) => void;
+    onFinally?: (context: BasePluginContext) => void;
+}
+interface CreateMessageEngineOptions {
+    initialMessages?: ChatMessage[];
+    /**
+     * 请求消息时，要包含的字段（白名单）。默认包含所有字段。
+     * 如果 `requestMessageFieldsExclude` 存在，会先取 `requestMessageFields` 中的字段，再排除 `requestMessageFieldsExclude` 中的字段
+     */
+    requestMessageFields?: string[];
+    /**
+     * 请求消息时，要排除的字段（黑名单）。默认会排除 `state`、`metadata`、`loading` 字段（这几个字段是给UI展示用的）。
+     * 如果 `requestMessageFields` 存在，会先取 `requestMessageFields` 中的字段，再排除 `requestMessageFieldsExclude` 中的字段
+     */
+    requestMessageFieldsExclude?: string[];
+    responseProvider?: ResponseProvider;
+    /**
+     * 全局的数据块处理钩子，在接收到每个响应数据块时触发。
+     * 注意：此钩子与插件中的 onCompletionChunk 有区别。
+     * 如果传入了此参数，默认的 chunk 处理逻辑不会自动执行，需要手动调用 runDefault 来执行默认处理逻辑。
+     */
+    onCompletionChunk?: (context: CompletionChunkContext, runDefault: () => void) => void;
+    plugins?: MessageEnginePlugin[];
+}
+
+declare const createNativeMessageAdapter: () => MessageStateAdapter;
+
+interface VueMessageStateAdapter extends MessageStateAdapter {
+    requestState: Ref<RequestState>;
+    processingState: Ref<RequestProcessingState | undefined>;
+    messages: Ref<ChatMessage[]>;
+    isProcessing: ComputedRef<boolean>;
+}
+declare const createVueMessageAdapter: () => VueMessageStateAdapter;
+
+declare const createMessageEngine: (adapter: MessageStateAdapter, options?: CreateMessageEngineOptions) => MessageEngine;
+
+declare const lengthPlugin: (options?: MessageEnginePlugin & {
+    continueContent?: string;
+}) => MessageEnginePlugin;
+
+declare const thinkingPlugin: (options?: MessageEnginePlugin) => MessageEnginePlugin;
+
+type AssistantMessageWithState = ChatMessage<Record<string, unknown>, {
+    toolCall?: Record<string, Record<string, unknown>>;
+}>;
+type ToolCallContext = BasePluginContext & {
+    assistantMessage: AssistantMessageWithState;
+    toolMessage: ChatMessage;
+};
+declare const toolPlugin: (options: MessageEnginePlugin & {
+    /**
+     * 获取工具列表的函数。会在请求大模型前调用。
+     */
+    getTools: () => Promise<ChatCompletionTool[]>;
+    /**
+     * 在处理包含 tool_calls 的响应前调用。
+     */
+    beforeCallTools?: (toolCalls: ChatCompletionMessageToolCall[], context: BasePluginContext & {
+        assistantMessage: AssistantMessageWithState;
+    }) => Promise<void>;
+    /**
+     * 执行单个工具调用并返回其文本结果的函数。
+     */
+    callTool: (toolCall: ChatCompletionMessageToolCall, context: ToolCallContext) => Promise<string | Record<string, any>> | AsyncGenerator<string | Record<string, any>>;
+    /**
+     * 工具调用开始时的回调函数。
+     * 触发时机：工具消息已创建并追加后，调用 callTool 之前触发。
+     * @param toolCall - 工具调用对象
+     * @param context - 插件上下文，包含当前工具消息
+     */
+    onToolCallStart?: (toolCall: ChatCompletionMessageToolCall, context: ToolCallContext) => void;
+    /**
+     * 工具调用结束时的回调函数。
+     * 触发时机：工具调用完成（成功、失败或取消）时触发。
+     * @param toolCall - 工具调用对象
+     * @param context - 插件上下文，包含当前工具消息、状态和错误信息
+     * @param context.status - 工具调用状态：'success' | 'failed' | 'cancelled'
+     * @param context.error - 当状态为 'failed' 或 'cancelled' 时，可能包含错误信息
+     */
+    onToolCallEnd?: (toolCall: ChatCompletionMessageToolCall, context: ToolCallContext & {
+        status: "success" | "failed" | "cancelled";
+        error?: Error;
+    }) => void;
+    /**
+     * 当请求被中止时用于工具调用取消的消息内容。
+     */
+    toolCallCancelledContent?: string;
+    /**
+     * 当工具调用执行失败（抛错或拒绝）时使用的消息内容。
+     */
+    toolCallFailedContent?: string;
+    /**
+     * 是否在请求前自动补充缺失的 tool 消息。
+     * 当 assistant 响应了 tool_calls 但未追加对应的 tool 消息时，
+     * 插件将自动补充"工具调用已取消"的 tool 消息。默认：false。
+     */
+    autoFillMissingToolMessages?: boolean;
+}) => MessageEnginePlugin;
+
+declare function normalizeToAsyncGenerator<T>(result: Promise<T> | AsyncGenerator<T> | Promise<AsyncGenerator<T>>): AsyncGenerator<T>;
+/**
+ * Merge delta data from completion responses
+ * Handles string concatenation, object merging, and array merging by index
+ *
+ * @param target - Target object to merge into
+ * @param source - Source object to merge from
+ * @returns Merged target object
+ */
+declare const combineDeltaData: (target: Record<string, any>, source: Record<string, any>) => Record<string, any>;
+
+export { type AfterRequestContext, type BasePluginContext, type BeforeRequestContext, type ChatMessage, type CompletionChunkContext, type CreateMessageEngineOptions, type DeepReadonly, type InternalMessageState, type MessageEngine, type MessageEnginePlugin, type MessageMutationRecipe, type MessageRequestBody, type MessageRuntime, type MessageStateAdapter, type MessageUpdateKind, type MessageUpdateKinds, type MutateMessageStateFn, type PublicMessageState, type RequestProcessingState, type RequestState, type ResponseProvider, type VueMessageStateAdapter, combineDeltaData, createMessageEngine, createNativeMessageAdapter, createVueMessageAdapter, lengthPlugin, normalizeToAsyncGenerator, thinkingPlugin, toolPlugin };

package/dist/core.js

@@ -0,0 +1 @@
+"use strict";var X=Object.defineProperty;var ge=Object.getOwnPropertyDescriptor;var fe=Object.getOwnPropertyNames;var de=Object.prototype.hasOwnProperty;var pe=(e,t)=>{for(var s in t)X(e,s,{get:t[s],enumerable:!0})},me=(e,t,s,n)=>{if(t&&typeof t=="object"||typeof t=="function")for(let o of fe(t))!de.call(e,o)&&o!==s&&X(e,o,{get:()=>t[o],enumerable:!(n=ge(t,o))||n.enumerable});return e};var Me=e=>me(X({},"__esModule",{value:!0}),e);var Te={};pe(Te,{combineDeltaData:()=>B,createMessageEngine:()=>ce,createNativeMessageAdapter:()=>he,createVueMessageAdapter:()=>ye,lengthPlugin:()=>H,normalizeToAsyncGenerator:()=>O,thinkingPlugin:()=>Q,toolPlugin:()=>le});module.exports=Me(Te);var Ce=e=>typeof e=="function"?null:Array.isArray(e)?e.length>0?new Set(e):null:new Set([e]),$=e=>{let t=new Set,s=(a,l)=>{try{a.listener(l)}catch(m){console.error("Error in message state subscriber:",m)}};return{notify:a=>{let l=new Set(Array.isArray(a)?a:[a]),m=e(),d=Array.from(t);for(let r of d){if(r.kinds){let u=!1;for(let y of r.kinds)if(l.has(y)){u=!0;break}if(!u)continue}s(r,m)}},subscribe:(a,l)=>{let m=typeof a=="function"?a:l;if(!m)throw new Error("subscribe listener is required");let d={kinds:Ce(a),listener:m};return t.add(d),s(d,e()),()=>{t.delete(d)}}}};var he=()=>{let e=!1,t,s=l=>{if(e)throw new Error("Message state adapter is already initialized");t={requestState:l.requestState,processingState:l.processingState,messages:[...l.messages]},e=!0},n=()=>{if(!e)throw new Error("Message state adapter is not initialized");return{requestState:t.requestState,processingState:t.processingState,messages:[...t.messages],isProcessing:t.requestState==="processing"}},o=$(n);return{initialize:s,getState:n,createMessage(l){return l},mutate:(l,m)=>{if(!e)throw new Error("Message state adapter is not initialized");let d=!1;m(t,()=>{d=!0}),d||o.notify(l)},subscribe:o.subscribe}};var w=require("vue");var se=e=>(0,w.reactive)(e),Y=e=>{let t=(0,w.isProxy)(e)?(0,w.toRaw)(e):e;if(Array.isArray(t))return t.map(s=>Y(s));if(t&&typeof t=="object"){let s={};for(let[n,o]of Object.entries(t))s[n]=Y(o);return s}return t},ye=()=>{let e=!1,t=(0,w.ref)("idle"),s=(0,w.ref)(void 0),n=(0,w.ref)([]),o=(0,w.computed)(()=>t.value==="processing"),a=u=>{if(e)throw new Error("Message state adapter is already initialized");t.value=u.requestState,s.value=u.processingState,n.value=u.messages.map(se),e=!0},l=u=>se(u),m=()=>{if(!e)throw new Error("Message state adapter is not initialized");return{requestState:t.value,processingState:s.value,messages:Y(n.value),isProcessing:o.value}},d=$(m);return{requestState:t,processingState:s,messages:n,isProcessing:o,initialize:a,getState:m,createMessage:l,mutate:(u,y)=>{if(!e)throw new Error("Message state adapter is not initialized");let T={get requestState(){return t.value},set requestState(S){t.value=S},get processingState(){return s.value},set processingState(S){s.value=S},get messages(){return n.value},set messages(S){n.value=S.map(l)}},p=!1;if(y(T,()=>{p=!0}),p)return;(Array.isArray(u)?u:[u]).includes("messages")&&(n.value=[...n.value]),d.notify(u)},subscribe:d.subscribe}};var H=(e={})=>{let{continueContent:t="Please continue with your previous answer.",...s}=e;return{name:"length",...s,onAfterRequest:async n=>{let{lastChoice:o,appendMessage:a,requestNext:l}=n;return o?.finish_reason==="length"&&(a({role:"user",content:t}),l()),s.onAfterRequest?.(n)}}};var Q=(e={})=>{let t=(s,n)=>!!s.thinking!==n||!!s.open!==n;return{name:"thinking",...e,onCompletionChunk(s){let{choice:n,currentMessage:o,updateCurrentMessage:a}=s,l=n,m=l?.message?.reasoning_content||l?.delta?.reasoning_content,d=typeof m=="string"&&m.trim()!=="";return d?o.state&&typeof o.state=="object"?t(o.state,d)&&a(r=>{r.state.thinking=!0,r.state.open=!0}):a(r=>{r.state={thinking:d,open:d}}):o.state&&typeof o.state=="object"&&"thinking"in o.state&&t(o.state,d)&&a(r=>{r.state.thinking=!1,r.state.open=!1}),e.onCompletionChunk?.(s)},onTurnEnd(s){let{currentTurn:n,mutate:o}=s,a=n.at(-1);return a?.state&&typeof a.state=="object"&&"thinking"in a.state&&t(a.state,!1)&&o("messages",()=>{a.state.thinking=!1,a.state.open=!1}),e.onTurnEnd?.(s)}}};var N=class extends Error{constructor(t){super(t),this.name="AbortError"}};function Se(e){if(e.aborted)return{promise:Promise.reject(new N(String(e.reason??"Aborted"))),cleanup:()=>{}};let t=null;return{promise:new Promise((o,a)=>{t=()=>{a(new N(String(e.reason??"Aborted")))},e.addEventListener("abort",t,{once:!0})}),cleanup:()=>{t&&(e.removeEventListener("abort",t),t=null)}}}function ae(e,t){let{promise:s,cleanup:n}=Se(t);return Promise.race([e,s]).finally(n)}function re(e,t){let s={};for(let n in e)t.includes(n)&&(s[n]=e[n]);return s}function ie(e,t){let s={};for(let n in e)t.includes(n)||(s[n]=e[n]);return s}async function*O(e){if(ne(e)){yield*e;return}let t=await e;if(ne(t)){yield*t;return}yield t}function ne(e){return e&&typeof e=="object"&&typeof e[Symbol.asyncIterator]=="function"}var Z=e=>typeof e=="object"&&e!==null,oe=e=>Z(e)&&typeof e.index=="number",B=(e,t)=>{for(let[s,n]of Object.entries(t)){let o=e[s];if(o)if(typeof o=="string"&&typeof n=="string")s==="type"&&o||(e[s]=o+n);else if(Array.isArray(o)&&Array.isArray(n))if(o.every(a=>oe(a))&&n.every(a=>oe(a))){let a=new Map(o.map(r=>[r.index,r])),l=new Map(n.map(r=>[r.index,r]));for(let[r,u]of l)if(a.has(r)){let y=a.get(r);a.set(r,B(y,u))}else a.set(r,u);let m=Math.max(...Array.from(a.keys()),-1)+1,d=m>o.length?Array.from({length:m}):o;for(let[r,u]of a)d[r]=u;e[s]=d}else e[s]=[...o,...n];else Z(o)&&Z(n)&&(e[s]=B(o,n));else e[s]=n}return e};function be({messages:e,cancelledContent:t,createMessage:s,mutate:n}){let o=[];for(let a=0;a<e.length;a++){let l=e[a];if(l.role==="assistant"&&l.tool_calls&&l.tool_calls.length>0){let m=new Set(l.tool_calls.map(u=>u.id)),d=new Set;for(let u=a+1;u<e.length;u++){let y=e[u];y.role==="tool"&&y.tool_call_id&&m.has(y.tool_call_id)&&d.add(y.tool_call_id)}let r=l.tool_calls.map(u=>u.id).filter(u=>!d.has(u));r.length>0&&o.push({insertAfterIndex:a,missingToolCallIds:r})}}o.length!==0&&n("messages",a=>{for(let l=o.length-1;l>=0;l--){let{insertAfterIndex:m,missingToolCallIds:d}=o[l],r=d.map(u=>s({role:"tool",tool_call_id:u,content:t}));a.messages.splice(m+1,0,...r)}})}var le=e=>{let{getTools:t,beforeCallTools:s,callTool:n,onToolCallStart:o,onToolCallEnd:a,toolCallCancelledContent:l="Tool call cancelled.",toolCallFailedContent:m="Tool call failed.",autoFillMissingToolMessages:d=!1,...r}=e,u=(p,M)=>{var C,S;return p.state??(p.state={}),(C=p.state).toolCall??(C.toolCall={}),(S=p.state.toolCall)[M]??(S[M]={}),p},y=(...p)=>{let[M,{assistantMessage:C,mutate:S}]=p;S("messages",()=>{let R=u(C,M.id);R.state.toolCall[M.id].status="running"}),o?.(...p)},T=(...p)=>{let[M,{status:C,assistantMessage:S,mutate:R}]=p;R("messages",()=>{let W=u(S,M.id);W.state.toolCall[M.id].status=C}),a?.(...p)};return{name:"tool",...r,onTurnStart:p=>{let{getState:M,createMessage:C,mutate:S}=p,R=M().messages;return d&&be({messages:R,cancelledContent:l,createMessage:C,mutate:S}),r.onTurnStart?.(p)},onBeforeRequest:async p=>{let{requestBody:M}=p,C=await t();return C&&C.length>0&&(M.tools=C),r.onBeforeRequest?.(p)},onAfterRequest:async p=>{let{currentMessage:M,lastChoice:C,appendMessage:S,abortSignal:R,setRequestState:W,requestNext:v,mutate:F,createMessage:E}=p;if(C?.finish_reason!=="tool_calls"||!M.tool_calls?.length)return;W("processing","calling-tools"),await s?.(M.tool_calls,{...p,assistantMessage:M});let U=M.tool_calls.map(async j=>{let G=Math.floor(Date.now()/1e3),J=!1,x=E({role:"tool",tool_call_id:j.id,content:"",metadata:{createdAt:G,updatedAt:G}});S(x);let K={...p,assistantMessage:M,toolMessage:x};y(j,K);try{let i=n(j,K),c=O(i);for await(let f of c)F("messages",()=>{if((typeof f=="string"&&f.length>0||f&&typeof f=="object"&&Object.keys(f).length>0)&&(J=!0),typeof f=="string")x.content+=f;else{let h={};try{let b=Array.isArray(x.content)?x.content.map(g=>g.text).join(""):x.content;h=JSON.parse(b||"{}")}catch(b){console.warn(b)}x.content=JSON.stringify(B(h,f))}x.metadata.updatedAt=Math.floor(Date.now()/1e3)});T(j,{...K,status:"success"})}catch(i){let c=i instanceof Error?i:new Error(String(i));if(R.aborted){T(j,{...K,status:"cancelled",error:c});return}console.error(i),J||F("messages",()=>{x.content=m,x.metadata.updatedAt=Math.floor(Date.now()/1e3)}),T(j,{...K,status:"failed",error:c})}});return await Promise.all(U),R.aborted||v(),r.onAfterRequest?.(p)}}};var Pe=async()=>{throw new Error("Response provider is not set")},Ae=e=>{let t=[];for(let s of e){if(s.name){let n=t.findIndex(o=>o.name===s.name);n!==-1&&t.splice(n,1)}t.push(s)}return t},z=(e,t)=>typeof e.disabled=="function"?e.disabled(t):!!e.disabled,ce=(e,t={})=>{let{initialMessages:s=[],requestMessageFields:n=[],requestMessageFieldsExclude:o=["state","metadata","loading"],responseProvider:a=Pe,onCompletionChunk:l,plugins:m=[]}=t,d={requestState:"idle",processingState:void 0,messages:[...s]};e.initialize(d);let r={currentTurn:[],customContext:{},abortController:null,responseProvider:a},u=[Q(),H()],y=Ae(u.concat(m)),T=()=>e.getState(),p=i=>e.createMessage(i),M=e.subscribe,C=e.mutate,S=i=>!i||Object.keys(i).length===0?!1:Object.values(i).some(c=>!!c),R=i=>{let c=i;return n.length&&(c=c.map(f=>re(f,n))),o.length&&(c=c.map(f=>ie(f,o))),c},W=i=>{Object.assign(r.customContext,i)},v=(i,c)=>{C("requestState",(f,h)=>{if(f.requestState===i&&f.processingState===c){h();return}f.requestState=i,f.processingState=i==="processing"?c??"requesting":void 0})},F=(...i)=>{let c=i.map(f=>p(f));return C("messages",f=>{f.messages.push(...c)}),r.currentTurn.push(...c),c},E=i=>({getState:T,createMessage:p,mutate:C,abortSignal:i,currentTurn:r.currentTurn,customContext:r.customContext,setRequestState:v,setCustomContext:W});async function U(i,c,f={}){v("processing","requesting");let h={messages:T().messages},b=E(c);for(let _ of y.filter(A=>!z(A,b)))await _.onBeforeRequest?.({...b,requestBody:h});h.messages=R(h.messages);let g={role:"assistant",content:"",loading:!0};[g]=F(g),f.setAssistantMessage?.(g);let P=i(h,c),k=O(P),I;for await(let _ of k){v("processing","completing"),C("messages",(q,V)=>{g.loading?g.loading=void 0:V()});let A=(_.choices||[]).find(q=>q.index===0)??_.choices?.[0];if(!A)continue;I=A;let D=()=>{C("messages",()=>{g.metadata||(g.metadata={});let{created:q,...V}=_;g.metadata.createdAt=q,g.metadata.updatedAt=Math.floor(Date.now()/1e3),Object.assign(g.metadata,V);let L="delta"in A&&S(A.delta)&&A.delta||"message"in A&&S(A.message)&&A.message||null;if(L?.role&&(g.role=L.role),L){let{role:xe,...ue}=L;B(g,ue)}})},ee=q=>{C("messages",()=>{q(g)})};if(l){let q=E(c);l({...q,chunk:_,choice:A,currentMessage:g,updateCurrentMessage:ee},D)}else D();let te=E(c);for(let q of y.filter(V=>!z(V,te)))q.onCompletionChunk?.({...te,abortSignal:c,chunk:_,choice:A,currentMessage:g,updateCurrentMessage:ee})}await j(g,i,c,I,f)}async function j(i,c,f,h,b){let g=!1,P=E(f),k=y.filter(I=>!z(I,P)).map(I=>{if(!I.onAfterRequest)return null;let _=D=>{F(...Array.isArray(D)?D:[D])},A=()=>{g=!0};return I.onAfterRequest({...P,currentMessage:i,lastChoice:h,appendMessage:_,requestNext:A})}).filter(I=>I!==null);await ae(Promise.all(k),f),g&&await U(c,f,b)}async function G(){let i=new AbortController;r.abortController=i,r.customContext={};let c=null,f=h=>{c=h};try{v("processing","requesting");let h=E(i.signal);for(let P of y.filter(k=>!z(k,h)))await P.onTurnStart?.(h);let b=r.responseProvider;try{await U(b,i.signal,{setAssistantMessage:f}),v("completed")}catch(P){if(i.signal.aborted||P instanceof N||P instanceof Error&&P.name==="AbortError")v("aborted");else throw P}let g=E(i.signal);for(let P of y.filter(k=>!z(k,g)))await P.onTurnEnd?.(g)}catch(h){v("error");let b=!1,g=E(i.signal);for(let P of y.filter(k=>!z(k,g)))P.onError&&(b=!0,P.onError({...g,error:h}));if(!b)throw h}finally{let h=E(i.signal);for(let b of y.filter(g=>!z(g,h)))try{b.onFinally?.(h)}catch(g){console.error(`Error in onFinally hook for plugin [${b.name||"Anonymous"}]:`,g)}r.abortController=null,r.currentTurn=[],C("messages",(b,g)=>{c?.loading?c.loading=void 0:g()})}}async function J(i){if(!i||!i.trim()){console.warn("Cannot send empty message");return}if(T().requestState==="processing"){console.warn("Cannot send message while processing is in progress");return}let c=Math.floor(Date.now()/1e3);F({role:"user",content:i.trim(),metadata:{createdAt:c,updatedAt:c}}),await G()}async function x(...i){if(T().requestState==="processing"){console.warn("Cannot send message while processing is in progress");return}F(...i),await G()}async function K(){r.abortController?.abort(),T().isProcessing&&await new Promise(i=>{let c=()=>{};c=M("requestState",f=>{f.isProcessing||(c(),i())})})}return{getState:T,subscribe:M,sendMessage:J,send:x,abort:K,setResponseProvider(i){r.responseProvider=i}}};0&&(module.exports={combineDeltaData,createMessageEngine,createNativeMessageAdapter,createVueMessageAdapter,lengthPlugin,normalizeToAsyncGenerator,thinkingPlugin,toolPlugin});

package/dist/core.mjs

@@ -0,0 +1 @@
+import{a as o,b as f,c as l,d as m,e as p,f as c,g as M,h as d}from"./chunk-EES4JUCD.mjs";var w=()=>{let s=!1,t,n=e=>{if(s)throw new Error("Message state adapter is already initialized");t={requestState:e.requestState,processingState:e.processingState,messages:[...e.messages]},s=!0},a=()=>{if(!s)throw new Error("Message state adapter is not initialized");return{requestState:t.requestState,processingState:t.processingState,messages:[...t.messages],isProcessing:t.requestState==="processing"}},r=o(a);return{initialize:n,getState:a,createMessage(e){return e},mutate:(e,g)=>{if(!s)throw new Error("Message state adapter is not initialized");let i=!1;g(t,()=>{i=!0}),i||r.notify(e)},subscribe:r.subscribe}};export{c as combineDeltaData,d as createMessageEngine,w as createNativeMessageAdapter,f as createVueMessageAdapter,l as lengthPlugin,p as normalizeToAsyncGenerator,m as thinkingPlugin,M as toolPlugin};

package/dist/index.d.mts

@@ -1,216 +1,7 @@
+import { a as AIModelConfig, c as ChatCompletionRequest, d as ChatCompletionResponse, o as StreamHandler, B as BaseModelProvider, l as ChatMessage, M as MaybePromise, T as ToolCall } from './types-CePh-Jcx.mjs';
+export { A as AIAdapterError, b as AIProvider, C as ChatCompletionOptions, e as ChatCompletionResponseChoice, f as ChatCompletionResponseMessage, g as ChatCompletionResponseUsage, h as ChatCompletionStreamResponse, i as ChatCompletionStreamResponseChoice, j as ChatCompletionStreamResponseDelta, k as ChatHistory, E as ErrorType, m as MessageMetadata, n as MessageRole, S as StreamEventType } from './types-CePh-Jcx.mjs';
 import { Ref, ComputedRef } from 'vue';
 
-/**
- * 模型Provider基类
- */
-declare abstract class BaseModelProvider {
-    protected config: AIModelConfig;
-    /**
-     * @param config AI模型配置
-     */
-    constructor(config: AIModelConfig);
-    /**
-     * 发送聊天请求并获取响应
-     * @param request 聊天请求参数
-     * @returns 聊天响应
-     */
-    abstract chat(request: ChatCompletionRequest): Promise<ChatCompletionResponse>;
-    /**
-     * 发送流式聊天请求并通过处理器处理响应
-     * @param request 聊天请求参数
-     * @param handler 流式响应处理器
-     */
-    abstract chatStream(request: ChatCompletionRequest, handler: StreamHandler): Promise<void>;
-    /**
-     * 更新配置
-     * @param config 新的AI模型配置
-     */
-    updateConfig(config: AIModelConfig): void;
-    /**
-     * 获取当前配置
-     * @returns AI模型配置
-     */
-    getConfig(): AIModelConfig;
-    /**
-     * 验证请求参数
-     * @param request 聊天请求参数
-     */
-    protected validateRequest(request: ChatCompletionRequest): void;
-}
-
-type MaybePromise<T> = T | Promise<T>;
-/**
- * 消息角色类型
- */
-type MessageRole = 'system' | 'user' | 'assistant';
-interface ToolCall {
-    index: number;
-    id: string;
-    type: 'function';
-    function: {
-        name: string;
-        arguments: string;
-        result?: string;
-    };
-}
-interface MessageMetadata {
-    createdAt?: number;
-    updatedAt?: number;
-    id?: string;
-    model?: string;
-    [key: string]: any;
-}
-/**
- * 聊天消息接口
- */
-interface ChatMessage {
-    role: string;
-    content: string;
-    reasoning_content?: string;
-    metadata?: MessageMetadata;
-    tool_calls?: ToolCall[];
-    tool_call_id?: string;
-    [key: string]: any;
-    [key: symbol]: any;
-}
-/**
- * 聊天历史记录
- */
-type ChatHistory = ChatMessage[];
-/**
- * 聊天完成请求选项
- */
-interface ChatCompletionOptions {
-    model?: string;
-    temperature?: number;
-    top_p?: number;
-    n?: number;
-    stream?: boolean;
-    max_tokens?: number;
-    signal?: AbortSignal;
-}
-/**
- * 聊天完成请求参数
- */
-interface ChatCompletionRequest {
-    messages: ChatMessage[];
-    options?: ChatCompletionOptions;
-}
-/**
- * 聊天完成响应消息
- */
-interface ChatCompletionResponseMessage {
-    role: MessageRole;
-    content: string;
-    [x: string]: unknown;
-}
-/**
- * 聊天完成响应选择
- */
-interface ChatCompletionResponseChoice {
-    index: number;
-    message: ChatCompletionResponseMessage;
-    finish_reason: string;
-}
-/**
- * 聊天完成响应使用情况
- */
-interface ChatCompletionResponseUsage {
-    prompt_tokens: number;
-    completion_tokens: number;
-    total_tokens: number;
-}
-/**
- * 聊天完成响应
- */
-interface ChatCompletionResponse {
-    id: string;
-    object: string;
-    created: number;
-    model: string;
-    choices: ChatCompletionResponseChoice[];
-    usage: ChatCompletionResponseUsage;
-}
-/**
- * 流式聊天完成响应增量
- */
-interface ChatCompletionStreamResponseDelta {
-    content?: string;
-    role?: MessageRole;
-    [x: string]: unknown;
-}
-/**
- * 流式聊天完成响应选择
- */
-interface ChatCompletionStreamResponseChoice {
-    index: number;
-    delta: ChatCompletionStreamResponseDelta;
-    finish_reason: string | null;
-}
-/**
- * 流式聊天完成响应
- */
-interface ChatCompletionStreamResponse {
-    id: string;
-    object: string;
-    created: number;
-    model: string;
-    choices: ChatCompletionStreamResponseChoice[];
-}
-/**
- * AI模型提供商类型
- */
-type AIProvider = 'openai' | 'deepseek' | 'custom';
-/**
- * AI模型配置接口
- */
-interface AIModelConfig {
-    provider: AIProvider;
-    providerImplementation?: BaseModelProvider;
-    apiKey?: string;
-    apiUrl?: string;
-    apiVersion?: string;
-    defaultModel?: string;
-    defaultOptions?: ChatCompletionOptions;
-}
-/**
- * 错误类型
- */
-declare enum ErrorType {
-    NETWORK_ERROR = "network_error",
-    AUTHENTICATION_ERROR = "authentication_error",
-    RATE_LIMIT_ERROR = "rate_limit_error",
-    SERVER_ERROR = "server_error",
-    MODEL_ERROR = "model_error",
-    TIMEOUT_ERROR = "timeout_error",
-    UNKNOWN_ERROR = "unknown_error"
-}
-/**
- * AI适配器错误
- */
-interface AIAdapterError {
-    type: ErrorType;
-    message: string;
-    statusCode?: number;
-    originalError?: object;
-}
-/**
- * 流式响应事件类型
- */
-declare enum StreamEventType {
-    DATA = "data",
-    ERROR = "error",
-    DONE = "done"
-}
-/**
- * 流式响应处理器
- */
-interface StreamHandler {
-    onData: (data: ChatCompletionStreamResponse) => void;
-    onError: (error: AIAdapterError) => void;
-    onDone: (finishReason?: string) => void;
-}
-
 /**
  * AI客户端类
  * 负责根据配置选择合适的提供商并处理请求
@@ -294,11 +85,12 @@ interface Tool {
     type: 'function';
     function: {
         name: string;
-        description: string;
+        description?: string;
         /**
          * function 的输入参数，以 JSON Schema 对象描述
          */
-        parameters: any;
+        parameters?: any;
+        inputSchema?: any;
     };
     [key: string]: any;
 }
@@ -352,6 +144,7 @@ interface ChatCompletion {
     choices: CompletionChoice[];
     usage?: Usage;
 }
+type ResponseProvider<T = ChatCompletion> = (requestBody: MessageRequestBody, abortSignal: AbortSignal) => Promise<T> | AsyncGenerator<T> | Promise<AsyncGenerator<T>>;
 interface UseMessageOptions {
     initialMessages?: ChatMessage[];
     /**
@@ -365,7 +158,7 @@ interface UseMessageOptions {
      */
     requestMessageFieldsExclude?: string[];
     plugins?: UseMessagePlugin[];
-    responseProvider: <T = ChatCompletion>(requestBody: MessageRequestBody, abortSignal: AbortSignal) => Promise<T> | AsyncGenerator<T> | Promise<AsyncGenerator<T>>;
+    responseProvider: ResponseProvider;
     /**
      * 全局的数据块处理钩子，在接收到每个响应数据块时触发。
      * 注意：此钩子与插件中的 onCompletionChunk 有区别。
@@ -663,21 +456,30 @@ declare function sseStreamToGenerator<T = any>(response: Response, options?: {
 
 declare const useConversation: (options: UseConversationOptions) => UseConversationReturn;
 
-declare const fallbackRolePlugin: (options?: UseMessagePlugin & {
-    fallbackRole?: string;
-}) => UseMessagePlugin;
-
 declare const lengthPlugin: (options?: UseMessagePlugin & {
     continueContent?: string;
 }) => UseMessagePlugin;
 
 declare const thinkingPlugin: (options?: UseMessagePlugin) => UseMessagePlugin;
 
-/**
- * 消息排除模式：从 messages 数组中直接移除消息。
- * 使用此模式时，消息会被完全从数组中移除，不会保留在 messages 中。
- */
-declare const EXCLUDE_MODE_REMOVE: "remove";
+interface UseMessageToolActionContext extends BasePluginContext {
+    assistantMessage: ChatMessage;
+    /**
+     * @deprecated use `assistantMessage` instead
+     */
+    currentMessage: ChatMessage;
+}
+interface UseMessageCallToolContext extends UseMessageToolActionContext {
+    toolMessage: ChatMessage;
+}
+interface UseMessageToolCallContext extends BasePluginContext {
+    assistantMessage: ChatMessage;
+    /**
+     * @deprecated use `assistantMessage` instead
+     */
+    primaryMessage: ChatMessage;
+    toolMessage: ChatMessage;
+}
 declare const toolPlugin: (options: UseMessagePlugin & {
     /**
      * 获取工具列表的函数。
@@ -686,25 +488,18 @@ declare const toolPlugin: (options: UseMessagePlugin & {
     /**
      * 在处理包含 tool_calls 的响应前调用。
      */
-    beforeCallTools?: (toolCalls: ToolCall[], context: BasePluginContext & {
-        currentMessage: ChatMessage;
-    }) => Promise<void>;
+    beforeCallTools?: (toolCalls: ToolCall[], context: UseMessageToolActionContext) => Promise<void>;
     /**
      * 执行单个工具调用并返回其文本结果的函数。
      */
-    callTool: (toolCall: ToolCall, context: BasePluginContext & {
-        currentMessage: ChatMessage;
-    }) => Promise<string | Record<string, any>> | AsyncGenerator<string | Record<string, any>>;
+    callTool: (toolCall: ToolCall, context: UseMessageCallToolContext) => Promise<string | Record<string, any>> | AsyncGenerator<string | Record<string, any>>;
     /**
      * 工具调用开始时的回调函数。
      * 触发时机：工具消息已创建并追加后，调用 callTool 之前触发。
      * @param toolCall - 工具调用对象
      * @param context - 插件上下文，包含当前工具消息
      */
-    onToolCallStart?: (toolCall: ToolCall, context: BasePluginContext & {
-        primaryMessage: ChatMessage;
-        toolMessage: ChatMessage;
-    }) => void;
+    onToolCallStart?: (toolCall: ToolCall, context: UseMessageToolCallContext) => void;
     /**
      * 工具调用结束时的回调函数。
      * 触发时机：工具调用完成（成功、失败或取消）时触发。
@@ -713,9 +508,7 @@ declare const toolPlugin: (options: UseMessagePlugin & {
      * @param context.status - 工具调用状态：'success' | 'failed' | 'cancelled'
      * @param context.error - 当状态为 'failed' 或 'cancelled' 时，可能包含错误信息
      */
-    onToolCallEnd?: (toolCall: ToolCall, context: BasePluginContext & {
-        primaryMessage: ChatMessage;
-        toolMessage: ChatMessage;
+    onToolCallEnd?: (toolCall: ToolCall, context: UseMessageToolCallContext & {
         status: "success" | "failed" | "cancelled";
         error?: Error;
     }) => void;
@@ -728,20 +521,13 @@ declare const toolPlugin: (options: UseMessagePlugin & {
      */
     toolCallFailedContent?: string;
     /**
-     * 是否在请求被中止时自动补充缺失的 tool 消息。
+     * 是否在请求前自动补充缺失的 tool 消息。
      * 当 assistant 响应了 tool_calls 但未追加对应的 tool 消息时，
      * 插件将自动补充"工具调用已取消"的 tool 消息。默认：false。
      */
     autoFillMissingToolMessages?: boolean;
-    /**
-     * 是否在下一轮对话中，排除包含 tool_calls 的 assistant 消息和对应的 tool 消息，只保留结果。
-     * - 当为 `true` 时，这些消息会被标记为不发送，不会包含在下一次请求的 messages 中。
-     * - 当为 `'remove'` 时，这些消息会直接从 messages 数组中移除。
-     * 默认：false。
-     */
-    excludeToolMessagesNextTurn?: boolean | typeof EXCLUDE_MODE_REMOVE;
 }) => UseMessagePlugin;
 
 declare const useMessage: (options: UseMessageOptions) => UseMessageReturn;
 
-export { type AIAdapterError, AIClient, type AIModelConfig, type AIProvider, BaseModelProvider, type BasePluginContext, type ChatCompletion, type ChatCompletionOptions, type ChatCompletionRequest, type ChatCompletionResponse, type ChatCompletionResponseChoice, type ChatCompletionResponseMessage, type ChatCompletionResponseUsage, type ChatCompletionStreamResponse, type ChatCompletionStreamResponseChoice, type ChatCompletionStreamResponseDelta, type ChatHistory, type ChatMessage, type Choice, type CompletionChoice, type Conversation, type ConversationInfo, type ConversationStorageStrategy, type DeltaChoice, EXCLUDE_MODE_REMOVE, ErrorType, type IndexedDBConfig, IndexedDBStrategy, type LocalStorageConfig, LocalStorageStrategy, type MaybePromise, type MessageMetadata, type MessageRequestBody, type MessageRole, OpenAIProvider, type RequestProcessingState, type RequestState, StreamEventType, type StreamHandler, type Tool, type ToolCall, type Usage, type UseConversationOptions, type UseConversationReturn, type UseMessageOptions, type UseMessagePlugin, type UseMessageReturn, extractTextFromResponse, fallbackRolePlugin, formatMessages, handleSSEStream, indexedDBStorageStrategyFactory, lengthPlugin, localStorageStrategyFactory, sseStreamToGenerator, thinkingPlugin, toolPlugin, useConversation, useMessage };
+export { AIClient, AIModelConfig, BaseModelProvider, type BasePluginContext, type ChatCompletion, ChatCompletionRequest, ChatCompletionResponse, ChatMessage, type Choice, type CompletionChoice, type Conversation, type ConversationInfo, type ConversationStorageStrategy, type DeltaChoice, type IndexedDBConfig, IndexedDBStrategy, type LocalStorageConfig, LocalStorageStrategy, MaybePromise, type MessageRequestBody, OpenAIProvider, type RequestProcessingState, type RequestState, type ResponseProvider, StreamHandler, type Tool, ToolCall, type Usage, type UseConversationOptions, type UseConversationReturn, type UseMessageCallToolContext, type UseMessageOptions, type UseMessagePlugin, type UseMessageReturn, type UseMessageToolActionContext, type UseMessageToolCallContext, extractTextFromResponse, formatMessages, handleSSEStream, indexedDBStorageStrategyFactory, lengthPlugin, localStorageStrategyFactory, sseStreamToGenerator, thinkingPlugin, toolPlugin, useConversation, useMessage };