@shumai-one/shumai-agent 0.4.2 → 0.4.3

package/bin/chunk-ptxcc9bv.js

@@ -0,0 +1,2 @@
+// @bun
+import{c,Q,I,S,T}from"./shumai-agent-app.js";var n=class{get supportsCreateIfNotExists(){return!0}get supportsTransactionalDdl(){return!1}get supportsReturning(){return!0}async acquireMigrationLock(){}async releaseMigrationLock(){}get supportsOutput(){return!0}},s=class{#e;#t=new o;#r;#a;constructor(e){this.#e={...e}}async init(){if(this.#r=this.#e.database,this.#a=new i(this.#r),this.#e.onCreateConnection)await this.#e.onCreateConnection(this.#a)}async acquireConnection(){return await this.#t.lock(),this.#a}async beginTransaction(e){await e.executeQuery(I.raw("begin"))}async commitTransaction(e){await e.executeQuery(I.raw("commit"))}async rollbackTransaction(e){await e.executeQuery(I.raw("rollback"))}async releaseConnection(){this.#t.unlock()}async destroy(){this.#r?.close()}},i=class{#e;constructor(e){this.#e=e}executeQuery(e){let{sql:r,parameters:a}=e,t=this.#e.prepare(r);return Promise.resolve({rows:t.all(a)})}async*streamQuery(){throw Error("Streaming query is not supported by SQLite driver.")}},o=class{#e;#t;async lock(){while(this.#e!==void 0)await this.#e;this.#e=new Promise((e)=>{this.#t=e})}unlock(){let e=this.#t;this.#e=void 0,this.#t=void 0,e?.()}},l=class{#e;constructor(e){this.#e=e}async getSchemas(){return[]}async getTables(e={withInternalKyselyTables:!1}){let r=this.#e.selectFrom("sqlite_schema").where("type","=","table").where("name","not like","sqlite_%").select("name").$castTo();if(!e.withInternalKyselyTables)r=r.where("name","!=",S).where("name","!=",T);let a=await r.execute();return Promise.all(a.map(({name:t})=>this.#t(t)))}async getMetadata(e){return{tables:await this.getTables(e)}}async#t(e){let r=this.#e,a=(await r.selectFrom("sqlite_master").where("name","=",e).select("sql").$castTo().execute())[0]?.sql?.split(/[\(\),]/)?.find((t)=>t.toLowerCase().includes("autoincrement"))?.split(/\s+/)?.[0]?.replace(/["`]/g,"");return{name:e,columns:(await r.selectFrom(c`pragma_table_info(${e})`.as("table_info")).select(["name","type","notnull","dflt_value"]).execute()).map((t)=>({name:t.name,dataType:t.type,isNullable:!t.notnull,isAutoIncrementing:t.name===a,hasDefaultValue:t.dflt_value!=null})),isView:!0}}},u=class extends Q{getCurrentParameterPlaceholder(){return"?"}getLeftIdentifierWrapper(){return'"'}getRightIdentifierWrapper(){return'"'}getAutoIncrement(){return"autoincrement"}},h=class{#e;constructor(e){this.#e={...e}}createDriver(){return new s(this.#e)}createQueryCompiler(){return new u}createAdapter(){return new n}createIntrospector(e){return new l(e)}};export{h as BunSqliteDialect};

package/bin/chunk-q7ypqj4c.js

@@ -0,0 +1,2 @@
+// @bun
+import{l,j}from"./shumai-agent-app.js";function W(e,y){if(typeof e==="string"&&typeof y==="string")return e.toLowerCase()===y.toLowerCase();return e===y}function T(e,y){if(typeof e!=="string")return y.includes(e);return y.some((O)=>typeof O==="string"&&e.toLowerCase()===O.toLowerCase())}function D(e,y){return!T(e,y)}function k(e,y){if(typeof e!=="string"||typeof y!=="string")return!1;return e.toLowerCase().includes(y.toLowerCase())}function B(e,y){if(typeof e!=="string"||typeof y!=="string")return!1;return e.toLowerCase().startsWith(y.toLowerCase())}function _(e,y){if(typeof e!=="string"||typeof y!=="string")return!1;return e.toLowerCase().endsWith(y.toLowerCase())}var x=(e,y)=>{let O=null,R=j({config:{adapterId:"memory",adapterName:"Memory Adapter",usePlural:!1,debugLogs:y?.debugLogs||!1,supportsArrays:!0,customTransformInput(d){if(d.options.advanced?.database?.generateId==="serial"&&d.field==="id"&&d.action==="create")return e[d.model].length+1;return d.data},transaction:async(d)=>{let b=structuredClone(e);try{return await d(R(O))}catch(E){throw Object.keys(e).forEach((A)=>{e[A]=b[A]}),E}}},adapter:({getFieldName:d,getDefaultFieldName:b,options:E,getModelName:A})=>{let S=(t,i,f)=>{if(!i)return t;return t.sort((o,c)=>{let g=d({model:f,field:i.field}),p=o[g],u=c[g],a=0;if(p==null&&u==null)a=0;else if(p==null)a=-1;else if(u==null)a=1;else if(typeof p==="string"&&typeof u==="string")a=p.localeCompare(u);else if(p instanceof Date&&u instanceof Date)a=p.getTime()-u.getTime();else if(typeof p==="number"&&typeof u==="number")a=p-u;else if(typeof p==="boolean"&&typeof u==="boolean")a=p===u?0:p?1:-1;else a=String(p).localeCompare(String(u));return i.direction==="asc"?a:-a})};function C(t,i,f,o){let c=(()=>{let u=e[i];if(!u)throw l.error(`[MemoryAdapter] Model ${i} not found in the DB`,Object.keys(e)),Error(`Model ${i} not found`);let a=(n,h)=>{let{field:s,value:r,operator:L,mode:v="sensitive"}=h,m=v==="insensitive"&&(typeof r==="string"||Array.isArray(r)&&r.every((M)=>typeof M==="string"));switch(L){case"in":if(!Array.isArray(r))throw Error("Value must be an array");if(m)return T(n[s],r);return r.includes(n[s]);case"not_in":if(!Array.isArray(r))throw Error("Value must be an array");if(m)return D(n[s],r);return!r.includes(n[s]);case"contains":if(m)return k(n[s],r);return n[s]?.includes(r);case"starts_with":if(m)return B(n[s],r);return n[s].startsWith(r);case"ends_with":if(m)return _(n[s],r);return n[s].endsWith(r);case"ne":return m?!W(n[s],r):n[s]!==r;case"gt":return r!=null&&Boolean(n[s]>r);case"gte":return r!=null&&Boolean(n[s]>=r);case"lt":return r!=null&&Boolean(n[s]<r);case"lte":return r!=null&&Boolean(n[s]<=r);default:if(m)return W(n[s],r);if(r===null)return n[s]==null;return n[s]===r}},w=u.filter((n)=>{if(!t.length||t.length===0)return!0;let h=a(n,t[0]);for(let s of t){let r=a(n,s);if(s.connector==="OR")h=h||r;else h=h&&r}return h});if(o?.length&&o.length>0)w=w.map((n)=>Object.fromEntries(Object.entries(n).filter(([h])=>o.includes(b({model:i,field:h})))));return w})();if(!f)return c;let g=new Map,p=new Map;for(let u of c){let a=String(u.id);if(!g.has(a)){let n={...u};for(let[h,s]of Object.entries(f)){let r=A(h);if(s.relation==="one-to-one")n[r]=null;else n[r]=[],p.set(`${a}-${h}`,new Set)}g.set(a,n)}let w=g.get(a);for(let[n,h]of Object.entries(f)){let s=A(n),r=e[s];if(!r)throw l.error(`[MemoryAdapter] JoinOption model ${s} not found in the DB`,Object.keys(e)),Error(`JoinOption model ${s} not found`);let L=r.filter((v)=>v[h.on.to]===u[h.on.from]);if(h.relation==="one-to-one")w[s]=L[0]||null;else{let v=p.get(`${a}-${n}`),m=h.limit??100,M=0;for(let I of L){if(M>=m)break;if(!v.has(I.id))w[s].push(I),v.add(I.id),M++}}}}return Array.from(g.values())}return{create:async({model:t,data:i})=>{if(E.advanced?.database?.generateId==="serial")i.id=e[A(t)].length+1;if(!e[t])e[t]=[];return e[t].push(i),i},findOne:async({model:t,where:i,select:f,join:o})=>{let c=C(i,t,o,f);if(o){let g=c;if(!g.length)return null;return g[0]}return c[0]||null},findMany:async({model:t,where:i,sortBy:f,limit:o,select:c,offset:g,join:p})=>{let u=C(i||[],t,p,c);if(p){let w=u;if(!w.length)return[];S(w,f,t);let n=w;if(g!==void 0)n=n.slice(g);if(o!==void 0)n=n.slice(0,o);return n}let a=S(u,f,t);if(g!==void 0)a=a.slice(g);if(o!==void 0)a=a.slice(0,o);return a||[]},count:async({model:t,where:i})=>{if(i)return C(i,t).length;return e[t].length},update:async({model:t,where:i,update:f})=>{let o=C(i,t);return o.forEach((c)=>{Object.assign(c,f)}),o[0]||null},delete:async({model:t,where:i})=>{let f=e[t],o=C(i,t);e[t]=f.filter((c)=>!o.includes(c))},deleteMany:async({model:t,where:i})=>{let f=e[t],o=C(i,t),c=0;return e[t]=f.filter((g)=>{if(o.includes(g))return c++,!1;return!o.includes(g)}),c},consumeOne:async({model:t,where:i})=>{let f=e[t],o=C(i,t)[0];if(!o)return null;return e[t]=f.filter((c)=>c!==o),o},updateMany({model:t,where:i,update:f}){let o=C(i,t);return o.forEach((c)=>{Object.assign(c,f)}),o[0]||null}}}});return(d)=>(O=d,R(d))};export{x as memoryAdapter};

package/bin/chunk-r8hcdypf.js

@@ -0,0 +1,2 @@
+// @bun
+import{xe,S,T,Ue}from"./shumai-agent-app.js";var m=class extends Ue{},p=class{#e;#t;constructor(e){this.#e={...e}}async init(){if(this.#t=new w(this.#e.database),this.#e.onCreateConnection)await this.#e.onCreateConnection(this.#t)}async acquireConnection(){return this.#t}async beginTransaction(){throw Error("D1 does not support interactive transactions. Use the D1 batch() API instead.")}async commitTransaction(){throw Error("D1 does not support interactive transactions. Use the D1 batch() API instead.")}async rollbackTransaction(){throw Error("D1 does not support interactive transactions. Use the D1 batch() API instead.")}async releaseConnection(){}async destroy(){}},w=class{#e;constructor(e){this.#e=e}async executeQuery(e){let t=await this.#e.prepare(e.sql).bind(...e.parameters).all(),a=t.meta.changes!=null?BigInt(t.meta.changes):void 0;return{insertId:t.meta.last_row_id===void 0||t.meta.last_row_id===null?void 0:BigInt(t.meta.last_row_id),rows:t?.results||[],numAffectedRows:a}}async*streamQuery(){throw Error("D1 does not support streaming queries.")}},d=class{#e;#t;constructor(e,t){this.#e=e,this.#t=t}async getSchemas(){return[]}async getTables(e={withInternalKyselyTables:!1}){let t=this.#e.selectFrom("sqlite_master").where("type","in",["table","view"]).where("name","not like","sqlite_%").where("name","not like","_cf_%").select(["name","type","sql"]).$castTo();if(!e.withInternalKyselyTables)t=t.where("name","!=",S).where("name","!=",T);let a=await t.execute();if(a.length===0)return[];let c=a.map((s)=>this.#t.prepare("SELECT * FROM pragma_table_info(?)").bind(s.name)),l=await this.#t.batch(c);return a.map((s,u)=>{let o=l[u]?.results??[],r=s.sql?.split(/[(),]/)?.find((n)=>n.toLowerCase().includes("autoincrement"))?.split(/\s+/)?.filter(Boolean)?.[0]?.replace(/["`]/g,"");if(!r){let n=o.filter((h)=>h.pk>0),i=n.length===1?n[0]:void 0;if(i&&i.type.toLowerCase()==="integer")r=i.name}return{name:s.name,isView:s.type==="view",columns:o.map((n)=>({name:n.name,dataType:n.type,isNullable:!n.notnull,isAutoIncrementing:n.name===r,hasDefaultValue:n.dflt_value!=null}))}})}async getMetadata(e){return{tables:await this.getTables(e)}}},y=class extends xe{},f=class{#e;constructor(e){this.#e={...e}}createDriver(){return new p(this.#e)}createQueryCompiler(){return new y}createAdapter(){return new m}createIntrospector(e){return new d(e,this.#e.database)}};export{f as D1SqliteDialect};

package/bin/chunk-rv0jkz1g.js

@@ -0,0 +1,8 @@
+// @bun
+import{E}from"./chunk-2mj5h53m.js";import{f}from"./chunk-phjnj6x5.js";import{m,Ee,fe}from"./shumai-agent-app.js";import{x,le}from"./chunk-p7sczpks.js";class g extends Error{code;diagnosticDetails;constructor(r,t,n){super(r);this.name="PiMessagesResponseError",this.code=t,this.diagnosticDetails=n}}function w(r){try{let t=JSON.parse(r),n=t?.error;return t&&typeof n==="object"&&n!==null&&!Array.isArray(n)?t:void 0}catch{return}}function b(r){return r.length>8192?`${r.slice(0,8192)}\u2026`:r}function I(r,t,n){let e=typeof n?.error?.message==="string"?n.error.message:void 0,s=typeof n?.error?.code==="string"?n.error.code:void 0,a=e??t,o=s?` (${s})`:"";return`${r.status} ${r.statusText}: ${a}${o}`}function k(r,t,n,e){let s=w(e),a=typeof s?.error?.code==="string"?s.error.code:void 0;return new g(I(n,e,s),a,{version:1,provider:r.provider,model:r.id,url:t.toString(),status:n.status,statusText:n.statusText,error:s?.error,body:s?void 0:b(e),timestampMs:Date.now()})}function y(){return{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}}}function l(r,t){if(!t)return;fe(r,{type:"pi_messages_rewrite",timestamp:Date.now(),details:{...t}})}function R(r){let t={role:"assistant",content:[],api:r.api,provider:r.provider,model:r.id,usage:y(),stopReason:"stop",timestamp:Date.now()},n=new Map;return(e)=>{switch(e.type){case"done":return Object.assign(t,{stopReason:e.reason,usage:e.usage,responseId:e.responseId}),l(t,e.rewrite),{type:"done",reason:e.reason,message:t};case"error":return Object.assign(t,{stopReason:e.reason,usage:e.usage,errorMessage:e.errorMessage,responseId:e.responseId}),l(t,e.rewrite),{type:"error",reason:e.reason,error:t};case"start":break;case"text_start":t.content[e.contentIndex]={type:"text",text:""};break;case"text_delta":t.content[e.contentIndex].text+=e.delta;break;case"text_end":Object.assign(t.content[e.contentIndex],{text:e.content,textSignature:e.contentSignature});break;case"thinking_start":t.content[e.contentIndex]={type:"thinking",thinking:""};break;case"thinking_delta":t.content[e.contentIndex].thinking+=e.delta;break;case"thinking_end":Object.assign(t.content[e.contentIndex],{thinking:e.content,thinkingSignature:e.contentSignature,redacted:e.redacted});break;case"toolcall_start":t.content[e.contentIndex]={type:"toolCall",id:e.id,name:e.toolName,arguments:{}},n.set(e.contentIndex,"");break;case"toolcall_delta":{let s=`${n.get(e.contentIndex)??""}${e.delta}`;n.set(e.contentIndex,s),t.content[e.contentIndex].arguments=E(s);break}case"toolcall_end":return Object.assign(t.content[e.contentIndex],e.toolCall),n.delete(e.contentIndex),{type:"toolcall_end",contentIndex:e.contentIndex,toolCall:t.content[e.contentIndex],partial:t}}return{...e,partial:t}}}async function*_(r){let t=new TextDecoder,n=r.getReader(),e="";try{while(!0){let{done:s,value:a}=await n.read();e+=s?t.decode():t.decode(a,{stream:!0}),e=e.replace(/\r\n/g,`
+`);let o=e.indexOf(`
+
+`);while(o!==-1){let c=h(e.slice(0,o));if(c)yield c;e=e.slice(o+2),o=e.indexOf(`
+
+`)}if(s)break}if(e.trim()){let s=h(e);if(s)yield s}}finally{n.releaseLock()}}function h(r){let t=r.split(`
+`).find((n)=>n.startsWith("data:"))?.slice(5).trim();return t&&t!=="[DONE]"?JSON.parse(t):void 0}function S(r,t,n){let e=n?"aborted":"error",s={role:"assistant",content:[],api:r.api,provider:r.provider,model:r.id,usage:y(),stopReason:e,errorMessage:t instanceof Error?t.message:String(t),timestamp:Date.now()};if(!n&&t instanceof g)fe(s,Ee("pi_messages_response_failure",t,t.diagnosticDetails));return{type:"error",reason:e,error:s}}function M(r,t){if(r)return r;return f("PI_CACHE_RETENTION",t)==="long"?"long":void 0}var D=(r,t,n)=>{let e=new m,s=R(r);return(async()=>{try{let a=n?.apiKey;if(!a)throw Error(`No API key provided for provider "${r.provider}"`);let o=new URL(`${r.baseUrl.replace(/\/+$/u,"")}/messages`);if(n?.debug)o.searchParams.set("debug","1");let c={model:r.id,context:t,options:{temperature:n?.temperature,maxTokens:n?.maxTokens,reasoning:n?.reasoning,cacheRetention:M(n?.cacheRetention,n?.env),sessionId:n?.sessionId,toolChoice:n?.toolChoice}},p=await n?.onPayload?.(c,r);if(p!==void 0)c=p;let i=await fetch(o,{method:"POST",headers:{authorization:`Bearer ${a}`,accept:"text/event-stream","content-type":"application/json",...le(n?.headers)},body:JSON.stringify(c),signal:n?.signal});if(await n?.onResponse?.({status:i.status,headers:x(i.headers)},r),!i.ok){let d=await i.text();throw k(r,o,i,d)}if(!i.body)throw Error(`${r.provider} response has no body`);for await(let d of _(i.body)){let u=s(d);if(e.push(u),u.type==="done"||u.type==="error")return}throw Error(`${r.provider} stream ended without a terminal event`)}catch(a){e.push(S(r,a,n?.signal?.aborted??!1))}})(),e},N=(r,t,n)=>{let e=n;return D(r,t,{...n,reasoning:n?.reasoning,toolChoice:e?.toolChoice,debug:e?.debug})};export{g as PiMessagesResponseError,D as stream,N as streamSimple};