feat(ui): add toasts with decorator

pyproject.toml

@@ -49,3 +49,4 @@ line-length = 120
 
 [tool.pyright]
 reportMissingModuleSource = false
+pythonVersion = "3.12"

src/lib/components/Chat.svelte

@@ -17,8 +17,11 @@
 
   async function refresh() {
     refreshing = true;
-    chain = await reloadChain();
-    refreshing = false;
+    try {
+      chain = await reloadChain();
+    } finally {
+      refreshing = false;
+    }
   }
 
   async function start() {

src/lib/py/load.py

@@ -3,15 +3,22 @@
 from importlib import reload
 from os import getenv
 from time import perf_counter
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Callable
 from zipfile import BadZipFile
 
 from micropip import uninstall
+from pyodide.ffi import create_once_callable
 
 if TYPE_CHECKING:
     from micropip import install
+
+    create_proxy = create_once_callable
+
+    def with_toast[T: Callable](message: str) -> Callable[[T], T]: ...
+
 else:
     from micropip import install as _install
+    from pyodide.ffi import create_proxy
 
     @wraps(_install)
     async def install(*args, **kwargs):
@@ -33,6 +40,8 @@ async def patch_promplate():
     patch_promplate()
 
 
+@with_toast("re-installing reasonify")
+@create_proxy
 async def reload_reasonify_chain():
     uninstall("reasonify-headless")
     await install(getenv("PACKAGE", "reasonify-headless"))
@@ -45,6 +54,8 @@ async def reload_reasonify_chain():
     return reasonify.chain
 
 
+@with_toast("installing reasonify")
+@create_proxy
 async def get_reasonify_chain(patch_promplate=patch_promplate):
     await gather(patch_promplate(), install(getenv("PACKAGE", "reasonify-headless")))
 

src/lib/py/load.ts

@@ -3,6 +3,7 @@ import VERSION from "./version";
 import { dev } from "$app/environment";
 import * as env from "$env/static/public";
 import { cacheSingleton } from "$lib/utils/cache";
+import { withToast } from "$lib/utils/toast";
 import { version } from "pyodide";
 
 const indexURL = ("PUBLIC_PYODIDE_INDEX_URL" in env) ? (env.PUBLIC_PYODIDE_INDEX_URL as string).replace("{}", version) : `https://cdn.jsdelivr.net/pyodide/v${version}/full/`;
@@ -12,5 +13,6 @@ export const getPy = cacheSingleton(async () => {
   const PACKAGE = dev ? `/whl/reasonify_headless-${VERSION}-py3-none-any.whl` : `reasonify-headless==${VERSION}`;
   const py = await loadPyodide({ indexURL, packages: ["micropip", "typing-extensions"], env: { PACKAGE }, args: ["-OO"] });
   pyodideReady.set(true);
+  py.globals.set("with_toast", withToast);
   return py;
 });

src/lib/tools/python/index.ts

@@ -6,6 +6,7 @@ import schema from "./schema.ts?raw";
 import { getPy } from "$lib/py";
 import { cacheSingleton } from "$lib/utils/cache";
 import { formatSchemaPrompt } from "$lib/utils/format";
+import { withToast } from "$lib/utils/toast";
 
 const instruction = `
 The "source" field must be a runnable python script.
@@ -26,5 +27,5 @@ export default {
   usage: "Run some python code. Open your mind. If something can be done through code, then don't use other tools.",
   schema: formatSchemaPrompt(schema),
   instruction,
-  run: async options => await (await getRun()).callKwargs(options),
+  run: withToast("running python code")(async options => await (await getRun()).callKwargs(options)),
 } as Tool;

src/lib/utils/toast.ts

@@ -0,0 +1,20 @@
+import type { PythonError } from "pyodide/ffi";
+
+import { toast } from "svelte-sonner";
+
+function displayError(error: any) {
+  if (error instanceof Error && error.name === "PythonError")
+    return (error as PythonError).message;
+  return String(error);
+}
+
+export function withToast<T extends (...args: any[]) => any>(message: string) {
+  return (anyFunction: T) => {
+    return (async (...args) => {
+      const promise: Promise<ReturnType<T>> = Promise.resolve(anyFunction(...args));
+      toast.promise(promise, { loading: message, success: message, error: displayError });
+      const result = await promise;
+      return result;
+    }) as T;
+  };
+}

src/routes/+layout.svelte

@@ -5,7 +5,7 @@
   import { Toaster } from "svelte-sonner";
 </script>
 
-<Toaster theme="dark" richColors />
+<Toaster theme="dark" richColors position="bottom-left" toastOptions={{ class: "ws-pre-wrap font-mono" }} />
 
 <slot />
 

src/routes/+page.svelte

@@ -6,6 +6,12 @@
   import { initChain } from "$lib/py";
   import { promplateReady, pyodideReady, reasonifyReady } from "$lib/stores";
   import { version } from "pyodide";
+  import { onDestroy } from "svelte";
+
+  onDestroy(() => {
+    $promplateReady = false;
+    $reasonifyReady = false;
+  });
 </script>
 
 {#if browser}