@marimo-team/frontend 0.25.1-dev43 → 0.25.1-dev45

package/dist/assets/{abnfDiagram-VCTEODGH-7LmalJX2.js → abnfDiagram-VCTEODGH-CSPJSGaM.js}

@@ -1 +1 @@
-import{n as e}from"./chunk-Y2CYZVJY-DsF7k-Jl.js";import{t}from"./src-Cyu5G-Tr.js";import"./chunk-DU6HZSFF-BlpgvKzl.js";import"./mermaid-Dj_W1msf.js";import{m as n,t as r}from"./mermaid-parser.core-3RKOfUXB.js";import{n as i,r as a,t as o}from"./chunk-SVP7TREG-rqVzqvaq.js";import{t as s}from"./chunk-JWPE2WC7-DVXcaiue.js";var c=n().RailroadAbnf.parser.LangiumParser,l=e(e=>{let t=e.alternatives.map(u);return t.length===1?t[0]:{type:`choice`,alternatives:t}},`transformAlternation`),u=e(e=>{let t=e.elements.map(f);return t.length===1?t[0]:{type:`sequence`,elements:t}},`transformConcatenation`),d=e(e=>{if(e.includes(`*`)){let[t,n]=e.split(`*`);return{min:t?parseInt(t,10):0,max:n?parseInt(n,10):1/0}}let t=parseInt(e,10);return{min:t,max:t}},`parseRepeat`),f=e(e=>{let t=p(e.primary);if(!e.repeat)return t;let{min:n,max:r}=d(e.repeat);return n===0&&r===1?{type:`optional`,element:t}:{type:`repetition`,element:t,min:n,max:r}},`transformElement`),p=e(e=>{switch(e.$type){case`AbnfStringLiteral`:return{type:`terminal`,value:e.value};case`AbnfNumVal`:return{type:`terminal`,value:e.value};case`AbnfRuleName`:return{type:`nonterminal`,name:e.name};case`AbnfGroup`:return l(e.element);case`AbnfOptionalGroup`:return{type:`optional`,element:l(e.element)};default:throw Error(`Unsupported ABNF primary node: ${e.$type}`)}},`transformPrimary`),m=e(e=>({name:e.name,definition:l(e.definition)}),`transformRule`),h=e(e=>{s(e,o),e.title&&o.setTitle(e.title),e.rules.map(e=>o.addRule(m(e)))},`populateDb`),g={parser:{parse:e(e=>{o.clear(),t.debug(`[ABNF Parser] Starting Langium parse`);let n=c.parse(e);if(n.lexerErrors.length>0||n.parserErrors.length>0)throw new r(n);let i=n.value;t.debug(`[ABNF Parser] Parsed rules:`,i.rules.length),h(i),t.debug(`[ABNF Parser] Parse complete`)},`parse`),parser:{yy:o}},db:o,renderer:a,styles:i};export{g as diagram};
+import{n as e}from"./chunk-Y2CYZVJY-DsF7k-Jl.js";import{t}from"./src-Cyu5G-Tr.js";import"./chunk-DU6HZSFF-BlpgvKzl.js";import"./mermaid-D_jPUVqz.js";import{m as n,t as r}from"./mermaid-parser.core-3RKOfUXB.js";import{n as i,r as a,t as o}from"./chunk-SVP7TREG-B0UfxhUP.js";import{t as s}from"./chunk-JWPE2WC7-DVXcaiue.js";var c=n().RailroadAbnf.parser.LangiumParser,l=e(e=>{let t=e.alternatives.map(u);return t.length===1?t[0]:{type:`choice`,alternatives:t}},`transformAlternation`),u=e(e=>{let t=e.elements.map(f);return t.length===1?t[0]:{type:`sequence`,elements:t}},`transformConcatenation`),d=e(e=>{if(e.includes(`*`)){let[t,n]=e.split(`*`);return{min:t?parseInt(t,10):0,max:n?parseInt(n,10):1/0}}let t=parseInt(e,10);return{min:t,max:t}},`parseRepeat`),f=e(e=>{let t=p(e.primary);if(!e.repeat)return t;let{min:n,max:r}=d(e.repeat);return n===0&&r===1?{type:`optional`,element:t}:{type:`repetition`,element:t,min:n,max:r}},`transformElement`),p=e(e=>{switch(e.$type){case`AbnfStringLiteral`:return{type:`terminal`,value:e.value};case`AbnfNumVal`:return{type:`terminal`,value:e.value};case`AbnfRuleName`:return{type:`nonterminal`,name:e.name};case`AbnfGroup`:return l(e.element);case`AbnfOptionalGroup`:return{type:`optional`,element:l(e.element)};default:throw Error(`Unsupported ABNF primary node: ${e.$type}`)}},`transformPrimary`),m=e(e=>({name:e.name,definition:l(e.definition)}),`transformRule`),h=e(e=>{s(e,o),e.title&&o.setTitle(e.title),e.rules.map(e=>o.addRule(m(e)))},`populateDb`),g={parser:{parse:e(e=>{o.clear(),t.debug(`[ABNF Parser] Starting Langium parse`);let n=c.parse(e);if(n.lexerErrors.length>0||n.parserErrors.length>0)throw new r(n);let i=n.value;t.debug(`[ABNF Parser] Parsed rules:`,i.rules.length),h(i),t.debug(`[ABNF Parser] Parse complete`)},`parse`),parser:{yy:o}},db:o,renderer:a,styles:i};export{g as diagram};

package/dist/assets/{add-cell-with-ai-DFp1d9S3.js → add-cell-with-ai-BFgdVPaN.js}

