@legit-sdk/assistant-ui 0.1.3 → 0.2.0

package/dist/index.d.ts

@@ -1,11 +1,11 @@
 import { AssistantCloud } from 'assistant-cloud';
 import { AssistantRuntime, ExportedMessageRepository, ThreadMessageLike } from '@assistant-ui/react';
-import { initLegitFs } from '@legit-sdk/core';
+import { openLegitFs } from '@legit-sdk/core';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ReactNode } from 'react';
 import { ModelMessage, StreamTextResult } from 'ai';
 
-type LegitFsInstance = Awaited<ReturnType<typeof initLegitFs>>;
+type LegitFsInstance = Awaited<ReturnType<typeof openLegitFs>>;
 type AssistantCloudConfig = {
     baseUrl: string;
     authToken: () => Promise<string | null>;

package/dist/index.js

@@ -2,4 +2,4 @@
 // Generated by esbuild
 
 var ve=Object.defineProperty;var Ee=r=>{throw TypeError(r)};var lt=(r,e,t)=>e in r?ve(r,e,{enumerable:!0,configurable:!0,writable:!0,value:t}):r[e]=t;var a=(r,e)=>ve(r,"name",{value:e,configurable:!0});var u=(r,e,t)=>lt(r,typeof e!="symbol"?e+"":e,t),Me=(r,e,t)=>e.has(r)||Ee("Cannot "+t);var G=(r,e,t)=>(Me(r,e,"read from private field"),t?t.call(r):e.get(r)),K=(r,e,t)=>e.has(r)?Ee("Cannot add the same private member more than once"):e instanceof WeakSet?e.add(r):e.set(r,t),V=(r,e,t,o)=>(Me(r,e,"write to private field"),o?o.call(r,t):e.set(r,t),t);var Oe=a(r=>{try{let t=r.split(".")[1];if(!t)throw new Error("Invalid JWT format");let o=t.replace(/-/g,"+").replace(/_/g,"/");for(;o.length%4!==0;)o+="=";let s=atob(o),i=JSON.parse(s).exp;if(!i||typeof i!="number")throw new Error('JWT does not contain a valid "exp" field');return i*1e3}catch(e){throw new Error("Unable to determine the token expiry: "+e)}},"getJwtExpiry"),Q,C,ge=(C=class{constructor(e){u(this,"strategy","jwt");u(this,"cachedToken",null);u(this,"tokenExpiry",null);K(this,Q);V(this,Q,e)}async getAuthHeaders(){let e=Date.now();if(this.cachedToken&&this.tokenExpiry&&this.tokenExpiry-e>30*1e3)return{Authorization:`Bearer ${this.cachedToken}`};let t=await G(this,Q).call(this);return t?(this.cachedToken=t,this.tokenExpiry=Oe(t),{Authorization:`Bearer ${t}`}):!1}readAuthHeaders(e){let t=e.get("Authorization");if(!t)return;let[o,s]=t.split(" ");if(o!=="Bearer"||!s)throw new Error("Invalid auth header received");this.cachedToken=s,this.tokenExpiry=Oe(s)}},Q=new WeakMap,a(C,"AssistantCloudJWTAuthStrategy"),C),X,Y,Z,A,Fe=(A=class{constructor(e,t,o){u(this,"strategy","api-key");K(this,X);K(this,Y);K(this,Z);V(this,X,e),V(this,Y,t),V(this,Z,o)}async getAuthHeaders(){return{Authorization:`Bearer ${G(this,X)}`,"Aui-User-Id":G(this,Y),"Aui-Workspace-Id":G(this,Z)}}readAuthHeaders(){}},X=new WeakMap,Y=new WeakMap,Z=new WeakMap,a(A,"AssistantCloudAPIKeyAuthStrategy"),A),oe="aui:refresh_token",S,De=(S=class{constructor(e){u(this,"strategy","anon");u(this,"baseUrl");u(this,"jwtStrategy");this.baseUrl=e,this.jwtStrategy=new ge(async()=>{let t=Date.now(),o=localStorage.getItem(oe),s=o?JSON.parse(o):void 0;if(s)if(new Date(s.expires_at).getTime()-t>30*1e3){let T=await fetch(`${this.baseUrl}/v1/auth/tokens/refresh`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({refresh_token:s.token})});if(T.ok){let z=await T.json(),{access_token:m,refresh_token:h}=z;return h&&localStorage.setItem(oe,JSON.stringify(h)),m}}else localStorage.removeItem(oe);let l=await fetch(`${this.baseUrl}/v1/auth/tokens/anonymous`,{method:"POST"});if(!l.ok)return null;let i=await l.json(),{access_token:n,refresh_token:c}=i;return!n||!c?null:(localStorage.setItem(oe,JSON.stringify(c)),n)})}async getAuthHeaders(){return this.jwtStrategy.getAuthHeaders()}readAuthHeaders(e){this.jwtStrategy.readAuthHeaders(e)}},a(S,"AssistantCloudAnonymousAuthStrategy"),S);var _,ct=(_=class extends Error{constructor(e){super(e),this.name="APIError"}},a(_,"CloudAPIError"),_),b,qe=(b=class{constructor(e){u(this,"_auth");u(this,"_baseUrl");if("authToken"in e)this._baseUrl=e.baseUrl,this._auth=new ge(e.authToken);else if("apiKey"in e)this._baseUrl="https://backend.assistant-api.com",this._auth=new Fe(e.apiKey,e.userId,e.workspaceId);else if("anonymous"in e)this._baseUrl=e.baseUrl,this._auth=new De(e.baseUrl);else throw new Error("Invalid configuration: Must provide authToken, apiKey, or anonymous configuration")}async initializeAuth(){return!!this._auth.getAuthHeaders()}async makeRawRequest(e,t={}){let o=await this._auth.getAuthHeaders();if(!o)throw new Error("Authorization failed");let s={...o,...t.headers,"Content-Type":"application/json"},l=new URLSearchParams;if(t.query)for(let[c,d]of Object.entries(t.query))d!==!1&&(d===!0?l.set(c,"true"):l.set(c,d.toString()));let i=new URL(`${this._baseUrl}/v1${e}`);i.search=l.toString();let n=await fetch(i,{method:t.method??"GET",headers:s,body:t.body?JSON.stringify(t.body):null});if(this._auth.readAuthHeaders(n.headers),!n.ok){let c=await n.text();try{let d=JSON.parse(c);throw new ct(d.message)}catch{throw new Error(`Request failed with status ${n.status}, ${c}`)}}return n}async makeRequest(e,t={}){return(await this.makeRawRequest(e,t)).json()}},a(b,"AssistantCloudAPI"),b);var R,Le=(R=class{constructor(e){this.cloud=e}async create(){return this.cloud.makeRequest("/auth/tokens",{method:"POST"})}},a(R,"AssistantCloudAuthTokens"),R);var g={toResponse(r,e){return new Response(g.toByteStream(r,e),{headers:e.headers??{}})},fromResponse(r,e){return g.fromByteStream(r.body,e)},toByteStream(r,e){return r.pipeThrough(e)},fromByteStream(r,e){return r.pipeThrough(e)}};var ne=a(function(){let r,e,t=new Promise((o,s)=>{r=o,e=s});if(!r||!e)throw new Error("Failed to create promise");return{promise:t,resolve:r,reject:e}},"promiseWithResolvers");var Ne=a(()=>{let r=[],e=!1,t,o,s=a(i=>{i.promise||(i.promise=i.reader.read().then(({done:n,value:c})=>{i.promise=void 0,n?(r.splice(r.indexOf(i),1),e&&r.length===0&&t.close()):t.enqueue(c),o?.resolve(),o=void 0}).catch(n=>{console.error(n),r.forEach(c=>{c.reader.cancel()}),r.length=0,t.error(n),o?.reject(n),o=void 0}))},"handlePull");return{readable:new ReadableStream({start(i){t=i},pull(){return o=ne(),r.forEach(i=>{s(i)}),o.promise},cancel(){r.forEach(i=>{i.reader.cancel()}),r.length=0}}),isSealed(){return e},seal(){e=!0,r.length===0&&t.close()},addStream(i){if(e)throw new Error("Cannot add streams after the run callback has settled.");let n={reader:i.getReader()};r.push(n),s(n)},enqueue(i){this.addStream(new ReadableStream({start(n){n.enqueue(i),n.close()}}))}}},"createMergeStream");var k,Ue=(k=class{constructor(e){u(this,"_controller");u(this,"_isClosed",!1);this._controller=e}append(e){return this._controller.enqueue({type:"text-delta",path:[],textDelta:e}),this}close(){this._isClosed||(this._isClosed=!0,this._controller.enqueue({type:"part-finish",path:[]}),this._controller.close())}},a(k,"TextStreamControllerImpl"),k),ye=a(r=>new ReadableStream({start(e){return r.start?.(new Ue(e))},pull(e){return r.pull?.(new Ue(e))},cancel(e){return r.cancel?.(e)}}),"createTextStream"),we=a(()=>{let r;return[ye({start(t){r=t}}),r]},"createTextStreamController");var P,$e=(P=class{constructor(e){u(this,"_isClosed",!1);u(this,"_mergeTask");u(this,"_argsTextController");this._controller=e;let t=ye({start:a(s=>{this._argsTextController=s},"start")}),o=!1;this._mergeTask=t.pipeTo(new WritableStream({write:a(s=>{switch(s.type){case"text-delta":o=!0,this._controller.enqueue(s);break;case"part-finish":o||this._controller.enqueue({type:"text-delta",textDelta:"{}",path:[]}),this._controller.enqueue({type:"tool-call-args-text-finish",path:[]});break;default:throw new Error(`Unexpected chunk type: ${s.type}`)}},"write")}))}get argsText(){return this._argsTextController}async setResponse(e){this._argsTextController.close(),await Promise.resolve(),this._controller.enqueue({type:"result",path:[],...e.artifact!==void 0?{artifact:e.artifact}:{},result:e.result,isError:e.isError??!1})}async close(){this._isClosed||(this._isClosed=!0,this._argsTextController.close(),await this._mergeTask,this._controller.enqueue({type:"part-finish",path:[]}),this._controller.close())}},a(P,"ToolCallStreamControllerImpl"),P),ut=a(r=>new ReadableStream({start(e){return r.start?.(new $e(e))},pull(e){return r.pull?.(new $e(e))},cancel(e){return r.cancel?.(e)}}),"createToolCallStream"),He=a(()=>{let r;return[ut({start(t){r=t}}),r]},"createToolCallStreamController");var I,ie=(I=class{constructor(){u(this,"value",-1)}up(){return++this.value}},a(I,"Counter"),I);var v,Je=(v=class extends TransformStream{constructor(e){super({transform(t,o){o.enqueue({...t,path:[e,...t.path]})}})}},a(v,"PathAppendEncoder"),v),E,er=(E=class extends TransformStream{constructor(e){super({transform(t,o){let{path:[s,...l]}=t;if(e!==s)throw new Error(`Path mismatch: expected ${e}, got ${s}`);o.enqueue({...t,path:l})}})}},a(E,"PathAppendDecoder"),E),M,je=(M=class extends TransformStream{constructor(e){let t=new ie,o=new Map;super({transform(s,l){s.type==="part-start"&&s.path.length===0&&o.set(t.up(),e.up());let[i,...n]=s.path;if(i===void 0){l.enqueue(s);return}let c=o.get(i);if(c===void 0)throw new Error("Path not found");l.enqueue({...s,path:[c,...n]})}})}},a(M,"PathMergeEncoder"),M);var O,Be=(O=class extends TransformStream{constructor(e,t,o){let[s,l]=ee(),i;super({start(n){return i=s.pipeTo(new WritableStream({write(c){n.enqueue(c)},abort(c){n.error(c)},close(){n.terminate()}})).catch(c=>{n.error(c)}),e.start?.(l)},transform(n){return e.transform?.(n,l)},async flush(){await e.flush?.(l),l.close(),await i}},t,o)}},a(O,"AssistantTransformStream"),O);var F,le=(F=class extends TransformStream{constructor(e){super();let t=e(super.readable);Object.defineProperty(this,"readable",{value:t,writable:!1})}},a(F,"PipeableTransformStream"),F);var p=(r=>(r.TextDelta="0",r.Data="2",r.Error="3",r.Annotation="8",r.ToolCall="9",r.ToolCallResult="a",r.StartToolCall="b",r.ToolCallArgsTextDelta="c",r.FinishMessage="d",r.FinishStep="e",r.StartStep="f",r.ReasoningDelta="g",r.Source="h",r.RedactedReasoning="i",r.ReasoningSignature="j",r.File="k",r.AuiUpdateStateOperations="aui-state",r.AuiTextDelta="aui-text-delta",r.AuiReasoningDelta="aui-reasoning-delta",r))(p||{});var D,We=(D=class extends TransformStream{constructor(){super({transform:a((e,t)=>{t.enqueue(`${e.type}:${JSON.stringify(e.value)}
-`)},"transform")})}},a(D,"DataStreamChunkEncoder"),D),q,pt=(q=class extends TransformStream{constructor(){super({transform:a((e,t)=>{let o=e.indexOf(":");if(o===-1)throw new Error("Invalid stream part");t.enqueue({type:e.slice(0,o),value:JSON.parse(e.slice(o+1))})},"transform")})}},a(q,"DataStreamChunkDecoder"),q);var L,ze=(L=class extends TransformStream{constructor(){let e=[];super({transform(t,o){if(t.type==="part-start"){if(t.path.length!==0){o.error(new Error("Nested parts are not supported"));return}e.push(t.part),o.enqueue(t);return}if(t.type==="text-delta"||t.type==="result"||t.type==="part-finish"||t.type==="tool-call-args-text-finish"){if(t.path.length!==1){o.error(new Error(`${t.type} chunks must have a path of length 1`));return}let s=t.path[0];if(s<0||s>=e.length){o.error(new Error(`Invalid path index: ${s}`));return}let l=e[s];o.enqueue({...t,meta:l});return}o.enqueue(t)}})}},a(L,"AssistantMetaTransformStream"),L);var N,Te=(N=class extends le{constructor(){super(t=>{let o=new TransformStream({transform(s,l){let i=s.type;switch(i){case"part-start":{let n=s.part;if(n.type==="tool-call"){let{type:c,...d}=n;l.enqueue({type:p.StartToolCall,value:d})}if(n.type==="source"){let{type:c,...d}=n;l.enqueue({type:p.Source,value:d})}break}case"text-delta":{let n=s.meta;switch(n.type){case"text":{n.parentId?l.enqueue({type:p.AuiTextDelta,value:{textDelta:s.textDelta,parentId:n.parentId}}):l.enqueue({type:p.TextDelta,value:s.textDelta});break}case"reasoning":{n.parentId?l.enqueue({type:p.AuiReasoningDelta,value:{reasoningDelta:s.textDelta,parentId:n.parentId}}):l.enqueue({type:p.ReasoningDelta,value:s.textDelta});break}case"tool-call":{l.enqueue({type:p.ToolCallArgsTextDelta,value:{toolCallId:n.toolCallId,argsTextDelta:s.textDelta}});break}default:throw new Error(`Unsupported part type for text-delta: ${n.type}`)}break}case"result":{let n=s.meta;if(n.type!=="tool-call")throw new Error(`Result chunk on non-tool-call part not supported: ${n.type}`);l.enqueue({type:p.ToolCallResult,value:{toolCallId:n.toolCallId,result:s.result,artifact:s.artifact,...s.isError?{isError:s.isError}:{}}});break}case"step-start":{let{type:n,...c}=s;l.enqueue({type:p.StartStep,value:c});break}case"step-finish":{let{type:n,...c}=s;l.enqueue({type:p.FinishStep,value:c});break}case"message-finish":{let{type:n,...c}=s;l.enqueue({type:p.FinishMessage,value:c});break}case"error":{l.enqueue({type:p.Error,value:s.error});break}case"annotations":{l.enqueue({type:p.Annotation,value:s.annotations});break}case"data":{l.enqueue({type:p.Data,value:s.data});break}case"update-state":{l.enqueue({type:p.AuiUpdateStateOperations,value:s.operations});break}case"tool-call-args-text-finish":case"part-finish":break;default:{let n=i;throw new Error(`Unsupported chunk type: ${n}`)}}}});return t.pipeThrough(new ze).pipeThrough(o).pipeThrough(new We).pipeThrough(new TextEncoderStream)});u(this,"headers",new Headers({"Content-Type":"text/plain; charset=utf-8","x-vercel-ai-data-stream":"v1"}))}},a(N,"DataStreamEncoder"),N),yr=[p.StartToolCall,p.ToolCall,p.TextDelta,p.ReasoningDelta,p.Source,p.Error,p.FinishStep,p.FinishMessage,p.AuiTextDelta,p.AuiReasoningDelta];var Ge=a((r,e=21)=>(t=e)=>{let o="",s=t|0;for(;s--;)o+=r[Math.random()*r.length|0];return o},"customAlphabet");var Ke=Ge("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",7);var x,dt=(x=class{constructor(e){u(this,"_state");u(this,"_parentId");this._state=e||{merger:Ne(),contentCounter:new ie}}get __internal_isClosed(){return this._state.merger.isSealed()}__internal_getReadable(){return this._state.merger.readable}__internal_subscribeToClose(e){this._state.closeSubscriber=e}_addPart(e,t){this._state.append&&(this._state.append.controller.close(),this._state.append=void 0),this.enqueue({type:"part-start",part:e,path:[]}),this._state.merger.addStream(t.pipeThrough(new Je(this._state.contentCounter.value)))}merge(e){this._state.merger.addStream(e.pipeThrough(new je(this._state.contentCounter)))}appendText(e){this._state.append?.kind!=="text"&&(this._state.append={kind:"text",controller:this.addTextPart()}),this._state.append.controller.append(e)}appendReasoning(e){this._state.append?.kind!=="reasoning"&&(this._state.append={kind:"reasoning",controller:this.addReasoningPart()}),this._state.append.controller.append(e)}addTextPart(){let[e,t]=we();return this._addPart({type:"text"},e),t}addReasoningPart(){let[e,t]=we();return this._addPart({type:"reasoning"},e),t}addToolCallPart(e){let t=typeof e=="string"?{toolName:e}:e,o=t.toolName,s=t.toolCallId??Ke(),[l,i]=He();return this._addPart({type:"tool-call",toolName:o,toolCallId:s,...this._parentId&&{parentId:this._parentId}},l),t.argsText!==void 0&&(i.argsText.append(t.argsText),i.argsText.close()),t.args!==void 0&&(i.argsText.append(JSON.stringify(t.args)),i.argsText.close()),t.response!==void 0&&i.setResponse(t.response),i}appendSource(e){this._addPart({...e,...this._parentId&&{parentId:this._parentId}},new ReadableStream({start(t){t.enqueue({type:"part-finish",path:[]}),t.close()}}))}appendFile(e){this._addPart(e,new ReadableStream({start(t){t.enqueue({type:"part-finish",path:[]}),t.close()}}))}enqueue(e){this._state.merger.enqueue(e),e.type==="part-start"&&e.path.length===0&&this._state.contentCounter.up()}withParentId(e){let t=new x(this._state);return t._parentId=e,t}close(){this._state.append?.controller?.close(),this._state.merger.seal(),this._state.closeSubscriber?.()}},a(x,"_AssistantStreamControllerImpl"),x);function Ve(r){let e=new dt;return a(async()=>{try{await r(e)}catch(o){throw e.__internal_isClosed||e.enqueue({type:"error",path:[],error:String(o)}),o}finally{e.__internal_isClosed||e.close()}},"runTask")(),e.__internal_getReadable()}a(Ve,"createAssistantStream");function ee(){let{resolve:r,promise:e}=ne(),t;return[Ve(s=>(t=s,t.__internal_subscribeToClose(r),e)),t]}a(ee,"createAssistantStreamController");var U,te=(U=class extends le{constructor(){super(e=>{let t=new Be({transform(o,s){s.appendText(o)}});return e.pipeThrough(new TextDecoderStream).pipeThrough(t)})}},a(U,"PlainTextDecoder"),U);var $,Qe=($=class{constructor(e){this.cloud=e}__internal_getAssistantOptions(e){return{api:this.cloud._baseUrl+"/v1/runs/stream",headers:a(async()=>{let t=await this.cloud._auth.getAuthHeaders();if(!t)throw new Error("Authorization failed");return{...t,Accept:"text/plain"}},"headers"),body:{assistant_id:e,response_format:"vercel-ai-data-stream/v1",thread_id:"unstable_todo"}}}async stream(e){let t=await this.cloud.makeRawRequest("/runs/stream",{method:"POST",headers:{Accept:"text/plain"},body:e});return g.fromResponse(t,new te)}},a($,"AssistantCloudRuns"),$);var H,Xe=(H=class{constructor(e){this.cloud=e}async list(e,t){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}/messages`,{query:t})}async create(e,t){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}/messages`,{method:"POST",body:t})}},a(H,"AssistantCloudThreadMessages"),H);var J,Ye=(J=class{constructor(e){u(this,"messages");this.cloud=e,this.messages=new Xe(e)}async list(e){return this.cloud.makeRequest("/threads",{query:e})}async create(e){return this.cloud.makeRequest("/threads",{method:"POST",body:e})}async update(e,t){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}`,{method:"PUT",body:t})}async delete(e){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}`,{method:"DELETE"})}},a(J,"AssistantCloudThreads"),J);var j,Ze=(j=class{constructor(e){this.cloud=e}async pdfToImages(e){return this.cloud.makeRequest("/files/pdf-to-images",{method:"POST",body:e})}async generatePresignedUploadUrl(e){return this.cloud.makeRequest("/files/attachments/generate-presigned-upload-url",{method:"POST",body:e})}},a(j,"AssistantCloudFiles"),j);var B,xe=(B=class{constructor(e){u(this,"threads");u(this,"auth");u(this,"runs");u(this,"files");let t=new qe(e);this.threads=new Ye(t),this.auth={tokens:new Le(t)},this.runs=new Qe(t),this.files=new Ze(t)}},a(B,"AssistantCloud"),B);import{initLegitFs as xt}from"@legit-sdk/core";import Ct from"memfs";import{createContext as mt,useContext as ht,useEffect as ft,useMemo as gt,useState as Ce,useCallback as ce}from"react";import{jsx as Tt}from"react/jsx-runtime";var re="main",y="/.legit/branches",tt="/.legit/commits",et=a(r=>(r.startsWith("/")&&(r=r.slice(1)),`${y}/${re}/${r}`),"resolveThreadPath"),yt=a(r=>r instanceof Error?r:new Error(String(r??"Unknown error")),"normalizeError"),rt=mt(void 0);function wt({children:r}){let[e,t]=Ce(null),[o,s]=Ce(!0),[l,i]=Ce(null);ft(()=>{let m=!1,h=a(async()=>{try{let f=await w();m||(t(f),i(null),s(!1))}catch(f){if(!m){let ae=yt(f);t(null),i(ae),s(!1)}}},"load");return s(!0),i(null),h(),()=>{m=!0}},[]);let n=ce(async(m,h)=>{if(e){let f=et(m);await e.promises.writeFile(f,h)}},[e]),c=ce(async m=>{if(e){let h=await W(re);if(!h)return;let f=h.find(fe=>fe.oid===m),ae=h.find(fe=>fe.oid===f?.parentOids[0]);if(!f||!ae)return;let Pe=f?.parentOids[1],Ie=ae?.originBranchOid;return!Pe||!Ie?void 0:{newOid:Pe,oldOid:Ie}}},[e]),d=ce(async(m,h)=>{if(e)return await at(m,h)},[e]),T=ce(async m=>(e&&(console.log("rollback",m),await st(re,m)),await ue(re)),[e]),z=gt(()=>({legitFs:e,loading:o,error:l,saveData:n,getMessageDiff:c,getPastState:d,rollback:T,threadId:re,resolvePath:et}),[e,o,l,n]);return Tt(rt.Provider,{value:z,children:r})}a(wt,"LegitFsProvider");function Ae(){let r=ht(rt);if(r===void 0)throw new Error("useLegitFs must be used within a LegitFsProvider");return r}a(Ae,"useLegitFs");var pe=null;async function w(){return pe||(pe=xt(Ct,"/"),window&&(window.legitFs=await pe)),pe}a(w,"getLegitFs");async function ot(){let r=await w();try{return(await r.promises.readdir(y)).filter(t=>!t.startsWith("."))}catch(e){if(e.code==="ENOENT")return await r.promises.mkdir(y,{recursive:!0}),[];throw e}}a(ot,"listBranches");async function ue(r){try{return await(await w()).promises.readFile(`${y}/${r}/.legit/head`,"utf8")}catch(e){throw console.error("Error reading head from path:",r,e),e}}a(ue,"readHead");async function st(r,e){try{await(await w()).promises.writeFile(`${y}/${r}/.legit/operationHead`,e,"utf8")}catch(t){throw console.error("Error writing head to path:",r,t),t}}a(st,"writeOperationHead");async function nt(r,e){try{await(await w()).promises.writeFile(`${y}/${r}/.legit/operation`,e,"utf8")}catch(t){throw console.error("Error writing operation to path:",r,t),t}}a(nt,"writeOperation");async function W(r){try{let t=await(await w()).promises.readFile(`${y}/${r}/.legit/operationHistory`,"utf8");return t.length>0?JSON.parse(t):JSON.parse("[]")}catch(e){throw console.error("Error reading operation history from path:",r,e),e}}a(W,"readOperationHistory");async function at(r,e){try{e.startsWith("/")&&(e=e.slice(1));let t=await w(),o=`${tt}/${r.slice(0,2)}/${r.slice(2)}/${e}`;return await t.promises.readFile(o,"utf8")}catch(t){throw console.error("Error reading past state from path:",r,t),t}}a(at,"readPastState");var de=a(async r=>{let e=[];for(let t of r){let o,s;try{o=JSON.parse(t.message),s={id:t.oid,height:e.length,parent_id:t.parentOids[0]===t.parentOids[1]?null:t.parentOids[0],metadata:{custom:{depending_branch_commit_id:t.originBranchOid??null}},...o},e.unshift(s)}catch{console.log("Skipping operation with invalid JSON message:",t);continue}}return e},"createUIMessages");var Se=class Se{async list(e,t){let o=await W(e);return{messages:(await de(o)).map((i,n)=>({...i,height:n,created_at:new Date(i.created_at),updated_at:new Date(i.updated_at)}))}}async create(e,t){let o=`msg-${Date.now()}-${Math.random().toString(36).slice(2)}`,s={created_at:new Date,updated_at:new Date,format:t.format,content:t.content};return await nt(e,JSON.stringify(s)),{message_id:o}}};a(Se,"LegitThreadMessages");var me=Se;var _e=class _e{constructor(){u(this,"messages",new me)}async list(e){return{threads:(await ot()).map(o=>({id:o,title:o,last_message_at:new Date,metadata:{},external_id:null,project_id:"",created_at:new Date,updated_at:new Date,workspace_id:"local-workspace",is_archived:!1}))}}async create(e){return{thread_id:"main"}}async update(e,t){}async delete(e){}};a(_e,"LegitThreads");var he=_e;var Re=class Re extends xe{constructor(t){super(t);u(this,"runtime");u(this,"legitFs",null);this.legitFs=t.legitFs,this.threads=new he;let o=this.runs.stream;this.runs.stream=async s=>{if(!this.runtime)throw new Error("LegitLocal runtime not set");let n=this.runtime.thread.__internal_threadBinding?.getState?.()?.adapters?.chatModel?.options??{},c=new Headers;c.set("Content-Type","application/json");try{let d=await fetch(n.api??"/api/chat",{method:"POST",headers:c,credentials:n.credentials??"same-origin",body:JSON.stringify({messages:s.messages,system:s.system,tools:s.tools,runConfig:n.runConfig??{}})});if(!d.ok)throw new Error(`Request failed with status ${d.status}`);return g.fromResponse(d,new te)}catch(d){return console.error("LegitLocal stream failed, falling back to default handler.",d),o.call(this.runs,s)}}}};a(Re,"LegitLocal");var be=Re;import{ExportedMessageRepository as At}from"@assistant-ui/react";var St=a(r=>{if(r&&typeof r=="object"){let e=r,t=e.custom&&typeof e.custom=="object"?e.custom:{};return{...e,custom:t}}return{custom:{}}},"normalizeMetadata"),_t="aui/v0";function bt(r){if(r.format!==_t)return null;let e=r.content;if(!e||typeof e!="object")return null;let t=e.role;if(t!=="assistant"&&t!=="user"&&t!=="system")return null;let o=e.content??[],s=typeof o=="string"||Array.isArray(o)?o:[],l=e.createdAt instanceof Date?e.createdAt:e.createdAt?new Date(e.createdAt):new Date(r.created_at),i=St(r.metadata);return{...e,role:t,content:s,id:e.id??r.id,createdAt:l,metadata:i}}a(bt,"toThreadMessageLike");async function ke({threadId:r,threadApi:e}){console.log("sync initiated");let t=await W(r),s=(await de(t)).map(bt).filter(l=>l!==null);e?.reset?.(s),e.import(At.fromArray(s))}a(ke,"syncThreadFromLegitFs");var it=a(r=>{if(typeof r=="string")return r;if(r===void 0)return"";try{return JSON.stringify(r)}catch{return""}},"serializeStructuredData");var se=a(r=>r==null?null:typeof r=="string"||typeof r=="number"||typeof r=="boolean"?r:Array.isArray(r)?r.map(e=>se(e)):typeof r=="object"?Object.fromEntries(Object.entries(r).map(([e,t])=>[e,se(t)])):JSON.stringify(r),"toJsonValue"),Rt=a((r,e)=>e?typeof r=="string"?{type:"error-text",value:r}:{type:"error-json",value:se(r)}:typeof r=="string"?{type:"text",value:r}:{type:"json",value:se(r)},"buildToolResultOutput"),kt=a(r=>{switch(r.type){case"text":return{type:"text",text:typeof r.text=="string"?r.text:""};case"tool-call":return{type:"tool-call",toolCallId:typeof r.toolCallId=="string"?r.toolCallId:`unknown-${Date.now()}`,toolName:typeof r.toolName=="string"?r.toolName:"unknown",input:se(r.args??r.input??{})};default:return null}},"mapAssistantContentPart"),Pt=a(r=>r.type!=="tool-result"?null:{type:"tool-result",toolCallId:typeof r.toolCallId=="string"?r.toolCallId:`unknown-${Date.now()}`,toolName:typeof r.toolName=="string"?r.toolName:"unknown",output:Rt(r.result??r.output,!!r.isError)},"mapToolContentPart"),js=a(r=>r.map(e=>{let t=Array.isArray(e.content)?e.content:[];switch(e.role){case"system":return{role:"system",content:t.map(s=>s.type==="text"&&typeof s.text=="string"?s.text:"").join("")};case"user":return{role:"user",content:t.map(s=>{switch(s.type){case"text":return{type:"text",text:typeof s.text=="string"?s.text:""};default:return null}}).filter(Boolean)};case"assistant":return{role:"assistant",content:t.map(s=>kt(s)).filter(Boolean)};case"tool":{let o=t.map(s=>Pt(s)).filter(Boolean);return o.length>0?{role:"tool",content:o}:null}default:return null}}).filter(e=>e!==null),"convertMessagesToModelMessages");async function Bs(r){let[e,t]=ee();(async()=>{try{let s=null,l=new Map;for await(let i of r.fullStream)switch(i.type){case"text-start":s=t.addTextPart();break;case"text-delta":s||(s=t.addTextPart()),s.append(i.text);break;case"text-end":s?.close(),s=null;break;case"tool-input-start":{let n=i.id;if(!l.has(n)){let c=t.addToolCallPart({toolCallId:n,toolName:i.toolName});l.set(n,c)}break}case"tool-input-delta":{let n=i.id,c=l.get(n);c&&c.argsText.append(i.delta);break}case"tool-input-end":{let n=i.id,c=l.get(n);c&&c.argsText.close();break}case"tool-call":{let n=i.toolCallId;if(!l.get(n)){let d=t.addToolCallPart({toolCallId:n,toolName:i.toolName,args:i.input});l.set(n,d)}break}case"tool-result":{let n=i.toolCallId,c=l.get(n);c&&c.setResponse({result:it(i.output),isError:!1});break}case"tool-error":{let n=i.toolCallId,c=l.get(n);c&&c.setResponse({result:it(i.error),isError:!0});break}case"start-step":{let n=`msg-${Date.now()}-${Math.random().toString(36).slice(2)}`;t.enqueue({type:"step-start",path:[],messageId:n});break}case"finish-step":{let n=i.usage;t.enqueue({type:"step-finish",path:[],finishReason:i.finishReason,usage:{promptTokens:n.promptTokens??n.prompt?.tokens??0,completionTokens:n.completionTokens??n.completion?.tokens??0},isContinued:!1});break}case"finish":{let n=i.totalUsage;t.enqueue({type:"message-finish",path:[],finishReason:i.finishReason,usage:{promptTokens:n.promptTokens??n.prompt?.tokens??0,completionTokens:n.completionTokens??n.completion?.tokens??0}});break}case"error":{t.enqueue({type:"error",path:[],error:i.error instanceof Error?i.error.message:typeof i.error=="string"?i.error:"Unknown error"});break}case"start":case"abort":case"raw":case"reasoning-start":case"reasoning-delta":case"reasoning-end":case"source":case"file":break}s?.close(),l.forEach(i=>i.close()),t.close()}catch(s){t.enqueue({type:"error",path:[],error:s instanceof Error?s.message:typeof s=="string"?s:"Unknown error"}),t.close()}})();let o=new Te;return g.toResponse(e,o)}a(Bs,"streamTextToDataStreamResponse");import{useEffect as It}from"react";import{useAssistantApi as vt}from"@assistant-ui/react";var Et=100;function Ys(){let{legitFs:r,threadId:e}=Ae(),o=vt().thread();return It(()=>{if(!r||!e)return;let s=!1,l,i=null,n=a(()=>{s||(l=setTimeout(c,Et))},"scheduleNext"),c=a(async()=>{try{let d=await ue(e);if(s||d===i)return;let{isRunning:T,isLoading:z}=o.getState();if(T||z)return;setTimeout(async()=>{await ke({threadId:e,threadApi:o}),i=d},100)}catch(d){d.code!=="ENOENT"&&console.error("Failed to poll Legit head",d)}finally{n()}},"tick");return n(),()=>{s=!0,l!==void 0&&clearTimeout(l)}},[r,o,e]),null}a(Ys,"LegitFsHeadPoller");export{Ys as LegitFsHeadPoller,wt as LegitFsProvider,be as LegitLocal,js as convertMessagesToModelMessages,Bs as streamTextToDataStreamResponse,ke as syncThreadFromLegitFs,Ae as useLegitFs};
+`)},"transform")})}},a(D,"DataStreamChunkEncoder"),D),q,pt=(q=class extends TransformStream{constructor(){super({transform:a((e,t)=>{let o=e.indexOf(":");if(o===-1)throw new Error("Invalid stream part");t.enqueue({type:e.slice(0,o),value:JSON.parse(e.slice(o+1))})},"transform")})}},a(q,"DataStreamChunkDecoder"),q);var L,ze=(L=class extends TransformStream{constructor(){let e=[];super({transform(t,o){if(t.type==="part-start"){if(t.path.length!==0){o.error(new Error("Nested parts are not supported"));return}e.push(t.part),o.enqueue(t);return}if(t.type==="text-delta"||t.type==="result"||t.type==="part-finish"||t.type==="tool-call-args-text-finish"){if(t.path.length!==1){o.error(new Error(`${t.type} chunks must have a path of length 1`));return}let s=t.path[0];if(s<0||s>=e.length){o.error(new Error(`Invalid path index: ${s}`));return}let l=e[s];o.enqueue({...t,meta:l});return}o.enqueue(t)}})}},a(L,"AssistantMetaTransformStream"),L);var N,Te=(N=class extends le{constructor(){super(t=>{let o=new TransformStream({transform(s,l){let i=s.type;switch(i){case"part-start":{let n=s.part;if(n.type==="tool-call"){let{type:c,...d}=n;l.enqueue({type:p.StartToolCall,value:d})}if(n.type==="source"){let{type:c,...d}=n;l.enqueue({type:p.Source,value:d})}break}case"text-delta":{let n=s.meta;switch(n.type){case"text":{n.parentId?l.enqueue({type:p.AuiTextDelta,value:{textDelta:s.textDelta,parentId:n.parentId}}):l.enqueue({type:p.TextDelta,value:s.textDelta});break}case"reasoning":{n.parentId?l.enqueue({type:p.AuiReasoningDelta,value:{reasoningDelta:s.textDelta,parentId:n.parentId}}):l.enqueue({type:p.ReasoningDelta,value:s.textDelta});break}case"tool-call":{l.enqueue({type:p.ToolCallArgsTextDelta,value:{toolCallId:n.toolCallId,argsTextDelta:s.textDelta}});break}default:throw new Error(`Unsupported part type for text-delta: ${n.type}`)}break}case"result":{let n=s.meta;if(n.type!=="tool-call")throw new Error(`Result chunk on non-tool-call part not supported: ${n.type}`);l.enqueue({type:p.ToolCallResult,value:{toolCallId:n.toolCallId,result:s.result,artifact:s.artifact,...s.isError?{isError:s.isError}:{}}});break}case"step-start":{let{type:n,...c}=s;l.enqueue({type:p.StartStep,value:c});break}case"step-finish":{let{type:n,...c}=s;l.enqueue({type:p.FinishStep,value:c});break}case"message-finish":{let{type:n,...c}=s;l.enqueue({type:p.FinishMessage,value:c});break}case"error":{l.enqueue({type:p.Error,value:s.error});break}case"annotations":{l.enqueue({type:p.Annotation,value:s.annotations});break}case"data":{l.enqueue({type:p.Data,value:s.data});break}case"update-state":{l.enqueue({type:p.AuiUpdateStateOperations,value:s.operations});break}case"tool-call-args-text-finish":case"part-finish":break;default:{let n=i;throw new Error(`Unsupported chunk type: ${n}`)}}}});return t.pipeThrough(new ze).pipeThrough(o).pipeThrough(new We).pipeThrough(new TextEncoderStream)});u(this,"headers",new Headers({"Content-Type":"text/plain; charset=utf-8","x-vercel-ai-data-stream":"v1"}))}},a(N,"DataStreamEncoder"),N),yr=[p.StartToolCall,p.ToolCall,p.TextDelta,p.ReasoningDelta,p.Source,p.Error,p.FinishStep,p.FinishMessage,p.AuiTextDelta,p.AuiReasoningDelta];var Ge=a((r,e=21)=>(t=e)=>{let o="",s=t|0;for(;s--;)o+=r[Math.random()*r.length|0];return o},"customAlphabet");var Ke=Ge("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",7);var x,dt=(x=class{constructor(e){u(this,"_state");u(this,"_parentId");this._state=e||{merger:Ne(),contentCounter:new ie}}get __internal_isClosed(){return this._state.merger.isSealed()}__internal_getReadable(){return this._state.merger.readable}__internal_subscribeToClose(e){this._state.closeSubscriber=e}_addPart(e,t){this._state.append&&(this._state.append.controller.close(),this._state.append=void 0),this.enqueue({type:"part-start",part:e,path:[]}),this._state.merger.addStream(t.pipeThrough(new Je(this._state.contentCounter.value)))}merge(e){this._state.merger.addStream(e.pipeThrough(new je(this._state.contentCounter)))}appendText(e){this._state.append?.kind!=="text"&&(this._state.append={kind:"text",controller:this.addTextPart()}),this._state.append.controller.append(e)}appendReasoning(e){this._state.append?.kind!=="reasoning"&&(this._state.append={kind:"reasoning",controller:this.addReasoningPart()}),this._state.append.controller.append(e)}addTextPart(){let[e,t]=we();return this._addPart({type:"text"},e),t}addReasoningPart(){let[e,t]=we();return this._addPart({type:"reasoning"},e),t}addToolCallPart(e){let t=typeof e=="string"?{toolName:e}:e,o=t.toolName,s=t.toolCallId??Ke(),[l,i]=He();return this._addPart({type:"tool-call",toolName:o,toolCallId:s,...this._parentId&&{parentId:this._parentId}},l),t.argsText!==void 0&&(i.argsText.append(t.argsText),i.argsText.close()),t.args!==void 0&&(i.argsText.append(JSON.stringify(t.args)),i.argsText.close()),t.response!==void 0&&i.setResponse(t.response),i}appendSource(e){this._addPart({...e,...this._parentId&&{parentId:this._parentId}},new ReadableStream({start(t){t.enqueue({type:"part-finish",path:[]}),t.close()}}))}appendFile(e){this._addPart(e,new ReadableStream({start(t){t.enqueue({type:"part-finish",path:[]}),t.close()}}))}enqueue(e){this._state.merger.enqueue(e),e.type==="part-start"&&e.path.length===0&&this._state.contentCounter.up()}withParentId(e){let t=new x(this._state);return t._parentId=e,t}close(){this._state.append?.controller?.close(),this._state.merger.seal(),this._state.closeSubscriber?.()}},a(x,"_AssistantStreamControllerImpl"),x);function Ve(r){let e=new dt;return a(async()=>{try{await r(e)}catch(o){throw e.__internal_isClosed||e.enqueue({type:"error",path:[],error:String(o)}),o}finally{e.__internal_isClosed||e.close()}},"runTask")(),e.__internal_getReadable()}a(Ve,"createAssistantStream");function ee(){let{resolve:r,promise:e}=ne(),t;return[Ve(s=>(t=s,t.__internal_subscribeToClose(r),e)),t]}a(ee,"createAssistantStreamController");var U,te=(U=class extends le{constructor(){super(e=>{let t=new Be({transform(o,s){s.appendText(o)}});return e.pipeThrough(new TextDecoderStream).pipeThrough(t)})}},a(U,"PlainTextDecoder"),U);var $,Qe=($=class{constructor(e){this.cloud=e}__internal_getAssistantOptions(e){return{api:this.cloud._baseUrl+"/v1/runs/stream",headers:a(async()=>{let t=await this.cloud._auth.getAuthHeaders();if(!t)throw new Error("Authorization failed");return{...t,Accept:"text/plain"}},"headers"),body:{assistant_id:e,response_format:"vercel-ai-data-stream/v1",thread_id:"unstable_todo"}}}async stream(e){let t=await this.cloud.makeRawRequest("/runs/stream",{method:"POST",headers:{Accept:"text/plain"},body:e});return g.fromResponse(t,new te)}},a($,"AssistantCloudRuns"),$);var H,Xe=(H=class{constructor(e){this.cloud=e}async list(e,t){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}/messages`,{query:t})}async create(e,t){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}/messages`,{method:"POST",body:t})}},a(H,"AssistantCloudThreadMessages"),H);var J,Ye=(J=class{constructor(e){u(this,"messages");this.cloud=e,this.messages=new Xe(e)}async list(e){return this.cloud.makeRequest("/threads",{query:e})}async create(e){return this.cloud.makeRequest("/threads",{method:"POST",body:e})}async update(e,t){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}`,{method:"PUT",body:t})}async delete(e){return this.cloud.makeRequest(`/threads/${encodeURIComponent(e)}`,{method:"DELETE"})}},a(J,"AssistantCloudThreads"),J);var j,Ze=(j=class{constructor(e){this.cloud=e}async pdfToImages(e){return this.cloud.makeRequest("/files/pdf-to-images",{method:"POST",body:e})}async generatePresignedUploadUrl(e){return this.cloud.makeRequest("/files/attachments/generate-presigned-upload-url",{method:"POST",body:e})}},a(j,"AssistantCloudFiles"),j);var B,xe=(B=class{constructor(e){u(this,"threads");u(this,"auth");u(this,"runs");u(this,"files");let t=new qe(e);this.threads=new Ye(t),this.auth={tokens:new Le(t)},this.runs=new Qe(t),this.files=new Ze(t)}},a(B,"AssistantCloud"),B);import{openLegitFs as xt}from"@legit-sdk/core";import Ct from"memfs";import{createContext as mt,useContext as ht,useEffect as ft,useMemo as gt,useState as Ce,useCallback as ce}from"react";import{jsx as Tt}from"react/jsx-runtime";var re="main",y="/.legit/branches",tt="/.legit/commits",et=a(r=>(r.startsWith("/")&&(r=r.slice(1)),`${y}/${re}/${r}`),"resolveThreadPath"),yt=a(r=>r instanceof Error?r:new Error(String(r??"Unknown error")),"normalizeError"),rt=mt(void 0);function wt({children:r}){let[e,t]=Ce(null),[o,s]=Ce(!0),[l,i]=Ce(null);ft(()=>{let m=!1,h=a(async()=>{try{let f=await w();m||(t(f),i(null),s(!1))}catch(f){if(!m){let ae=yt(f);t(null),i(ae),s(!1)}}},"load");return s(!0),i(null),h(),()=>{m=!0}},[]);let n=ce(async(m,h)=>{if(e){let f=et(m);await e.promises.writeFile(f,h)}},[e]),c=ce(async m=>{if(e){let h=await W(re);if(!h)return;let f=h.find(fe=>fe.oid===m),ae=h.find(fe=>fe.oid===f?.parentOids[0]);if(!f||!ae)return;let Pe=f?.parentOids[1],Ie=ae?.originBranchOid;return!Pe||!Ie?void 0:{newOid:Pe,oldOid:Ie}}},[e]),d=ce(async(m,h)=>{if(e)return await at(m,h)},[e]),T=ce(async m=>(e&&(console.log("rollback",m),await st(re,m)),await ue(re)),[e]),z=gt(()=>({legitFs:e,loading:o,error:l,saveData:n,getMessageDiff:c,getPastState:d,rollback:T,threadId:re,resolvePath:et}),[e,o,l,n]);return Tt(rt.Provider,{value:z,children:r})}a(wt,"LegitFsProvider");function Ae(){let r=ht(rt);if(r===void 0)throw new Error("useLegitFs must be used within a LegitFsProvider");return r}a(Ae,"useLegitFs");var pe=null;async function w(){return pe||(pe=xt({storageFs:Ct,gitRoot:"/"}),window&&(window.legitFs=await pe)),pe}a(w,"getLegitFs");async function ot(){let r=await w();try{return(await r.promises.readdir(y)).filter(t=>!t.startsWith("."))}catch(e){if(e.code==="ENOENT")return await r.promises.mkdir(y,{recursive:!0}),[];throw e}}a(ot,"listBranches");async function ue(r){try{return await(await w()).promises.readFile(`${y}/${r}/.legit/head`,"utf8")}catch(e){throw console.error("Error reading head from path:",r,e),e}}a(ue,"readHead");async function st(r,e){try{await(await w()).promises.writeFile(`${y}/${r}/.legit/operationHead`,e,"utf8")}catch(t){throw console.error("Error writing head to path:",r,t),t}}a(st,"writeOperationHead");async function nt(r,e){try{await(await w()).promises.writeFile(`${y}/${r}/.legit/operation`,e,"utf8")}catch(t){throw console.error("Error writing operation to path:",r,t),t}}a(nt,"writeOperation");async function W(r){try{let t=await(await w()).promises.readFile(`${y}/${r}/.legit/operationHistory`,"utf8");return t.length>0?JSON.parse(t):JSON.parse("[]")}catch(e){throw console.error("Error reading operation history from path:",r,e),e}}a(W,"readOperationHistory");async function at(r,e){try{e.startsWith("/")&&(e=e.slice(1));let t=await w(),o=`${tt}/${r.slice(0,2)}/${r.slice(2)}/${e}`;return await t.promises.readFile(o,"utf8")}catch(t){throw console.error("Error reading past state from path:",r,t),t}}a(at,"readPastState");var de=a(async r=>{let e=[];for(let t of r){let o,s;try{o=JSON.parse(t.message),s={id:t.oid,height:e.length,parent_id:t.parentOids[0]===t.parentOids[1]?null:t.parentOids[0],metadata:{custom:{depending_branch_commit_id:t.originBranchOid??null}},...o},e.unshift(s)}catch{console.log("Skipping operation with invalid JSON message:",t);continue}}return e},"createUIMessages");var Se=class Se{async list(e,t){let o=await W(e);return{messages:(await de(o)).map((i,n)=>({...i,height:n,created_at:new Date(i.created_at),updated_at:new Date(i.updated_at)}))}}async create(e,t){let o=`msg-${Date.now()}-${Math.random().toString(36).slice(2)}`,s={created_at:new Date,updated_at:new Date,format:t.format,content:t.content};return await nt(e,JSON.stringify(s)),{message_id:o}}};a(Se,"LegitThreadMessages");var me=Se;var _e=class _e{constructor(){u(this,"messages",new me)}async list(e){return{threads:(await ot()).map(o=>({id:o,title:o,last_message_at:new Date,metadata:{},external_id:null,project_id:"",created_at:new Date,updated_at:new Date,workspace_id:"local-workspace",is_archived:!1}))}}async create(e){return{thread_id:"main"}}async update(e,t){}async delete(e){}};a(_e,"LegitThreads");var he=_e;var Re=class Re extends xe{constructor(t){super(t);u(this,"runtime");u(this,"legitFs",null);this.legitFs=t.legitFs,this.threads=new he;let o=this.runs.stream;this.runs.stream=async s=>{if(!this.runtime)throw new Error("LegitLocal runtime not set");let n=this.runtime.thread.__internal_threadBinding?.getState?.()?.adapters?.chatModel?.options??{},c=new Headers;c.set("Content-Type","application/json");try{let d=await fetch(n.api??"/api/chat",{method:"POST",headers:c,credentials:n.credentials??"same-origin",body:JSON.stringify({messages:s.messages,system:s.system,tools:s.tools,runConfig:n.runConfig??{}})});if(!d.ok)throw new Error(`Request failed with status ${d.status}`);return g.fromResponse(d,new te)}catch(d){return console.error("LegitLocal stream failed, falling back to default handler.",d),o.call(this.runs,s)}}}};a(Re,"LegitLocal");var be=Re;import{ExportedMessageRepository as At}from"@assistant-ui/react";var St=a(r=>{if(r&&typeof r=="object"){let e=r,t=e.custom&&typeof e.custom=="object"?e.custom:{};return{...e,custom:t}}return{custom:{}}},"normalizeMetadata"),_t="aui/v0";function bt(r){if(r.format!==_t)return null;let e=r.content;if(!e||typeof e!="object")return null;let t=e.role;if(t!=="assistant"&&t!=="user"&&t!=="system")return null;let o=e.content??[],s=typeof o=="string"||Array.isArray(o)?o:[],l=e.createdAt instanceof Date?e.createdAt:e.createdAt?new Date(e.createdAt):new Date(r.created_at),i=St(r.metadata);return{...e,role:t,content:s,id:e.id??r.id,createdAt:l,metadata:i}}a(bt,"toThreadMessageLike");async function ke({threadId:r,threadApi:e}){console.log("sync initiated");let t=await W(r),s=(await de(t)).map(bt).filter(l=>l!==null);e?.reset?.(s),e.import(At.fromArray(s))}a(ke,"syncThreadFromLegitFs");var it=a(r=>{if(typeof r=="string")return r;if(r===void 0)return"";try{return JSON.stringify(r)}catch{return""}},"serializeStructuredData");var se=a(r=>r==null?null:typeof r=="string"||typeof r=="number"||typeof r=="boolean"?r:Array.isArray(r)?r.map(e=>se(e)):typeof r=="object"?Object.fromEntries(Object.entries(r).map(([e,t])=>[e,se(t)])):JSON.stringify(r),"toJsonValue"),Rt=a((r,e)=>e?typeof r=="string"?{type:"error-text",value:r}:{type:"error-json",value:se(r)}:typeof r=="string"?{type:"text",value:r}:{type:"json",value:se(r)},"buildToolResultOutput"),kt=a(r=>{switch(r.type){case"text":return{type:"text",text:typeof r.text=="string"?r.text:""};case"tool-call":return{type:"tool-call",toolCallId:typeof r.toolCallId=="string"?r.toolCallId:`unknown-${Date.now()}`,toolName:typeof r.toolName=="string"?r.toolName:"unknown",input:se(r.args??r.input??{})};default:return null}},"mapAssistantContentPart"),Pt=a(r=>r.type!=="tool-result"?null:{type:"tool-result",toolCallId:typeof r.toolCallId=="string"?r.toolCallId:`unknown-${Date.now()}`,toolName:typeof r.toolName=="string"?r.toolName:"unknown",output:Rt(r.result??r.output,!!r.isError)},"mapToolContentPart"),js=a(r=>r.map(e=>{let t=Array.isArray(e.content)?e.content:[];switch(e.role){case"system":return{role:"system",content:t.map(s=>s.type==="text"&&typeof s.text=="string"?s.text:"").join("")};case"user":return{role:"user",content:t.map(s=>{switch(s.type){case"text":return{type:"text",text:typeof s.text=="string"?s.text:""};default:return null}}).filter(Boolean)};case"assistant":return{role:"assistant",content:t.map(s=>kt(s)).filter(Boolean)};case"tool":{let o=t.map(s=>Pt(s)).filter(Boolean);return o.length>0?{role:"tool",content:o}:null}default:return null}}).filter(e=>e!==null),"convertMessagesToModelMessages");async function Bs(r){let[e,t]=ee();(async()=>{try{let s=null,l=new Map;for await(let i of r.fullStream)switch(i.type){case"text-start":s=t.addTextPart();break;case"text-delta":s||(s=t.addTextPart()),s.append(i.text);break;case"text-end":s?.close(),s=null;break;case"tool-input-start":{let n=i.id;if(!l.has(n)){let c=t.addToolCallPart({toolCallId:n,toolName:i.toolName});l.set(n,c)}break}case"tool-input-delta":{let n=i.id,c=l.get(n);c&&c.argsText.append(i.delta);break}case"tool-input-end":{let n=i.id,c=l.get(n);c&&c.argsText.close();break}case"tool-call":{let n=i.toolCallId;if(!l.get(n)){let d=t.addToolCallPart({toolCallId:n,toolName:i.toolName,args:i.input});l.set(n,d)}break}case"tool-result":{let n=i.toolCallId,c=l.get(n);c&&c.setResponse({result:it(i.output),isError:!1});break}case"tool-error":{let n=i.toolCallId,c=l.get(n);c&&c.setResponse({result:it(i.error),isError:!0});break}case"start-step":{let n=`msg-${Date.now()}-${Math.random().toString(36).slice(2)}`;t.enqueue({type:"step-start",path:[],messageId:n});break}case"finish-step":{let n=i.usage;t.enqueue({type:"step-finish",path:[],finishReason:i.finishReason,usage:{promptTokens:n.promptTokens??n.prompt?.tokens??0,completionTokens:n.completionTokens??n.completion?.tokens??0},isContinued:!1});break}case"finish":{let n=i.totalUsage;t.enqueue({type:"message-finish",path:[],finishReason:i.finishReason,usage:{promptTokens:n.promptTokens??n.prompt?.tokens??0,completionTokens:n.completionTokens??n.completion?.tokens??0}});break}case"error":{t.enqueue({type:"error",path:[],error:i.error instanceof Error?i.error.message:typeof i.error=="string"?i.error:"Unknown error"});break}case"start":case"abort":case"raw":case"reasoning-start":case"reasoning-delta":case"reasoning-end":case"source":case"file":break}s?.close(),l.forEach(i=>i.close()),t.close()}catch(s){t.enqueue({type:"error",path:[],error:s instanceof Error?s.message:typeof s=="string"?s:"Unknown error"}),t.close()}})();let o=new Te;return g.toResponse(e,o)}a(Bs,"streamTextToDataStreamResponse");import{useEffect as It}from"react";import{useAssistantApi as vt}from"@assistant-ui/react";var Et=100;function Ys(){let{legitFs:r,threadId:e}=Ae(),o=vt().thread();return It(()=>{if(!r||!e)return;let s=!1,l,i=null,n=a(()=>{s||(l=setTimeout(c,Et))},"scheduleNext"),c=a(async()=>{try{let d=await ue(e);if(s||d===i)return;let{isRunning:T,isLoading:z}=o.getState();if(T||z)return;setTimeout(async()=>{await ke({threadId:e,threadApi:o}),i=d},100)}catch(d){d.code!=="ENOENT"&&console.error("Failed to poll Legit head",d)}finally{n()}},"tick");return n(),()=>{s=!0,l!==void 0&&clearTimeout(l)}},[r,o,e]),null}a(Ys,"LegitFsHeadPoller");export{Ys as LegitFsHeadPoller,wt as LegitFsProvider,be as LegitLocal,js as convertMessagesToModelMessages,Bs as streamTextToDataStreamResponse,ke as syncThreadFromLegitFs,Ae as useLegitFs};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@legit-sdk/assistant-ui",
-  "version": "0.1.3",
+  "version": "0.2.0",
   "description": "Legit integration for Assistant UI",
   "type": "module",
   "main": "./dist/index.js",
@@ -46,7 +46,7 @@
   },
   "dependencies": {
     "ai": "^5.0.59",
-    "@legit-sdk/core": "0.1.11"
+    "@legit-sdk/core": "0.2.0"
   },
   "publishConfig": {
     "access": "public"