@shumai-one/shumai-agent 0.2.0 → 0.2.2

package/bin/chunk-m6rw11kp.js

@@ -0,0 +1,2 @@
+// @bun
+import{p as b,q as E}from"./chunk-2t47accr.js";import{v as S}from"./chunk-cpmyq7y4.js";import{w as L,x as M}from"./chunk-b35kbsh7.js";import{Cc as I}from"./chunk-fvmb8en4.js";import"./chunk-22m0y0fs.js";var P=async(B,J,D)=>{let F={api:B.api,provider:B.provider,model:B.id,output:[],stopReason:"stop",timestamp:Date.now()};try{let G=D?.apiKey;if(!G)throw Error(`No API key for provider: ${B.provider}`);let V=v(B,G,D?.headers),Q=z(B,J),X=await D?.onPayload?.(Q,B);if(X!==void 0)Q=X;let N={...D?.signal?{signal:D.signal}:{},...D?.timeoutMs!==void 0?{timeout:D.timeoutMs}:{},maxRetries:D?.maxRetries??0},{data:O,response:q}=await V.chat.completions.create(Q,N).withResponse();await D?.onResponse?.({status:q.status,headers:L(q.headers)},B);let Y=O;if(F.responseId=Y.id,Y.usage)F.usage=C(Y.usage,B);let Z=Y.choices[0];if(Z){let _=Z.message.content;if(typeof _==="string"&&_.length>0)F.output.push({type:"text",text:_});for(let $ of Z.message.images??[]){let A=typeof $.image_url==="string"?$.image_url:$.image_url?.url;if(!A?.startsWith("data:"))continue;let f=A.match(/^data:([^;]+);base64,(.+)$/);if(!f)continue;F.output.push({type:"image",mimeType:f[1],data:f[2]})}}return F}catch(G){return F.stopReason=D?.signal?.aborted?"aborted":"error",F.errorMessage=E(b(G)),F}};function v(B,J,D){return new I({apiKey:J,baseURL:B.baseUrl,dangerouslyAllowBrowser:!0,defaultHeaders:M({...B.headers,...D})})}function z(B,J){let D=J.input.map((F)=>{if(F.type==="text")return{type:"text",text:S(F.text)};return{type:"image_url",image_url:{url:`data:${F.mimeType};base64,${F.data}`}}});return{model:B.id,messages:[{role:"user",content:D}],stream:!1,modalities:B.output.includes("text")?["image","text"]:["image"]}}function C(B,J){let D=B.prompt_tokens||0,F=B.prompt_tokens_details?.cached_tokens||0,G=B.prompt_tokens_details?.cache_write_tokens||0,V=G>0?Math.max(0,F-G):F,Q=Math.max(0,D-V-G),X=B.completion_tokens||0,N={input:Q,output:X,cacheRead:V,cacheWrite:G,totalTokens:Q+X+V+G,cost:{input:J.cost.input/1e6*Q,output:J.cost.output/1e6*X,cacheRead:J.cost.cacheRead/1e6*V,cacheWrite:J.cost.cacheWrite/1e6*G,total:0}};return N.cost.total=N.cost.input+N.cost.output+N.cost.cacheRead+N.cost.cacheWrite,N}export{P as generateImages};