sage-agent-internal 0.2.5__tar.gz → 0.2.6__tar.gz

{sage_agent_internal-0.2.5 → sage_agent_internal-0.2.6}/.gitignore

@@ -127,3 +127,4 @@ dmypy.json
 *.ipynb
 
 .env
+/ui-tests/test-results.json

{sage_agent_internal-0.2.5 → sage_agent_internal-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sage_agent_internal
-Version: 0.2.5
+Version: 0.2.6
 Summary: Sage Agent - Your Jupyter Notebook Assistant
 Project-URL: Homepage, https://github.com/AlpineX-Labs/jupyter-sage-agent-internal
 Project-URL: Bug Tracker, https://github.com/AlpineX-Labs/jupyter-sage-agent-internal/issues

{sage_agent_internal-0.2.5 → sage_agent_internal-0.2.6}/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sage-agent-internal",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Sage Agent - Your Jupyter Notebook Assistant",
   "keywords": [
     "jupyter",

{sage_agent_internal-0.2.5 → sage_agent_internal-0.2.6}/sage_agent_internal/_version.py

@@ -1,4 +1,4 @@
 # This file is auto-generated by Hatchling. As such, do not:
 #   - modify
 #   - track in version control e.g. be sure to add to .gitignore
-__version__ = VERSION = '0.2.5'
+__version__ = VERSION = '0.2.6'

{sage_agent_internal-0.2.5 → sage_agent_internal-0.2.6}/sage_agent_internal/labextension/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sage-agent-internal",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Sage Agent - Your Jupyter Notebook Assistant",
   "keywords": [
     "jupyter",
@@ -127,7 +127,7 @@
     "outputDir": "sage_agent_internal/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.c97bd67ec0a2caecdfcb.js",
+      "load": "static/remoteEntry.4a1093a43fdce6a4f6e0.js",
       "extension": "./extension",
       "style": "./style"
     }

{sage_agent_internal-0.2.5 → sage_agent_internal-0.2.6}/sage_agent_internal/labextension/schemas/sage-agent-internal/package.json.orig

@@ -1,6 +1,6 @@
 {
   "name": "sage-agent-internal",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Sage Agent - Your Jupyter Notebook Assistant",
   "keywords": [
     "jupyter",

sage_agent_internal-0.2.5/sage_agent_internal/labextension/static/remoteEntry.c97bd67ec0a2caecdfcb.js → sage_agent_internal-0.2.6/sage_agent_internal/labextension/static/remoteEntry.4a1093a43fdce6a4f6e0.js

@@ -1 +1 @@
-var _JUPYTERLAB;(()=>{"use strict";var e,r,t,a,n,o,i,f,d,l,u,c,s,p,h,b,m,g,v,y,j,w,P,k,S,O={484:(e,r,t)=>{var a={"./index":()=>Promise.all([t.e(353),t.e(747),t.e(122),t.e(973),t.e(760),t.e(345),t.e(379),t.e(839),t.e(968)]).then(()=>()=>t(9040)),"./extension":()=>Promise.all([t.e(353),t.e(747),t.e(122),t.e(973),t.e(760),t.e(345),t.e(379),t.e(839),t.e(968)]).then(()=>()=>t(9040)),"./style":()=>t.e(363).then(()=>()=>t(8363))},n=(e,r)=>(t.R=r,r=t.o(a,e)?a[e]():Promise.resolve().then(()=>{throw new Error('Module "'+e+'" does not exist in container.')}),t.R=void 0,r),o=(e,r)=>{if(t.S){var a="default",n=t.S[a];if(n&&n!==e)throw new Error("Container initialization failed as it has already been initialized with a different share scope");return t.S[a]=e,t.I(a,r)}};t.d(r,{get:()=>n,init:()=>o})}},x={};function _(e){var r=x[e];if(void 0!==r)return r.exports;var t=x[e]={id:e,exports:{}};return O[e].call(t.exports,t,t.exports,_),t.exports}_.m=O,_.c=x,_.n=e=>{var r=e&&e.__esModule?()=>e.default:()=>e;return _.d(r,{a:r}),r},r=Object.getPrototypeOf?e=>Object.getPrototypeOf(e):e=>e.__proto__,_.t=function(t,a){if(1&a&&(t=this(t)),8&a)return t;if("object"==typeof t&&t){if(4&a&&t.__esModule)return t;if(16&a&&"function"==typeof t.then)return t}var n=Object.create(null);_.r(n);var o={};e=e||[null,r({}),r([]),r(r)];for(var i=2&a&&t;("object"==typeof i||"function"==typeof i)&&!~e.indexOf(i);i=r(i))Object.getOwnPropertyNames(i).forEach(e=>o[e]=()=>t[e]);return o.default=()=>t,_.d(n,o),n},_.d=(e,r)=>{for(var t in r)_.o(r,t)&&!_.o(e,t)&&Object.defineProperty(e,t,{enumerable:!0,get:r[t]})},_.f={},_.e=e=>Promise.all(Object.keys(_.f).reduce((r,t)=>(_.f[t](e,r),r),[])),_.u=e=>e+"."+{122:"3c2c200586e65203b674",220:"d384edfd5bbcfda33a2a",262:"3fc4ad88e6f8c273d4ad",353:"2c285601a208c3735929",363:"6fc8d18eedbf137940f2",364:"c2fb82c6f484097228c6",439:"3cb9ac7022c8a6074647",481:"96b91d018df31d54d173",553:"b7906877d98a62265533",635:"8c395c854aa2b140b668",713:"103cb5ae248be385b08a",742:"b204789055ded4e07883",747:"b8e9485ebcbb3066eb4b",821:"b58f29814f5900f1f6f8",845:"cc2bc80fbfea6d20a396",888:"09382cc93a2ad3932b31",968:"980219bfdaa6b20f56cb",973:"016a586759b281b79f91"}[e]+".js?v="+{122:"3c2c200586e65203b674",220:"d384edfd5bbcfda33a2a",262:"3fc4ad88e6f8c273d4ad",353:"2c285601a208c3735929",363:"6fc8d18eedbf137940f2",364:"c2fb82c6f484097228c6",439:"3cb9ac7022c8a6074647",481:"96b91d018df31d54d173",553:"b7906877d98a62265533",635:"8c395c854aa2b140b668",713:"103cb5ae248be385b08a",742:"b204789055ded4e07883",747:"b8e9485ebcbb3066eb4b",821:"b58f29814f5900f1f6f8",845:"cc2bc80fbfea6d20a396",888:"09382cc93a2ad3932b31",968:"980219bfdaa6b20f56cb",973:"016a586759b281b79f91"}[e],_.g=function(){if("object"==typeof globalThis)return globalThis;try{return this||new Function("return this")()}catch(e){if("object"==typeof window)return window}}(),_.o=(e,r)=>Object.prototype.hasOwnProperty.call(e,r),t={},a="sage-agent-internal:",_.l=(e,r,n,o)=>{if(t[e])t[e].push(r);else{var i,f;if(void 0!==n)for(var d=document.getElementsByTagName("script"),l=0;l<d.length;l++){var u=d[l];if(u.getAttribute("src")==e||u.getAttribute("data-webpack")==a+n){i=u;break}}i||(f=!0,(i=document.createElement("script")).charset="utf-8",i.timeout=120,_.nc&&i.setAttribute("nonce",_.nc),i.setAttribute("data-webpack",a+n),i.src=e),t[e]=[r];var c=(r,a)=>{i.onerror=i.onload=null,clearTimeout(s);var n=t[e];if(delete t[e],i.parentNode&&i.parentNode.removeChild(i),n&&n.forEach(e=>e(a)),r)return r(a)},s=setTimeout(c.bind(null,void 0,{type:"timeout",target:i}),12e4);i.onerror=c.bind(null,i.onerror),i.onload=c.bind(null,i.onload),f&&document.head.appendChild(i)}},_.r=e=>{"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},(()=>{_.S={};var e={},r={};_.I=(t,a)=>{a||(a=[]);var n=r[t];if(n||(n=r[t]={}),!(a.indexOf(n)>=0)){if(a.push(n),e[t])return e[t];_.o(_.S,t)||(_.S[t]={});var o=_.S[t],i="sage-agent-internal",f=(e,r,t,a)=>{var n=o[e]=o[e]||{},f=n[r];(!f||!f.loaded&&(!a!=!f.eager?a:i>f.from))&&(n[r]={get:t,from:i,eager:!!a})},d=[];return"default"===t&&(f("@anthropic-ai/sdk","0.54.0",()=>_.e(742).then(()=>()=>_(2742))),f("best-effort-json-parser","1.2.1",()=>_.e(439).then(()=>()=>_(7820))),f("codemirror-merge-alpinex","6.10.4",()=>Promise.all([_.e(353),_.e(635),_.e(760),_.e(839)]).then(()=>()=>_(7635))),f("codemirror","6.0.2",()=>Promise.all([_.e(353),_.e(821),_.e(747),_.e(760)]).then(()=>()=>_(8821))),f("diff2html","3.4.52",()=>Promise.all([_.e(122),_.e(379)]).then(()=>()=>_(2122))),f("diff","7.0.0",()=>_.e(713).then(()=>()=>_(5713))),f("diff","0",()=>_.e(481).then(()=>()=>_(7481))),f("marked","15.0.12",()=>_.e(364).then(()=>()=>_(4364))),f("partial-json","0.1.7",()=>_.e(220).then(()=>()=>_(9220))),f("react-bootstrap","2.10.10",()=>Promise.all([_.e(553),_.e(345),_.e(628)]).then(()=>()=>_(4553))),f("rxjs","7.8.2",()=>_.e(888).then(()=>()=>_(2888))),f("sage-agent-internal","0.2.5",()=>Promise.all([_.e(353),_.e(747),_.e(122),_.e(973),_.e(760),_.e(345),_.e(379),_.e(839),_.e(968)]).then(()=>()=>_(9040))),f("uuid","11.1.0",()=>_.e(262).then(()=>()=>_(8262)))),e[t]=d.length?Promise.all(d).then(()=>e[t]=1):1}}})(),(()=>{var e;_.g.importScripts&&(e=_.g.location+"");var r=_.g.document;if(!e&&r&&(r.currentScript&&"SCRIPT"===r.currentScript.tagName.toUpperCase()&&(e=r.currentScript.src),!e)){var t=r.getElementsByTagName("script");if(t.length)for(var a=t.length-1;a>-1&&(!e||!/^http(s?):/.test(e));)e=t[a--].src}if(!e)throw new Error("Automatic publicPath is not supported in this browser");e=e.replace(/^blob:/,"").replace(/#.*$/,"").replace(/\?.*$/,"").replace(/\/[^\/]+$/,"/"),_.p=e})(),n=e=>{var r=e=>e.split(".").map(e=>+e==e?+e:e),t=/^([^-+]+)?(?:-([^+]+))?(?:\+(.+))?$/.exec(e),a=t[1]?r(t[1]):[];return t[2]&&(a.length++,a.push.apply(a,r(t[2]))),t[3]&&(a.push([]),a.push.apply(a,r(t[3]))),a},o=(e,r)=>{e=n(e),r=n(r);for(var t=0;;){if(t>=e.length)return t<r.length&&"u"!=(typeof r[t])[0];var a=e[t],o=(typeof a)[0];if(t>=r.length)return"u"==o;var i=r[t],f=(typeof i)[0];if(o!=f)return"o"==o&&"n"==f||"s"==f||"u"==o;if("o"!=o&&"u"!=o&&a!=i)return a<i;t++}},i=e=>{var r=e[0],t="";if(1===e.length)return"*";if(r+.5){t+=0==r?">=":-1==r?"<":1==r?"^":2==r?"~":r>0?"=":"!=";for(var a=1,n=1;n<e.length;n++)a--,t+="u"==(typeof(f=e[n]))[0]?"-":(a>0?".":"")+(a=2,f);return t}var o=[];for(n=1;n<e.length;n++){var f=e[n];o.push(0===f?"not("+d()+")":1===f?"("+d()+" || "+d()+")":2===f?o.pop()+" "+o.pop():i(f))}return d();function d(){return o.pop().replace(/^\((.+)\)$/,"$1")}},f=(e,r)=>{if(0 in e){r=n(r);var t=e[0],a=t<0;a&&(t=-t-1);for(var o=0,i=1,d=!0;;i++,o++){var l,u,c=i<e.length?(typeof e[i])[0]:"";if(o>=r.length||"o"==(u=(typeof(l=r[o]))[0]))return!d||("u"==c?i>t&&!a:""==c!=a);if("u"==u){if(!d||"u"!=c)return!1}else if(d)if(c==u)if(i<=t){if(l!=e[i])return!1}else{if(a?l>e[i]:l<e[i])return!1;l!=e[i]&&(d=!1)}else if("s"!=c&&"n"!=c){if(a||i<=t)return!1;d=!1,i--}else{if(i<=t||u<c!=a)return!1;d=!1}else"s"!=c&&"n"!=c&&(d=!1,i--)}}var s=[],p=s.pop.bind(s);for(o=1;o<e.length;o++){var h=e[o];s.push(1==h?p()|p():2==h?p()&p():h?f(h,r):!p())}return!!p()},d=(e,r)=>e&&_.o(e,r),l=e=>(e.loaded=1,e.get()),u=e=>Object.keys(e).reduce((r,t)=>(e[t].eager&&(r[t]=e[t]),r),{}),c=(e,r,t,a)=>{var n=a?u(e[r]):e[r];return(r=Object.keys(n).reduce((e,r)=>!f(t,r)||e&&!o(e,r)?e:r,0))&&n[r]},s=(e,r,t)=>{var a=t?u(e[r]):e[r];return Object.keys(a).reduce((e,r)=>!e||!a[e].loaded&&o(e,r)?r:e,0)},p=(e,r,t,a)=>"Unsatisfied version "+t+" from "+(t&&e[r][t].from)+" of shared singleton module "+r+" (required "+i(a)+")",h=(e,r,t,a,n)=>{var o=e[t];return"No satisfying version ("+i(a)+")"+(n?" for eager consumption":"")+" of shared module "+t+" found in shared scope "+r+".\nAvailable versions: "+Object.keys(o).map(e=>e+" from "+o[e].from).join(", ")},b=e=>{throw new Error(e)},m=e=>{"undefined"!=typeof console&&console.warn&&console.warn(e)},v=(e,r,t)=>t?t():((e,r)=>b("Shared module "+r+" doesn't exist in shared scope "+e))(e,r),y=(g=e=>function(r,t,a,n,o){var i=_.I(r);return i&&i.then&&!a?i.then(e.bind(e,r,_.S[r],t,!1,n,o)):e(r,_.S[r],t,a,n,o)})((e,r,t,a,n,o)=>{if(!d(r,t))return v(e,t,o);var i=c(r,t,n,a);return i?l(i):o?o():void b(h(r,e,t,n,a))}),j=g((e,r,t,a,n,o)=>{if(!d(r,t))return v(e,t,o);var i=s(r,t,a);return f(n,i)||m(p(r,t,i,n)),l(r[t][i])}),w={},P={195:()=>j("default","@codemirror/state",!1,[1,6,2,0]),2084:()=>j("default","@codemirror/language",!1,[1,6,0,0]),3708:()=>j("default","@lezer/common",!1,[1,1,0,0]),5024:()=>j("default","@codemirror/view",!1,[1,6,9,6]),3345:()=>j("default","react",!1,[1,18,2,0]),3379:()=>y("default","diff",!1,[1,7,0,0],()=>_.e(713).then(()=>()=>_(5713))),1839:()=>j("default","@lezer/highlight",!1,[1,1,0,0]),113:()=>y("default","diff",!1,[1,8,0,1],()=>_.e(481).then(()=>()=>_(7481))),135:()=>y("default","@anthropic-ai/sdk",!1,[2,0,54,0],()=>_.e(742).then(()=>()=>_(2742))),389:()=>j("default","@jupyterlab/statedb",!1,[1,4,4,6]),1336:()=>y("default","diff2html",!1,[1,3,4,51],()=>()=>_(2122)),2311:()=>y("default","react-bootstrap",!1,[1,2,10,10],()=>Promise.all([_.e(553),_.e(628)]).then(()=>()=>_(4553))),3139:()=>y("default","codemirror-merge-alpinex",!1,[1,6,10,4],()=>_.e(635).then(()=>()=>_(7635))),3427:()=>j("default","@jupyterlab/docmanager",!1,[1,4,4,6]),3550:()=>y("default","marked",!1,[1,15,0,11],()=>_.e(364).then(()=>()=>_(4364))),4602:()=>j("default","@lumino/signaling",!1,[1,2,0,0]),4744:()=>y("default","uuid",!1,[1,11,1,0],()=>_.e(262).then(()=>()=>_(8262))),5256:()=>j("default","@lumino/widgets",!1,[1,2,3,1,,"alpha",0]),5280:()=>j("default","@jupyterlab/extensionmanager",!1,[1,4,4,6]),6750:()=>y("default","rxjs",!1,[1,7,8,2],()=>_.e(888).then(()=>()=>_(2888))),7332:()=>y("default","codemirror",!1,[1,6,0,2],()=>_.e(821).then(()=>()=>_(8821))),7570:()=>y("default","partial-json",!1,[2,0,1,7],()=>_.e(220).then(()=>()=>_(9220))),8118:()=>j("default","@jupyterlab/codemirror",!1,[1,4,4,6]),8276:()=>y("default","best-effort-json-parser",!1,[1,1,1,3],()=>_.e(439).then(()=>()=>_(7820))),8361:()=>j("default","@jupyterlab/settingregistry",!1,[1,4,4,6]),8531:()=>j("default","@jupyterlab/notebook",!1,[1,4,4,6]),9159:()=>j("default","@jupyterlab/ui-components",!1,[1,4,4,6]),9715:()=>j("default","@jupyterlab/apputils",!1,[1,4,5,6]),7628:()=>j("default","react-dom",!1,[1,18,2,0])},k={345:[3345],379:[3379],628:[7628],760:[195,2084,3708,5024],839:[1839],968:[113,135,389,1336,2311,3139,3427,3550,4602,4744,5256,5280,6750,7332,7570,8118,8276,8361,8531,9159,9715]},S={},_.f.consumes=(e,r)=>{_.o(k,e)&&k[e].forEach(e=>{if(_.o(w,e))return r.push(w[e]);if(!S[e]){var t=r=>{w[e]=0,_.m[e]=t=>{delete _.c[e],t.exports=r()}};S[e]=!0;var a=r=>{delete w[e],_.m[e]=t=>{throw delete _.c[e],r}};try{var n=P[e]();n.then?r.push(w[e]=n.then(t).catch(a)):t(n)}catch(e){a(e)}}})},(()=>{_.b=document.baseURI||self.location.href;var e={431:0};_.f.j=(r,t)=>{var a=_.o(e,r)?e[r]:void 0;if(0!==a)if(a)t.push(a[2]);else if(/^(345|379|628|760|839)$/.test(r))e[r]=0;else{var n=new Promise((t,n)=>a=e[r]=[t,n]);t.push(a[2]=n);var o=_.p+_.u(r),i=new Error;_.l(o,t=>{if(_.o(e,r)&&(0!==(a=e[r])&&(e[r]=void 0),a)){var n=t&&("load"===t.type?"missing":t.type),o=t&&t.target&&t.target.src;i.message="Loading chunk "+r+" failed.\n("+n+": "+o+")",i.name="ChunkLoadError",i.type=n,i.request=o,a[1](i)}},"chunk-"+r,r)}};var r=(r,t)=>{var a,n,[o,i,f]=t,d=0;if(o.some(r=>0!==e[r])){for(a in i)_.o(i,a)&&(_.m[a]=i[a]);f&&f(_)}for(r&&r(t);d<o.length;d++)n=o[d],_.o(e,n)&&e[n]&&e[n][0](),e[n]=0},t=self.webpackChunksage_agent_internal=self.webpackChunksage_agent_internal||[];t.forEach(r.bind(null,0)),t.push=r.bind(null,t.push.bind(t))})(),_.nc=void 0;var E=_(484);(_JUPYTERLAB=void 0===_JUPYTERLAB?{}:_JUPYTERLAB)["sage-agent-internal"]=E})();
+var _JUPYTERLAB;(()=>{"use strict";var e,r,t,a,n,o,i,f,d,l,u,c,s,p,h,b,m,g,v,y,j,w,P,k,S,O={484:(e,r,t)=>{var a={"./index":()=>Promise.all([t.e(353),t.e(747),t.e(122),t.e(973),t.e(760),t.e(345),t.e(379),t.e(839),t.e(968)]).then(()=>()=>t(9040)),"./extension":()=>Promise.all([t.e(353),t.e(747),t.e(122),t.e(973),t.e(760),t.e(345),t.e(379),t.e(839),t.e(968)]).then(()=>()=>t(9040)),"./style":()=>t.e(363).then(()=>()=>t(8363))},n=(e,r)=>(t.R=r,r=t.o(a,e)?a[e]():Promise.resolve().then(()=>{throw new Error('Module "'+e+'" does not exist in container.')}),t.R=void 0,r),o=(e,r)=>{if(t.S){var a="default",n=t.S[a];if(n&&n!==e)throw new Error("Container initialization failed as it has already been initialized with a different share scope");return t.S[a]=e,t.I(a,r)}};t.d(r,{get:()=>n,init:()=>o})}},x={};function _(e){var r=x[e];if(void 0!==r)return r.exports;var t=x[e]={id:e,exports:{}};return O[e].call(t.exports,t,t.exports,_),t.exports}_.m=O,_.c=x,_.n=e=>{var r=e&&e.__esModule?()=>e.default:()=>e;return _.d(r,{a:r}),r},r=Object.getPrototypeOf?e=>Object.getPrototypeOf(e):e=>e.__proto__,_.t=function(t,a){if(1&a&&(t=this(t)),8&a)return t;if("object"==typeof t&&t){if(4&a&&t.__esModule)return t;if(16&a&&"function"==typeof t.then)return t}var n=Object.create(null);_.r(n);var o={};e=e||[null,r({}),r([]),r(r)];for(var i=2&a&&t;("object"==typeof i||"function"==typeof i)&&!~e.indexOf(i);i=r(i))Object.getOwnPropertyNames(i).forEach(e=>o[e]=()=>t[e]);return o.default=()=>t,_.d(n,o),n},_.d=(e,r)=>{for(var t in r)_.o(r,t)&&!_.o(e,t)&&Object.defineProperty(e,t,{enumerable:!0,get:r[t]})},_.f={},_.e=e=>Promise.all(Object.keys(_.f).reduce((r,t)=>(_.f[t](e,r),r),[])),_.u=e=>e+"."+{122:"3c2c200586e65203b674",220:"d384edfd5bbcfda33a2a",262:"3fc4ad88e6f8c273d4ad",353:"2c285601a208c3735929",363:"6fc8d18eedbf137940f2",364:"c2fb82c6f484097228c6",439:"3cb9ac7022c8a6074647",481:"96b91d018df31d54d173",553:"b7906877d98a62265533",635:"8c395c854aa2b140b668",713:"103cb5ae248be385b08a",742:"b204789055ded4e07883",747:"b8e9485ebcbb3066eb4b",821:"b58f29814f5900f1f6f8",845:"cc2bc80fbfea6d20a396",888:"09382cc93a2ad3932b31",968:"980219bfdaa6b20f56cb",973:"016a586759b281b79f91"}[e]+".js?v="+{122:"3c2c200586e65203b674",220:"d384edfd5bbcfda33a2a",262:"3fc4ad88e6f8c273d4ad",353:"2c285601a208c3735929",363:"6fc8d18eedbf137940f2",364:"c2fb82c6f484097228c6",439:"3cb9ac7022c8a6074647",481:"96b91d018df31d54d173",553:"b7906877d98a62265533",635:"8c395c854aa2b140b668",713:"103cb5ae248be385b08a",742:"b204789055ded4e07883",747:"b8e9485ebcbb3066eb4b",821:"b58f29814f5900f1f6f8",845:"cc2bc80fbfea6d20a396",888:"09382cc93a2ad3932b31",968:"980219bfdaa6b20f56cb",973:"016a586759b281b79f91"}[e],_.g=function(){if("object"==typeof globalThis)return globalThis;try{return this||new Function("return this")()}catch(e){if("object"==typeof window)return window}}(),_.o=(e,r)=>Object.prototype.hasOwnProperty.call(e,r),t={},a="sage-agent-internal:",_.l=(e,r,n,o)=>{if(t[e])t[e].push(r);else{var i,f;if(void 0!==n)for(var d=document.getElementsByTagName("script"),l=0;l<d.length;l++){var u=d[l];if(u.getAttribute("src")==e||u.getAttribute("data-webpack")==a+n){i=u;break}}i||(f=!0,(i=document.createElement("script")).charset="utf-8",i.timeout=120,_.nc&&i.setAttribute("nonce",_.nc),i.setAttribute("data-webpack",a+n),i.src=e),t[e]=[r];var c=(r,a)=>{i.onerror=i.onload=null,clearTimeout(s);var n=t[e];if(delete t[e],i.parentNode&&i.parentNode.removeChild(i),n&&n.forEach(e=>e(a)),r)return r(a)},s=setTimeout(c.bind(null,void 0,{type:"timeout",target:i}),12e4);i.onerror=c.bind(null,i.onerror),i.onload=c.bind(null,i.onload),f&&document.head.appendChild(i)}},_.r=e=>{"undefined"!=typeof Symbol&&Symbol.toStringTag&&Object.defineProperty(e,Symbol.toStringTag,{value:"Module"}),Object.defineProperty(e,"__esModule",{value:!0})},(()=>{_.S={};var e={},r={};_.I=(t,a)=>{a||(a=[]);var n=r[t];if(n||(n=r[t]={}),!(a.indexOf(n)>=0)){if(a.push(n),e[t])return e[t];_.o(_.S,t)||(_.S[t]={});var o=_.S[t],i="sage-agent-internal",f=(e,r,t,a)=>{var n=o[e]=o[e]||{},f=n[r];(!f||!f.loaded&&(!a!=!f.eager?a:i>f.from))&&(n[r]={get:t,from:i,eager:!!a})},d=[];return"default"===t&&(f("@anthropic-ai/sdk","0.54.0",()=>_.e(742).then(()=>()=>_(2742))),f("best-effort-json-parser","1.2.1",()=>_.e(439).then(()=>()=>_(7820))),f("codemirror-merge-alpinex","6.10.4",()=>Promise.all([_.e(353),_.e(635),_.e(760),_.e(839)]).then(()=>()=>_(7635))),f("codemirror","6.0.2",()=>Promise.all([_.e(353),_.e(821),_.e(747),_.e(760)]).then(()=>()=>_(8821))),f("diff2html","3.4.52",()=>Promise.all([_.e(122),_.e(379)]).then(()=>()=>_(2122))),f("diff","7.0.0",()=>_.e(713).then(()=>()=>_(5713))),f("diff","0",()=>_.e(481).then(()=>()=>_(7481))),f("marked","15.0.12",()=>_.e(364).then(()=>()=>_(4364))),f("partial-json","0.1.7",()=>_.e(220).then(()=>()=>_(9220))),f("react-bootstrap","2.10.10",()=>Promise.all([_.e(553),_.e(345),_.e(628)]).then(()=>()=>_(4553))),f("rxjs","7.8.2",()=>_.e(888).then(()=>()=>_(2888))),f("sage-agent-internal","0.2.6",()=>Promise.all([_.e(353),_.e(747),_.e(122),_.e(973),_.e(760),_.e(345),_.e(379),_.e(839),_.e(968)]).then(()=>()=>_(9040))),f("uuid","11.1.0",()=>_.e(262).then(()=>()=>_(8262)))),e[t]=d.length?Promise.all(d).then(()=>e[t]=1):1}}})(),(()=>{var e;_.g.importScripts&&(e=_.g.location+"");var r=_.g.document;if(!e&&r&&(r.currentScript&&"SCRIPT"===r.currentScript.tagName.toUpperCase()&&(e=r.currentScript.src),!e)){var t=r.getElementsByTagName("script");if(t.length)for(var a=t.length-1;a>-1&&(!e||!/^http(s?):/.test(e));)e=t[a--].src}if(!e)throw new Error("Automatic publicPath is not supported in this browser");e=e.replace(/^blob:/,"").replace(/#.*$/,"").replace(/\?.*$/,"").replace(/\/[^\/]+$/,"/"),_.p=e})(),n=e=>{var r=e=>e.split(".").map(e=>+e==e?+e:e),t=/^([^-+]+)?(?:-([^+]+))?(?:\+(.+))?$/.exec(e),a=t[1]?r(t[1]):[];return t[2]&&(a.length++,a.push.apply(a,r(t[2]))),t[3]&&(a.push([]),a.push.apply(a,r(t[3]))),a},o=(e,r)=>{e=n(e),r=n(r);for(var t=0;;){if(t>=e.length)return t<r.length&&"u"!=(typeof r[t])[0];var a=e[t],o=(typeof a)[0];if(t>=r.length)return"u"==o;var i=r[t],f=(typeof i)[0];if(o!=f)return"o"==o&&"n"==f||"s"==f||"u"==o;if("o"!=o&&"u"!=o&&a!=i)return a<i;t++}},i=e=>{var r=e[0],t="";if(1===e.length)return"*";if(r+.5){t+=0==r?">=":-1==r?"<":1==r?"^":2==r?"~":r>0?"=":"!=";for(var a=1,n=1;n<e.length;n++)a--,t+="u"==(typeof(f=e[n]))[0]?"-":(a>0?".":"")+(a=2,f);return t}var o=[];for(n=1;n<e.length;n++){var f=e[n];o.push(0===f?"not("+d()+")":1===f?"("+d()+" || "+d()+")":2===f?o.pop()+" "+o.pop():i(f))}return d();function d(){return o.pop().replace(/^\((.+)\)$/,"$1")}},f=(e,r)=>{if(0 in e){r=n(r);var t=e[0],a=t<0;a&&(t=-t-1);for(var o=0,i=1,d=!0;;i++,o++){var l,u,c=i<e.length?(typeof e[i])[0]:"";if(o>=r.length||"o"==(u=(typeof(l=r[o]))[0]))return!d||("u"==c?i>t&&!a:""==c!=a);if("u"==u){if(!d||"u"!=c)return!1}else if(d)if(c==u)if(i<=t){if(l!=e[i])return!1}else{if(a?l>e[i]:l<e[i])return!1;l!=e[i]&&(d=!1)}else if("s"!=c&&"n"!=c){if(a||i<=t)return!1;d=!1,i--}else{if(i<=t||u<c!=a)return!1;d=!1}else"s"!=c&&"n"!=c&&(d=!1,i--)}}var s=[],p=s.pop.bind(s);for(o=1;o<e.length;o++){var h=e[o];s.push(1==h?p()|p():2==h?p()&p():h?f(h,r):!p())}return!!p()},d=(e,r)=>e&&_.o(e,r),l=e=>(e.loaded=1,e.get()),u=e=>Object.keys(e).reduce((r,t)=>(e[t].eager&&(r[t]=e[t]),r),{}),c=(e,r,t,a)=>{var n=a?u(e[r]):e[r];return(r=Object.keys(n).reduce((e,r)=>!f(t,r)||e&&!o(e,r)?e:r,0))&&n[r]},s=(e,r,t)=>{var a=t?u(e[r]):e[r];return Object.keys(a).reduce((e,r)=>!e||!a[e].loaded&&o(e,r)?r:e,0)},p=(e,r,t,a)=>"Unsatisfied version "+t+" from "+(t&&e[r][t].from)+" of shared singleton module "+r+" (required "+i(a)+")",h=(e,r,t,a,n)=>{var o=e[t];return"No satisfying version ("+i(a)+")"+(n?" for eager consumption":"")+" of shared module "+t+" found in shared scope "+r+".\nAvailable versions: "+Object.keys(o).map(e=>e+" from "+o[e].from).join(", ")},b=e=>{throw new Error(e)},m=e=>{"undefined"!=typeof console&&console.warn&&console.warn(e)},v=(e,r,t)=>t?t():((e,r)=>b("Shared module "+r+" doesn't exist in shared scope "+e))(e,r),y=(g=e=>function(r,t,a,n,o){var i=_.I(r);return i&&i.then&&!a?i.then(e.bind(e,r,_.S[r],t,!1,n,o)):e(r,_.S[r],t,a,n,o)})((e,r,t,a,n,o)=>{if(!d(r,t))return v(e,t,o);var i=c(r,t,n,a);return i?l(i):o?o():void b(h(r,e,t,n,a))}),j=g((e,r,t,a,n,o)=>{if(!d(r,t))return v(e,t,o);var i=s(r,t,a);return f(n,i)||m(p(r,t,i,n)),l(r[t][i])}),w={},P={195:()=>j("default","@codemirror/state",!1,[1,6,2,0]),2084:()=>j("default","@codemirror/language",!1,[1,6,0,0]),3708:()=>j("default","@lezer/common",!1,[1,1,0,0]),5024:()=>j("default","@codemirror/view",!1,[1,6,9,6]),3345:()=>j("default","react",!1,[1,18,2,0]),3379:()=>y("default","diff",!1,[1,7,0,0],()=>_.e(713).then(()=>()=>_(5713))),1839:()=>j("default","@lezer/highlight",!1,[1,1,0,0]),113:()=>y("default","diff",!1,[1,8,0,1],()=>_.e(481).then(()=>()=>_(7481))),135:()=>y("default","@anthropic-ai/sdk",!1,[2,0,54,0],()=>_.e(742).then(()=>()=>_(2742))),389:()=>j("default","@jupyterlab/statedb",!1,[1,4,4,6]),1336:()=>y("default","diff2html",!1,[1,3,4,51],()=>()=>_(2122)),2311:()=>y("default","react-bootstrap",!1,[1,2,10,10],()=>Promise.all([_.e(553),_.e(628)]).then(()=>()=>_(4553))),3139:()=>y("default","codemirror-merge-alpinex",!1,[1,6,10,4],()=>_.e(635).then(()=>()=>_(7635))),3427:()=>j("default","@jupyterlab/docmanager",!1,[1,4,4,6]),3550:()=>y("default","marked",!1,[1,15,0,11],()=>_.e(364).then(()=>()=>_(4364))),4602:()=>j("default","@lumino/signaling",!1,[1,2,0,0]),4744:()=>y("default","uuid",!1,[1,11,1,0],()=>_.e(262).then(()=>()=>_(8262))),5256:()=>j("default","@lumino/widgets",!1,[1,2,3,1,,"alpha",0]),5280:()=>j("default","@jupyterlab/extensionmanager",!1,[1,4,4,6]),6750:()=>y("default","rxjs",!1,[1,7,8,2],()=>_.e(888).then(()=>()=>_(2888))),7332:()=>y("default","codemirror",!1,[1,6,0,2],()=>_.e(821).then(()=>()=>_(8821))),7570:()=>y("default","partial-json",!1,[2,0,1,7],()=>_.e(220).then(()=>()=>_(9220))),8118:()=>j("default","@jupyterlab/codemirror",!1,[1,4,4,6]),8276:()=>y("default","best-effort-json-parser",!1,[1,1,1,3],()=>_.e(439).then(()=>()=>_(7820))),8361:()=>j("default","@jupyterlab/settingregistry",!1,[1,4,4,6]),8531:()=>j("default","@jupyterlab/notebook",!1,[1,4,4,6]),9159:()=>j("default","@jupyterlab/ui-components",!1,[1,4,4,6]),9715:()=>j("default","@jupyterlab/apputils",!1,[1,4,5,6]),7628:()=>j("default","react-dom",!1,[1,18,2,0])},k={345:[3345],379:[3379],628:[7628],760:[195,2084,3708,5024],839:[1839],968:[113,135,389,1336,2311,3139,3427,3550,4602,4744,5256,5280,6750,7332,7570,8118,8276,8361,8531,9159,9715]},S={},_.f.consumes=(e,r)=>{_.o(k,e)&&k[e].forEach(e=>{if(_.o(w,e))return r.push(w[e]);if(!S[e]){var t=r=>{w[e]=0,_.m[e]=t=>{delete _.c[e],t.exports=r()}};S[e]=!0;var a=r=>{delete w[e],_.m[e]=t=>{throw delete _.c[e],r}};try{var n=P[e]();n.then?r.push(w[e]=n.then(t).catch(a)):t(n)}catch(e){a(e)}}})},(()=>{_.b=document.baseURI||self.location.href;var e={431:0};_.f.j=(r,t)=>{var a=_.o(e,r)?e[r]:void 0;if(0!==a)if(a)t.push(a[2]);else if(/^(345|379|628|760|839)$/.test(r))e[r]=0;else{var n=new Promise((t,n)=>a=e[r]=[t,n]);t.push(a[2]=n);var o=_.p+_.u(r),i=new Error;_.l(o,t=>{if(_.o(e,r)&&(0!==(a=e[r])&&(e[r]=void 0),a)){var n=t&&("load"===t.type?"missing":t.type),o=t&&t.target&&t.target.src;i.message="Loading chunk "+r+" failed.\n("+n+": "+o+")",i.name="ChunkLoadError",i.type=n,i.request=o,a[1](i)}},"chunk-"+r,r)}};var r=(r,t)=>{var a,n,[o,i,f]=t,d=0;if(o.some(r=>0!==e[r])){for(a in i)_.o(i,a)&&(_.m[a]=i[a]);f&&f(_)}for(r&&r(t);d<o.length;d++)n=o[d],_.o(e,n)&&e[n]&&e[n][0](),e[n]=0},t=self.webpackChunksage_agent_internal=self.webpackChunksage_agent_internal||[];t.forEach(r.bind(null,0)),t.push=r.bind(null,t.push.bind(t))})(),_.nc=void 0;var E=_(484);(_JUPYTERLAB=void 0===_JUPYTERLAB?{}:_JUPYTERLAB)["sage-agent-internal"]=E})();

sage_agent_internal-0.2.6/ui-tests/README.md

@@ -0,0 +1,22 @@
+
+# Running Sage_AI Tests
+
+## Step 1: Configure `config.ts`
+Copy the example config file:
+
+```bash
+cp tests/config-example.ts tests/config.ts
+```
+## Step 2: Add Configs
+
+Update `tests/config.ts` with your API key and other required configs.
+
+## Step 3: Install & Run Tests
+
+Navigate to the UI tests folder, install dependencies, and run the tests:
+
+```bash
+cd jupyter-sage-agent-internal/ui-tests
+jlpm install
+jlpm run test
+```

sage_agent_internal-0.2.6/ui-tests/package.json

@@ -0,0 +1,22 @@
+{
+  "name": "sage-agent-internal-ui-tests",
+  "version": "1.0.0",
+  "description": "JupyterLab sage-agent-internal Integration Tests",
+  "private": true,
+  "scripts": {
+    "start": "jupyter lab --config jupyter_server_test_config.py",
+    "clean:screenshots": "node -e \"const fs = require('fs'); const path = require('path'); const screenshotsDir = path.join(__dirname, 'screenshots'); if (fs.existsSync(screenshotsDir)) { fs.rmSync(screenshotsDir, { recursive: true, force: true }); console.log('Screenshots directory cleaned'); } else { console.log('Screenshots directory does not exist'); }\"",
+    "test": "npm run clean:screenshots && jlpm playwright test sage_diff_interactions",
+    "test:update": "npm run clean:screenshots && jlpm playwright test --update-snapshots",
+    "test:llm": "npm run clean:screenshots && jlpm playwright test sage_llm_test",
+    "test:diffs": "npm run clean:screenshots && jlpm playwright test sage_diff_interactions",
+    "test:headed": "npm run clean:screenshots && jlpm playwright test --headed",
+    "test:debug": "npm run clean:screenshots && jlpm playwright test --debug",
+    "setup": "echo '✅ Setup complete! Please edit tests/config.ts with your API key.'"
+  },
+  "devDependencies": {
+    "@jupyterlab/galata": "^5.0.5",
+    "@playwright/test": "^1.37.0",
+    "@types/node": "^20.0.0"
+  }
+}

sage_agent_internal-0.2.6/ui-tests/playwright.config.js

@@ -0,0 +1,27 @@
+/**
+ * Configuration for Playwright using default from @jupyterlab/galata
+ */
+const baseConfig = require('@jupyterlab/galata/lib/playwright-config');
+
+module.exports = {
+  ...baseConfig,
+  // Add 1 retry for each test
+  retries: 1,
+  use: {
+    ...baseConfig.use,
+    headless: true,
+    viewport: { width: 1920, height: 1080 }
+  },
+  // Add multiple reporters for better CI integration
+  reporter: [
+    ['html'], // Keep the existing HTML reporter
+    ['json', { outputFile: 'test-results.json' }], // Add JSON reporter for easier parsing
+    ['list'] // Add list reporter for console output
+  ],
+  webServer: {
+    command: 'jlpm start',
+    url: 'http://localhost:8888/lab',
+    timeout: 120 * 1000,
+    reuseExistingServer: !process.env.CI
+  }
+};

sage_agent_internal-0.2.6/ui-tests/setup-tests.bat

@@ -0,0 +1,38 @@
+@echo off
+REM Sage LLM Test Setup Script for Windows
+REM This script helps set up the testing environment
+
+echo 🚀 Setting up Sage LLM Test Environment
+
+REM Configuration is now handled via tests/config.ts
+echo 📝 Configuration is managed in tests/config.ts
+echo ⚠️  IMPORTANT: Please edit tests/config.ts and set your API key before running tests!
+echo    Update SAGE_API_KEY in tests/config.ts with your actual API key
+
+REM Check if node_modules exists
+if not exist "node_modules" (
+    echo 📦 Installing dependencies...
+    npm install
+) else (
+    echo ✅ Dependencies already installed
+)
+
+REM Create screenshots directory
+if not exist "screenshots" (
+    echo 📁 Creating screenshots directory...
+    mkdir screenshots\states\idle
+    mkdir screenshots\states\diff_approval
+    mkdir screenshots\states\interaction
+    mkdir screenshots\states\generation
+    mkdir screenshots\test_runs
+) else (
+    echo ✅ Screenshots directory already exists
+)
+
+echo.
+echo 🎯 Setup complete! To run the tests:
+echo    1. Make sure your API key is set in tests/config.ts
+echo    2. Start JupyterLab: npm run start
+echo    3. In another terminal, run tests: npm test
+echo.
+echo 📸 Screenshots will be saved in the screenshots\ directory

sage_agent_internal-0.2.6/ui-tests/setup-tests.sh

@@ -0,0 +1,36 @@
+#!/bin/bash
+
+# Sage LLM Test Setup Script
+# This script helps set up the testing environment
+
+echo "🚀 Setting up Sage LLM Test Environment"
+
+# Configuration is now handled via tests/config.ts
+echo "📝 Configuration is managed in tests/config.ts"
+echo "⚠️  IMPORTANT: Please edit tests/config.ts and set your API key before running tests!"
+echo "   Update SAGE_API_KEY in tests/config.ts with your actual API key"
+
+# Check if node_modules exists
+if [ ! -d "node_modules" ]; then
+    echo "📦 Installing dependencies..."
+    npm install
+else
+    echo "✅ Dependencies already installed"
+fi
+
+# Create screenshots directory
+if [ ! -d "screenshots" ]; then
+    echo "📁 Creating screenshots directory..."
+    mkdir -p screenshots/states/{idle,diff_approval,interaction,generation}
+    mkdir -p screenshots/test_runs
+else
+    echo "✅ Screenshots directory already exists"
+fi
+
+echo ""
+echo "🎯 Setup complete! To run the tests:"
+echo "   1. Make sure your API key is set in tests/config.ts"
+echo "   2. Start JupyterLab: npm run start"
+echo "   3. In another terminal, run tests: npm test"
+echo ""
+echo "📸 Screenshots will be saved in the screenshots/ directory"

sage_agent_internal-0.2.6/ui-tests/tests/README_SAGE_LLM_TESTS.md

@@ -0,0 +1,133 @@
+# Sage LLM State Testing
+
+This directory contains comprehensive tests for all LLM interaction states in the Sage Agent extension.
+
+## Quick Setup
+
+### Windows:
+```bash
+.\setup-tests.bat
+```
+
+### Linux/Mac:
+```bash
+chmod +x setup-tests.sh
+./setup-tests.sh
+```
+
+### Manual Setup:
+1. Edit `tests/config.ts` and set your `SAGE_API_KEY`
+2. Install dependencies: `npm install`
+3. Create screenshots directory: `mkdir -p screenshots`
+
+## Running Tests
+
+1. **Start JupyterLab** (in one terminal):
+   ```bash
+   npm run start
+   ```
+
+2. **Run Tests** (in another terminal):
+   ```bash
+   npm test                    # Run all tests
+   npm test sage_llm_test      # Run only LLM state tests
+   ```
+
+## Configuration
+
+**IMPORTANT:** Before running tests, you must set your API key in one of these ways:
+
+## Configuration
+
+Edit `tests/config.ts` and set your API key:
+```typescript
+const CONFIG: Config = {
+  SAGE_API_KEY: 'your-actual-api-key-here',  // Replace with your actual API key
+  CLAUDE_MODEL_URL: 'https://sage.alpinex.ai:8760',
+  CLAUDE_MODEL_ID: 'claude-sonnet-4-20250514',
+  // ... other config
+};
+```
+
+## Test Categories
+
+### 🏠 Idle States
+- Empty interface (no interactions)
+- Finished chat without diffs
+- Page reload with chat history
+- New chat creation
+
+### 🔄 Diff Approval States
+- **Single File Diffs**: Approval → Accept/Reject → Execute
+- **Multi-File Diffs**: Approval → Accept/Reject → Execute
+
+### ⚡ Interaction States
+- Context picker activation
+- Cell execution requests
+- Tool calls and function execution
+- Waiting for user replies
+
+### 🤖 Generation States
+- Active LLM generation
+- Tool execution with loading
+- Long-running generation tracking
+
+## Screenshots
+
+All screenshots are automatically captured and organized in:
+```
+screenshots/
+├── test_runs/
+│   └── [timestamp]/         # Screenshots from each test run
+└── states/                  # Organized by state category
+    ├── idle/
+    ├── diff_approval/
+    ├── interaction/
+    └── generation/
+```
+
+## Test Prompts
+
+The tests use specific prompts designed to trigger different states:
+
+- **Simple Query**: `"What is the current time?"` (no diffs)
+- **Single Diff**: `"Please fix this Python syntax error: print('hello world'"` 
+- **Multi-File Diff**: `"Create a new React component with TypeScript interfaces and CSS styling"`
+- **Tool Call**: `"Check the current git status and list recent commits"`
+- **Context Analysis**: `"Analyze the current codebase structure and suggest improvements"`
+- **Long Generation**: `"Write comprehensive documentation for this entire project..."`
+
+## Troubleshooting
+
+### Common Issues:
+
+1. **API Key Error**: Make sure `SAGE_API_KEY` is set correctly
+2. **JupyterLab Not Starting**: Check if port 8888 is available
+3. **Tests Timing Out**: Increase timeout values in config
+4. **Screenshots Not Saving**: Check directory permissions
+
+### Debug Mode:
+Set `headless: false` in playwright config to see browser interactions.
+
+## Test Development
+
+To add new states or modify existing tests:
+
+1. **Add new prompts** in the `TEST_PROMPTS` object
+2. **Create new test cases** following the existing pattern
+3. **Update selectors** if UI elements change
+4. **Add new screenshot categories** as needed
+
+## Performance Notes
+
+- Tests run with visual browser by default (for debugging)
+- Screenshots are full-page captures (may be large)
+- Long generation tests may take several minutes
+- Each test run creates a timestamped screenshot directory
+
+## CI/CD Integration
+
+For automated testing:
+- Set `headless: true` in playwright config
+- Use environment variables for sensitive data
+- Archive screenshot artifacts for failed tests

sage_agent_internal-0.2.6/ui-tests/tests/SAGE_LLM_TESTING_PLAN.md

@@ -0,0 +1,238 @@
+# Sage LLM State Testing Plan
+
+## Overview
+This document outlines the comprehensive testing strategy for all LLM interaction states in the Sage Agent. The tests will capture screenshots at each state and validate UI components to ensure proper functionality across all interaction scenarios.
+
+## Test States Overview
+
+### 1. Idle States
+- **None**: Empty chat interface, no interactions
+- **No diff but finished chat**: Chat completed without code diffs
+- **Reloaded page chat**: Page refreshed with existing chat history
+- **Create new chat**: Fresh chat session initiated
+
+### 2. Diff Approval States
+
+#### Single Diff States:
+- **Approval**: Single file diff awaiting user approval
+- **Accepted**: User accepted single diff
+- **Rejected**: User rejected single diff  
+- **Run**: Single diff being executed
+
+#### Multi Diff States:
+- **Approval**: Multiple file diffs awaiting user approval
+- **Accepted**: User accepted multi-file diffs
+- **Rejected**: User rejected multi-file diffs
+- **Run**: Multi-file diffs being executed
+
+### 3. Interaction States
+- **Context picker open**: File/context selection interface displayed
+- **Run cell request**: Cell execution in progress
+- **Generating States**: LLM generating response
+- **Running tool call**: Tool/function execution
+- **Waiting for user reply**: LLM awaiting user input
+
+### 4. Generating States
+- **Generating**: LLM actively generating response
+- **Running tool call**: Tool execution with loading indicators
+- **Waiting for user reply**: Paused state awaiting user interaction
+
+## Test Implementation Strategy
+
+### Setup Requirements
+1. **Environment Configuration**
+   - Configuration via `tests/config.ts` file with API keys and model settings
+   - Automated API key injection via settings UI
+   - Playwright browser automation
+
+2. **Screenshot Management**
+   - Organized folder structure: `screenshots/states/[category]/[state]/`
+   - Full-page screenshots for comprehensive state capture
+   - Timestamped filenames for test run identification
+
+3. **State Triggering**
+   - Specific prompts designed to trigger each state
+   - Automated UI interactions to navigate through states
+   - Validation of state transitions
+
+### Test Structure
+
+#### Core Test File: `sage_llm_test.spec.ts`
+- Main orchestrator for all state tests
+- Handles environment setup and teardown
+- Manages screenshot capture and organization
+
+#### Test Categories:
+1. **Idle State Tests** (`idle_states.spec.ts`)
+2. **Diff Approval Tests** (`diff_approval.spec.ts`)  
+3. **Interaction State Tests** (`interaction_states.spec.ts`)
+4. **Generation State Tests** (`generation_states.spec.ts`)
+
+### Prompts for State Triggering
+
+#### Single-line Diffs
+```
+"Please fix this Python syntax error: print('hello world'"
+```
+
+#### Multiple Edits (Single File)
+```
+"Refactor this function to be more readable and add error handling"
+```
+
+#### Multi-file Diffs
+```
+"Create a new React component with TypeScript interfaces and CSS styling"
+```
+
+#### Context Picker
+```
+"Analyze the current codebase structure and suggest improvements"
+```
+
+#### Tool Calls
+```
+"Check the current git status and create a new branch"
+```
+
+#### Long Generation
+```
+"Write a comprehensive documentation for this entire project including setup, usage, and API reference"
+```
+
+### Screenshot Organization
+```
+screenshots/
+├── states/
+│   ├── idle/
+│   │   ├── none/
+│   │   ├── finished_no_diff/
+│   │   ├── reloaded_page/
+│   │   └── new_chat/
+│   ├── diff_approval/
+│   │   ├── single/
+│   │   │   ├── approval/
+│   │   │   ├── accepted/
+│   │   │   ├── rejected/
+│   │   │   └── run/
+│   │   └── multi/
+│   │       ├── approval/
+│   │       ├── accepted/
+│   │       ├── rejected/
+│   │       └── run/
+│   ├── interaction/
+│   │   ├── context_picker/
+│   │   ├── run_cell/
+│   │   └── generating/
+│   └── generation/
+│       ├── generating/
+│       ├── tool_call/
+│       └── waiting_reply/
+└── test_runs/
+    └── [timestamp]/
+        └── [copied from states/]
+```
+
+### Validation Strategy
+
+#### Component Visibility Tests
+- Chat container presence and visibility
+- Input field accessibility
+- Send button state validation
+- Loading indicators during generation
+- Diff approval UI elements
+- Error states and messages
+
+#### State Transition Validation
+- Proper state changes after user actions
+- Loading states during async operations
+- Error recovery mechanisms
+- UI responsiveness during long operations
+
+#### Screenshot Comparison
+- Visual regression testing capabilities
+- Baseline screenshot establishment
+- Automated difference detection
+- Manual review flags for significant changes
+
+### Environment Setup
+
+#### Required Configuration
+```typescript
+// tests/config.ts file
+const CONFIG = {
+  SAGE_API_KEY: 'your_api_key_here',
+  CLAUDE_MODEL_URL: 'https://sage.alpinex.ai:8760',
+  CLAUDE_MODEL_ID: 'claude-sonnet-4-20250514',
+  DATABASE_URL: 'your_database_url_here'
+};
+```
+
+#### API Key Injection Strategy
+1. Load configuration from tests/config.ts at test startup
+2. Navigate to Settings UI programmatically
+3. Inject API key via form automation
+4. Validate successful configuration
+5. Proceed with state testing
+
+### Execution Flow
+
+#### Pre-Test Setup
+1. Launch JupyterLab with Sage extension
+2. Wait for full application load
+3. Configure API settings automatically
+4. Create clean chat environment
+5. Initialize screenshot directories
+
+#### Test Execution
+1. Execute state-specific prompts
+2. Wait for state transitions
+3. Capture full-page screenshots
+4. Validate component states
+5. Record test metrics and timing
+
+#### Post-Test Cleanup
+1. Archive screenshots with timestamps
+2. Generate test report with state coverage
+3. Clean up temporary files
+4. Reset application state for next run
+
+### Success Criteria
+
+#### Functional Validation
+- All states successfully triggered
+- UI components behave as expected
+- No crashes or errors during state transitions
+- Proper error handling for failed operations
+
+#### Visual Validation
+- Screenshots capture complete UI state
+- All relevant UI elements visible
+- Consistent styling across states
+- Loading indicators work properly
+
+#### Performance Validation
+- State transitions complete within reasonable timeframes
+- No memory leaks during extended testing
+- Responsive UI during long operations
+- Proper cleanup after test completion
+
+## Implementation Notes
+
+### Technical Considerations
+- Use Playwright's built-in waiting mechanisms
+- Implement robust error handling for flaky operations
+- Add retry logic for network-dependent operations
+- Maintain state isolation between tests
+
+### Maintenance Strategy
+- Regular baseline screenshot updates
+- Prompt refinement based on model updates
+- UI selector updates for component changes
+- Documentation updates for new states
+
+### Future Enhancements
+- Automated state coverage reporting
+- Integration with CI/CD pipeline
+- Performance benchmarking across states
+- Cross-browser compatibility testing