@opentiny/tiny-robot-kit 0.3.1-alpha.4 → 0.4.0-alpha.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.d.mts +303 -416
- package/dist/index.d.ts +303 -416
- package/dist/index.js +2 -3
- package/dist/index.mjs +2 -3
- package/package.json +2 -2
package/dist/index.js
CHANGED
|
@@ -1,3 +1,2 @@
|
|
|
1
|
-
"use strict";var
|
|
2
|
-
|
|
3
|
-
`);a=A.pop()||"";for(let f of A)if(f.trim()!==""){if(f.trim()==="data: [DONE]"){d&&(c=d),e.onDone(c);continue}try{let R=f.match(/^data: (.+)$/m);if(!R)continue;let h=JSON.parse(R[1]);e.onData(h),d=h.choices?.[0]?.finish_reason||void 0}catch(R){console.error("Error parsing SSE message:",R)}}}(a.trim()==="data: [DONE]"||t?.aborted)&&(t?.aborted&&(c="aborted"),e.onDone(c))}catch(o){if(t?.aborted)return;throw o}}function H(r){return r.map(e=>typeof e=="object"&&"role"in e&&"content"in e?{role:e.role,content:String(e.content),...e.name?{name:e.name}:{}}:typeof e=="string"?{role:"user",content:e}:{role:"user",content:String(e)})}function j(r){return!r.choices||!r.choices.length?"":r.choices[0].message?.content||""}var S=class extends E{constructor(t){super(t);this.defaultModel="gpt-3.5-turbo";this.baseURL=t.apiUrl||"https://api.openai.com/v1",this.apiKey=t.apiKey||"",t.defaultModel&&(this.defaultModel=t.defaultModel),this.apiKey||console.warn("API key is not provided. Authentication will likely fail.")}async chat(t){try{this.validateRequest(t);let i={model:t.options?.model||this.config.defaultModel||this.defaultModel,messages:t.messages,...t.options,stream:!1},u={method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(i)};this.apiKey&&Object.assign(u.headers,{Authorization:`Bearer ${this.apiKey}`});let a=await fetch(`${this.baseURL}/chat/completions`,u);if(!a.ok){let c=await a.text();throw new Error(`HTTP error! status: ${a.status}, details: ${c}`)}return await a.json()}catch(i){throw D(i)}}async chatStream(t,i){let{signal:u,...a}=t.options||{};try{this.validateRequest(t);let c={model:t.options?.model||this.config.defaultModel||this.defaultModel,messages:t.messages,...a,stream:!0},d={method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${this.apiKey}`,Accept:"text/event-stream"},body:JSON.stringify(c),signal:u};this.apiKey&&Object.assign(d.headers,{Authorization:`Bearer ${this.apiKey}`});let o=await fetch(`${this.baseURL}/chat/completions`,d);if(!o.ok){let p=await o.text();throw new Error(`HTTP error! status: ${o.status}, details: ${p}`)}await I(o,i,u)}catch(c){if(u?.aborted)return;i.onError(D(c))}}updateConfig(t){super.updateConfig(t),t.apiUrl&&(this.baseURL=t.apiUrl),t.apiKey&&(this.apiKey=t.apiKey),t.defaultModel&&(this.defaultModel=t.defaultModel)}};var b=class{constructor(e){this.config=e,this.provider=this.createProvider(e)}createProvider(e){if(e.provider==="custom"&&"providerImplementation"in e)return e.providerImplementation;switch(e.provider){case"deepseek":let t={defaultModel:"deepseek-chat",apiUrl:"https://api.deepseek.com/v1"};return new S({...t,...e});case"openai":default:return new S(e)}}async chat(e){return this.provider.chat(e)}async chatStream(e,t){let i={...e,options:{...e.options,stream:!0}};return this.provider.chatStream(i,t)}getConfig(){return{...this.config}}updateConfig(e){this.config={...this.config,...e},e.provider&&e.provider!==this.config.provider?this.provider=this.createProvider(this.config):this.provider.updateConfig(this.config)}};var v=require("vue"),F=(c=>(c.INIT="init",c.PROCESSING="processing",c.STREAMING="streaming",c.FINISHED="finished",c.ABORTED="aborted",c.ERROR="error",c))(F||{}),P=["processing","streaming"],V=["finished","aborted","error"];function U(r){let{client:e,useStreamByDefault:t=!0,errorMessage:i="\u8BF7\u6C42\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5",initialMessages:u=[]}=r,a=(0,v.ref)([...u]),c=(0,v.ref)(""),d=(0,v.ref)(t),o=null,p=(0,v.reactive)({status:"init",errorMsg:null}),m=s=>{let n=r.events?.onReceiveData,g=!1;if(n&&n(s,a,()=>{g=!0}),!g){let C={role:"assistant",content:s.choices[0].message.content};a.value.push(C)}},A=async s=>{let n=await e.chat({messages:(0,v.toRaw)(a.value),options:{stream:!1,signal:s.signal}});m(n)},f=s=>{let n=r.events?.onReceiveData,g=!1;if(n&&n(s,a,()=>{g=!0}),!g){a.value[a.value.length-1].role==="user"&&a.value.push({role:"assistant",content:""});let C=s.choices?.[0];C&&C.delta.content&&(a.value[a.value.length-1].content+=C.delta.content)}},R=async s=>{await e.chatStream({messages:(0,v.toRaw)(a.value),options:{stream:!0,signal:s.signal}},{onData:n=>{p.status="streaming",f(n)},onError:n=>{p.status="error",p.errorMsg=i,console.error("Stream request error:",n)},onDone:n=>{let g=r.events?.onFinish,C=!1;if(g&&g(n,{messages:a,messageState:p},()=>{C=!0}),!C){if(n==="aborted"||p.status==="aborted")return;p.status="finished"}}})},h=async()=>{p.status="processing",p.errorMsg=null,o=new AbortController;try{d.value?await R(o):await A(o)}catch(s){p.errorMsg=i,p.status="error",console.error("Send message error:",s)}finally{o=null}};return{messages:a,messageState:p,inputMessage:c,useStream:d,sendMessage:async(s=c.value,n=!0)=>{if(P.includes(p.status)||!s||typeof s=="string"&&!s.trim()||Array.isArray(s)&&s.length===0)return;let g={role:"user",content:s};a.value.push(g),n&&(c.value=""),await h()},send:async()=>{P.includes(p.status)||await h()},clearMessages:()=>{a.value=[],p.errorMsg=null},addMessage:s=>{Array.isArray(s)?a.value.push(...s):a.value.push(s)},abortRequest:()=>{o&&(o.abort(),o=null,p.status="aborted")},retryRequest:async s=>{s===0||!a.value[s]||a.value[s].role==="user"||(a.value.splice(s),await h())}}}var O=require("vue");var x=class{constructor(e="tiny-robot-ai-conversations"){this.storageKey=e}saveConversations(e){try{localStorage.setItem(this.storageKey,JSON.stringify(e))}catch(t){console.error("\u4FDD\u5B58\u4F1A\u8BDD\u5931\u8D25:",t)}}loadConversations(){try{let e=localStorage.getItem(this.storageKey);return e?JSON.parse(e):[]}catch(e){return console.error("\u52A0\u8F7D\u4F1A\u8BDD\u5931\u8D25:",e),[]}}};function Q(){return Date.now().toString(36)+Math.random().toString(36).substring(2,9)}function X(r){let{client:e,storage:t=new x,autoSave:i=!0,allowEmpty:u=!1,useStreamByDefault:a=!0,errorMessage:c="\u8BF7\u6C42\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5",events:d}=r,o=(0,O.reactive)({conversations:[],currentId:null,loading:!1}),p=!1,m=U({client:e,useStreamByDefault:a,errorMessage:c,initialMessages:[],events:{onReceiveData:d?.onReceiveData,onFinish:d?.onFinish}});(0,O.watch)(()=>m.messages.value,l=>{if(o.currentId&&l.length>0){let s=o.conversations.findIndex(n=>n.id===o.currentId);s!==-1&&(o.conversations[s].messages=[...l],o.conversations[s].updatedAt=Date.now(),i&&M())}},{deep:!0});let A=(l="\u65B0\u4F1A\u8BDD",s={})=>{if(!u&&m.messages.value.length===0&&o.currentId)return o.currentId;let n=Q(),g={id:n,title:l,createdAt:Date.now(),updatedAt:Date.now(),messages:[],metadata:s};return o.conversations.unshift(g),f(n),i&&M(),n},f=l=>{let s=o.conversations.find(n=>n.id===l);s&&(o.currentId=l,m.clearMessages(),s.messages.length>0&&s.messages.forEach(n=>m.addMessage(n)))},R=l=>{let s=o.conversations.findIndex(n=>n.id===l);s!==-1&&(o.conversations.splice(s,1),o.currentId===l&&(o.conversations.length>0?f(o.conversations[0].id):(o.currentId=null,m.clearMessages())),i&&M())},h=(l,s)=>{let n=o.conversations.find(g=>g.id===l);n&&(n.title=s,n.updatedAt=Date.now(),i&&M())},N=(l,s)=>{let n=o.conversations.find(g=>g.id===l);n&&(n.metadata={...n.metadata,...s},n.updatedAt=Date.now(),i&&M())},M=async()=>{try{await t.saveConversations(o.conversations)}catch(l){console.error("\u4FDD\u5B58\u4F1A\u8BDD\u5931\u8D25:",l)}},w=async()=>{o.loading=!0;try{let l=await t.loadConversations();o.conversations=l,l.length>0&&!o.currentId&&f(l[0].id),!p&&d?.onLoaded&&(p=!0,d.onLoaded(l))}catch(l){console.error("\u52A0\u8F7D\u4F1A\u8BDD\u5931\u8D25:",l)}finally{o.loading=!1}},_=async l=>{let s=o.conversations.find(n=>n.id===l);if(!s||s.messages.length<2)return s?.title||"\u65B0\u4F1A\u8BDD";try{let n={role:"system",content:"\u8BF7\u6839\u636E\u4EE5\u4E0B\u5BF9\u8BDD\u5185\u5BB9\uFF0C\u751F\u6210\u4E00\u4E2A\u7B80\u77ED\u7684\u6807\u9898\uFF08\u4E0D\u8D85\u8FC720\u4E2A\u5B57\u7B26\uFF09\u3002\u53EA\u9700\u8981\u8FD4\u56DE\u6807\u9898\u6587\u672C\uFF0C\u4E0D\u9700\u8981\u4EFB\u4F55\u89E3\u91CA\u6216\u989D\u5916\u5185\u5BB9\u3002"},g=s.messages.slice(0,Math.min(4,s.messages.length)),K=(await e.chat({messages:[n,...g],options:{stream:!1,max_tokens:30}})).choices[0].message.content.trim();return h(l,K),K}catch(n){return console.error("\u751F\u6210\u6807\u9898\u5931\u8D25:",n),s.title}},k=()=>o.currentId&&o.conversations.find(l=>l.id===o.currentId)||null;return w(),{state:o,messageManager:m,createConversation:A,switchConversation:f,deleteConversation:R,updateTitle:h,updateMetadata:N,saveConversations:M,loadConversations:w,generateTitle:_,getCurrentConversation:k}}0&&(module.exports={AIClient,BaseModelProvider,ErrorType,FinalStatus,GeneratingStatus,LocalStorageStrategy,OpenAIProvider,STATUS,StreamEventType,extractTextFromResponse,formatMessages,handleSSEStream,useConversation,useMessage});
|
|
1
|
+
"use strict";var K=Object.defineProperty;var le=Object.getOwnPropertyDescriptor;var ce=Object.getOwnPropertyNames;var ue=Object.prototype.hasOwnProperty;var ge=(e,t)=>{for(var n in t)K(e,n,{get:t[n],enumerable:!0})},fe=(e,t,n,s)=>{if(t&&typeof t=="object"||typeof t=="function")for(let o of ce(t))!ue.call(e,o)&&o!==n&&K(e,o,{get:()=>t[o],enumerable:!(s=le(t,o))||s.enumerable});return e};var de=e=>fe(K({},"__esModule",{value:!0}),e);var Te={};ge(Te,{EXCLUDE_MODE_REMOVE:()=>oe,createSSEStreamGenerator:()=>Y,fallbackRolePlugin:()=>J,lengthPlugin:()=>X,thinkingPlugin:()=>$,toolPlugin:()=>ye,useConversation:()=>be,useMessage:()=>L});module.exports=de(Te);function z(e="The operation was aborted"){let t=new Error(e);return t.name="AbortError",t}async function*Y(e,t={}){let n=e.body?.getReader();if(!n)throw new Error("ReadableStream not supported");let{signal:s}=t,o=new TextDecoder,a="",i=()=>{n.cancel()};s?.addEventListener("abort",i);try{for(;;){if(s?.aborted)throw z();let u;try{u=await n.read()}catch(w){throw s?.aborted?z():w}let{done:y,value:d}=u;if(y){if(s?.aborted)throw z();return}let M=o.decode(d,{stream:!0});a+=M;let A=a.split(`
|
|
2
|
+
`);a=A.pop()||"";for(let w of A)if(w.trim()!==""&&w.startsWith("data: ")){let v=w.slice(6);if(v==="[DONE]")return;try{yield JSON.parse(v)}catch(T){console.warn("Failed to parse SSE data:",v,T)}}}}finally{s?.removeEventListener("abort",i),n.releaseLock()}}var F=require("vue");var q=require("vue");var J=(e={})=>{let{fallbackRole:t="assistant",...n}=e;return{name:"fallbackRole",...n,onBeforeRequest(s){let{requestBody:o,messages:a}=s;return o.messages=a.map(i=>({...i,role:i.role||t})),n.onBeforeRequest?.(s)}}};var X=(e={})=>{let{continueContent:t="Please continue with your previous answer.",...n}=e;return{name:"length",...n,onAfterRequest:async s=>{let{lastChoice:o,appendMessage:a,requestNext:i}=s;return o?.finish_reason==="length"&&(a({role:"user",content:t}),i()),n.onAfterRequest?.(s)}}};var $=(e={})=>({name:"thinking",...e,onCompletionChunk(t){let{choice:n,currentMessage:s}=t,a=typeof(n?.message?.reasoning_content||n?.delta?.reasoning_content)=="string";return s.state?s.state.thinking=a:s.state={thinking:a},e.onCompletionChunk?.(t)},onTurnEnd(t){let n=t.currentTurn.slice(-1)[0];return n?.state&&(n.state.thinking=void 0),e.onTurnEnd?.(t)}});var se=require("vue");var B=class extends Error{constructor(t){super(t),this.name="AbortError"}};function pe(e){if(e.aborted)return{promise:Promise.reject(new B(String(e.reason??"Aborted"))),cleanup:()=>{}};let t=null;return{promise:new Promise((o,a)=>{t=()=>{a(new B(String(e.reason??"Aborted")))},e.addEventListener("abort",t,{once:!0})}),cleanup:()=>{t&&(e.removeEventListener("abort",t),t=null)}}}function te(e,t){let{promise:n,cleanup:s}=pe(t);return Promise.race([e,n]).finally(s)}function ne(e,t){let n={};for(let s of t)s in e&&(n[s]=e[s]);return n}async function*W(e){if(Z(e)){yield*e;return}let t=await e;if(Z(t)){yield*t;return}yield t}function Z(e){return e&&typeof e=="object"&&typeof e[Symbol.asyncIterator]=="function"}var Q=e=>typeof e=="object"&&e!==null,ee=e=>Q(e)&&typeof e.index=="number",G=(e,t)=>{for(let[n,s]of Object.entries(t)){let o=e[n];if(o)if(typeof o=="string"&&typeof s=="string")n==="type"&&o||(e[n]=o+s);else if(Array.isArray(o)&&Array.isArray(s))if(o.every(a=>ee(a))&&s.every(a=>ee(a))){let a=new Map(o.map(d=>[d.index,d])),i=new Map(s.map(d=>[d.index,d]));for(let[d,M]of i)if(a.has(d)){let A=a.get(d);a.set(d,G(A,M))}else a.set(d,M);let u=Math.max(...Array.from(a.keys()),-1)+1,y=u>o.length?Array.from({length:u}):o;for(let[d,M]of a)y[d]=M;e[n]=y}else e[n]=[...o,...s];else Q(o)&&Q(s)&&(e[n]=G(o,s));else e[n]=s}return e};function me(e,t){let n=[];for(let s=0;s<e.length;s++){let o=e[s];if(o.role==="assistant"&&o.tool_calls&&o.tool_calls.length>0){let a=new Set(o.tool_calls.map(y=>y.id)),i=new Set;for(let y=s+1;y<e.length;y++){let d=e[y];d.role==="tool"&&d.tool_call_id&&a.has(d.tool_call_id)&&i.add(d.tool_call_id)}let u=o.tool_calls.map(y=>y.id).filter(y=>!i.has(y));u.length>0&&n.push({insertAfterIndex:s,missingToolCallIds:u})}}for(let s=n.length-1;s>=0;s--){let{insertAfterIndex:o,missingToolCallIds:a}=n[s],i=a.map(u=>({role:"tool",tool_call_id:u,content:t}));e.splice(o+1,0,...i)}}var oe="remove";function he(e,t,n,s){let o=e.filter(i=>i[n]),a=new Set(o.flatMap(i=>i.tool_calls?.map(u=>u.id)??[]));if(t===oe)for(let i=e.length-1;i>=0;i--){let u=e[i];(u[n]||u[s]||u.tool_call_id&&a.has(u.tool_call_id))&&e.splice(i,1)}else if(t===!0)for(let i of e)(i[n]||i.tool_call_id&&a.has(i.tool_call_id))&&(i[s]=!0,delete i[n])}var ye=e=>{let{getTools:t,beforeCallTools:n,callTool:s,onToolCallStart:o,onToolCallEnd:a,toolCallCancelledContent:i="Tool call cancelled.",toolCallFailedContent:u="Tool call failed.",autoFillMissingToolMessages:y=!1,excludeToolMessagesNextTurn:d=!1,...M}=e,A=Symbol("doNotSendNextTurn"),w=Symbol("doNotSend"),v=(...C)=>{let[p,{primaryMessage:r}]=C;r.state.toolCall[p.id].status="running",o?.(...C)},T=(...C)=>{let[p,{status:r,primaryMessage:l}]=C;l.state.toolCall[p.id].status=r,a?.(...C)};return{name:"tool",...M,onTurnStart:C=>{let{messages:p}=C;return y&&me(p,i),d&&he(p,d,A,w),M.onTurnStart?.(C)},onBeforeRequest:async C=>{let{messages:p,requestBody:r}=C;d===!0&&(r.messages=p.filter(g=>!g[w]));let l=await t?.();return l&&l.length>0&&(r.tools=l),M.onBeforeRequest?.(C)},onAfterRequest:async C=>{let{currentMessage:p,lastChoice:r,appendMessage:l,abortSignal:g,setRequestState:f,requestNext:x}=C;if(r?.finish_reason!=="tool_calls"||!p.tool_calls?.length)return;d&&(p[A]=!0),f("processing","calling-tools"),await n?.(p.tool_calls,{...C,currentMessage:p});let E=p.tool_calls.map(async S=>{let D=Math.floor(Date.now()/1e3),O=(0,se.reactive)({role:"tool",tool_call_id:S.id,content:"",metadata:{createdAt:D,updatedAt:D}});l(O);let c={...C,primaryMessage:p,toolMessage:O};v(S,c);try{let h=s(S,c),b=W(h);for await(let _ of b){if(typeof _=="string")O.content+=_;else{let m={};try{m=JSON.parse(O.content||"{}")}catch(R){console.warn(R)}O.content=JSON.stringify(G(m,_))}O.metadata.updatedAt=Math.floor(Date.now()/1e3)}T(S,{...c,status:"success"})}catch(h){let b=h instanceof Error?h:new Error(String(h));if(g.aborted){T(S,{...c,status:"cancelled",error:b});return}console.error(h),O.content.length===0&&(O.content=u),T(S,{...c,status:"failed",error:b})}});return await Promise.all(E),x(),M.onAfterRequest?.(C)},onCompletionChunk:C=>{var r,l,g;let{currentMessage:p}=C;if(Array.isArray(p.tool_calls))for(let f of p.tool_calls)p.state?.toolCall?.[f.id]?.status||(p.state??(p.state={}),(r=p.state).toolCall??(r.toolCall={}),(l=p.state.toolCall)[g=f.id]??(l[g]={}));return M.onCompletionChunk?.(C)}}};var Ce=e=>{let t=[];for(let n of e){if(n.name){let s=t.findIndex(o=>o.name===n.name);s!==-1&&t.splice(s,1)}t.push(n)}return t},L=e=>{let{initialMessages:t=[],requestMessageFields:n=["role","content","tool_calls","tool_call_id"],plugins:s=[],onCompletionChunk:o}=e,a=(0,q.ref)("idle"),i=(0,q.ref)(void 0),u=(0,q.ref)(t),y=(0,q.ref)(e.responseProvider),d=null,M=[],A={},w=[J(),$(),X()],v=Ce(w.concat(s)),T=(0,q.computed)(()=>a.value==="processing"),C=async c=>{if(!c||!c.trim()){console.warn("Cannot send empty message");return}if(T.value){console.warn("Cannot send message while processing is in progress");return}let h=Math.floor(Date.now()/1e3);u.value.push({role:"user",content:c.trim(),metadata:{createdAt:h,updatedAt:h}}),M.push(u.value[u.value.length-1]),await E()},p=async(...c)=>{if(T.value){console.warn("Cannot send message while processing is in progress");return}u.value.push(...c),M.push(...c),await E()},r=c=>c.map(h=>ne(h,n)),l=(c,h)=>{a.value=c,c==="processing"?i.value=h||"requesting":i.value=void 0},g=c=>{Object.assign(A,c)},f=()=>({messages:u.value,currentTurn:M,requestState:a.value,processingState:i.value,requestMessageFields:n,plugins:v,setRequestState:l,customContext:A,setCustomContext:g}),x=async(c,h)=>{l("processing","requesting");let b=new Proxy({messages:r(u.value)},{set(k,P,I){return P==="messages"?(k.messages=r(I),!0):(k[P]=I,!0)}}),_=f();for(let k of v.filter(P=>!P.disabled))await k.onBeforeRequest?.({..._,abortSignal:h,requestBody:b});let m=(0,q.reactive)({role:"",content:"",loading:!0});O(m);let R,V=c(b,h),U=W(V);for await(let k of U){l("processing","completing"),m.loading&&(m.loading=void 0);let P=k.choices?.find(N=>N.index===0);if(P){R=P;let N=()=>{m.metadata||(m.metadata={});let{created:j,...ie}=k;m.metadata.createdAt=j,m.metadata.updatedAt=Math.floor(Date.now()/1e3),Object.assign(m.metadata,ie),G(m,P.message||P.delta)};if(o){let j=f();o({...j,abortSignal:h,chunk:k,currentMessage:m},N)}else N()}let I=f();for(let N of v.filter(j=>!j.disabled))N.onCompletionChunk?.({...I,abortSignal:h,chunk:k,choice:P,currentMessage:m})}await D(m,c,h,R)},E=async()=>{let c=new AbortController;d=c;let h=y.value;A={};try{l("processing","requesting");let b=f();for(let m of v.filter(R=>!R.disabled))await m.onTurnStart?.({...b,abortSignal:c.signal});try{await x(h,c.signal),l("completed")}catch(m){if(c.signal.aborted||m instanceof B||m instanceof Error&&m.name==="AbortError")l("aborted");else throw m}let _=f();for(let m of v.filter(R=>!R.disabled))await m.onTurnEnd?.({..._,abortSignal:c.signal})}catch(b){l("error");let _=f();for(let m of v.filter(R=>!R.disabled))m.onError?.({..._,abortSignal:c.signal,error:b});throw b}finally{d=null,M.slice(-1)[0]&&(M.slice(-1)[0].loading=void 0),M=[]}},S=async()=>{d?.abort(),T.value&&await new Promise(c=>{let h=(0,q.watch)(T,b=>{b||(h(),c())},{immediate:!0})})},D=async(c,h,b,_)=>{let m=!1,R=f(),V=v.filter(U=>!U.disabled).map(U=>{if(!U.onAfterRequest)return null;let k=I=>{O(I)},P=()=>{m=!0};return U.onAfterRequest({...R,abortSignal:b,currentMessage:c,lastChoice:_,appendMessage:k,requestNext:P})}).filter(U=>U!==null);await te(Promise.all(V),b),m&&await x(h,b)},O=c=>{let h=Array.isArray(c)?c:[c];u.value.push(...h),M.push(...h)};return{requestState:a,processingState:i,messages:u,responseProvider:y,isProcessing:T,sendMessage:C,send:p,abortRequest:S}};var H=require("vue");function ae(e,t=200,n=!1,s=!0,o=!1){return Me(ve(t,n,s,o),e)}function Me(e,t){function n(...s){return new Promise((o,a)=>{Promise.resolve(e(()=>t.apply(this,s),{fn:t,thisArg:this,args:s})).then(o).catch(a)})}return n}var re=()=>{};function ve(...e){let t=0,n,s=!0,o=re,a,i,u,y,d;!(0,H.isRef)(e[0])&&typeof e[0]=="object"?{delay:i,trailing:u=!0,leading:y=!0,rejectOnCancel:d=!1}=e[0]:[i,u=!0,y=!0,d=!1]=e;let M=()=>{n&&(clearTimeout(n),n=void 0,o(),o=re)};return w=>{let v=(0,H.toValue)(i),T=Date.now()-t,C=()=>a=w();return M(),v<=0?(t=Date.now(),C()):(T>v?(t=Date.now(),(y||!s)&&C()):u&&(a=new Promise((p,r)=>{o=d?r:p,n=setTimeout(()=>{t=Date.now(),s=!0,p(C()),M()},Math.max(0,v-T))})),!y&&!n&&(n=setTimeout(()=>s=!0,v)),s=!1,a)}}var be=e=>{let t=(0,F.ref)([]),n=new Map,s=new Map,o=(0,F.ref)(null),a=(0,F.computed)(()=>{let r=o.value;if(!r)return null;let l=t.value.find(f=>f.id===r);if(!l)return null;let g=n.get(r);return g?{...l,engine:g}:null}),i=r=>{if(!e.storage?.saveMessages)return;let l=r||o.value,g=t.value.find(x=>x.id===l);if(!g)return;g.updatedAt=Date.now(),e.storage?.saveConversation?.(g);let f=n.get(g.id);f&&e.storage.saveMessages(g.id,f.messages.value)},u=(r,l)=>{if(!e.autoSaveMessages||!e.storage?.saveMessages)return;let g=s.get(r);g&&g();let f=e.autoSaveThrottle??1e3,x=ae(()=>{i(r)},f,!0,!0),E=(0,F.watch)(l.messages,x,{deep:!0});s.set(r,E)},y=r=>{let l=s.get(r);l&&(l(),s.delete(r))};e.storage?.loadConversations&&Promise.resolve(e.storage.loadConversations()).then(r=>{t.value=r}).catch(r=>{console.error("[useConversation] loadConversations failed:",r)});let d=async(r,l)=>{let g=n.get(r);if(g)return g;let f=l?.initialMessages??e.useMessageOptions.initialMessages??[];if(e.storage?.loadMessages)try{f=await e.storage.loadMessages(r)}catch(E){console.error("[useConversation] loadMessages failed:",E)}let x=L({...e.useMessageOptions,...l,initialMessages:f});return n.set(r,x),u(r,x),x};function M(){return Date.now().toString(36)+Math.random().toString(36).substring(2,9)}let A=r=>{let{id:l=M(),title:g,metadata:f,useMessageOptions:x}=r||{},E=Date.now(),S={id:l,title:g,createdAt:E,updatedAt:E,metadata:f};t.value.unshift(S);let D=L({...e.useMessageOptions,...x});return n.set(l,D),u(l,D),e.storage?.saveConversation?.(S),e.storage?.saveMessages?.(l,D.messages.value),o.value=l,a.value},w=async r=>r?o.value===r?a.value:t.value.find(g=>g.id===r)?(await d(r),n.forEach((g,f)=>{if(f===r)return;g.isProcessing?.value||(y(f),n.delete(f))}),o.value=r,a.value):null:null;return{conversations:t,activeConversationId:o,activeConversation:a,createConversation:A,switchConversation:w,deleteConversation:async r=>{let l=t.value.findIndex(f=>f.id===r);if(l===-1)return;if(await n.get(r)?.abortRequest(),y(r),n.delete(r),t.value.splice(l,1),e.storage?.deleteConversation?.(r),o.value===r){let f=t.value[0];f?await w(f.id):o.value=null}},updateConversationTitle:(r,l)=>{let g=t.value.find(f=>f.id===r);g&&(g.title=l,g.updatedAt=Date.now(),e.storage?.saveConversation?.(g))},saveMessages:i,sendMessage:r=>{a.value?.engine.sendMessage(r)},abortActiveRequest:async()=>{await a.value?.engine.abortRequest()}}};0&&(module.exports={EXCLUDE_MODE_REMOVE,createSSEStreamGenerator,fallbackRolePlugin,lengthPlugin,thinkingPlugin,toolPlugin,useConversation,useMessage});
|
package/dist/index.mjs
CHANGED
|
@@ -1,3 +1,2 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
`);a=y.pop()||"";for(let f of y)if(f.trim()!==""){if(f.trim()==="data: [DONE]"){d&&(c=d),e.onDone(c);continue}try{let v=f.match(/^data: (.+)$/m);if(!v)continue;let h=JSON.parse(v[1]);e.onData(h),d=h.choices?.[0]?.finish_reason||void 0}catch(v){console.error("Error parsing SSE message:",v)}}}(a.trim()==="data: [DONE]"||o?.aborted)&&(o?.aborted&&(c="aborted"),e.onDone(c))}catch(s){if(o?.aborted)return;throw s}}function B(n){return n.map(e=>typeof e=="object"&&"role"in e&&"content"in e?{role:e.role,content:String(e.content),...e.name?{name:e.name}:{}}:typeof e=="string"?{role:"user",content:e}:{role:"user",content:String(e)})}function H(n){return!n.choices||!n.choices.length?"":n.choices[0].message?.content||""}var S=class extends A{constructor(o){super(o);this.defaultModel="gpt-3.5-turbo";this.baseURL=o.apiUrl||"https://api.openai.com/v1",this.apiKey=o.apiKey||"",o.defaultModel&&(this.defaultModel=o.defaultModel),this.apiKey||console.warn("API key is not provided. Authentication will likely fail.")}async chat(o){try{this.validateRequest(o);let l={model:o.options?.model||this.config.defaultModel||this.defaultModel,messages:o.messages,...o.options,stream:!1},g={method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(l)};this.apiKey&&Object.assign(g.headers,{Authorization:`Bearer ${this.apiKey}`});let a=await fetch(`${this.baseURL}/chat/completions`,g);if(!a.ok){let c=await a.text();throw new Error(`HTTP error! status: ${a.status}, details: ${c}`)}return await a.json()}catch(l){throw I(l)}}async chatStream(o,l){let{signal:g,...a}=o.options||{};try{this.validateRequest(o);let c={model:o.options?.model||this.config.defaultModel||this.defaultModel,messages:o.messages,...a,stream:!0},d={method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${this.apiKey}`,Accept:"text/event-stream"},body:JSON.stringify(c),signal:g};this.apiKey&&Object.assign(d.headers,{Authorization:`Bearer ${this.apiKey}`});let s=await fetch(`${this.baseURL}/chat/completions`,d);if(!s.ok){let p=await s.text();throw new Error(`HTTP error! status: ${s.status}, details: ${p}`)}await b(s,l,g)}catch(c){if(g?.aborted)return;l.onError(I(c))}}updateConfig(o){super.updateConfig(o),o.apiUrl&&(this.baseURL=o.apiUrl),o.apiKey&&(this.apiKey=o.apiKey),o.defaultModel&&(this.defaultModel=o.defaultModel)}};var x=class{constructor(e){this.config=e,this.provider=this.createProvider(e)}createProvider(e){if(e.provider==="custom"&&"providerImplementation"in e)return e.providerImplementation;switch(e.provider){case"deepseek":let o={defaultModel:"deepseek-chat",apiUrl:"https://api.deepseek.com/v1"};return new S({...o,...e});case"openai":default:return new S(e)}}async chat(e){return this.provider.chat(e)}async chatStream(e,o){let l={...e,options:{...e.options,stream:!0}};return this.provider.chatStream(l,o)}getConfig(){return{...this.config}}updateConfig(e){this.config={...this.config,...e},e.provider&&e.provider!==this.config.provider?this.provider=this.createProvider(this.config):this.provider.updateConfig(this.config)}};import{reactive as j,ref as O,toRaw as N}from"vue";var F=(c=>(c.INIT="init",c.PROCESSING="processing",c.STREAMING="streaming",c.FINISHED="finished",c.ABORTED="aborted",c.ERROR="error",c))(F||{}),_=["processing","streaming"],re=["finished","aborted","error"];function k(n){let{client:e,useStreamByDefault:o=!0,errorMessage:l="\u8BF7\u6C42\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5",initialMessages:g=[]}=n,a=O([...g]),c=O(""),d=O(o),s=null,p=j({status:"init",errorMsg:null}),m=t=>{let r=n.events?.onReceiveData,u=!1;if(r&&r(t,a,()=>{u=!0}),!u){let C={role:"assistant",content:t.choices[0].message.content};a.value.push(C)}},y=async t=>{let r=await e.chat({messages:N(a.value),options:{stream:!1,signal:t.signal}});m(r)},f=t=>{let r=n.events?.onReceiveData,u=!1;if(r&&r(t,a,()=>{u=!0}),!u){a.value[a.value.length-1].role==="user"&&a.value.push({role:"assistant",content:""});let C=t.choices?.[0];C&&C.delta.content&&(a.value[a.value.length-1].content+=C.delta.content)}},v=async t=>{await e.chatStream({messages:N(a.value),options:{stream:!0,signal:t.signal}},{onData:r=>{p.status="streaming",f(r)},onError:r=>{p.status="error",p.errorMsg=l,console.error("Stream request error:",r)},onDone:r=>{let u=n.events?.onFinish,C=!1;if(u&&u(r,{messages:a,messageState:p},()=>{C=!0}),!C){if(r==="aborted"||p.status==="aborted")return;p.status="finished"}}})},h=async()=>{p.status="processing",p.errorMsg=null,s=new AbortController;try{d.value?await v(s):await y(s)}catch(t){p.errorMsg=l,p.status="error",console.error("Send message error:",t)}finally{s=null}};return{messages:a,messageState:p,inputMessage:c,useStream:d,sendMessage:async(t=c.value,r=!0)=>{if(_.includes(p.status)||!t||typeof t=="string"&&!t.trim()||Array.isArray(t)&&t.length===0)return;let u={role:"user",content:t};a.value.push(u),r&&(c.value=""),await h()},send:async()=>{_.includes(p.status)||await h()},clearMessages:()=>{a.value=[],p.errorMsg=null},addMessage:t=>{Array.isArray(t)?a.value.push(...t):a.value.push(t)},abortRequest:()=>{s&&(s.abort(),s=null,p.status="aborted")},retryRequest:async t=>{t===0||!a.value[t]||a.value[t].role==="user"||(a.value.splice(t),await h())}}}import{reactive as L,watch as $}from"vue";var w=class{constructor(e="tiny-robot-ai-conversations"){this.storageKey=e}saveConversations(e){try{localStorage.setItem(this.storageKey,JSON.stringify(e))}catch(o){console.error("\u4FDD\u5B58\u4F1A\u8BDD\u5931\u8D25:",o)}}loadConversations(){try{let e=localStorage.getItem(this.storageKey);return e?JSON.parse(e):[]}catch(e){return console.error("\u52A0\u8F7D\u4F1A\u8BDD\u5931\u8D25:",e),[]}}};function G(){return Date.now().toString(36)+Math.random().toString(36).substring(2,9)}function le(n){let{client:e,storage:o=new w,autoSave:l=!0,allowEmpty:g=!1,useStreamByDefault:a=!0,errorMessage:c="\u8BF7\u6C42\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u91CD\u8BD5",events:d}=n,s=L({conversations:[],currentId:null,loading:!1}),p=!1,m=k({client:e,useStreamByDefault:a,errorMessage:c,initialMessages:[],events:{onReceiveData:d?.onReceiveData,onFinish:d?.onFinish}});$(()=>m.messages.value,i=>{if(s.currentId&&i.length>0){let t=s.conversations.findIndex(r=>r.id===s.currentId);t!==-1&&(s.conversations[t].messages=[...i],s.conversations[t].updatedAt=Date.now(),l&&R())}},{deep:!0});let y=(i="\u65B0\u4F1A\u8BDD",t={})=>{if(!g&&m.messages.value.length===0&&s.currentId)return s.currentId;let r=G(),u={id:r,title:i,createdAt:Date.now(),updatedAt:Date.now(),messages:[],metadata:t};return s.conversations.unshift(u),f(r),l&&R(),r},f=i=>{let t=s.conversations.find(r=>r.id===i);t&&(s.currentId=i,m.clearMessages(),t.messages.length>0&&t.messages.forEach(r=>m.addMessage(r)))},v=i=>{let t=s.conversations.findIndex(r=>r.id===i);t!==-1&&(s.conversations.splice(t,1),s.currentId===i&&(s.conversations.length>0?f(s.conversations[0].id):(s.currentId=null,m.clearMessages())),l&&R())},h=(i,t)=>{let r=s.conversations.find(u=>u.id===i);r&&(r.title=t,r.updatedAt=Date.now(),l&&R())},T=(i,t)=>{let r=s.conversations.find(u=>u.id===i);r&&(r.metadata={...r.metadata,...t},r.updatedAt=Date.now(),l&&R())},R=async()=>{try{await o.saveConversations(s.conversations)}catch(i){console.error("\u4FDD\u5B58\u4F1A\u8BDD\u5931\u8D25:",i)}},E=async()=>{s.loading=!0;try{let i=await o.loadConversations();s.conversations=i,i.length>0&&!s.currentId&&f(i[0].id),!p&&d?.onLoaded&&(p=!0,d.onLoaded(i))}catch(i){console.error("\u52A0\u8F7D\u4F1A\u8BDD\u5931\u8D25:",i)}finally{s.loading=!1}},D=async i=>{let t=s.conversations.find(r=>r.id===i);if(!t||t.messages.length<2)return t?.title||"\u65B0\u4F1A\u8BDD";try{let r={role:"system",content:"\u8BF7\u6839\u636E\u4EE5\u4E0B\u5BF9\u8BDD\u5185\u5BB9\uFF0C\u751F\u6210\u4E00\u4E2A\u7B80\u77ED\u7684\u6807\u9898\uFF08\u4E0D\u8D85\u8FC720\u4E2A\u5B57\u7B26\uFF09\u3002\u53EA\u9700\u8981\u8FD4\u56DE\u6807\u9898\u6587\u672C\uFF0C\u4E0D\u9700\u8981\u4EFB\u4F55\u89E3\u91CA\u6216\u989D\u5916\u5185\u5BB9\u3002"},u=t.messages.slice(0,Math.min(4,t.messages.length)),U=(await e.chat({messages:[r,...u],options:{stream:!1,max_tokens:30}})).choices[0].message.content.trim();return h(i,U),U}catch(r){return console.error("\u751F\u6210\u6807\u9898\u5931\u8D25:",r),t.title}},P=()=>s.currentId&&s.conversations.find(i=>i.id===s.currentId)||null;return E(),{state:s,messageManager:m,createConversation:y,switchConversation:f,deleteConversation:v,updateTitle:h,updateMetadata:T,saveConversations:R,loadConversations:E,generateTitle:D,getCurrentConversation:P}}export{x as AIClient,A as BaseModelProvider,K as ErrorType,re as FinalStatus,_ as GeneratingStatus,w as LocalStorageStrategy,S as OpenAIProvider,F as STATUS,q as StreamEventType,H as extractTextFromResponse,B as formatMessages,b as handleSSEStream,le as useConversation,k as useMessage};
|
|
1
|
+
function L(e="The operation was aborted"){let n=new Error(e);return n.name="AbortError",n}async function*se(e,n={}){let t=e.body?.getReader();if(!t)throw new Error("ReadableStream not supported");let{signal:s}=n,r=new TextDecoder,a="",i=()=>{t.cancel()};s?.addEventListener("abort",i);try{for(;;){if(s?.aborted)throw L();let u;try{u=await t.read()}catch(w){throw s?.aborted?L():w}let{done:y,value:d}=u;if(y){if(s?.aborted)throw L();return}let M=r.decode(d,{stream:!0});a+=M;let A=a.split(`
|
|
2
|
+
`);a=A.pop()||"";for(let w of A)if(w.trim()!==""&&w.startsWith("data: ")){let v=w.slice(6);if(v==="[DONE]")return;try{yield JSON.parse(v)}catch(T){console.warn("Failed to parse SSE data:",v,T)}}}}finally{s?.removeEventListener("abort",i),t.releaseLock()}}import{computed as ye,ref as te,watch as Ce}from"vue";import{computed as ce,reactive as ue,ref as j,watch as ge}from"vue";var K=(e={})=>{let{fallbackRole:n="assistant",...t}=e;return{name:"fallbackRole",...t,onBeforeRequest(s){let{requestBody:r,messages:a}=s;return r.messages=a.map(i=>({...i,role:i.role||n})),t.onBeforeRequest?.(s)}}};var z=(e={})=>{let{continueContent:n="Please continue with your previous answer.",...t}=e;return{name:"length",...t,onAfterRequest:async s=>{let{lastChoice:r,appendMessage:a,requestNext:i}=s;return r?.finish_reason==="length"&&(a({role:"user",content:n}),i()),t.onAfterRequest?.(s)}}};var J=(e={})=>({name:"thinking",...e,onCompletionChunk(n){let{choice:t,currentMessage:s}=n,a=typeof(t?.message?.reasoning_content||t?.delta?.reasoning_content)=="string";return s.state?s.state.thinking=a:s.state={thinking:a},e.onCompletionChunk?.(n)},onTurnEnd(n){let t=n.currentTurn.slice(-1)[0];return t?.state&&(t.state.thinking=void 0),e.onTurnEnd?.(n)}});import{reactive as re}from"vue";var I=class extends Error{constructor(n){super(n),this.name="AbortError"}};function oe(e){if(e.aborted)return{promise:Promise.reject(new I(String(e.reason??"Aborted"))),cleanup:()=>{}};let n=null;return{promise:new Promise((r,a)=>{n=()=>{a(new I(String(e.reason??"Aborted")))},e.addEventListener("abort",n,{once:!0})}),cleanup:()=>{n&&(e.removeEventListener("abort",n),n=null)}}}function Q(e,n){let{promise:t,cleanup:s}=oe(n);return Promise.race([e,t]).finally(s)}function Y(e,n){let t={};for(let s of n)s in e&&(t[s]=e[s]);return t}async function*G(e){if(X(e)){yield*e;return}let n=await e;if(X(n)){yield*n;return}yield n}function X(e){return e&&typeof e=="object"&&typeof e[Symbol.asyncIterator]=="function"}var H=e=>typeof e=="object"&&e!==null,$=e=>H(e)&&typeof e.index=="number",N=(e,n)=>{for(let[t,s]of Object.entries(n)){let r=e[t];if(r)if(typeof r=="string"&&typeof s=="string")t==="type"&&r||(e[t]=r+s);else if(Array.isArray(r)&&Array.isArray(s))if(r.every(a=>$(a))&&s.every(a=>$(a))){let a=new Map(r.map(d=>[d.index,d])),i=new Map(s.map(d=>[d.index,d]));for(let[d,M]of i)if(a.has(d)){let A=a.get(d);a.set(d,N(A,M))}else a.set(d,M);let u=Math.max(...Array.from(a.keys()),-1)+1,y=u>r.length?Array.from({length:u}):r;for(let[d,M]of a)y[d]=M;e[t]=y}else e[t]=[...r,...s];else H(r)&&H(s)&&(e[t]=N(r,s));else e[t]=s}return e};function ae(e,n){let t=[];for(let s=0;s<e.length;s++){let r=e[s];if(r.role==="assistant"&&r.tool_calls&&r.tool_calls.length>0){let a=new Set(r.tool_calls.map(y=>y.id)),i=new Set;for(let y=s+1;y<e.length;y++){let d=e[y];d.role==="tool"&&d.tool_call_id&&a.has(d.tool_call_id)&&i.add(d.tool_call_id)}let u=r.tool_calls.map(y=>y.id).filter(y=>!i.has(y));u.length>0&&t.push({insertAfterIndex:s,missingToolCallIds:u})}}for(let s=t.length-1;s>=0;s--){let{insertAfterIndex:r,missingToolCallIds:a}=t[s],i=a.map(u=>({role:"tool",tool_call_id:u,content:n}));e.splice(r+1,0,...i)}}var ie="remove";function le(e,n,t,s){let r=e.filter(i=>i[t]),a=new Set(r.flatMap(i=>i.tool_calls?.map(u=>u.id)??[]));if(n===ie)for(let i=e.length-1;i>=0;i--){let u=e[i];(u[t]||u[s]||u.tool_call_id&&a.has(u.tool_call_id))&&e.splice(i,1)}else if(n===!0)for(let i of e)(i[t]||i.tool_call_id&&a.has(i.tool_call_id))&&(i[s]=!0,delete i[t])}var Re=e=>{let{getTools:n,beforeCallTools:t,callTool:s,onToolCallStart:r,onToolCallEnd:a,toolCallCancelledContent:i="Tool call cancelled.",toolCallFailedContent:u="Tool call failed.",autoFillMissingToolMessages:y=!1,excludeToolMessagesNextTurn:d=!1,...M}=e,A=Symbol("doNotSendNextTurn"),w=Symbol("doNotSend"),v=(...C)=>{let[p,{primaryMessage:o}]=C;o.state.toolCall[p.id].status="running",r?.(...C)},T=(...C)=>{let[p,{status:o,primaryMessage:l}]=C;l.state.toolCall[p.id].status=o,a?.(...C)};return{name:"tool",...M,onTurnStart:C=>{let{messages:p}=C;return y&&ae(p,i),d&&le(p,d,A,w),M.onTurnStart?.(C)},onBeforeRequest:async C=>{let{messages:p,requestBody:o}=C;d===!0&&(o.messages=p.filter(g=>!g[w]));let l=await n?.();return l&&l.length>0&&(o.tools=l),M.onBeforeRequest?.(C)},onAfterRequest:async C=>{let{currentMessage:p,lastChoice:o,appendMessage:l,abortSignal:g,setRequestState:f,requestNext:x}=C;if(o?.finish_reason!=="tool_calls"||!p.tool_calls?.length)return;d&&(p[A]=!0),f("processing","calling-tools"),await t?.(p.tool_calls,{...C,currentMessage:p});let E=p.tool_calls.map(async S=>{let q=Math.floor(Date.now()/1e3),O=re({role:"tool",tool_call_id:S.id,content:"",metadata:{createdAt:q,updatedAt:q}});l(O);let c={...C,primaryMessage:p,toolMessage:O};v(S,c);try{let h=s(S,c),b=G(h);for await(let _ of b){if(typeof _=="string")O.content+=_;else{let m={};try{m=JSON.parse(O.content||"{}")}catch(R){console.warn(R)}O.content=JSON.stringify(N(m,_))}O.metadata.updatedAt=Math.floor(Date.now()/1e3)}T(S,{...c,status:"success"})}catch(h){let b=h instanceof Error?h:new Error(String(h));if(g.aborted){T(S,{...c,status:"cancelled",error:b});return}console.error(h),O.content.length===0&&(O.content=u),T(S,{...c,status:"failed",error:b})}});return await Promise.all(E),x(),M.onAfterRequest?.(C)},onCompletionChunk:C=>{var o,l,g;let{currentMessage:p}=C;if(Array.isArray(p.tool_calls))for(let f of p.tool_calls)p.state?.toolCall?.[f.id]?.status||(p.state??(p.state={}),(o=p.state).toolCall??(o.toolCall={}),(l=p.state.toolCall)[g=f.id]??(l[g]={}));return M.onCompletionChunk?.(C)}}};var fe=e=>{let n=[];for(let t of e){if(t.name){let s=n.findIndex(r=>r.name===t.name);s!==-1&&n.splice(s,1)}n.push(t)}return n},V=e=>{let{initialMessages:n=[],requestMessageFields:t=["role","content","tool_calls","tool_call_id"],plugins:s=[],onCompletionChunk:r}=e,a=j("idle"),i=j(void 0),u=j(n),y=j(e.responseProvider),d=null,M=[],A={},w=[K(),J(),z()],v=fe(w.concat(s)),T=ce(()=>a.value==="processing"),C=async c=>{if(!c||!c.trim()){console.warn("Cannot send empty message");return}if(T.value){console.warn("Cannot send message while processing is in progress");return}let h=Math.floor(Date.now()/1e3);u.value.push({role:"user",content:c.trim(),metadata:{createdAt:h,updatedAt:h}}),M.push(u.value[u.value.length-1]),await E()},p=async(...c)=>{if(T.value){console.warn("Cannot send message while processing is in progress");return}u.value.push(...c),M.push(...c),await E()},o=c=>c.map(h=>Y(h,t)),l=(c,h)=>{a.value=c,c==="processing"?i.value=h||"requesting":i.value=void 0},g=c=>{Object.assign(A,c)},f=()=>({messages:u.value,currentTurn:M,requestState:a.value,processingState:i.value,requestMessageFields:t,plugins:v,setRequestState:l,customContext:A,setCustomContext:g}),x=async(c,h)=>{l("processing","requesting");let b=new Proxy({messages:o(u.value)},{set(k,P,U){return P==="messages"?(k.messages=o(U),!0):(k[P]=U,!0)}}),_=f();for(let k of v.filter(P=>!P.disabled))await k.onBeforeRequest?.({..._,abortSignal:h,requestBody:b});let m=ue({role:"",content:"",loading:!0});O(m);let R,W=c(b,h),D=G(W);for await(let k of D){l("processing","completing"),m.loading&&(m.loading=void 0);let P=k.choices?.find(F=>F.index===0);if(P){R=P;let F=()=>{m.metadata||(m.metadata={});let{created:B,...ne}=k;m.metadata.createdAt=B,m.metadata.updatedAt=Math.floor(Date.now()/1e3),Object.assign(m.metadata,ne),N(m,P.message||P.delta)};if(r){let B=f();r({...B,abortSignal:h,chunk:k,currentMessage:m},F)}else F()}let U=f();for(let F of v.filter(B=>!B.disabled))F.onCompletionChunk?.({...U,abortSignal:h,chunk:k,choice:P,currentMessage:m})}await q(m,c,h,R)},E=async()=>{let c=new AbortController;d=c;let h=y.value;A={};try{l("processing","requesting");let b=f();for(let m of v.filter(R=>!R.disabled))await m.onTurnStart?.({...b,abortSignal:c.signal});try{await x(h,c.signal),l("completed")}catch(m){if(c.signal.aborted||m instanceof I||m instanceof Error&&m.name==="AbortError")l("aborted");else throw m}let _=f();for(let m of v.filter(R=>!R.disabled))await m.onTurnEnd?.({..._,abortSignal:c.signal})}catch(b){l("error");let _=f();for(let m of v.filter(R=>!R.disabled))m.onError?.({..._,abortSignal:c.signal,error:b});throw b}finally{d=null,M.slice(-1)[0]&&(M.slice(-1)[0].loading=void 0),M=[]}},S=async()=>{d?.abort(),T.value&&await new Promise(c=>{let h=ge(T,b=>{b||(h(),c())},{immediate:!0})})},q=async(c,h,b,_)=>{let m=!1,R=f(),W=v.filter(D=>!D.disabled).map(D=>{if(!D.onAfterRequest)return null;let k=U=>{O(U)},P=()=>{m=!0};return D.onAfterRequest({...R,abortSignal:b,currentMessage:c,lastChoice:_,appendMessage:k,requestNext:P})}).filter(D=>D!==null);await Q(Promise.all(W),b),m&&await x(h,b)},O=c=>{let h=Array.isArray(c)?c:[c];u.value.push(...h),M.push(...h)};return{requestState:a,processingState:i,messages:u,responseProvider:y,isProcessing:T,sendMessage:C,send:p,abortRequest:S}};import{isRef as de,toValue as pe}from"vue";function ee(e,n=200,t=!1,s=!0,r=!1){return me(he(n,t,s,r),e)}function me(e,n){function t(...s){return new Promise((r,a)=>{Promise.resolve(e(()=>n.apply(this,s),{fn:n,thisArg:this,args:s})).then(r).catch(a)})}return t}var Z=()=>{};function he(...e){let n=0,t,s=!0,r=Z,a,i,u,y,d;!de(e[0])&&typeof e[0]=="object"?{delay:i,trailing:u=!0,leading:y=!0,rejectOnCancel:d=!1}=e[0]:[i,u=!0,y=!0,d=!1]=e;let M=()=>{t&&(clearTimeout(t),t=void 0,r(),r=Z)};return w=>{let v=pe(i),T=Date.now()-n,C=()=>a=w();return M(),v<=0?(n=Date.now(),C()):(T>v?(n=Date.now(),(y||!s)&&C()):u&&(a=new Promise((p,o)=>{r=d?o:p,t=setTimeout(()=>{n=Date.now(),s=!0,p(C()),M()},Math.max(0,v-T))})),!y&&!t&&(t=setTimeout(()=>s=!0,v)),s=!1,a)}}var Be=e=>{let n=te([]),t=new Map,s=new Map,r=te(null),a=ye(()=>{let o=r.value;if(!o)return null;let l=n.value.find(f=>f.id===o);if(!l)return null;let g=t.get(o);return g?{...l,engine:g}:null}),i=o=>{if(!e.storage?.saveMessages)return;let l=o||r.value,g=n.value.find(x=>x.id===l);if(!g)return;g.updatedAt=Date.now(),e.storage?.saveConversation?.(g);let f=t.get(g.id);f&&e.storage.saveMessages(g.id,f.messages.value)},u=(o,l)=>{if(!e.autoSaveMessages||!e.storage?.saveMessages)return;let g=s.get(o);g&&g();let f=e.autoSaveThrottle??1e3,x=ee(()=>{i(o)},f,!0,!0),E=Ce(l.messages,x,{deep:!0});s.set(o,E)},y=o=>{let l=s.get(o);l&&(l(),s.delete(o))};e.storage?.loadConversations&&Promise.resolve(e.storage.loadConversations()).then(o=>{n.value=o}).catch(o=>{console.error("[useConversation] loadConversations failed:",o)});let d=async(o,l)=>{let g=t.get(o);if(g)return g;let f=l?.initialMessages??e.useMessageOptions.initialMessages??[];if(e.storage?.loadMessages)try{f=await e.storage.loadMessages(o)}catch(E){console.error("[useConversation] loadMessages failed:",E)}let x=V({...e.useMessageOptions,...l,initialMessages:f});return t.set(o,x),u(o,x),x};function M(){return Date.now().toString(36)+Math.random().toString(36).substring(2,9)}let A=o=>{let{id:l=M(),title:g,metadata:f,useMessageOptions:x}=o||{},E=Date.now(),S={id:l,title:g,createdAt:E,updatedAt:E,metadata:f};n.value.unshift(S);let q=V({...e.useMessageOptions,...x});return t.set(l,q),u(l,q),e.storage?.saveConversation?.(S),e.storage?.saveMessages?.(l,q.messages.value),r.value=l,a.value},w=async o=>o?r.value===o?a.value:n.value.find(g=>g.id===o)?(await d(o),t.forEach((g,f)=>{if(f===o)return;g.isProcessing?.value||(y(f),t.delete(f))}),r.value=o,a.value):null:null;return{conversations:n,activeConversationId:r,activeConversation:a,createConversation:A,switchConversation:w,deleteConversation:async o=>{let l=n.value.findIndex(f=>f.id===o);if(l===-1)return;if(await t.get(o)?.abortRequest(),y(o),t.delete(o),n.value.splice(l,1),e.storage?.deleteConversation?.(o),r.value===o){let f=n.value[0];f?await w(f.id):r.value=null}},updateConversationTitle:(o,l)=>{let g=n.value.find(f=>f.id===o);g&&(g.title=l,g.updatedAt=Date.now(),e.storage?.saveConversation?.(g))},saveMessages:i,sendMessage:o=>{a.value?.engine.sendMessage(o)},abortActiveRequest:async()=>{await a.value?.engine.abortRequest()}}};export{ie as EXCLUDE_MODE_REMOVE,se as createSSEStreamGenerator,K as fallbackRolePlugin,z as lengthPlugin,J as thinkingPlugin,Re as toolPlugin,Be as useConversation,V as useMessage};
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@opentiny/tiny-robot-kit",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.4.0-alpha.1",
|
|
4
4
|
"publishConfig": {
|
|
5
5
|
"access": "public"
|
|
6
6
|
},
|
|
@@ -26,5 +26,5 @@
|
|
|
26
26
|
"peerDependencies": {
|
|
27
27
|
"vue": ">=3.0.0"
|
|
28
28
|
},
|
|
29
|
-
"gitHead": "
|
|
29
|
+
"gitHead": "db8d5d524ec974b61544c464ad4a246beec8d364"
|
|
30
30
|
}
|