@@ -1,4 +1,4 @@
-import{a as e,t}from"./rolldown-runtime-B0Z9INg1.js";import{a as n,c as r,d as i,f as a,l as o,r as s,t as c}from"./useEvent-Dc9HkKI2.js";import{t as l}from"./react-CUdNIagt.js";import{Bt as u,D as d,Fa as f,Fi as p,In as m,Pi as h,Qr as g,Ti as _,Va as v,Z as y,Zn as b,_a as x,b as S,bo as C,da as ee,f as te,ft as ne,ir as w,ja as T,ji as E,qa as re,ti as ie,yo as ae}from"./cells-CzugpTQu.js";import{A as oe,B as se,C as ce,D as le,E as ue,I as de,L as fe,N as pe,P as D,R as me,S as he,T as ge,W as _e,_ as ve,j as O,k as ye,v as be,w as xe,x as Se,y as Ce}from"./schemas-BL-NsLlF.js";import{t as we}from"./compiler-runtime-BQ9S8nyo.js";import{t as Te}from"./config-BxZgoFvi.js";import{P as Ee,m as De,t as Oe}from"./ai-model-dropdown-CJdHRDLy.js";import{f as ke,g as k}from"./useEventListener-BdZUApHd.js";import{a as Ae}from"./session-DXKMv4Bs.js";import{A as je,O as Me,o as Ne}from"./config-DpGPGznL.js";import{t as Pe}from"./cn-ldLVeaqm.js";import{t as Fe}from"./jsx-runtime-B74pBk57.js";import{a as Ie,c as Le,p as Re,r as ze,t as Be}from"./dropdown-menu-VuyoeQOD.js";import{t as Ve}from"./tooltip-DR2Z2N4w.js";import{t as He}from"./button-BoklZTGL.js";import{t as Ue}from"./createLucideIcon-rHn2EdoM.js";import{i as We,r as Ge}from"./paths-CzfpItSz.js";import{r as Ke}from"./useNumberFormatter-C4-_TnkF.js";import{a as qe,n as Je,t as Ye}from"./process-output-gzSUJLRt.js";import{r as Xe}from"./input-BDakodZ2.js";import{t as Ze}from"./use-toast-CC89xOwH.js";import{Gt as Qe,Qt as $e,Tt as et,Wt as tt,Zt as nt,at as rt,dt as it,en as at,qt as ot,st,ut as ct,wt as lt}from"./dist-BczecT6V.js";import{t as ut}from"./createReducer-BPvK7C_R.js";import{h as dt,n as ft,o as pt,r as mt}from"./dist-Ccyblc8N.js";import{t as ht}from"./requests-yTHMxhVa.js";import{n as gt}from"./dist-DBqBVprh.js";import{t as _t}from"./uuid-k9DpjH68.js";import{r as vt}from"./useTheme-DcnXlv0_.js";import{a as yt}from"./errors-D9vyuMar.js";import{Cr as bt,Ct as xt,Di as St,Gn as Ct,Ia as wt,Qr as Tt,S as Et,Sr as Dt,_a as Ot,br as kt,ir as At,ja as jt,k as Mt,mi as Nt,o as Pt,qt as Ft,vr as It,w as Lt,wr as Rt,xa as zt,xr as Bt,yr as Vt}from"./index-DqLAGQek.js";import{i as Ht,r as Ut,t as Wt}from"./error-entries-CDSsy76p.js";var Gt=Ue(`ban`,[[`circle`,{cx:`12`,cy:`12`,r:`10`,key:`1mglay`}],[`path`,{d:`M4.929 4.929 19.07 19.071`,key:`196cmz`}]]),Kt=Ue(`circle-stop`,[[`circle`,{cx:`12`,cy:`12`,r:`10`,key:`1mglay`}],[`rect`,{x:`9`,y:`9`,width:`6`,height:`6`,rx:`1`,key:`1ssd4o`}]]),qt=Ue(`paperclip`,[[`path`,{d:`m16 6-8.414 8.586a2 2 0 0 0 2.829 2.829l8.414-8.586a4 4 0 1 0-5.657-5.657l-8.379 8.551a6 6 0 1 0 8.485 8.485l8.379-8.551`,key:`1miecu`}]]),Jt=Ue(`send-horizontal`,[[`path`,{d:`M3.714 3.048a.498.498 0 0 0-.683.627l2.843 7.627a2 2 0 0 1 0 1.396l-2.842 7.627a.498.498 0 0 0 .682.627l18-8.5a.5.5 0 0 0 0-.904z`,key:`117uat`}],[`path`,{d:`M6 12h16`,key:`s4cdu5`}]]),A=we(),Yt=`vercel.ai.error`,Xt=Symbol.for(Yt),Zt,Qt,j=class e extends (Qt=Error,Zt=Xt,Qt){constructor({name:e,message:t,cause:n}){super(t),this[Zt]=!0,this.name=e,this.cause=n}static isInstance(t){return e.hasMarker(t,Yt)}static hasMarker(e,t){let n=Symbol.for(t);return typeof e==`object`&&!!e&&n in e&&typeof e[n]==`boolean`&&e[n]===!0}};function $t(e){return e==null?`unknown error`:typeof e==`string`?e:e instanceof Error?e.toString():JSON.stringify(e)}var en=`AI_InvalidArgumentError`,tn=`vercel.ai.error.${en}`,nn=Symbol.for(tn),rn,an,on=class extends (an=j,rn=nn,an){constructor({message:e,cause:t,argument:n}){super({name:en,message:e,cause:t}),this[rn]=!0,this.argument=n}static isInstance(e){return j.hasMarker(e,tn)}},sn=`AI_JSONParseError`,cn=`vercel.ai.error.${sn}`,ln=Symbol.for(cn),un,dn,fn=class extends (dn=j,un=ln,dn){constructor({text:e,cause:t}){super({name:sn,message:`JSON parsing failed: Text: ${e}.
+import{a as e,t}from"./rolldown-runtime-B0Z9INg1.js";import{a as n,c as r,d as i,f as a,l as o,r as s,t as c}from"./useEvent-Dc9HkKI2.js";import{t as l}from"./react-CUdNIagt.js";import{Bt as u,D as d,Fa as f,Fi as p,In as m,Pi as h,Qr as g,Ti as _,Va as v,Z as y,Zn as b,_a as x,b as S,bo as C,da as ee,f as te,ft as ne,ir as w,ja as T,ji as E,qa as re,ti as ie,yo as ae}from"./cells-CzugpTQu.js";import{A as oe,B as se,C as ce,D as le,E as ue,I as de,L as fe,N as pe,P as D,R as me,S as he,T as ge,W as _e,_ as ve,j as O,k as ye,v as be,w as xe,x as Se,y as Ce}from"./schemas-BL-NsLlF.js";import{t as we}from"./compiler-runtime-BQ9S8nyo.js";import{t as Te}from"./config-BxZgoFvi.js";import{P as Ee,m as De,t as Oe}from"./ai-model-dropdown-DJ2elMLM.js";import{f as ke,g as k}from"./useEventListener-BdZUApHd.js";import{a as Ae}from"./session-DXKMv4Bs.js";import{A as je,O as Me,o as Ne}from"./config-DpGPGznL.js";import{t as Pe}from"./cn-ldLVeaqm.js";import{t as Fe}from"./jsx-runtime-B74pBk57.js";import{a as Ie,c as Le,p as Re,r as ze,t as Be}from"./dropdown-menu-VuyoeQOD.js";import{t as Ve}from"./tooltip-DR2Z2N4w.js";import{t as He}from"./button-BoklZTGL.js";import{t as Ue}from"./createLucideIcon-rHn2EdoM.js";import{i as We,r as Ge}from"./paths-CzfpItSz.js";import{r as Ke}from"./useNumberFormatter-C4-_TnkF.js";import{a as qe,n as Je,t as Ye}from"./process-output-Cg2ta3vY.js";import{r as Xe}from"./input-BDakodZ2.js";import{t as Ze}from"./use-toast-CC89xOwH.js";import{Gt as Qe,Qt as $e,Tt as et,Wt as tt,Zt as nt,at as rt,dt as it,en as at,qt as ot,st,ut as ct,wt as lt}from"./dist-BczecT6V.js";import{t as ut}from"./createReducer-BPvK7C_R.js";import{h as dt,n as ft,o as pt,r as mt}from"./dist-Ccyblc8N.js";import{t as ht}from"./requests-yTHMxhVa.js";import{n as gt}from"./dist-DBqBVprh.js";import{t as _t}from"./uuid-k9DpjH68.js";import{r as vt}from"./useTheme-DcnXlv0_.js";import{a as yt}from"./errors-D9vyuMar.js";import{C as bt,Cr as xt,Di as St,Gn as Ct,La as wt,Ma as Tt,O as Et,Qr as Dt,Sr as Ot,St as kt,_a as At,a as jt,br as Mt,ir as Nt,mi as Pt,qt as Ft,vr as It,wr as Lt,x as Rt,xa as zt,xr as Bt,yr as Vt}from"./index-DafnX1Xx.js";import{i as Ht,r as Ut,t as Wt}from"./error-entries-CDSsy76p.js";var Gt=Ue(`ban`,[[`circle`,{cx:`12`,cy:`12`,r:`10`,key:`1mglay`}],[`path`,{d:`M4.929 4.929 19.07 19.071`,key:`196cmz`}]]),Kt=Ue(`circle-stop`,[[`circle`,{cx:`12`,cy:`12`,r:`10`,key:`1mglay`}],[`rect`,{x:`9`,y:`9`,width:`6`,height:`6`,rx:`1`,key:`1ssd4o`}]]),qt=Ue(`paperclip`,[[`path`,{d:`m16 6-8.414 8.586a2 2 0 0 0 2.829 2.829l8.414-8.586a4 4 0 1 0-5.657-5.657l-8.379 8.551a6 6 0 1 0 8.485 8.485l8.379-8.551`,key:`1miecu`}]]),Jt=Ue(`send-horizontal`,[[`path`,{d:`M3.714 3.048a.498.498 0 0 0-.683.627l2.843 7.627a2 2 0 0 1 0 1.396l-2.842 7.627a.498.498 0 0 0 .682.627l18-8.5a.5.5 0 0 0 0-.904z`,key:`117uat`}],[`path`,{d:`M6 12h16`,key:`s4cdu5`}]]),A=we(),Yt=`vercel.ai.error`,Xt=Symbol.for(Yt),Zt,Qt,j=class e extends (Qt=Error,Zt=Xt,Qt){constructor({name:e,message:t,cause:n}){super(t),this[Zt]=!0,this.name=e,this.cause=n}static isInstance(t){return e.hasMarker(t,Yt)}static hasMarker(e,t){let n=Symbol.for(t);return typeof e==`object`&&!!e&&n in e&&typeof e[n]==`boolean`&&e[n]===!0}};function $t(e){return e==null?`unknown error`:typeof e==`string`?e:e instanceof Error?e.toString():JSON.stringify(e)}var en=`AI_InvalidArgumentError`,tn=`vercel.ai.error.${en}`,nn=Symbol.for(tn),rn,an,on=class extends (an=j,rn=nn,an){constructor({message:e,cause:t,argument:n}){super({name:en,message:e,cause:t}),this[rn]=!0,this.argument=n}static isInstance(e){return j.hasMarker(e,tn)}},sn=`AI_JSONParseError`,cn=`vercel.ai.error.${sn}`,ln=Symbol.for(cn),un,dn,fn=class extends (dn=j,un=ln,dn){constructor({text:e,cause:t}){super({name:sn,message:`JSON parsing failed: Text: ${e}.
 Error message: ${$t(t)}`,cause:t}),this[un]=!0,this.text=e}static isInstance(e){return j.hasMarker(e,cn)}},pn=`AI_TypeValidationError`,mn=`vercel.ai.error.${pn}`,hn=Symbol.for(mn),gn,_n,vn=class e extends (_n=j,gn=hn,_n){constructor({value:e,cause:t,context:n}){let r=`Type validation failed`;if(n?.field&&(r+=` for ${n.field}`),n?.entityName||n?.entityId){r+=` (`;let e=[];n.entityName&&e.push(n.entityName),n.entityId&&e.push(`id: "${n.entityId}"`),r+=e.join(`, `),r+=`)`}super({name:pn,message:`${r}: Value: ${JSON.stringify(e)}.
 Error message: ${$t(t)}`,cause:t}),this[gn]=!0,this.value=e,this.context=n}static isInstance(e){return j.hasMarker(e,mn)}static wrap({value:t,cause:n,context:r}){return e.isInstance(n)&&n.value===t&&n.context?.field===r?.field&&n.context?.entityName===r?.entityName&&n.context?.entityId===r?.entityId?n:new e({value:t,cause:n,context:r})}},yn=class extends Error{constructor(e,t){super(e),this.name=`ParseError`,this.type=t.type,this.field=t.field,this.value=t.value,this.line=t.line}},bn=10,xn=13,Sn=32;function Cn(e){}function wn(e){if(typeof e==`function`)throw TypeError("`config` must be an object, got a function instead. Did you mean `createParser({onEvent: fn})`?");let{onEvent:t=Cn,onError:n=Cn,onRetry:r=Cn,onComment:i,maxBufferSize:a}=e,o=[],s=0,c=!0,l,u=``,d=0,f,p=!1;function m(e){if(p)throw Error("Cannot feed parser: it was terminated after exceeding the configured max buffer size. Call `reset()` to resume parsing.");if(c&&(c=!1,e.charCodeAt(0)===239&&e.charCodeAt(1)===187&&e.charCodeAt(2)===191&&(e=e.slice(3))),o.length===0){let t=g(e);t!==``&&(o.push(t),s=t.length),h();return}if(e.indexOf(`
 `)===-1&&e.indexOf(`\r`)===-1){o.push(e),s+=e.length,h();return}o.push(e);let t=o.join(``);o.length=0,s=0;let n=g(t);n!==``&&(o.push(n),s=n.length),h()}function h(){a!==void 0&&(s+u.length<=a||(p=!0,o.length=0,s=0,l=void 0,u=``,d=0,f=void 0,n(new yn(`Buffered data exceeded max buffer size of ${a} characters`,{type:`max-buffer-size-exceeded`}))))}function g(e){let n=0;if(e.indexOf(`\r`)===-1){let r=e.indexOf(`
@@ -41,7 +41,7 @@ ${t}`,d++;break;case`id`:l=t.includes(`\0`)?void 0:t;break;case`retry`:/^\d+$/.t
     - SQL cells: use mo.sql(f"""{content}""") function to insert content. If a database engine is specified, use mo.sql(f"""{content}""", engine=engine) instead.
 
     Returns:
-    - A result object containing standard tool metadata.`},fo=O({edit:O({type:ve([`update_cell`,`add_cell`,`delete_cell`]),code:D().optional(),position:O({type:ve([`relative`,`column_end`,`notebook_end`]).optional(),cellId:D().optional(),columnIndex:oe().optional(),before:he().optional()})})}),po=class{name=`edit_notebook_tool`;description=uo;schema=fo;outputSchema=qa;mode=[`agent`];handler=async({edit:e},t)=>{let{addStagedCell:n,createNewCell:r,store:i}=t;switch(e.type){case`update_cell`:{let{position:t,code:r}=e,a=t.cellId;if(!a)throw new Y(`Cell ID is required for update_cell`,`CELL_ID_REQUIRED_FOR_UPDATE_CELL`,!0,`Provide a cell ID to update`);if(!r)throw new Y(`Code is required for update_cell`,`CODE_REQUIRED_FOR_UPDATE_CELL`,!0,`Provide the new code to update the cell`);let o=i.get(S);this.validateCellIdExists(a,o);let s=this.getCellEditorView(a,o);At(a);let c=i.get(Z).get(a);if(c?.type===`add_cell`){ne(s,r);break}let l=s.state.doc.toString(),u=c?.previousCode??l;if(r===u)break;n({cellId:a,edit:{type:`update_cell`,previousCode:u}}),ne(s,r);break}case`add_cell`:{let{position:t,code:a}=e,o=`__end__`,s=!1,c=ae.create(),l=i.get(S);switch(t.type){case`relative`:{let e=t.cellId;if(!e)throw new Y(`Cell ID is required for add_cell with relative position`,`TOOL_ERROR`,!0,`Provide a cell ID to add the cell before`);if(t.before===void 0)throw new Y(`Before is required for add_cell with relative position`,`TOOL_ERROR`,!0,`Provide a boolean value for before, true to add before cell ${e}, false to add after the cell`);this.validateCellIdExists(e,l),o=e,s=t.before;break}case`column_end`:if(!t.columnIndex)throw new Y(`Column index is required for add_cell with column_end position`,`TOOL_ERROR`,!0,`Provide a column index to add the cell at the end of`);o={type:`__end__`,columnId:this.getColumnId(t.columnIndex,l)}}r({cellId:o,before:s,code:a,newCellId:c}),n({cellId:c,edit:{type:`add_cell`}}),At(c);break}case`delete_cell`:{let{position:t}=e,r=t.cellId;if(!r)throw new Y(`Cell ID is required for delete_cell`,`CELL_ID_REQUIRED_FOR_DELETE_CELL`,!0,`Provide a cell ID to delete`);let a=i.get(S);this.validateCellIdExists(r,a),n({cellId:r,edit:{type:`delete_cell`,previousCode:this.getCellEditorView(r,a).state.doc.toString()}}),At(r);break}}return{status:`success`,next_steps:[`If you need to perform more edits, call this tool again.`,`You should use the lint notebook tool to check for errors and lint issues. Fix them by editing the notebook.`,`You should use the run stale cells tool to run the cells that have been edited or newly added. This allows you to see the output of the cells and fix any errors.`]}};validateCellIdExists(e,t){if(!t.cellIds.getColumns().some(t=>t.idSet.has(e)))throw new Y(`Cell not found`,`CELL_NOT_FOUND`,!1,`Check which cells exist in the notebook`)}getColumnId(e,t){let n=t.cellIds.getColumns();if(e<0||e>=n.length)throw new Y(`Column index is out of range`,`COLUMN_INDEX_OUT_OF_RANGE`,!0,`Choose a column index between 0 and the number of columns in the notebook (0-based)`);return n[e].id}getCellEditorView(e,t){let n=t.cellHandles[e].current;if(!n?.editorView)throw new Y(`Cell editor not found`,`CELL_EDITOR_NOT_FOUND`,!1,`Internal error, ask the user to report this error`);return n.editorView}};function mo(e,t){return e?!!([`image/`,`video/`,`audio/`,`application/pdf`].some(t=>e.startsWith(t))||[`svg`,`vega`].some(t=>e.includes(t))||e===`text/html`&&[`<img`,`<video`,`<audio`,`<iframe`,`<canvas`,`<svg`,`<marimo-ui-element`].some(e=>t.includes(e))):!1}var ho=class extends bt{title=`Cell Outputs`;mentionPrefix=`@`;contextType=`cell-output`;store;constructor(e){super(),this.store=e}getItems(){let e=this.store.get(S),t=[];for(let n of e.cellIds.inOrderIds){let r=vo(n,e),i=r.cellOutput;if(!i)continue;let a=i.output.mimetype||`unknown`;t.push({uri:this.asURI(n),name:r.cellName,type:this.contextType,description:`Cell output (${a})`,data:{...r,cellOutput:i}})}return t}formatCompletion(e){let{cellOutput:t,cellName:n,cellCode:r}=e.data;return{label:`@${n}`,displayLabel:n,detail:`${t.outputType} output`,boost:t.outputType===`media`?kt.HIGH:kt.MEDIUM,type:this.contextType,section:Bt.CELL_OUTPUT,apply:`@${n}`,info:()=>{let e=document.createElement(`div`);e.classList.add(`mo-cm-tooltip`,`docs-documentation`,`min-w-[300px]`,`max-w-[500px]`,`flex`,`flex-col`,`gap-2`);let i=document.createElement(`div`);i.classList.add(`flex`,`flex-col`,`gap-1`);let a=document.createElement(`div`);a.classList.add(`font-bold`,`text-base`),a.textContent=n,i.append(a);let o=document.createElement(`div`);if(o.classList.add(`text-sm`,`text-muted-foreground`),i.append(o),e.append(i),r){let t=document.createElement(`div`);t.classList.add(`text-xs`,`font-medium`,`text-muted-foreground`),t.textContent=`Code:`,e.append(t);let n=document.createElement(`div`);n.classList.add(`text-xs`,`font-mono`,`bg-muted`,`p-2`,`rounded`,`max-h-20`,`overflow-y-auto`),n.textContent=r.slice(0,200)+(r.length>200?`...`:``),e.append(n)}if(t.processedContent){let n=document.createElement(`div`);n.classList.add(`text-xs`,`font-medium`,`text-muted-foreground`,`mt-2`),n.textContent=`Output Preview:`,e.append(n);let r=document.createElement(`div`);r.classList.add(`text-xs`,`bg-muted`,`p-2`,`rounded`,`max-h-24`,`overflow-y-auto`,`mb-2`),r.textContent=t.processedContent.slice(0,300)+(t.processedContent.length>300?`...`:``),e.append(r)}if(t.outputType===`media`){let t=document.createElement(`div`);t.classList.add(`text-xs`,`text-muted-foreground`,`italic`,`mb-2`),t.textContent=`A screenshot of the output will be attached`,e.append(t)}return e}}}formatContext(e){let{cellOutput:t,cellName:n,cellId:r,cellCode:i}=e.data,a={name:n,cellId:r,outputType:t.outputType,mimetype:t.output.mimetype},o=`Cell Code:\n${i}\n\n`;return t.outputType===`text`&&t.processedContent?o+=`Output:\n${t.processedContent}`:t.outputType===`media`&&(o+=`Media Output: Contains ${t.output.mimetype} content`,t.imageUrl&&(o+=`\nImage URL: ${t.imageUrl}`)),Dt({type:this.contextType,data:a,details:o})}async getAttachments(e){let t=e[0].data.cellId,n=e[0].data.cellName;return go(e.map(e=>e.data.cellOutput),t,n)}};async function go(e,t,n){let r=e.filter(e=>e.shouldDownloadImage&&e.outputType===`media`);if(r.length===0)return[];let i=r.flatMap(e=>{let r=document.getElementById(C.create(t));return r?{cellId:t,cellName:n,mimetype:e.output.mimetype,element:r}:(k.warn(`Output element not found for cell ${t}`),[])});try{return await Promise.all(i.map(async e=>({type:`file`,filename:`${n}-output-screenshot`,mediaType:`image/png`,url:await Ft(e.element)})))}catch(e){return k.error(`Error downloading cell output images:`,e),[]}}function _o(e){if(Ct(e))return null;let t=e.mimetype,n=mo(t,String(e.data))?`media`:`text`,r=Ye(e),i,a=!1;return n===`media`&&(typeof e.data==`string`&&e.data.startsWith(`data:`)||typeof e.data==`string`&&e.data.startsWith(`http`)?i=e.data:a=!0),{processedContent:r,imageUrl:i,shouldDownloadImage:a,outputType:n,output:e}}function vo(e,t,n){let{includeConsoleOutput:r=!1}=n||{},i=t.cellRuntime[e],a=t.cellData[e],o=t.cellIds.inOrderIds.indexOf(e),s=m(a.name,o),c;r&&(c=i.consoleOutputs.map(e=>_o(e)).filter(e=>e!==null));let l,u=i.output;return u&&!Ct(u)&&(l=_o(u)??void 0),{cellId:e,cellName:s,cellCode:a.code,cellOutput:l,consoleOutputs:c}}var yo=200,bo=3e4,xo=2e3,So=3e3,Co=4e4,wo={baseDescription:`Run cells in the current notebook that are stale. Stale cells are cells that have been edited or newly added. You can run this tool after editing the notebook or when requested. The output of the ran cells will be returned alongside metadata.`},To=O({type:ue(`file`),url:D(),mediaType:D()}),Eo=class{name=`run_stale_cells_tool`;description=wo;schema=O({});outputSchema=qa.extend({cellsToOutput:pe(D(),O({consoleOutput:D().optional(),cellOutput:D().optional(),consoleAttachments:Se(To).optional(),cellAttachments:Se(To).optional()}).nullable()).optional()});mode=[`agent`];postExecutionDelay;constructor(e){this.postExecutionDelay=e?.postExecutionDelay??yo}handler=async(e,t)=>{let{prepareForRun:n,sendRun:r,store:i}=t,a=i.get(S),o=ee(a);if(o.length===0)return{status:`success`,message:`No stale cells found.`};if(await St({cellIds:o,sendRun:r,prepareForRun:n,notebook:a}),!await this.waitForCellsToFinish(i,o,bo,this.postExecutionDelay))return{status:`success`,message:`No output was returned because some cells have not finished executing`};let s=i.get(S),c=new Map,l=!1,u=!1,d=0;for(let e of o){let t=vo(e,s,{includeConsoleOutput:!0}),n=t.cellOutput,r=t.consoleOutputs,i=r&&r.length>0;if((n&&this.outputHasErrors(n)||i&&r.some(e=>this.outputHasErrors(e)))&&(l=!0),!n&&!i){c.set(e,null);continue}if(d>=Co){c.set(e,{cellOutput:`Cell executed (output omitted due to context limits).`});continue}let a,o;n&&(a=this.formatOutputString(n),d+=a.length),i&&(u=!0,o=r.map(e=>this.formatOutputString(e)).join(`
+    - A result object containing standard tool metadata.`},fo=O({edit:O({type:ve([`update_cell`,`add_cell`,`delete_cell`]),code:D().optional(),position:O({type:ve([`relative`,`column_end`,`notebook_end`]).optional(),cellId:D().optional(),columnIndex:oe().optional(),before:he().optional()})})}),po=class{name=`edit_notebook_tool`;description=uo;schema=fo;outputSchema=qa;mode=[`agent`];handler=async({edit:e},t)=>{let{addStagedCell:n,createNewCell:r,store:i}=t;switch(e.type){case`update_cell`:{let{position:t,code:r}=e,a=t.cellId;if(!a)throw new Y(`Cell ID is required for update_cell`,`CELL_ID_REQUIRED_FOR_UPDATE_CELL`,!0,`Provide a cell ID to update`);if(!r)throw new Y(`Code is required for update_cell`,`CODE_REQUIRED_FOR_UPDATE_CELL`,!0,`Provide the new code to update the cell`);let o=i.get(S);this.validateCellIdExists(a,o);let s=this.getCellEditorView(a,o);Nt(a);let c=i.get(Z).get(a);if(c?.type===`add_cell`){ne(s,r);break}let l=s.state.doc.toString(),u=c?.previousCode??l;if(r===u)break;n({cellId:a,edit:{type:`update_cell`,previousCode:u}}),ne(s,r);break}case`add_cell`:{let{position:t,code:a}=e,o=`__end__`,s=!1,c=ae.create(),l=i.get(S);switch(t.type){case`relative`:{let e=t.cellId;if(!e)throw new Y(`Cell ID is required for add_cell with relative position`,`TOOL_ERROR`,!0,`Provide a cell ID to add the cell before`);if(t.before===void 0)throw new Y(`Before is required for add_cell with relative position`,`TOOL_ERROR`,!0,`Provide a boolean value for before, true to add before cell ${e}, false to add after the cell`);this.validateCellIdExists(e,l),o=e,s=t.before;break}case`column_end`:if(!t.columnIndex)throw new Y(`Column index is required for add_cell with column_end position`,`TOOL_ERROR`,!0,`Provide a column index to add the cell at the end of`);o={type:`__end__`,columnId:this.getColumnId(t.columnIndex,l)}}r({cellId:o,before:s,code:a,newCellId:c}),n({cellId:c,edit:{type:`add_cell`}}),Nt(c);break}case`delete_cell`:{let{position:t}=e,r=t.cellId;if(!r)throw new Y(`Cell ID is required for delete_cell`,`CELL_ID_REQUIRED_FOR_DELETE_CELL`,!0,`Provide a cell ID to delete`);let a=i.get(S);this.validateCellIdExists(r,a),n({cellId:r,edit:{type:`delete_cell`,previousCode:this.getCellEditorView(r,a).state.doc.toString()}}),Nt(r);break}}return{status:`success`,next_steps:[`If you need to perform more edits, call this tool again.`,`You should use the lint notebook tool to check for errors and lint issues. Fix them by editing the notebook.`,`You should use the run stale cells tool to run the cells that have been edited or newly added. This allows you to see the output of the cells and fix any errors.`]}};validateCellIdExists(e,t){if(!t.cellIds.getColumns().some(t=>t.idSet.has(e)))throw new Y(`Cell not found`,`CELL_NOT_FOUND`,!1,`Check which cells exist in the notebook`)}getColumnId(e,t){let n=t.cellIds.getColumns();if(e<0||e>=n.length)throw new Y(`Column index is out of range`,`COLUMN_INDEX_OUT_OF_RANGE`,!0,`Choose a column index between 0 and the number of columns in the notebook (0-based)`);return n[e].id}getCellEditorView(e,t){let n=t.cellHandles[e].current;if(!n?.editorView)throw new Y(`Cell editor not found`,`CELL_EDITOR_NOT_FOUND`,!1,`Internal error, ask the user to report this error`);return n.editorView}};function mo(e,t){return e?!!([`image/`,`video/`,`audio/`,`application/pdf`].some(t=>e.startsWith(t))||[`svg`,`vega`].some(t=>e.includes(t))||e===`text/html`&&[`<img`,`<video`,`<audio`,`<iframe`,`<canvas`,`<svg`,`<marimo-ui-element`].some(e=>t.includes(e))):!1}var ho=class extends xt{title=`Cell Outputs`;mentionPrefix=`@`;contextType=`cell-output`;store;constructor(e){super(),this.store=e}getItems(){let e=this.store.get(S),t=[];for(let n of e.cellIds.inOrderIds){let r=vo(n,e),i=r.cellOutput;if(!i)continue;let a=i.output.mimetype||`unknown`;t.push({uri:this.asURI(n),name:r.cellName,type:this.contextType,description:`Cell output (${a})`,data:{...r,cellOutput:i}})}return t}formatCompletion(e){let{cellOutput:t,cellName:n,cellCode:r}=e.data;return{label:`@${n}`,displayLabel:n,detail:`${t.outputType} output`,boost:t.outputType===`media`?Mt.HIGH:Mt.MEDIUM,type:this.contextType,section:Bt.CELL_OUTPUT,apply:`@${n}`,info:()=>{let e=document.createElement(`div`);e.classList.add(`mo-cm-tooltip`,`docs-documentation`,`min-w-[300px]`,`max-w-[500px]`,`flex`,`flex-col`,`gap-2`);let i=document.createElement(`div`);i.classList.add(`flex`,`flex-col`,`gap-1`);let a=document.createElement(`div`);a.classList.add(`font-bold`,`text-base`),a.textContent=n,i.append(a);let o=document.createElement(`div`);if(o.classList.add(`text-sm`,`text-muted-foreground`),i.append(o),e.append(i),r){let t=document.createElement(`div`);t.classList.add(`text-xs`,`font-medium`,`text-muted-foreground`),t.textContent=`Code:`,e.append(t);let n=document.createElement(`div`);n.classList.add(`text-xs`,`font-mono`,`bg-muted`,`p-2`,`rounded`,`max-h-20`,`overflow-y-auto`),n.textContent=r.slice(0,200)+(r.length>200?`...`:``),e.append(n)}if(t.processedContent){let n=document.createElement(`div`);n.classList.add(`text-xs`,`font-medium`,`text-muted-foreground`,`mt-2`),n.textContent=`Output Preview:`,e.append(n);let r=document.createElement(`div`);r.classList.add(`text-xs`,`bg-muted`,`p-2`,`rounded`,`max-h-24`,`overflow-y-auto`,`mb-2`),r.textContent=t.processedContent.slice(0,300)+(t.processedContent.length>300?`...`:``),e.append(r)}if(t.outputType===`media`){let t=document.createElement(`div`);t.classList.add(`text-xs`,`text-muted-foreground`,`italic`,`mb-2`),t.textContent=`A screenshot of the output will be attached`,e.append(t)}return e}}}formatContext(e){let{cellOutput:t,cellName:n,cellId:r,cellCode:i}=e.data,a={name:n,cellId:r,outputType:t.outputType,mimetype:t.output.mimetype},o=`Cell Code:\n${i}\n\n`;return t.outputType===`text`&&t.processedContent?o+=`Output:\n${t.processedContent}`:t.outputType===`media`&&(o+=`Media Output: Contains ${t.output.mimetype} content`,t.imageUrl&&(o+=`\nImage URL: ${t.imageUrl}`)),Ot({type:this.contextType,data:a,details:o})}async getAttachments(e){let t=e[0].data.cellId,n=e[0].data.cellName;return go(e.map(e=>e.data.cellOutput),t,n)}};async function go(e,t,n){let r=e.filter(e=>e.shouldDownloadImage&&e.outputType===`media`);if(r.length===0)return[];let i=r.flatMap(e=>{let r=document.getElementById(C.create(t));return r?{cellId:t,cellName:n,mimetype:e.output.mimetype,element:r}:(k.warn(`Output element not found for cell ${t}`),[])});try{return await Promise.all(i.map(async e=>({type:`file`,filename:`${n}-output-screenshot`,mediaType:`image/png`,url:await Ft(e.element)})))}catch(e){return k.error(`Error downloading cell output images:`,e),[]}}function _o(e){if(Ct(e))return null;let t=e.mimetype,n=mo(t,String(e.data))?`media`:`text`,r=Ye(e),i,a=!1;return n===`media`&&(typeof e.data==`string`&&e.data.startsWith(`data:`)||typeof e.data==`string`&&e.data.startsWith(`http`)?i=e.data:a=!0),{processedContent:r,imageUrl:i,shouldDownloadImage:a,outputType:n,output:e}}function vo(e,t,n){let{includeConsoleOutput:r=!1}=n||{},i=t.cellRuntime[e],a=t.cellData[e],o=t.cellIds.inOrderIds.indexOf(e),s=m(a.name,o),c;r&&(c=i.consoleOutputs.map(e=>_o(e)).filter(e=>e!==null));let l,u=i.output;return u&&!Ct(u)&&(l=_o(u)??void 0),{cellId:e,cellName:s,cellCode:a.code,cellOutput:l,consoleOutputs:c}}var yo=200,bo=3e4,xo=2e3,So=3e3,Co=4e4,wo={baseDescription:`Run cells in the current notebook that are stale. Stale cells are cells that have been edited or newly added. You can run this tool after editing the notebook or when requested. The output of the ran cells will be returned alongside metadata.`},To=O({type:ue(`file`),url:D(),mediaType:D()}),Eo=class{name=`run_stale_cells_tool`;description=wo;schema=O({});outputSchema=qa.extend({cellsToOutput:pe(D(),O({consoleOutput:D().optional(),cellOutput:D().optional(),consoleAttachments:Se(To).optional(),cellAttachments:Se(To).optional()}).nullable()).optional()});mode=[`agent`];postExecutionDelay;constructor(e){this.postExecutionDelay=e?.postExecutionDelay??yo}handler=async(e,t)=>{let{prepareForRun:n,sendRun:r,store:i}=t,a=i.get(S),o=ee(a);if(o.length===0)return{status:`success`,message:`No stale cells found.`};if(await St({cellIds:o,sendRun:r,prepareForRun:n,notebook:a}),!await this.waitForCellsToFinish(i,o,bo,this.postExecutionDelay))return{status:`success`,message:`No output was returned because some cells have not finished executing`};let s=i.get(S),c=new Map,l=!1,u=!1,d=0;for(let e of o){let t=vo(e,s,{includeConsoleOutput:!0}),n=t.cellOutput,r=t.consoleOutputs,i=r&&r.length>0;if((n&&this.outputHasErrors(n)||i&&r.some(e=>this.outputHasErrors(e)))&&(l=!0),!n&&!i){c.set(e,null);continue}if(d>=Co){c.set(e,{cellOutput:`Cell executed (output omitted due to context limits).`});continue}let a,o;n&&(a=this.formatOutputString(n),d+=a.length),i&&(u=!0,o=r.map(e=>this.formatOutputString(e)).join(`
 `),o=this.truncateString(o,xo),d+=o.length),c.set(e,{cellOutput:a,consoleOutput:o})}if(c.size===0)return{status:`success`,message:`Stale cells have been run. No output was returned.`};let f=[`Review the output of the cells. The CellId is the key of the result object.`,l?`There are errors in the cells. Please fix them by using the edit notebook tool and the given CellIds.`:`You may edit the notebook further with the given CellIds.`];return{status:`success`,cellsToOutput:Object.fromEntries(c),message:u?`Console output represents the stdout or stderr of the cell (eg. print statements).`:void 0,next_steps:f}};outputHasErrors(e){return e.output.mimetype===`application/vnd.marimo+error`||e.output.mimetype===`application/vnd.marimo+traceback`}formatOutputString(e){let{outputType:t,processedContent:n,imageUrl:r,output:i}=e;if(t===`media`){let e=`Media Output: Contains ${i.mimetype} content`;return r?`${e}\nImage URL: ${r}`:e}if(i.mimetype===`text/html`)return`HTML Output: text/html content (${(typeof i.data==`string`?i.data.length:JSON.stringify(i.data).length).toLocaleString()} chars). Full output visible in notebook UI.`;let a=this.outputHasErrors(e)?So:xo,o=n;return o||=typeof i.data==`string`?i.data:JSON.stringify(i.data),`Output:\n${this.truncateString(o,a)}`}truncateString(e,t){return e.length<=t?e:`${e.slice(0,t)}\n\n[TRUNCATED: ${e.length.toLocaleString()} → ${t.toLocaleString()} chars. Full output visible in the notebook UI.]`}async waitForCellsToFinish(e,t,r,i){let a=e=>t.every(t=>{let n=e.cellRuntime[t];return n?.status!==`running`&&n?.status!==`queued`}),o=async()=>(i>0&&await new Promise(e=>setTimeout(e,i)),!0);if(a(e.get(S)))return await o();try{return await Promise.race([n(S,a),new Promise((e,t)=>setTimeout(()=>t(Error(`timeout`)),r))]),await o()}catch{return!1}}};function Do(e){let t=e.baseDescription;return e.whenToUse&&(t+=`\n\n## When to use:\n- ${e.whenToUse.join(`
 - `)}`),e.avoidIf&&(t+=`\n\n## Avoid if:\n- ${e.avoidIf.join(`
 - `)}`),e.prerequisites&&(t+=`\n\n## Prerequisites:\n- ${e.prerequisites.join(`
@@ -50,12 +50,12 @@ ${t}`,d++;break;case`id`:l=t.includes(`\0`)?void 0:t;break;case`retry`:/^\d+$/.t
 
 `)),e.errorData.some(e=>e.type===`sql-error`)){let r=Vt(e.cellId,t);r&&n.push(`Database schema:\n${r}`)}return n.join(`
 
-`)}function jo(e,t){return Dt({type:`error`,data:{name:e.cellName||`Cell ${e.cellId}`,cellId:e.cellId},details:Ao(e,t)})}function Mo(e){return e.errorData.length===1?Wt(e.errorData[0]):e.errorData.length>1?`${e.errorData.length} errors`:e.tracebackHtml?y(e.tracebackHtml).trim().split(`
-`).find(e=>e.trim())?.trim()||`traceback`:`error`}function No(e){return`Error: ${e.cellName||`Cell ${e.cellId}`}`}var Po=new _(`error`,`errors`),Fo=class extends bt{title=`Errors`;mentionPrefix=`@`;contextType=`error`;store;constructor(e){super(),this.store=e}getItems(){let e=Ht(this.store);if(e.length===0)return[];let t=[{uri:this.asURI(`all`),name:`Errors`,type:this.contextType,data:{type:`all-errors`,errors:e},description:`All errors in the notebook`}];for(let n of e)t.push({uri:this.asURI(n.cellId),name:No(n),type:this.contextType,data:{type:`cell-error`,error:n},description:Mo(n)});return t}formatCompletion(e){if(e.data.type===`all-errors`){let t=e.data.errors.length;return{label:`@Errors`,displayLabel:`Errors`,detail:`${t} ${Po.pluralize(t)}`,type:`error`,apply:`@error://all`,section:Bt.ERROR,info:()=>{let e=document.createElement(`div`);e.classList.add(`mo-cm-tooltip`,`docs-documentation`,`min-w-[200px]`,`flex`,`flex-col`,`gap-1`,`p-2`);let n=document.createElement(`div`);n.classList.add(`flex`,`flex-col`,`gap-1`);let r=document.createElement(`div`);r.classList.add(`font-bold`,`text-base`),r.textContent=`Errors`,n.append(r);let i=document.createElement(`div`);return i.classList.add(`text-sm`,`text-muted-foreground`),i.textContent=`${t} ${Po.pluralize(t)}`,n.append(i),e.append(n),e}}}if(e.data.type===`cell-error`){let{error:t}=e.data;return{...this.createBasicCompletion(e,{detail:Mo(t),type:`error`}),section:Bt.ERROR}}return{label:`Error`,displayLabel:`Error`,section:Bt.ERROR}}formatContext(e){return(e.data.type===`all-errors`?e.data.errors:[e.data.error]).map(e=>jo(e,this.store)).join(`
+`)}function jo(e,t){return Ot({type:`error`,data:{name:e.cellName||`Cell ${e.cellId}`,cellId:e.cellId},details:Ao(e,t)})}function Mo(e){return e.errorData.length===1?Wt(e.errorData[0]):e.errorData.length>1?`${e.errorData.length} errors`:e.tracebackHtml?y(e.tracebackHtml).trim().split(`
+`).find(e=>e.trim())?.trim()||`traceback`:`error`}function No(e){return`Error: ${e.cellName||`Cell ${e.cellId}`}`}var Po=new _(`error`,`errors`),Fo=class extends xt{title=`Errors`;mentionPrefix=`@`;contextType=`error`;store;constructor(e){super(),this.store=e}getItems(){let e=Ht(this.store);if(e.length===0)return[];let t=[{uri:this.asURI(`all`),name:`Errors`,type:this.contextType,data:{type:`all-errors`,errors:e},description:`All errors in the notebook`}];for(let n of e)t.push({uri:this.asURI(n.cellId),name:No(n),type:this.contextType,data:{type:`cell-error`,error:n},description:Mo(n)});return t}formatCompletion(e){if(e.data.type===`all-errors`){let t=e.data.errors.length;return{label:`@Errors`,displayLabel:`Errors`,detail:`${t} ${Po.pluralize(t)}`,type:`error`,apply:`@error://all`,section:Bt.ERROR,info:()=>{let e=document.createElement(`div`);e.classList.add(`mo-cm-tooltip`,`docs-documentation`,`min-w-[200px]`,`flex`,`flex-col`,`gap-1`,`p-2`);let n=document.createElement(`div`);n.classList.add(`flex`,`flex-col`,`gap-1`);let r=document.createElement(`div`);r.classList.add(`font-bold`,`text-base`),r.textContent=`Errors`,n.append(r);let i=document.createElement(`div`);return i.classList.add(`text-sm`,`text-muted-foreground`),i.textContent=`${t} ${Po.pluralize(t)}`,n.append(i),e.append(n),e}}}if(e.data.type===`cell-error`){let{error:t}=e.data;return{...this.createBasicCompletion(e,{detail:Mo(t),type:`error`}),section:Bt.ERROR}}return{label:`Error`,displayLabel:`Error`,section:Bt.ERROR}}formatContext(e){return(e.data.type===`all-errors`?e.data.errors:[e.data.error]).map(e=>jo(e,this.store)).join(`
 
-`)}},Io=x(),Lo={maxDepth:3,maxResults:20,defaultResultsLimit:5,includeDirectories:!1},Ro=class extends bt{title=`Files`;mentionPrefix=`#`;contextType=`file`;apiRequests;config;constructor(e,t=Lo){super(),this.apiRequests=e,this.config=t}createCompletionSource(){return async e=>{let t=e.matchBefore(/#[^\s#]*/);if(!t)return null;let n=t.text;if(!n.startsWith(`#`))return null;let r=n.slice(1);if(r.length===0)return this.getDefaultCompletions(t);try{let e=(await this.searchFiles(r)||[]).map(e=>{let t={uri:this.asURI(e.path),name:e.name,type:this.contextType,description:e.isDirectory?`Directory`:`File`,data:{path:e.path,isDirectory:e.isDirectory}};return this.formatCompletion(t)});return{from:t.from,options:e}}catch(e){return k.error(`Failed to search files:`,e),null}}}searchFiles=async(e,t={})=>{let{maxDepth:n,maxResults:r,includeDirectories:i}={...this.config,...t};return(await this.apiRequests.sendSearchFiles({query:e,includeFiles:!0,includeDirectories:i,depth:n,limit:r})).files};async getDefaultCompletions(e){try{for(let t of[`py`,`md`,`csv`])try{let n=await this.searchFiles(t,{maxDepth:1,maxResults:5});if(n&&n.length>0){let t=n.map(e=>{let t={uri:this.asURI(e.path),name:e.name,type:this.contextType,description:e.isDirectory?`Directory`:`File`,data:{path:e.path,isDirectory:e.isDirectory}};return this.formatCompletion(t)});return{from:e.from,options:t}}}catch(e){k.error(`Failed to get default file completions:`,e)}return{from:e.from,options:[]}}catch(e){return k.error(`Failed to get default file completions:`,e),null}}getItems(){return[]}formatCompletion(e){let{data:t,name:n}=e,r=t.isDirectory?`📁`:`📄`;return{...this.createBasicCompletion(e),type:`file`,section:Bt.FILE,boost:t.isDirectory?kt.MEDIUM:kt.LOW,detail:t.path,displayLabel:`${r} ${n}`,apply:async(e,r,i,a)=>{let o=e.state.facet(Io)?.addAttachment;if(!o){k.warn(`No addAttachment callback provided`);return}let s=await this.apiRequests.sendFileDetails({path:t.path}).catch(e=>(Ze({title:`Failed to get file details`,description:e.message}),null));if(!s)return;let c=s.mimeType||`text/plain`,l=Nt({contents:s.contents??``,mimeType:c,isBase64:s.isBase64??!1});o(new File([l],n,{type:c})),e.dispatch({changes:{from:i,to:a,insert:``}}),pt(e)},info:()=>{let e=document.createElement(`div`);e.classList.add(`flex`,`flex-col`,`gap-1`,`p-2`);let r=document.createElement(`div`);r.classList.add(`font-bold`),r.textContent=n,e.append(r);let i=document.createElement(`div`);return i.classList.add(`text-xs`,`text-muted-foreground`),i.textContent=t.path,e.append(i),e}}}formatContext(e){let{data:t,name:n}=e;return Dt({type:this.contextType,data:{name:n,path:t.path,isDirectory:t.isDirectory},details:t.isDirectory?`Directory containing files and subdirectories`:`File`})}},zo=class extends bt{title=`Tables`;mentionPrefix=`@`;contextType=`data`;tablesMap;constructor(e){super(),this.tablesMap=e}getItems(){return[...this.tablesMap.entries()].map(([e,t])=>({uri:this.asURI(e),name:e,type:this.contextType,description:t.source===`memory`?`in-memory`:t.source,data:t}))}formatContext(e){let{data:t}=e,{columns:n,source:r,num_rows:i,num_columns:a,name:o,variable_name:s}=t,c=[i==null?void 0:`${i} rows`,a==null?void 0:`${a} columns`].filter(Boolean).join(`, `),l=``;if(c&&(l+=`Shape: ${c}\n`),s&&(l+=`Variable: ${s}\n`),n&&n.length>0){l+=`Columns:
-`;for(let e of n){let t=`  ${e.name} (${e.type})`;if(e.sample_values&&e.sample_values.length>0){let n=e.sample_values.slice(0,3).map(e=>e===null?`null`:String(e)).join(`, `);t+=` - samples: [${n}]`}l+=`${t}\n`}}return Dt({type:this.contextType,data:{name:o,source:r??`unknown`},details:l.trim()})}formatCompletion(e){let t=e.data.name,n=e.data;return{label:`@${t}`,displayLabel:t,detail:n.source===`memory`?`in-memory`:n.source,boost:n.source_type===`local`?kt.LOCAL_TABLE:kt.REMOTE_TABLE,type:p(n),apply:`@${t}`,section:{name:p(n)===`dataframe`?`Dataframe`:`Table`,rank:Bt.TABLE.rank},info:()=>this.createTableInfoElement(t,n)}}createTableInfoElement(e,t){let n=document.createElement(`div`);n.classList.add(`mo-cm-tooltip`,`docs-documentation`,`min-w-[200px]`,`flex`,`flex-col`,`gap-1`,`p-2`);let r=document.createElement(`div`);r.classList.add(`flex`,`flex-col`,`gap-1`);let i=document.createElement(`div`);if(i.classList.add(`font-bold`,`text-base`),i.textContent=e,r.append(i),t.source){let e=document.createElement(`div`);e.classList.add(`text-xs`,`text-muted-foreground`),e.textContent=`Source: ${t.source}`,r.append(e)}n.append(r);let a=[t.num_rows==null?void 0:`${t.num_rows} rows`,t.num_columns==null?void 0:`${t.num_columns} columns`].filter(Boolean).join(`, `);if(a){let e=document.createElement(`div`);e.classList.add(`text-xs`,`bg-muted`,`px-2`,`py-1`,`rounded`),e.textContent=a,n.append(e)}if(t.columns&&t.columns.length>0){let e=document.createElement(`div`);e.classList.add(`overflow-auto`,`max-h-60`);let r=document.createElement(`table`);r.classList.add(`w-full`,`text-xs`,`border-collapse`);let i=r.insertRow();i.classList.add(`bg-muted`);let a=i.insertCell();a.classList.add(`p-2`,`font-medium`,`text-left`),a.textContent=`Column`;let o=i.insertCell();o.classList.add(`p-2`,`font-medium`,`text-left`),o.textContent=`Type`;let s=t.columns.some(e=>this.getItemMetadata(t,e)!==void 0),c;s&&(c=i.insertCell(),c.classList.add(`p-2`,`font-medium`,`text-left`),c.textContent=`Metadata`),t.columns.forEach((e,n)=>{let i=r.insertRow();i.classList.add(n%2==0?`bg-background`:`bg-muted/30`);let a=i.insertCell();a.classList.add(`p-2`,`font-mono`),a.textContent=e.name;let o=i.insertCell();if(o.classList.add(`p-2`,`text-muted-foreground`),o.textContent=e.type,s){let n=i.insertCell();n.classList.add(`p-2`);let r=this.getItemMetadata(t,e);r&&n.append(r)}}),e.append(r),n.append(e)}return n}getItemMetadata(e,t){let n=e.primary_keys?.includes(t.name),r=e.indexes?.includes(t.name);if(n||r){let e=document.createElement(`span`);e.textContent=n?`PK`:`IDX`;let t=n?`bg-blue-100 text-blue-800 dark:bg-blue-900 dark:text-blue-200`:`bg-amber-100 text-amber-800 dark:bg-amber-900 dark:text-amber-200`;return e.classList.add(`text-xs`,`px-1.5`,`py-0.5`,`rounded-full`,`font-medium`,...t.split(` `)),e}}},Bo=class extends bt{title=`Variables`;mentionPrefix=`@`;contextType=`variable`;variables;tablesMap;constructor(e,t){super(),this.variables=e,this.tablesMap=t}getItems(){let e=new Set(this.tablesMap.keys());return Object.entries(this.variables).flatMap(([t,n])=>e.has(t)?[]:[{uri:this.asURI(t),name:t,type:this.contextType,description:n.dataType??``,dataType:n.dataType,data:{variable:n}}])}formatCompletion(e){let{data:t}=e,{variable:n}=t;return{label:`@${n.name}`,displayLabel:n.name,detail:n.dataType??``,type:this.contextType,section:Bt.VARIABLE,info:()=>w(n)}}formatContext(e){let{data:t}=e,{variable:n}=t;return Dt({type:this.contextType,data:{name:n.name,dataType:n.dataType},details:n.value==null?void 0:JSON.stringify(n.value)})}};function Vo(e){let t=e.get(h),n=e.get(E),r=e.get(ie);return new Rt().register(new zo(n)).register(new Bo(r,n)).register(new Fo(e)).register(new ho(e)).register(new It(t.connectionsMap,n))}function Ho(){return new Ro(ht())}function Uo({input:e}){let t=``;return e.includes(`@`)&&(t=Zo(e),k.debug(`Included context`,t)),{includeOtherCode:b(``),context:{plainText:t,schema:[],variables:[]}}}var Wo=`data-marimo-context`,Go={marimo:{source:`context`}};function Ko(e){return e.type===`file`&&e.providerMetadata?.marimo?.source===Go.marimo.source}function qo(e){return{...e,providerMetadata:{...e.providerMetadata,marimo:{...e.providerMetadata?.marimo,...Go.marimo}}}}async function Jo(e){if(!e.includes(`@`))return{plainText:``,contextIds:[],attachments:[]};let t=Vo(s),n=t.parseAllContextIds(e);if(n.length===0)return{plainText:``,contextIds:[],attachments:[]};let r=t.formatContextForAI(n),i=[];try{i=(await t.getAttachmentsForContext(n)).map(qo)}catch(e){k.error(`Error getting attachments:`,e)}return{plainText:r,contextIds:n,attachments:i}}async function Yo(e){let{plainText:t,contextIds:n,attachments:r}=await Jo(e),i=null;return t.trim()&&(i={type:Wo,data:{plainText:t,contextIds:n.map(String)}}),{contextPart:i,attachments:r}}async function Xo({input:e}){let{plainText:t,attachments:n}=await Jo(e);return{body:{includeOtherCode:b(``),context:{plainText:t,schema:[],variables:[]}},attachments:n}}function Zo(e){let t=Vo(s),n=t.parseAllContextIds(e);return t.formatContextForAI(n)}function Qo(e,t=[]){return n=>{let r=e.map(e=>n.matchBefore(e)).find(Boolean);return!r||r&&r.from===r.to&&!n.explicit?null:{from:r?.from,options:[...t]}}}function $o(e){if(e.current?.view){let t=e.current.view.state.selection.main.from;e.current.view.dispatch({changes:{from:t,to:t,insert:`@`},selection:{anchor:t+1,head:t+1}}),e.current.view.focus(),dt(e.current.view)}}var es=new Set([`openai`,`google`,`anthropic`]),ts=[`image/*`,`text/*`],ns=52428800;function rs(e){return e.length>50?`${e.slice(0,50)}...`:e}async function is(e){return await Promise.all(e.map(async e=>({type:`file`,mediaType:e.type,filename:e.name,url:await Pt(e,`dataUrl`)})))}function as(e){if(e.length===0)return!1;let t=e.at(-1);if(!t||t.role!==`assistant`||!t.parts)return!1;let n=t.parts;return n.length!==0&&n[n.length-1].type===`reasoning`}function os(e){return e.map(e=>e.type===`text`?e.text:``).join(`
-`)}async function ss(e){let t=await Xo({input:os(e.flatMap(e=>e.parts))});function n(e,n){return n?{...e,parts:[...e.parts,...t.attachments]}:e}return{...t.body,uiMessages:e.map((t,r)=>n(t,r===e.length-1))}}async function cs({invokeAiTool:e,addToolOutput:t,toolCall:n,toolContext:r}){try{if(ko.has(n.toolName)){let e=await ko.invoke({toolName:n.toolName,rawArgs:n.input,toolContext:r});t({tool:n.toolName,toolCallId:n.toolCallId,output:e.result||e.error})}else{let r=await e({toolName:n.toolName,arguments:n.input??{}});t({tool:n.toolName,toolCallId:n.toolCallId,output:r.result||r.error})}}catch(e){k.error(`Tool call failed:`,e),t({tool:n.toolName,toolCallId:n.toolCallId,output:`Error: ${e instanceof Error?e.message:String(e)}`})}}function ls(e){let t=e.at(-1);if(!t||t.role!==`assistant`)return!1;let n=t.parts.at(-1);return!n||!V(n)?!1:ya({messages:e})||va({messages:e})||n.state===`output-denied`&&!n.providerExecuted}function us(e){let t=e.at(-1);return!t||t.role!==`assistant`?!1:t.parts.some(e=>V(e)&&ds(e.state))}function ds(e){switch(e){case`approval-requested`:case`input-streaming`:case`input-available`:return!0;case`approval-responded`:case`output-available`:case`output-error`:case`output-denied`:return!1;default:return e}}function fs(){let e=(0,A.c)(7),t;e[0]===Symbol.for(`react.memo_cache_sentinel`)?(t=[],e[0]=t):t=e[0];let[n,r]=(0,U.useState)(t),i;e[1]===Symbol.for(`react.memo_cache_sentinel`)?(i=e=>{if(e.length!==0){if(e.reduce(ps,0)>ns){Ze({title:`File size exceeded`,description:`Attachments must be under 50 MB`,variant:`danger`});return}r(t=>[...t,...e])}},e[1]=i):i=e[1];let a=c(i),o;e[2]===Symbol.for(`react.memo_cache_sentinel`)?(o=()=>r([]),e[2]=o):o=e[2];let s=o,l;e[3]===Symbol.for(`react.memo_cache_sentinel`)?(l=e=>r(t=>t.filter(t=>t!==e)),e[3]=l):l=e[3];let u=l,d;return e[4]!==a||e[5]!==n?(d={files:n,addFiles:a,clearFiles:s,removeFile:u},e[4]=a,e[5]=n,e[6]=d):d=e[6],d}function ps(e,t){return e+t.size}function ms(e){return e.isError?!1:e.isAbort?!0:!e.hasPendingToolCalls&&!e.hasUnresolvedToolCalls}function hs(){let[e,t]=(0,U.useState)([]),n=(0,U.useRef)([]),r=(0,U.useRef)(!1);return n.current=e,r.current=e.length>0,{messages:e,enqueue:c(e=>{t(t=>{let i=[...t,{id:_t(),parts:e}];return n.current=i,r.current=!0,i})}),flushNext:c(e=>{let i=n.current;if(i.length===0)return;let[a,...o]=i;n.current=o,r.current=o.length>0,t(o),e(a.parts)}),clear:c(()=>{n.current=[],r.current=!1,t([])}),hasQueuedRef:r}}var gs=e=>{let t=(0,A.c)(6),{attachment:n}=e;if(n.mediaType?.startsWith(`image/`)){let e=n.filename||`Attachment`,r;return t[0]!==n.url||t[1]!==e?(r=(0,W.jsx)(`img`,{src:n.url,alt:e,className:`max-h-[100px] max-w-[100px] object-contain mb-1.5`}),t[0]=n.url,t[1]=e,t[2]=r):r=t[2],r}let r;t[3]===Symbol.for(`react.memo_cache_sentinel`)?(r=(0,W.jsx)(zt,{className:`h-3 w-3 mt-0.5`}),t[3]=r):r=t[3];let i=n.filename||`Attachment`,a;return t[4]===i?a=t[5]:(a=(0,W.jsxs)(`div`,{className:`flex flex-row gap-1 items-center text-xs`,children:[r,i]}),t[4]=i,t[5]=a),a},_s=e=>{let t=(0,A.c)(12),{file:n,className:r,onRemove:i}=e,[a,o]=(0,U.useState)(!1),s;t[0]===r?s=t[1]:(s=Pe(`py-1 px-1.5 bg-muted rounded-md cursor-pointer flex flex-row gap-1 items-center text-xs`,r),t[0]=r,t[1]=s);let c,l;t[2]===Symbol.for(`react.memo_cache_sentinel`)?(c=()=>o(!0),l=()=>o(!1),t[2]=c,t[3]=l):(c=t[2],l=t[3]);let u;t[4]!==n||t[5]!==a||t[6]!==i?(u=a?(0,W.jsx)(Ke,{className:`h-3 w-3 mt-0.5`,onClick:i}):xs(n),t[4]=n,t[5]=a,t[6]=i,t[7]=u):u=t[7];let d;return t[8]!==n.name||t[9]!==s||t[10]!==u?(d=(0,W.jsxs)(`div`,{className:s,onMouseEnter:c,onMouseLeave:l,children:[u,n.name]}),t[8]=n.name,t[9]=s,t[10]=u,t[11]=d):d=t[11],d},vs=e=>{let t=(0,A.c)(12),{isLoading:n,onStop:r,onSendClick:i,isEmpty:a,showStopLabel:o}=e,s=o!==void 0&&o,c;t[0]===s?c=t[1]:(c=s?(0,W.jsxs)(`div`,{className:`flex flex-row items-center gap-1 px-1.5`,children:[(0,W.jsx)(`span`,{className:`text-xs text-error`,children:`Stop`}),(0,W.jsx)(Tt,{size:`small`})]}):(0,W.jsx)(Kt,{className:`size-4 text-error`}),t[0]=s,t[1]=c);let l=c,u=n?`Stop`:`Submit`,d=n?r:i,f=!n&&a,p;t[2]!==n||t[3]!==l?(p=n?l:(0,W.jsx)(Jt,{className:`h-3.5 w-3.5`}),t[2]=n,t[3]=l,t[4]=p):p=t[4];let m;t[5]!==d||t[6]!==f||t[7]!==p?(m=(0,W.jsx)(He,{variant:`text`,size:`sm`,className:`h-6 min-w-6 p-0 hover:bg-muted/30 cursor-pointer`,onClick:d,disabled:f,children:p}),t[5]=d,t[6]=f,t[7]=p,t[8]=m):m=t[8];let h;return t[9]!==u||t[10]!==m?(h=(0,W.jsx)(Ve,{content:u,children:m}),t[9]=u,t[10]=m,t[11]=h):h=t[11],h},ys=e=>{let t=(0,A.c)(4),{handleAddContext:n,isLoading:r}=e,i;t[0]===Symbol.for(`react.memo_cache_sentinel`)?(i=(0,W.jsx)(wt,{className:`h-3.5 w-3.5`}),t[0]=i):i=t[0];let a;return t[1]!==n||t[2]!==r?(a=(0,W.jsx)(Ve,{content:`Add context`,children:(0,W.jsx)(He,{variant:`text`,size:`icon`,onClick:n,disabled:r,children:i})}),t[1]=n,t[2]=r,t[3]=a):a=t[3],a},bs=e=>{let t=(0,A.c)(15),{fileInputRef:n,isLoading:r,onAddFiles:i}=e,a;t[0]===n?a=t[1]:(a=()=>n.current?.click(),t[0]=n,t[1]=a);let o;t[2]===Symbol.for(`react.memo_cache_sentinel`)?(o=(0,W.jsx)(qt,{className:`h-3.5 w-3.5`}),t[2]=o):o=t[2];let s;t[3]!==r||t[4]!==a?(s=(0,W.jsx)(Ve,{content:`Attach a file`,children:(0,W.jsx)(He,{variant:`text`,size:`icon`,onClick:a,disabled:r,children:o})}),t[3]=r,t[4]=a,t[5]=s):s=t[5];let c;t[6]===i?c=t[7]:(c=e=>{e.target.files&&i([...e.target.files])},t[6]=i,t[7]=c);let l;t[8]===Symbol.for(`react.memo_cache_sentinel`)?(l=ts.join(`,`),t[8]=l):l=t[8];let u;t[9]!==n||t[10]!==c?(u=(0,W.jsx)(Xe,{ref:n,type:`file`,multiple:!0,hidden:!0,onChange:c,accept:l}),t[9]=n,t[10]=c,t[11]=u):u=t[11];let d;return t[12]!==s||t[13]!==u?(d=(0,W.jsxs)(W.Fragment,{children:[s,u]}),t[12]=s,t[13]=u,t[14]=d):d=t[14],d};function xs(e){let t=`h-3 w-3 mt-0.5`;return e.type.startsWith(`image/`)?(0,W.jsx)(Ot,{className:t}):e.type.startsWith(`text/`)?(0,W.jsx)(Ge,{className:t}):(0,W.jsx)(zt,{className:t})}var Ss=e=>{let t=(0,A.c)(15),{icon:n,title:r,subtitle:i,href:a}=e,o;t[0]===r?o=t[1]:(o=(0,W.jsx)(`span`,{className:`truncate font-medium`,children:r}),t[0]=r,t[1]=o);let s;t[2]===i?s=t[3]:(s=i&&(0,W.jsxs)(`span`,{className:`truncate opacity-70`,children:[`(`,i,`)`]}),t[2]=i,t[3]=s);let c;t[4]!==n||t[5]!==o||t[6]!==s?(c=(0,W.jsxs)(W.Fragment,{children:[n,o,s]}),t[4]=n,t[5]=o,t[6]=s,t[7]=c):c=t[7];let l=c;if(a&&Ae(a)){let e=i??r,n;return t[8]!==l||t[9]!==a||t[10]!==e?(n=(0,W.jsx)(`a`,{href:a,target:`_blank`,rel:`noopener noreferrer`,title:e,className:`inline-flex max-w-full items-center gap-1.5 rounded-md border bg-muted/50 px-2 py-1 my-1 text-xs text-muted-foreground hover:bg-muted hover:text-foreground transition-colors`,children:l}),t[8]=l,t[9]=a,t[10]=e,t[11]=n):n=t[11],n}let u=i??r,d;return t[12]!==l||t[13]!==u?(d=(0,W.jsx)(`div`,{className:`inline-flex max-w-full items-center gap-1.5 rounded-md border bg-muted/50 px-2 py-1 my-1 text-xs text-muted-foreground`,title:u,children:l}),t[12]=l,t[13]=u,t[14]=d):d=t[14],d},Cs=nt.define(),Q=$e.define({create(){return new Map},update(e,t){let n=!1,r=e;for(let i of t.effects)if(i.is(Cs)){n||=(r=new Map(e),!0);for(let[e,t]of i.value)r.set(e,t)}return r}}),ws=$e.define({create(){return{onResourceClick:void 0,onResourceMouseOver:void 0,onResourceMouseOut:void 0,onPromptSubmit:void 0}},update(e){return e}}),Ts=nt.define();$e.define({create(){return new Map},update(e,t){let n=!1,r=e;for(let i of t.effects)if(i.is(Ts)){n||=(r=new Map(e),!0);for(let[e,t]of i.value)r.set(e,t)}return r}});var Es=(e,t)=>async n=>{let r=n.matchBefore(/@(\w+)?/);if(!r||r.from===r.to&&!n.explicit)return null;let i=await e();if(i.length===0)return null;let a=Cs.of(new Map(i.map(e=>[e.uri,e])));n.view&&n.view.dispatch({effects:a});let o=i.map(e=>({label:`@${e.name}`,displayLabel:e.name,detail:e.uri,info:e.description||void 0,type:e.mimeType?`constant`:`variable`,boost:e.description?100:0,...t?.(e),apply:(t,n,r,i)=>{t.dispatch({changes:{from:r,to:i,insert:`@${e.uri} `}})}}));return{from:r.from,options:o}},Ds=/@[\w-]+:\/\/(?!\/)[^\s]+/;function $(e){return e.matchAll(new RegExp(Ds.source,`g`))}var Os=class extends it{resource;view;constructor(e,t){super(),this.resource=e,this.view=t}eq(e){return e.resource.uri===this.resource.uri}toDOM(){let e=document.createElement(`span`);e.className=`cm-resource-widget`,e.textContent=`@${this.resource.name}`;let t=this.view.state.field(ws,!1),n=t?.onResourceClick,r=t?.onResourceMouseOver,i=t?.onResourceMouseOut;return n&&(e.style.cursor=`pointer`,e.addEventListener(`click`,e=>{e.preventDefault(),e.stopPropagation(),n(this.resource)})),r&&e.addEventListener(`mouseover`,e=>{e.preventDefault(),e.stopPropagation(),r(this.resource)}),i&&e.addEventListener(`mouseout`,e=>{e.preventDefault(),e.stopPropagation(),i(this.resource)}),e}},ks=class extends it{uri;view;constructor(e,t){super(),this.uri=e,this.view=t}eq(e){return e.uri===this.uri}toDOM(){let e=document.createElement(`span`);return e.className=`cm-not-found-resource-widget`,e.textContent=`@${this.uri.split(`://`)[1]??this.uri}`,e}};function As(e){let t=e.state.field(Q),n=[];for(let{from:r,to:i}of e.visibleRanges){let a=e.state.doc.sliceString(r,i);for(let i of $(a)){let a=r+i.index,o=i[0].slice(1),s=t.get(o);s?n.push(rt.replace({widget:new Os(s,e)}).range(a,a+i[0].length)):n.push(rt.replace({widget:new ks(o,e)}).range(a,a+i[0].length))}}return rt.set(n)}var js=ct.fromClass(class{decorations;constructor(e){this.decorations=As(e)}update(e){(e.docChanged||e.viewportChanged||e.transactions.some(e=>e.effects.some(e=>e.is(Cs))))&&(this.decorations=As(e.view))}},{decorations:e=>e.decorations});function Ms(e){let t=document.createElement(`div`);t.className=`cm-tooltip-cursor`;let n=document.createElement(`div`);if(n.className=`cm-tooltip-cursor-title`,n.textContent=`${e.name} (${e.uri})`,t.appendChild(n),e.description){let n=document.createElement(`div`);n.className=`cm-tooltip-cursor-description`,n.textContent=e.description,t.appendChild(n)}if(e.mimeType){let n=document.createElement(`div`);n.className=`cm-tooltip-cursor-mimetype`,n.textContent=e.mimeType,t.appendChild(n)}return{dom:t}}function Ns(e,t,n,r=0){for(let i of $(e)){let e=r+i.index,a=e+i[0].length;if(t>=e&&t<=a){let t=i[0].slice(1),r=n.get(t);if(r)return{resource:r,start:e,end:a}}}return null}function Ps(e){return lt((t,n)=>{let{from:r,text:i}=t.state.doc.lineAt(n),a=Ns(i,n,t.state.field(Q),r);return a?{pos:a.start,end:a.end,above:!0,create(){return(e.createTooltip??Ms)(a.resource)}}:null})}function Fs(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,i.to),o=Array.from($(a));if(o.length===0)return null;for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);if(t===a&&n.has(o))return{from:r,to:a,uri:o}}return null}function Is(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t).to,a=r.sliceString(t,i),o=Array.from($(a));if(o.length===0)return null;let s=o[0];if(!s||s.index!==0)return null;let c=t,l=t+s[0].length,u=s[0].slice(1);return n.has(u)?{from:c,to:l,uri:u}:null}function Ls(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,i.to),o=Array.from($(a));for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);if(t>r&&t<a&&n.has(o))return{from:r,to:a,uri:o}}return null}function Rs(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,t),o=Array.from($(a)),s=null;for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);n.has(o)&&a<=t&&(t>a?s=a:t===a&&(s=r))}return s}function zs(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,i.to),o=Array.from($(a));for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);if(n.has(o)&&r>=t){if(t<r)return r;if(t===r)return a}}return null}var Bs=tt.transactionFilter.of(e=>{let t=e.annotation(at.userEvent);if(!t)return e;let n=e.startState.selection.main;if(!n.empty)return e;if(t===`input.type`||t.startsWith(`input`)){if(e.changes.empty)return e;let t=e.startState.field(Q),n=e.startState.doc,r=!1,i=0,a=[];if(e.changes.iterChanges((e,o,s,c,l)=>{let u=l.toString(),d=e;if(/^\s+$/.test(u)){a.push({from:e,to:o,insert:u});return}let f=n.lineAt(d),p=n.sliceString(f.from,f.to),m=Array.from($(p)),h=!1,g=!1;for(let e of m){let r=f.from+e.index,i=r+e[0].length,a=e[0].slice(1);if(t.has(a)){if(d===r){let e=d>0?n.sliceString(d-1,d):``;e!==``&&e!==` `&&e!==`
+`)}},Io=x(),Lo={maxDepth:3,maxResults:20,defaultResultsLimit:5,includeDirectories:!1},Ro=class extends xt{title=`Files`;mentionPrefix=`#`;contextType=`file`;apiRequests;config;constructor(e,t=Lo){super(),this.apiRequests=e,this.config=t}createCompletionSource(){return async e=>{let t=e.matchBefore(/#[^\s#]*/);if(!t)return null;let n=t.text;if(!n.startsWith(`#`))return null;let r=n.slice(1);if(r.length===0)return this.getDefaultCompletions(t);try{let e=(await this.searchFiles(r)||[]).map(e=>{let t={uri:this.asURI(e.path),name:e.name,type:this.contextType,description:e.isDirectory?`Directory`:`File`,data:{path:e.path,isDirectory:e.isDirectory}};return this.formatCompletion(t)});return{from:t.from,options:e}}catch(e){return k.error(`Failed to search files:`,e),null}}}searchFiles=async(e,t={})=>{let{maxDepth:n,maxResults:r,includeDirectories:i}={...this.config,...t};return(await this.apiRequests.sendSearchFiles({query:e,includeFiles:!0,includeDirectories:i,depth:n,limit:r})).files};async getDefaultCompletions(e){try{for(let t of[`py`,`md`,`csv`])try{let n=await this.searchFiles(t,{maxDepth:1,maxResults:5});if(n&&n.length>0){let t=n.map(e=>{let t={uri:this.asURI(e.path),name:e.name,type:this.contextType,description:e.isDirectory?`Directory`:`File`,data:{path:e.path,isDirectory:e.isDirectory}};return this.formatCompletion(t)});return{from:e.from,options:t}}}catch(e){k.error(`Failed to get default file completions:`,e)}return{from:e.from,options:[]}}catch(e){return k.error(`Failed to get default file completions:`,e),null}}getItems(){return[]}formatCompletion(e){let{data:t,name:n}=e,r=t.isDirectory?`📁`:`📄`;return{...this.createBasicCompletion(e),type:`file`,section:Bt.FILE,boost:t.isDirectory?Mt.MEDIUM:Mt.LOW,detail:t.path,displayLabel:`${r} ${n}`,apply:async(e,r,i,a)=>{let o=e.state.facet(Io)?.addAttachment;if(!o){k.warn(`No addAttachment callback provided`);return}let s=await this.apiRequests.sendFileDetails({path:t.path}).catch(e=>(Ze({title:`Failed to get file details`,description:e.message}),null));if(!s)return;let c=s.mimeType||`text/plain`,l=Pt({contents:s.contents??``,mimeType:c,isBase64:s.isBase64??!1});o(new File([l],n,{type:c})),e.dispatch({changes:{from:i,to:a,insert:``}}),pt(e)},info:()=>{let e=document.createElement(`div`);e.classList.add(`flex`,`flex-col`,`gap-1`,`p-2`);let r=document.createElement(`div`);r.classList.add(`font-bold`),r.textContent=n,e.append(r);let i=document.createElement(`div`);return i.classList.add(`text-xs`,`text-muted-foreground`),i.textContent=t.path,e.append(i),e}}}formatContext(e){let{data:t,name:n}=e;return Ot({type:this.contextType,data:{name:n,path:t.path,isDirectory:t.isDirectory},details:t.isDirectory?`Directory containing files and subdirectories`:`File`})}},zo=class extends xt{title=`Tables`;mentionPrefix=`@`;contextType=`data`;tablesMap;constructor(e){super(),this.tablesMap=e}getItems(){return[...this.tablesMap.entries()].map(([e,t])=>({uri:this.asURI(e),name:e,type:this.contextType,description:t.source===`memory`?`in-memory`:t.source,data:t}))}formatContext(e){let{data:t}=e,{columns:n,source:r,num_rows:i,num_columns:a,name:o,variable_name:s}=t,c=[i==null?void 0:`${i} rows`,a==null?void 0:`${a} columns`].filter(Boolean).join(`, `),l=``;if(c&&(l+=`Shape: ${c}\n`),s&&(l+=`Variable: ${s}\n`),n&&n.length>0){l+=`Columns:
+`;for(let e of n){let t=`  ${e.name} (${e.type})`;if(e.sample_values&&e.sample_values.length>0){let n=e.sample_values.slice(0,3).map(e=>e===null?`null`:String(e)).join(`, `);t+=` - samples: [${n}]`}l+=`${t}\n`}}return Ot({type:this.contextType,data:{name:o,source:r??`unknown`},details:l.trim()})}formatCompletion(e){let t=e.data.name,n=e.data;return{label:`@${t}`,displayLabel:t,detail:n.source===`memory`?`in-memory`:n.source,boost:n.source_type===`local`?Mt.LOCAL_TABLE:Mt.REMOTE_TABLE,type:p(n),apply:`@${t}`,section:{name:p(n)===`dataframe`?`Dataframe`:`Table`,rank:Bt.TABLE.rank},info:()=>this.createTableInfoElement(t,n)}}createTableInfoElement(e,t){let n=document.createElement(`div`);n.classList.add(`mo-cm-tooltip`,`docs-documentation`,`min-w-[200px]`,`flex`,`flex-col`,`gap-1`,`p-2`);let r=document.createElement(`div`);r.classList.add(`flex`,`flex-col`,`gap-1`);let i=document.createElement(`div`);if(i.classList.add(`font-bold`,`text-base`),i.textContent=e,r.append(i),t.source){let e=document.createElement(`div`);e.classList.add(`text-xs`,`text-muted-foreground`),e.textContent=`Source: ${t.source}`,r.append(e)}n.append(r);let a=[t.num_rows==null?void 0:`${t.num_rows} rows`,t.num_columns==null?void 0:`${t.num_columns} columns`].filter(Boolean).join(`, `);if(a){let e=document.createElement(`div`);e.classList.add(`text-xs`,`bg-muted`,`px-2`,`py-1`,`rounded`),e.textContent=a,n.append(e)}if(t.columns&&t.columns.length>0){let e=document.createElement(`div`);e.classList.add(`overflow-auto`,`max-h-60`);let r=document.createElement(`table`);r.classList.add(`w-full`,`text-xs`,`border-collapse`);let i=r.insertRow();i.classList.add(`bg-muted`);let a=i.insertCell();a.classList.add(`p-2`,`font-medium`,`text-left`),a.textContent=`Column`;let o=i.insertCell();o.classList.add(`p-2`,`font-medium`,`text-left`),o.textContent=`Type`;let s=t.columns.some(e=>this.getItemMetadata(t,e)!==void 0),c;s&&(c=i.insertCell(),c.classList.add(`p-2`,`font-medium`,`text-left`),c.textContent=`Metadata`),t.columns.forEach((e,n)=>{let i=r.insertRow();i.classList.add(n%2==0?`bg-background`:`bg-muted/30`);let a=i.insertCell();a.classList.add(`p-2`,`font-mono`),a.textContent=e.name;let o=i.insertCell();if(o.classList.add(`p-2`,`text-muted-foreground`),o.textContent=e.type,s){let n=i.insertCell();n.classList.add(`p-2`);let r=this.getItemMetadata(t,e);r&&n.append(r)}}),e.append(r),n.append(e)}return n}getItemMetadata(e,t){let n=e.primary_keys?.includes(t.name),r=e.indexes?.includes(t.name);if(n||r){let e=document.createElement(`span`);e.textContent=n?`PK`:`IDX`;let t=n?`bg-blue-100 text-blue-800 dark:bg-blue-900 dark:text-blue-200`:`bg-amber-100 text-amber-800 dark:bg-amber-900 dark:text-amber-200`;return e.classList.add(`text-xs`,`px-1.5`,`py-0.5`,`rounded-full`,`font-medium`,...t.split(` `)),e}}},Bo=class extends xt{title=`Variables`;mentionPrefix=`@`;contextType=`variable`;variables;tablesMap;constructor(e,t){super(),this.variables=e,this.tablesMap=t}getItems(){let e=new Set(this.tablesMap.keys());return Object.entries(this.variables).flatMap(([t,n])=>e.has(t)?[]:[{uri:this.asURI(t),name:t,type:this.contextType,description:n.dataType??``,dataType:n.dataType,data:{variable:n}}])}formatCompletion(e){let{data:t}=e,{variable:n}=t;return{label:`@${n.name}`,displayLabel:n.name,detail:n.dataType??``,type:this.contextType,section:Bt.VARIABLE,info:()=>w(n)}}formatContext(e){let{data:t}=e,{variable:n}=t;return Ot({type:this.contextType,data:{name:n.name,dataType:n.dataType},details:n.value==null?void 0:JSON.stringify(n.value)})}};function Vo(e){let t=e.get(h),n=e.get(E),r=e.get(ie);return new Lt().register(new zo(n)).register(new Bo(r,n)).register(new Fo(e)).register(new ho(e)).register(new It(t.connectionsMap,n))}function Ho(){return new Ro(ht())}function Uo({input:e}){let t=``;return e.includes(`@`)&&(t=Zo(e),k.debug(`Included context`,t)),{includeOtherCode:b(``),context:{plainText:t,schema:[],variables:[]}}}var Wo=`data-marimo-context`,Go={marimo:{source:`context`}};function Ko(e){return e.type===`file`&&e.providerMetadata?.marimo?.source===Go.marimo.source}function qo(e){return{...e,providerMetadata:{...e.providerMetadata,marimo:{...e.providerMetadata?.marimo,...Go.marimo}}}}async function Jo(e){if(!e.includes(`@`))return{plainText:``,contextIds:[],attachments:[]};let t=Vo(s),n=t.parseAllContextIds(e);if(n.length===0)return{plainText:``,contextIds:[],attachments:[]};let r=t.formatContextForAI(n),i=[];try{i=(await t.getAttachmentsForContext(n)).map(qo)}catch(e){k.error(`Error getting attachments:`,e)}return{plainText:r,contextIds:n,attachments:i}}async function Yo(e){let{plainText:t,contextIds:n,attachments:r}=await Jo(e),i=null;return t.trim()&&(i={type:Wo,data:{plainText:t,contextIds:n.map(String)}}),{contextPart:i,attachments:r}}async function Xo({input:e}){let{plainText:t,attachments:n}=await Jo(e);return{body:{includeOtherCode:b(``),context:{plainText:t,schema:[],variables:[]}},attachments:n}}function Zo(e){let t=Vo(s),n=t.parseAllContextIds(e);return t.formatContextForAI(n)}function Qo(e,t=[]){return n=>{let r=e.map(e=>n.matchBefore(e)).find(Boolean);return!r||r&&r.from===r.to&&!n.explicit?null:{from:r?.from,options:[...t]}}}function $o(e){if(e.current?.view){let t=e.current.view.state.selection.main.from;e.current.view.dispatch({changes:{from:t,to:t,insert:`@`},selection:{anchor:t+1,head:t+1}}),e.current.view.focus(),dt(e.current.view)}}var es=new Set([`openai`,`google`,`anthropic`]),ts=[`image/*`,`text/*`],ns=52428800;function rs(e){return e.length>50?`${e.slice(0,50)}...`:e}async function is(e){return await Promise.all(e.map(async e=>({type:`file`,mediaType:e.type,filename:e.name,url:await jt(e,`dataUrl`)})))}function as(e){if(e.length===0)return!1;let t=e.at(-1);if(!t||t.role!==`assistant`||!t.parts)return!1;let n=t.parts;return n.length!==0&&n[n.length-1].type===`reasoning`}function os(e){return e.map(e=>e.type===`text`?e.text:``).join(`
+`)}async function ss(e){let t=await Xo({input:os(e.flatMap(e=>e.parts))});function n(e,n){return n?{...e,parts:[...e.parts,...t.attachments]}:e}return{...t.body,uiMessages:e.map((t,r)=>n(t,r===e.length-1))}}async function cs({invokeAiTool:e,addToolOutput:t,toolCall:n,toolContext:r}){try{if(ko.has(n.toolName)){let e=await ko.invoke({toolName:n.toolName,rawArgs:n.input,toolContext:r});t({tool:n.toolName,toolCallId:n.toolCallId,output:e.result||e.error})}else{let r=await e({toolName:n.toolName,arguments:n.input??{}});t({tool:n.toolName,toolCallId:n.toolCallId,output:r.result||r.error})}}catch(e){k.error(`Tool call failed:`,e),t({tool:n.toolName,toolCallId:n.toolCallId,output:`Error: ${e instanceof Error?e.message:String(e)}`})}}function ls(e){let t=e.at(-1);if(!t||t.role!==`assistant`)return!1;let n=t.parts.at(-1);return!n||!V(n)?!1:ya({messages:e})||va({messages:e})||n.state===`output-denied`&&!n.providerExecuted}function us(e){let t=e.at(-1);return!t||t.role!==`assistant`?!1:t.parts.some(e=>V(e)&&ds(e.state))}function ds(e){switch(e){case`approval-requested`:case`input-streaming`:case`input-available`:return!0;case`approval-responded`:case`output-available`:case`output-error`:case`output-denied`:return!1;default:return e}}function fs(){let e=(0,A.c)(7),t;e[0]===Symbol.for(`react.memo_cache_sentinel`)?(t=[],e[0]=t):t=e[0];let[n,r]=(0,U.useState)(t),i;e[1]===Symbol.for(`react.memo_cache_sentinel`)?(i=e=>{if(e.length!==0){if(e.reduce(ps,0)>ns){Ze({title:`File size exceeded`,description:`Attachments must be under 50 MB`,variant:`danger`});return}r(t=>[...t,...e])}},e[1]=i):i=e[1];let a=c(i),o;e[2]===Symbol.for(`react.memo_cache_sentinel`)?(o=()=>r([]),e[2]=o):o=e[2];let s=o,l;e[3]===Symbol.for(`react.memo_cache_sentinel`)?(l=e=>r(t=>t.filter(t=>t!==e)),e[3]=l):l=e[3];let u=l,d;return e[4]!==a||e[5]!==n?(d={files:n,addFiles:a,clearFiles:s,removeFile:u},e[4]=a,e[5]=n,e[6]=d):d=e[6],d}function ps(e,t){return e+t.size}function ms(e){return e.isError?!1:e.isAbort?!0:!e.hasPendingToolCalls&&!e.hasUnresolvedToolCalls}function hs(){let[e,t]=(0,U.useState)([]),n=(0,U.useRef)([]),r=(0,U.useRef)(!1);return n.current=e,r.current=e.length>0,{messages:e,enqueue:c(e=>{t(t=>{let i=[...t,{id:_t(),parts:e}];return n.current=i,r.current=!0,i})}),flushNext:c(e=>{let i=n.current;if(i.length===0)return;let[a,...o]=i;n.current=o,r.current=o.length>0,t(o),e(a.parts)}),clear:c(()=>{n.current=[],r.current=!1,t([])}),hasQueuedRef:r}}var gs=e=>{let t=(0,A.c)(6),{attachment:n}=e;if(n.mediaType?.startsWith(`image/`)){let e=n.filename||`Attachment`,r;return t[0]!==n.url||t[1]!==e?(r=(0,W.jsx)(`img`,{src:n.url,alt:e,className:`max-h-[100px] max-w-[100px] object-contain mb-1.5`}),t[0]=n.url,t[1]=e,t[2]=r):r=t[2],r}let r;t[3]===Symbol.for(`react.memo_cache_sentinel`)?(r=(0,W.jsx)(zt,{className:`h-3 w-3 mt-0.5`}),t[3]=r):r=t[3];let i=n.filename||`Attachment`,a;return t[4]===i?a=t[5]:(a=(0,W.jsxs)(`div`,{className:`flex flex-row gap-1 items-center text-xs`,children:[r,i]}),t[4]=i,t[5]=a),a},_s=e=>{let t=(0,A.c)(12),{file:n,className:r,onRemove:i}=e,[a,o]=(0,U.useState)(!1),s;t[0]===r?s=t[1]:(s=Pe(`py-1 px-1.5 bg-muted rounded-md cursor-pointer flex flex-row gap-1 items-center text-xs`,r),t[0]=r,t[1]=s);let c,l;t[2]===Symbol.for(`react.memo_cache_sentinel`)?(c=()=>o(!0),l=()=>o(!1),t[2]=c,t[3]=l):(c=t[2],l=t[3]);let u;t[4]!==n||t[5]!==a||t[6]!==i?(u=a?(0,W.jsx)(Ke,{className:`h-3 w-3 mt-0.5`,onClick:i}):xs(n),t[4]=n,t[5]=a,t[6]=i,t[7]=u):u=t[7];let d;return t[8]!==n.name||t[9]!==s||t[10]!==u?(d=(0,W.jsxs)(`div`,{className:s,onMouseEnter:c,onMouseLeave:l,children:[u,n.name]}),t[8]=n.name,t[9]=s,t[10]=u,t[11]=d):d=t[11],d},vs=e=>{let t=(0,A.c)(12),{isLoading:n,onStop:r,onSendClick:i,isEmpty:a,showStopLabel:o}=e,s=o!==void 0&&o,c;t[0]===s?c=t[1]:(c=s?(0,W.jsxs)(`div`,{className:`flex flex-row items-center gap-1 px-1.5`,children:[(0,W.jsx)(`span`,{className:`text-xs text-error`,children:`Stop`}),(0,W.jsx)(Dt,{size:`small`})]}):(0,W.jsx)(Kt,{className:`size-4 text-error`}),t[0]=s,t[1]=c);let l=c,u=n?`Stop`:`Submit`,d=n?r:i,f=!n&&a,p;t[2]!==n||t[3]!==l?(p=n?l:(0,W.jsx)(Jt,{className:`h-3.5 w-3.5`}),t[2]=n,t[3]=l,t[4]=p):p=t[4];let m;t[5]!==d||t[6]!==f||t[7]!==p?(m=(0,W.jsx)(He,{variant:`text`,size:`sm`,className:`h-6 min-w-6 p-0 hover:bg-muted/30 cursor-pointer`,onClick:d,disabled:f,children:p}),t[5]=d,t[6]=f,t[7]=p,t[8]=m):m=t[8];let h;return t[9]!==u||t[10]!==m?(h=(0,W.jsx)(Ve,{content:u,children:m}),t[9]=u,t[10]=m,t[11]=h):h=t[11],h},ys=e=>{let t=(0,A.c)(4),{handleAddContext:n,isLoading:r}=e,i;t[0]===Symbol.for(`react.memo_cache_sentinel`)?(i=(0,W.jsx)(wt,{className:`h-3.5 w-3.5`}),t[0]=i):i=t[0];let a;return t[1]!==n||t[2]!==r?(a=(0,W.jsx)(Ve,{content:`Add context`,children:(0,W.jsx)(He,{variant:`text`,size:`icon`,onClick:n,disabled:r,children:i})}),t[1]=n,t[2]=r,t[3]=a):a=t[3],a},bs=e=>{let t=(0,A.c)(15),{fileInputRef:n,isLoading:r,onAddFiles:i}=e,a;t[0]===n?a=t[1]:(a=()=>n.current?.click(),t[0]=n,t[1]=a);let o;t[2]===Symbol.for(`react.memo_cache_sentinel`)?(o=(0,W.jsx)(qt,{className:`h-3.5 w-3.5`}),t[2]=o):o=t[2];let s;t[3]!==r||t[4]!==a?(s=(0,W.jsx)(Ve,{content:`Attach a file`,children:(0,W.jsx)(He,{variant:`text`,size:`icon`,onClick:a,disabled:r,children:o})}),t[3]=r,t[4]=a,t[5]=s):s=t[5];let c;t[6]===i?c=t[7]:(c=e=>{e.target.files&&i([...e.target.files])},t[6]=i,t[7]=c);let l;t[8]===Symbol.for(`react.memo_cache_sentinel`)?(l=ts.join(`,`),t[8]=l):l=t[8];let u;t[9]!==n||t[10]!==c?(u=(0,W.jsx)(Xe,{ref:n,type:`file`,multiple:!0,hidden:!0,onChange:c,accept:l}),t[9]=n,t[10]=c,t[11]=u):u=t[11];let d;return t[12]!==s||t[13]!==u?(d=(0,W.jsxs)(W.Fragment,{children:[s,u]}),t[12]=s,t[13]=u,t[14]=d):d=t[14],d};function xs(e){let t=`h-3 w-3 mt-0.5`;return e.type.startsWith(`image/`)?(0,W.jsx)(At,{className:t}):e.type.startsWith(`text/`)?(0,W.jsx)(Ge,{className:t}):(0,W.jsx)(zt,{className:t})}var Ss=e=>{let t=(0,A.c)(15),{icon:n,title:r,subtitle:i,href:a}=e,o;t[0]===r?o=t[1]:(o=(0,W.jsx)(`span`,{className:`truncate font-medium`,children:r}),t[0]=r,t[1]=o);let s;t[2]===i?s=t[3]:(s=i&&(0,W.jsxs)(`span`,{className:`truncate opacity-70`,children:[`(`,i,`)`]}),t[2]=i,t[3]=s);let c;t[4]!==n||t[5]!==o||t[6]!==s?(c=(0,W.jsxs)(W.Fragment,{children:[n,o,s]}),t[4]=n,t[5]=o,t[6]=s,t[7]=c):c=t[7];let l=c;if(a&&Ae(a)){let e=i??r,n;return t[8]!==l||t[9]!==a||t[10]!==e?(n=(0,W.jsx)(`a`,{href:a,target:`_blank`,rel:`noopener noreferrer`,title:e,className:`inline-flex max-w-full items-center gap-1.5 rounded-md border bg-muted/50 px-2 py-1 my-1 text-xs text-muted-foreground hover:bg-muted hover:text-foreground transition-colors`,children:l}),t[8]=l,t[9]=a,t[10]=e,t[11]=n):n=t[11],n}let u=i??r,d;return t[12]!==l||t[13]!==u?(d=(0,W.jsx)(`div`,{className:`inline-flex max-w-full items-center gap-1.5 rounded-md border bg-muted/50 px-2 py-1 my-1 text-xs text-muted-foreground`,title:u,children:l}),t[12]=l,t[13]=u,t[14]=d):d=t[14],d},Cs=nt.define(),Q=$e.define({create(){return new Map},update(e,t){let n=!1,r=e;for(let i of t.effects)if(i.is(Cs)){n||=(r=new Map(e),!0);for(let[e,t]of i.value)r.set(e,t)}return r}}),ws=$e.define({create(){return{onResourceClick:void 0,onResourceMouseOver:void 0,onResourceMouseOut:void 0,onPromptSubmit:void 0}},update(e){return e}}),Ts=nt.define();$e.define({create(){return new Map},update(e,t){let n=!1,r=e;for(let i of t.effects)if(i.is(Ts)){n||=(r=new Map(e),!0);for(let[e,t]of i.value)r.set(e,t)}return r}});var Es=(e,t)=>async n=>{let r=n.matchBefore(/@(\w+)?/);if(!r||r.from===r.to&&!n.explicit)return null;let i=await e();if(i.length===0)return null;let a=Cs.of(new Map(i.map(e=>[e.uri,e])));n.view&&n.view.dispatch({effects:a});let o=i.map(e=>({label:`@${e.name}`,displayLabel:e.name,detail:e.uri,info:e.description||void 0,type:e.mimeType?`constant`:`variable`,boost:e.description?100:0,...t?.(e),apply:(t,n,r,i)=>{t.dispatch({changes:{from:r,to:i,insert:`@${e.uri} `}})}}));return{from:r.from,options:o}},Ds=/@[\w-]+:\/\/(?!\/)[^\s]+/;function $(e){return e.matchAll(new RegExp(Ds.source,`g`))}var Os=class extends it{resource;view;constructor(e,t){super(),this.resource=e,this.view=t}eq(e){return e.resource.uri===this.resource.uri}toDOM(){let e=document.createElement(`span`);e.className=`cm-resource-widget`,e.textContent=`@${this.resource.name}`;let t=this.view.state.field(ws,!1),n=t?.onResourceClick,r=t?.onResourceMouseOver,i=t?.onResourceMouseOut;return n&&(e.style.cursor=`pointer`,e.addEventListener(`click`,e=>{e.preventDefault(),e.stopPropagation(),n(this.resource)})),r&&e.addEventListener(`mouseover`,e=>{e.preventDefault(),e.stopPropagation(),r(this.resource)}),i&&e.addEventListener(`mouseout`,e=>{e.preventDefault(),e.stopPropagation(),i(this.resource)}),e}},ks=class extends it{uri;view;constructor(e,t){super(),this.uri=e,this.view=t}eq(e){return e.uri===this.uri}toDOM(){let e=document.createElement(`span`);return e.className=`cm-not-found-resource-widget`,e.textContent=`@${this.uri.split(`://`)[1]??this.uri}`,e}};function As(e){let t=e.state.field(Q),n=[];for(let{from:r,to:i}of e.visibleRanges){let a=e.state.doc.sliceString(r,i);for(let i of $(a)){let a=r+i.index,o=i[0].slice(1),s=t.get(o);s?n.push(rt.replace({widget:new Os(s,e)}).range(a,a+i[0].length)):n.push(rt.replace({widget:new ks(o,e)}).range(a,a+i[0].length))}}return rt.set(n)}var js=ct.fromClass(class{decorations;constructor(e){this.decorations=As(e)}update(e){(e.docChanged||e.viewportChanged||e.transactions.some(e=>e.effects.some(e=>e.is(Cs))))&&(this.decorations=As(e.view))}},{decorations:e=>e.decorations});function Ms(e){let t=document.createElement(`div`);t.className=`cm-tooltip-cursor`;let n=document.createElement(`div`);if(n.className=`cm-tooltip-cursor-title`,n.textContent=`${e.name} (${e.uri})`,t.appendChild(n),e.description){let n=document.createElement(`div`);n.className=`cm-tooltip-cursor-description`,n.textContent=e.description,t.appendChild(n)}if(e.mimeType){let n=document.createElement(`div`);n.className=`cm-tooltip-cursor-mimetype`,n.textContent=e.mimeType,t.appendChild(n)}return{dom:t}}function Ns(e,t,n,r=0){for(let i of $(e)){let e=r+i.index,a=e+i[0].length;if(t>=e&&t<=a){let t=i[0].slice(1),r=n.get(t);if(r)return{resource:r,start:e,end:a}}}return null}function Ps(e){return lt((t,n)=>{let{from:r,text:i}=t.state.doc.lineAt(n),a=Ns(i,n,t.state.field(Q),r);return a?{pos:a.start,end:a.end,above:!0,create(){return(e.createTooltip??Ms)(a.resource)}}:null})}function Fs(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,i.to),o=Array.from($(a));if(o.length===0)return null;for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);if(t===a&&n.has(o))return{from:r,to:a,uri:o}}return null}function Is(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t).to,a=r.sliceString(t,i),o=Array.from($(a));if(o.length===0)return null;let s=o[0];if(!s||s.index!==0)return null;let c=t,l=t+s[0].length,u=s[0].slice(1);return n.has(u)?{from:c,to:l,uri:u}:null}function Ls(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,i.to),o=Array.from($(a));for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);if(t>r&&t<a&&n.has(o))return{from:r,to:a,uri:o}}return null}function Rs(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,t),o=Array.from($(a)),s=null;for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);n.has(o)&&a<=t&&(t>a?s=a:t===a&&(s=r))}return s}function zs(e,t){let n=e.field(Q),r=e.doc,i=r.lineAt(t),a=r.sliceString(i.from,i.to),o=Array.from($(a));for(let e of o){let r=i.from+e.index,a=r+e[0].length,o=e[0].slice(1);if(n.has(o)&&r>=t){if(t<r)return r;if(t===r)return a}}return null}var Bs=tt.transactionFilter.of(e=>{let t=e.annotation(at.userEvent);if(!t)return e;let n=e.startState.selection.main;if(!n.empty)return e;if(t===`input.type`||t.startsWith(`input`)){if(e.changes.empty)return e;let t=e.startState.field(Q),n=e.startState.doc,r=!1,i=0,a=[];if(e.changes.iterChanges((e,o,s,c,l)=>{let u=l.toString(),d=e;if(/^\s+$/.test(u)){a.push({from:e,to:o,insert:u});return}let f=n.lineAt(d),p=n.sliceString(f.from,f.to),m=Array.from($(p)),h=!1,g=!1;for(let e of m){let r=f.from+e.index,i=r+e[0].length,a=e[0].slice(1);if(t.has(a)){if(d===r){let e=d>0?n.sliceString(d-1,d):``;e!==``&&e!==` `&&e!==`
 `&&e!==`	`&&(h=!0)}if(d===i){let e=d<n.length?n.sliceString(d,d+1):``;(e===``||e!==` `&&e!==`
 `&&e!==`	`)&&(g=!0)}}}let _=u;h&&(_=` ${_}`,i+=1),g&&(_=` ${_}`,i+=1),_!==u&&(r=!0),a.push({from:e,to:o,insert:_})}),r)return[{changes:a,selection:e.selection?{anchor:e.selection.main.anchor+i,head:e.selection.main.head+i}:void 0,scrollIntoView:e.scrollIntoView}]}if(t===`delete.backward`||t===`delete.forward`){let r=null;return t===`delete.backward`?r=Fs(e.startState,n.head):t===`delete.forward`&&(r=Is(e.startState,n.head)),r?[{changes:{from:r.from,to:r.to,insert:``},selection:{anchor:r.from},scrollIntoView:!0}]:e}if(e.selection?.main.empty&&n.empty&&!t.startsWith(`input`)&&t!==`input.type`){let t=n.head,r=e.selection.main.head,i=null;if(r<t){let n=Ls(e.startState,t);i=n&&t>n.from?n.from:Rs(e.startState,t)}else if(r>t){let n=Ls(e.startState,t);i=n&&t<n.to?n.to:zs(e.startState,t)}if(i!==null&&i!==r)return[{selection:{anchor:i},scrollIntoView:!0}]}return e}),Vs=150;function Hs(e,t){let n=new Set;for(let r of $(e)){let e=r[0].slice(1);t.has(e)||n.add(e)}return[...n]}function Us(e){return[...e].sort().join(`
 `)}function Ws(e){return!$(e).next().done}function Gs(e){let t=[],n=[];return e.changes.iterChanges((r,i,a,o)=>{let s=e.startState.doc.lineAt(r),c=e.startState.doc.lineAt(i===r?r:i-1);t.push(e.startState.doc.sliceString(s.from,c.to));let l=e.state.doc.lineAt(a),u=e.state.doc.lineAt(o===a?a:o-1);n.push(e.state.doc.sliceString(l.from,u.to))}),{before:t.join(`
@@ -63,14 +63,14 @@ ${t}`,d++;break;case`id`:l=t.includes(`\0`)?void 0:t;break;case`retry`:/^\d+$/.t
 `)}}function Ks(e,t){let n=t.debounceMs??Vs,{logger:r,resolve:i}=t;return class{view;timer;running=!1;rerun=!1;lastKey=``;destroyed=!1;constructor(e){this.view=e,this.maybeSchedule(e)}update(e){if(e.docChanged){let t=Gs(e);if(Ws(t.before)){this.maybeSchedule(e.view);return}this.maybeSchedule(e.view,t.after)}}destroy(){this.destroyed=!0,this.clearTimer()}maybeSchedule(e,t){let r=t??e.state.doc.toString();if(!Ws(r)){t===void 0&&(this.lastKey=``,this.clearTimer());return}let i=Us(Hs(r,e.state.field(Q)));if(i===``){t===void 0&&(this.lastKey=``,this.clearTimer());return}if(i===this.lastKey){this.clearTimer();return}this.clearTimer(),this.timer=setTimeout(()=>{this.timer=void 0,this.run(e)},n)}clearTimer(){this.timer!==void 0&&(clearTimeout(this.timer),this.timer=void 0)}async run(t){if(!this.destroyed){if(this.running){this.rerun=!0;return}this.running=!0;try{do{this.rerun=!1;let n=Hs(t.state.doc.toString(),t.state.field(Q));if(n.length===0){this.lastKey=``;break}this.lastKey=Us(n);let r=i?await i(n):await e();if(this.destroyed)return;let a=new Set(n),o=r.filter(e=>a.has(e.uri));if(o.length===0){if(this.rerun)continue;break}t.dispatch({effects:Cs.of(new Map(o.map(e=>[e.uri,e])))}),this.rerun=!0}while(this.rerun)}catch(e){this.lastKey=``,r?.error(`Failed to sync resources:`,e)}finally{this.running=!1}}}}}function qs(e,t={}){return ct.fromClass(Ks(e,t))}var Js=st.baseTheme({".cm-resource-widget":{background:`light-dark(rgba(86,156,214,0.08),rgba(86,156,214,0.18))`,borderRadius:`4px`,padding:`2px 4px`,color:`light-dark(#2367a2,#7ecbff)`,fontWeight:`500`},".cm-resource-widget:hover":{background:`light-dark(rgba(86,156,214,0.18),rgba(86,156,214,0.28))`},".cm-not-found-resource-widget":{background:`light-dark(rgba(151,151,151,0.08),rgba(151,151,151,0.18))`,borderRadius:`4px`,padding:`2px 4px`,color:`light-dark(#b00020,#ff8080)`,fontWeight:`500`}}),Ys=`_none_`,Xs=[{uri:``,name:`No resources`,type:Ys,data:{}}],Zs={info:`Variables, dataframes, and tables will appear here.`,apply:()=>{}};function Qs(e){let t=Vo(e).getAllItems();return t.length===0?Xs:t}function $s(e){let{language:t,store:n,onAddFiles:r}=e,i=()=>Qs(n);return[t.data.of({autocomplete:Es(async()=>i(),e=>e.type===Ys?Zs:Vo(n).getProvider(e.type)?.formatCompletion(e)||{})}),ot.high(et.of([{key:`Tab`,preventDefault:!0,run:e=>ft(e)}])),Io.of({addAttachment:e=>r?.([e])}),...r?[t.data.of({autocomplete:Ho().createCompletionSource()})]:[],js,Bs,qs(i,{logger:k}),Q.init(()=>{let e=Qs(n);return new Map(e.map(e=>[e.uri,e]))}),Js,Ps({createTooltip:e=>{let t=Vo(n).getProvider(e.type);if(!t)return k.warn(`No provider found for resource`,e),ec(e.description||e.name);let r=t.formatCompletion(e),i=e.description||e.name;if(!r?.info)return ec(i);if(typeof r.info==`string`)return ec(r.info);let a=r.info(r);return a?`dom`in a?{dom:a.dom}:`then`in a?(k.warn(`info is a promise. This is not supported`,a),ec(i)):{dom:a}:ec(i)}})]}function ec(e){let t=document.createElement(`div`);return t.innerHTML=e,{dom:t}}var tc={accept:`Keep change`,decline:`Revert change`},nc=e=>{let{handleAcceptCompletion:t,handleDeclineCompletion:n,isLoading:r,hasCompletion:i}=e;return e=>{let a=ke()?e.metaKey:e.ctrlKey;a&&e.key===`Enter`&&!r&&i&&t(),(e.key===`Delete`||e.key===`Backspace`)&&a&&e.shiftKey&&(n(),e.preventDefault(),e.stopPropagation())}},rc=e=>{let t=(0,A.c)(14),{isLoading:n,onAccept:r,onDecline:i,runCell:a,acceptLabel:o,declineLabel:s,size:c}=e,l;t[0]!==o||t[1]!==n||t[2]!==r||t[3]!==a||t[4]!==c?(l=(0,W.jsx)(ic,{isLoading:n,onAccept:r,size:c,runCell:a,label:o}),t[0]=o,t[1]=n,t[2]=r,t[3]=a,t[4]=c,t[5]=l):l=t[5];let u;t[6]!==s||t[7]!==n||t[8]!==i||t[9]!==c?(u=(0,W.jsx)(ac,{onDecline:i,size:c,label:s,disabled:n}),t[6]=s,t[7]=n,t[8]=i,t[9]=c,t[10]=u):u=t[10];let d;return t[11]!==l||t[12]!==u?(d=(0,W.jsxs)(W.Fragment,{children:[l,u]}),t[11]=l,t[12]=u,t[13]=d):d=t[13],d},ic=e=>{let t=(0,A.c)(33),{isLoading:n,onAccept:r,multipleCompletions:i,size:a,buttonStyles:o,acceptShortcut:s,runCell:c,playButtonStyles:l,borderless:u,label:d}=e,f=i!==void 0&&i,p=a===void 0?`sm`:a,m=u!==void 0&&u,h;t[0]!==r||t[1]!==c?(h=()=>{r(),c&&c()},t[0]=r,t[1]=c,t[2]=h):h=t[2];let g=h,_=d??(f?`Accept all`:`Accept`);if(c){let e=`h-6 text-(--grass-11) bg-(--grass-3)/60
     hover:bg-(--grass-3) dark:bg-(--grass-4)/80 dark:hover:bg-(--grass-3) font-semibold
     active:bg-(--grass-5) dark:active:bg-(--grass-4)
-    border-(--green-6) border hover:shadow-xs rounded-r-none ${m&&`border-none rounded-md rounded-r-none`} ${o}`,i;t[3]===s?i=t[4]:(i=s&&(0,W.jsx)(xt,{className:`ml-1 inline`,shortcut:s}),t[3]=s,t[4]=i);let a;t[5]!==n||t[6]!==r||t[7]!==p||t[8]!==e||t[9]!==i||t[10]!==_?(a=(0,W.jsxs)(He,{variant:`text`,size:p,disabled:n,onClick:r,className:e,children:[_,i]}),t[5]=n,t[6]=r,t[7]=p,t[8]=e,t[9]=i,t[10]=_,t[11]=a):a=t[11];let c=`${_} and run ${f?`all cells`:`cell`}`,u=`h-6 text-(--grass-11) bg-(--grass-3)/60
+    border-(--green-6) border hover:shadow-xs rounded-r-none ${m&&`border-none rounded-md rounded-r-none`} ${o}`,i;t[3]===s?i=t[4]:(i=s&&(0,W.jsx)(kt,{className:`ml-1 inline`,shortcut:s}),t[3]=s,t[4]=i);let a;t[5]!==n||t[6]!==r||t[7]!==p||t[8]!==e||t[9]!==i||t[10]!==_?(a=(0,W.jsxs)(He,{variant:`text`,size:p,disabled:n,onClick:r,className:e,children:[_,i]}),t[5]=n,t[6]=r,t[7]=p,t[8]=e,t[9]=i,t[10]=_,t[11]=a):a=t[11];let c=`${_} and run ${f?`all cells`:`cell`}`,u=`h-6 text-(--grass-11) bg-(--grass-3)/60
     hover:bg-(--grass-3) dark:bg-(--grass-4)/80 dark:hover:bg-(--grass-3) font-semibold
     active:bg-(--grass-5) dark:active:bg-(--grass-4)
     border-(--green-6) border hover:shadow-xs rounded-l-none px-1.5 ${m&&`border-0 border-l`} ${l}`,d;t[12]===Symbol.for(`react.memo_cache_sentinel`)?(d=(0,W.jsx)(qe,{className:`h-2.5 w-2.5`}),t[12]=d):d=t[12];let h;t[13]!==g||t[14]!==n||t[15]!==p||t[16]!==u?(h=(0,W.jsx)(He,{variant:`text`,size:p,disabled:n,onClick:g,className:u,children:d}),t[13]=g,t[14]=n,t[15]=p,t[16]=u,t[17]=h):h=t[17];let v;t[18]!==h||t[19]!==c?(v=(0,W.jsx)(Ve,{content:c,children:h}),t[18]=h,t[19]=c,t[20]=v):v=t[20];let y;return t[21]!==v||t[22]!==a?(y=(0,W.jsxs)(`div`,{className:`flex`,children:[a,v]}),t[21]=v,t[22]=a,t[23]=y):y=t[23],y}let v=`h-6 text-(--grass-11) bg-(--grass-3)/60
     hover:bg-(--grass-3) dark:bg-(--grass-4)/80 dark:hover:bg-(--grass-3) font-semibold
     active:bg-(--grass-5) dark:active:bg-(--grass-4)
-    border-(--green-6) border hover:shadow-xs rounded px-3 ${o}`,y;t[24]===s?y=t[25]:(y=s&&(0,W.jsx)(xt,{className:`ml-1 inline`,shortcut:s}),t[24]=s,t[25]=y);let b;return t[26]!==n||t[27]!==r||t[28]!==p||t[29]!==v||t[30]!==y||t[31]!==_?(b=(0,W.jsxs)(He,{variant:`text`,size:p,disabled:n,onClick:r,className:v,children:[_,y]}),t[26]=n,t[27]=r,t[28]=p,t[29]=v,t[30]=y,t[31]=_,t[32]=b):b=t[32],b},ac=e=>{let t=(0,A.c)(9),{onDecline:n,disabled:r,multipleCompletions:i,size:a,className:o,declineShortcut:s,borderless:c,label:l}=e,u=r!==void 0&&r,d=i!==void 0&&i,f=a===void 0?`sm`:a,p=`h-6 text-(--red-10) bg-(--red-3)/60 hover:bg-(--red-3)
+    border-(--green-6) border hover:shadow-xs rounded px-3 ${o}`,y;t[24]===s?y=t[25]:(y=s&&(0,W.jsx)(kt,{className:`ml-1 inline`,shortcut:s}),t[24]=s,t[25]=y);let b;return t[26]!==n||t[27]!==r||t[28]!==p||t[29]!==v||t[30]!==y||t[31]!==_?(b=(0,W.jsxs)(He,{variant:`text`,size:p,disabled:n,onClick:r,className:v,children:[_,y]}),t[26]=n,t[27]=r,t[28]=p,t[29]=v,t[30]=y,t[31]=_,t[32]=b):b=t[32],b},ac=e=>{let t=(0,A.c)(9),{onDecline:n,disabled:r,multipleCompletions:i,size:a,className:o,declineShortcut:s,borderless:c,label:l}=e,u=r!==void 0&&r,d=i!==void 0&&i,f=a===void 0?`sm`:a,p=`h-6 text-(--red-10) bg-(--red-3)/60 hover:bg-(--red-3)
     dark:bg-(--red-4)/80 dark:hover:bg-(--red-3) rounded px-3 font-semibold
     active:bg-(--red-5) dark:active:bg-(--red-4)
-    border-(--red-6) border hover:shadow-xs ${c!==void 0&&c&&`border-none rounded-md`} ${o}`,m=l??`Reject${d?` all`:``}`,h;t[0]===s?h=t[1]:(h=s&&(0,W.jsx)(xt,{className:`ml-1 inline`,shortcut:s}),t[0]=s,t[1]=h);let g;return t[2]!==u||t[3]!==n||t[4]!==f||t[5]!==p||t[6]!==m||t[7]!==h?(g=(0,W.jsxs)(He,{variant:`text`,size:f,disabled:u,onClick:n,className:p,children:[m,h]}),t[2]=u,t[3]=n,t[4]=f,t[5]=p,t[6]=m,t[7]=h,t[8]=g):g=t[8],g},oc=class{activeAttempt=null;cancel(){this.activeAttempt&&(this.activeAttempt.cancelled=!0)}async run({prepare:e,submit:t,onError:n}){if(this.activeAttempt!==null)return;let r={cancelled:!1};this.activeAttempt=r;try{let n=await e();if(r.cancelled)return;await t(n)}catch(e){r.cancelled||n(e)}finally{this.activeAttempt=null}}},sc=Me(`marimo:ai-language`,`python`,g),cc=`marimo:ai-prompt-history`,lc=new f(Se(D()),()=>[]),uc=({onClose:e})=>{let t=i(),[n,a]=(0,U.useState)(``),{acceptOwnedStagedCells:s,beginStagedCellGeneration:l,discardOwnedStagedCells:u,finishStagedCellGeneration:d,hasOwnedStagedCells:f,onData:p}=co(t),[m,h]=r(sc),g=Ne(),_=(0,U.useRef)(null),v=(0,U.useRef)(new oc).current,y=(0,U.useRef)(null),{files:b,addFiles:x,removeFile:S}=fs(),C=o(Te),ee=(0,U.useMemo)(()=>new ga({api:g.getAiURL(`completion`).toString(),headers:()=>g.headers(),prepareSendMessagesRequest:async e=>{let t={...await ss(e.messages),id:e.id,code:``,prompt:``,language:m};return{api:g.getAiURL(`completion`).toString(),body:t}}}),[m,g]),te=c(e=>{d(!1),Ze({title:`Generate with AI failed`,description:yt(e)})}),{sendMessage:ne,stop:w,status:T}=Ia({throttle:100,transport:ee,onData:p,onError:te,onFinish:({isAbort:e,isDisconnect:t,isError:n,finishReason:r})=>{d(!e&&!t&&!n&&r===`stop`)}}),E=T===`streaming`||T===`submitted`,ie=f(),ae=c(()=>{v.cancel(),d(!1),w()}),oe=c(()=>{v.cancel(),u(),w()});(0,U.useEffect)(()=>()=>{ae()},[ae]);let se=C?.models?.edit_model||`openai/gpt-4o`,ce=Ee.parse(se).providerId,le=es.has(ce),ue=async()=>{E||await v.run({prepare:async()=>(_.current?.view&&Ga(_.current.view),b?await is(b):void 0),submit:async e=>{l(),await ne({text:n,files:e})},onError:te})},de=(0,W.jsxs)(W.Fragment,{children:[(0,W.jsx)(Mt,{className:`size-4 mr-2`}),`Python`]}),fe=(0,W.jsxs)(W.Fragment,{children:[(0,W.jsx)(We,{className:`size-4 mr-2`}),`SQL`]}),pe=(0,W.jsxs)(Be,{modal:!1,children:[(0,W.jsxs)(Re,{className:`flex items-center justify-between h-7 text-xs px-2 py-0.5 border rounded-md hover:text-accent-foreground`,"data-testid":`language-button`,children:[m===`python`?de:fe,(0,W.jsx)(jt,{className:`ml-1 h-3.5 w-3.5 text-muted-foreground/70`})]}),(0,W.jsxs)(ze,{align:`center`,children:[(0,W.jsx)(`div`,{className:`px-2 py-1 text-sm text-muted-foreground`,children:`Select language`}),(0,W.jsx)(Le,{}),(0,W.jsx)(Ie,{onClick:()=>h(`python`),children:de}),(0,W.jsx)(Ie,{onClick:()=>h(`sql`),children:fe})]})]}),D=()=>{s()&&e()},me=()=>{oe(),_.current?.view?.focus()},he=()=>{oe(),e()},ge=(0,W.jsxs)(`div`,{className:`flex items-center px-3`,children:[(0,W.jsx)(re,{className:`size-4 text-(--blue-11) mr-2`}),(0,W.jsx)(dc,{inputRef:_,onClose:he,value:n,onChange:e=>{a(e)},onSubmit:ue,onKeyDown:nc({handleAcceptCompletion:D,handleDeclineCompletion:me,isLoading:E,hasCompletion:ie})}),(0,W.jsx)(He,{variant:`text`,size:`sm`,className:`mb-0 px-1`,onClick:he,children:(0,W.jsx)(Ke,{className:`size-4`})})]}),_e=(0,W.jsxs)(`div`,{className:`px-3 pt-1 flex flex-row items-center justify-between`,children:[(0,W.jsxs)(`div`,{className:`flex items-center gap-2`,children:[(0,W.jsx)(Oe,{triggerClassName:`h-7 text-xs max-w-64`,iconSize:`small`,forRole:`edit`,showAddCustomModelDocs:!0}),pe,(0,W.jsx)(De,{})]}),(0,W.jsxs)(`div`,{className:`flex flex-row items-center`,children:[b.length>0&&(0,W.jsx)(`div`,{className:`flex flex-row gap-1 flex-wrap pr-1`,children:b?.map((e,t)=>(0,W.jsx)(_s,{file:e,onRemove:()=>S(e)},`${e.name}-${t}`))}),(0,W.jsx)(ys,{handleAddContext:()=>$o(_),isLoading:E}),le&&(0,W.jsx)(bs,{fileInputRef:y,isLoading:E,onAddFiles:x}),(0,W.jsx)(vs,{isLoading:E,onStop:ae,onSendClick:ue,isEmpty:!n.trim(),showStopLabel:!0})]})]});return(0,W.jsxs)(`div`,{className:Pe(`flex flex-col w-full py-2`),children:[ge,_e]})},dc=e=>{let t=(0,A.c)(35),{value:n,placeholder:r,inputRef:a,className:o,onChange:s,onSubmit:l,onKeyDown:u,onClose:d,onAddFiles:f,additionalCompletions:p,maxHeight:m}=e,h=l,g=d,_=i(),{theme:v}=vt(),y;t[0]===p?y=t[1]:(y=e=>p?Qo([p.triggerCompletionRegex],p.completions)(e):null,t[0]=p,t[1]=y);let b=c(y),x,S,C,ee;if(t[2]!==b||t[3]!==f||t[4]!==_){let e=gt();t[9]===Symbol.for(`react.memo_cache_sentinel`)?(x=mt({}),t[9]=x):x=t[9],S=e,C=$s({language:e.language,store:_,onAddFiles:f}),ee=e.language.data.of({autocomplete:b}),t[2]=b,t[3]=f,t[4]=_,t[5]=x,t[6]=S,t[7]=C,t[8]=ee}else x=t[5],S=t[6],C=t[7],ee=t[8];let te;t[10]===Symbol.for(`react.memo_cache_sentinel`)?(te=Ka({storage:{load:fc,save:pc}}),t[10]=te):te=t[10];let ne;t[11]===Symbol.for(`react.memo_cache_sentinel`)?(ne=Lt(),t[11]=ne):ne=t[11];let w;t[12]===h?w=t[13]:(w=ot.highest(et.of([{preventDefault:!0,stopPropagation:!0,any:(e,t)=>{let n=t.metaKey||t.ctrlKey||t.shiftKey;if(t.key===`Enter`&&!n)return h(t,e.state.doc.toString()),t.preventDefault(),t.stopPropagation(),!0;if(t.key===`Enter`&&t.shiftKey){let n=e.state.selection.main.from;return e.dispatch({changes:{from:n,to:n,insert:`
-`},selection:{anchor:n+1,head:n+1}}),t.preventDefault(),t.stopPropagation(),!0}return!1}}])),t[12]=h,t[13]=w);let T;t[14]===g?T=t[15]:(T=et.of([{key:`Escape`,preventDefault:!0,stopPropagation:!0,run:()=>(g(),!0)}]),t[14]=g,t[15]=T);let E;t[16]!==x||t[17]!==S||t[18]!==C||t[19]!==ee||t[20]!==w||t[21]!==T?(E=[x,S,C,ee,te,st.lineWrapping,ne,w,T],t[16]=x,t[17]=S,t[18]=C,t[19]=ee,t[20]=w,t[21]=T,t[22]=E):E=t[22];let re=E,ie;t[23]===o?ie=t[24]:(ie=Pe(`flex-1 font-sans overflow-auto my-1`,o),t[23]=o,t[24]=ie);let ae=v===`dark`?`dark`:`light`,oe=r||`Generate with AI, @ to include context`,se;return t[25]!==re||t[26]!==a||t[27]!==m||t[28]!==s||t[29]!==u||t[30]!==ie||t[31]!==ae||t[32]!==oe||t[33]!==n?(se=(0,W.jsx)(Et,{ref:a,className:ie,width:`100%`,maxHeight:m,value:n,basicSetup:!1,extensions:re,onChange:s,onKeyDown:u,theme:ae,placeholder:oe}),t[25]=re,t[26]=a,t[27]=m,t[28]=s,t[29]=u,t[30]=ie,t[31]=ae,t[32]=oe,t[33]=n,t[34]=se):se=t[34],se};function fc(){return lc.get(cc)}function pc(e){return lc.set(cc,e)}export{Z as A,ga as B,hs as C,Yo as D,Ko as E,Xa as F,Zr as G,Xi as H,Qa as I,Kt as J,Jt as K,to as L,oo as M,ro as N,Vo as O,so as P,Ga as R,fs as S,Uo as T,fa as U,oa as V,Qi as W,Gt as Y,cs as _,rc as a,as as b,ys as c,_s as d,vs as f,rs as g,is as h,ic as i,no as j,ko as k,bs as l,es as m,dc as n,ac as o,Ss as p,qt as q,tc as r,nc as s,uc as t,gs as u,ls as v,$o as w,ms as x,us as y,Ia as z};
+    border-(--red-6) border hover:shadow-xs ${c!==void 0&&c&&`border-none rounded-md`} ${o}`,m=l??`Reject${d?` all`:``}`,h;t[0]===s?h=t[1]:(h=s&&(0,W.jsx)(kt,{className:`ml-1 inline`,shortcut:s}),t[0]=s,t[1]=h);let g;return t[2]!==u||t[3]!==n||t[4]!==f||t[5]!==p||t[6]!==m||t[7]!==h?(g=(0,W.jsxs)(He,{variant:`text`,size:f,disabled:u,onClick:n,className:p,children:[m,h]}),t[2]=u,t[3]=n,t[4]=f,t[5]=p,t[6]=m,t[7]=h,t[8]=g):g=t[8],g},oc=class{activeAttempt=null;cancel(){this.activeAttempt&&(this.activeAttempt.cancelled=!0)}async run({prepare:e,submit:t,onError:n}){if(this.activeAttempt!==null)return;let r={cancelled:!1};this.activeAttempt=r;try{let n=await e();if(r.cancelled)return;await t(n)}catch(e){r.cancelled||n(e)}finally{this.activeAttempt=null}}},sc=Me(`marimo:ai-language`,`python`,g),cc=`marimo:ai-prompt-history`,lc=new f(Se(D()),()=>[]),uc=({onClose:e})=>{let t=i(),[n,a]=(0,U.useState)(``),{acceptOwnedStagedCells:s,beginStagedCellGeneration:l,discardOwnedStagedCells:u,finishStagedCellGeneration:d,hasOwnedStagedCells:f,onData:p}=co(t),[m,h]=r(sc),g=Ne(),_=(0,U.useRef)(null),v=(0,U.useRef)(new oc).current,y=(0,U.useRef)(null),{files:b,addFiles:x,removeFile:S}=fs(),C=o(Te),ee=(0,U.useMemo)(()=>new ga({api:g.getAiURL(`completion`).toString(),headers:()=>g.headers(),prepareSendMessagesRequest:async e=>{let t={...await ss(e.messages),id:e.id,code:``,prompt:``,language:m};return{api:g.getAiURL(`completion`).toString(),body:t}}}),[m,g]),te=c(e=>{d(!1),Ze({title:`Generate with AI failed`,description:yt(e)})}),{sendMessage:ne,stop:w,status:T}=Ia({throttle:100,transport:ee,onData:p,onError:te,onFinish:({isAbort:e,isDisconnect:t,isError:n,finishReason:r})=>{d(!e&&!t&&!n&&r===`stop`)}}),E=T===`streaming`||T===`submitted`,ie=f(),ae=c(()=>{v.cancel(),d(!1),w()}),oe=c(()=>{v.cancel(),u(),w()});(0,U.useEffect)(()=>()=>{ae()},[ae]);let se=C?.models?.edit_model||`openai/gpt-4o`,ce=Ee.parse(se).providerId,le=es.has(ce),ue=async()=>{E||await v.run({prepare:async()=>(_.current?.view&&Ga(_.current.view),b?await is(b):void 0),submit:async e=>{l(),await ne({text:n,files:e})},onError:te})},de=(0,W.jsxs)(W.Fragment,{children:[(0,W.jsx)(Et,{className:`size-4 mr-2`}),`Python`]}),fe=(0,W.jsxs)(W.Fragment,{children:[(0,W.jsx)(We,{className:`size-4 mr-2`}),`SQL`]}),pe=(0,W.jsxs)(Be,{modal:!1,children:[(0,W.jsxs)(Re,{className:`flex items-center justify-between h-7 text-xs px-2 py-0.5 border rounded-md hover:text-accent-foreground`,"data-testid":`language-button`,children:[m===`python`?de:fe,(0,W.jsx)(Tt,{className:`ml-1 h-3.5 w-3.5 text-muted-foreground/70`})]}),(0,W.jsxs)(ze,{align:`center`,children:[(0,W.jsx)(`div`,{className:`px-2 py-1 text-sm text-muted-foreground`,children:`Select language`}),(0,W.jsx)(Le,{}),(0,W.jsx)(Ie,{onClick:()=>h(`python`),children:de}),(0,W.jsx)(Ie,{onClick:()=>h(`sql`),children:fe})]})]}),D=()=>{s()&&e()},me=()=>{oe(),_.current?.view?.focus()},he=()=>{oe(),e()},ge=(0,W.jsxs)(`div`,{className:`flex items-center px-3`,children:[(0,W.jsx)(re,{className:`size-4 text-(--blue-11) mr-2`}),(0,W.jsx)(dc,{inputRef:_,onClose:he,value:n,onChange:e=>{a(e)},onSubmit:ue,onKeyDown:nc({handleAcceptCompletion:D,handleDeclineCompletion:me,isLoading:E,hasCompletion:ie})}),(0,W.jsx)(He,{variant:`text`,size:`sm`,className:`mb-0 px-1`,onClick:he,children:(0,W.jsx)(Ke,{className:`size-4`})})]}),_e=(0,W.jsxs)(`div`,{className:`px-3 pt-1 flex flex-row items-center justify-between`,children:[(0,W.jsxs)(`div`,{className:`flex items-center gap-2`,children:[(0,W.jsx)(Oe,{triggerClassName:`h-7 text-xs max-w-64`,iconSize:`small`,forRole:`edit`,showAddCustomModelDocs:!0}),pe,(0,W.jsx)(De,{})]}),(0,W.jsxs)(`div`,{className:`flex flex-row items-center`,children:[b.length>0&&(0,W.jsx)(`div`,{className:`flex flex-row gap-1 flex-wrap pr-1`,children:b?.map((e,t)=>(0,W.jsx)(_s,{file:e,onRemove:()=>S(e)},`${e.name}-${t}`))}),(0,W.jsx)(ys,{handleAddContext:()=>$o(_),isLoading:E}),le&&(0,W.jsx)(bs,{fileInputRef:y,isLoading:E,onAddFiles:x}),(0,W.jsx)(vs,{isLoading:E,onStop:ae,onSendClick:ue,isEmpty:!n.trim(),showStopLabel:!0})]})]});return(0,W.jsxs)(`div`,{className:Pe(`flex flex-col w-full py-2`),children:[ge,_e]})},dc=e=>{let t=(0,A.c)(35),{value:n,placeholder:r,inputRef:a,className:o,onChange:s,onSubmit:l,onKeyDown:u,onClose:d,onAddFiles:f,additionalCompletions:p,maxHeight:m}=e,h=l,g=d,_=i(),{theme:v}=vt(),y;t[0]===p?y=t[1]:(y=e=>p?Qo([p.triggerCompletionRegex],p.completions)(e):null,t[0]=p,t[1]=y);let b=c(y),x,S,C,ee;if(t[2]!==b||t[3]!==f||t[4]!==_){let e=gt();t[9]===Symbol.for(`react.memo_cache_sentinel`)?(x=mt({}),t[9]=x):x=t[9],S=e,C=$s({language:e.language,store:_,onAddFiles:f}),ee=e.language.data.of({autocomplete:b}),t[2]=b,t[3]=f,t[4]=_,t[5]=x,t[6]=S,t[7]=C,t[8]=ee}else x=t[5],S=t[6],C=t[7],ee=t[8];let te;t[10]===Symbol.for(`react.memo_cache_sentinel`)?(te=Ka({storage:{load:fc,save:pc}}),t[10]=te):te=t[10];let ne;t[11]===Symbol.for(`react.memo_cache_sentinel`)?(ne=bt(),t[11]=ne):ne=t[11];let w;t[12]===h?w=t[13]:(w=ot.highest(et.of([{preventDefault:!0,stopPropagation:!0,any:(e,t)=>{let n=t.metaKey||t.ctrlKey||t.shiftKey;if(t.key===`Enter`&&!n)return h(t,e.state.doc.toString()),t.preventDefault(),t.stopPropagation(),!0;if(t.key===`Enter`&&t.shiftKey){let n=e.state.selection.main.from;return e.dispatch({changes:{from:n,to:n,insert:`
+`},selection:{anchor:n+1,head:n+1}}),t.preventDefault(),t.stopPropagation(),!0}return!1}}])),t[12]=h,t[13]=w);let T;t[14]===g?T=t[15]:(T=et.of([{key:`Escape`,preventDefault:!0,stopPropagation:!0,run:()=>(g(),!0)}]),t[14]=g,t[15]=T);let E;t[16]!==x||t[17]!==S||t[18]!==C||t[19]!==ee||t[20]!==w||t[21]!==T?(E=[x,S,C,ee,te,st.lineWrapping,ne,w,T],t[16]=x,t[17]=S,t[18]=C,t[19]=ee,t[20]=w,t[21]=T,t[22]=E):E=t[22];let re=E,ie;t[23]===o?ie=t[24]:(ie=Pe(`flex-1 font-sans overflow-auto my-1`,o),t[23]=o,t[24]=ie);let ae=v===`dark`?`dark`:`light`,oe=r||`Generate with AI, @ to include context`,se;return t[25]!==re||t[26]!==a||t[27]!==m||t[28]!==s||t[29]!==u||t[30]!==ie||t[31]!==ae||t[32]!==oe||t[33]!==n?(se=(0,W.jsx)(Rt,{ref:a,className:ie,width:`100%`,maxHeight:m,value:n,basicSetup:!1,extensions:re,onChange:s,onKeyDown:u,theme:ae,placeholder:oe}),t[25]=re,t[26]=a,t[27]=m,t[28]=s,t[29]=u,t[30]=ie,t[31]=ae,t[32]=oe,t[33]=n,t[34]=se):se=t[34],se};function fc(){return lc.get(cc)}function pc(e){return lc.set(cc,e)}export{Z as A,ga as B,hs as C,Yo as D,Ko as E,Xa as F,Zr as G,Xi as H,Qa as I,Kt as J,Jt as K,to as L,oo as M,ro as N,Vo as O,so as P,Ga as R,fs as S,Uo as T,fa as U,oa as V,Qi as W,Gt as Y,cs as _,rc as a,as as b,ys as c,_s as d,vs as f,rs as g,is as h,ic as i,no as j,ko as k,bs as l,es as m,dc as n,ac as o,Ss as p,qt as q,tc as r,nc as s,uc as t,gs as u,ls as v,$o as w,ms as x,us as y,Ia as z};