WIP: serve Jupyternaut avatar to make it accessible from UI

Author: dlqqq

jupyter_ai_jupyternaut/extension_app.py

@@ -2,6 +2,8 @@
 from asyncio import get_event_loop_policy
 from jupyter_server.extension.application import ExtensionApp
 from jupyter_server.serverapp import ServerApp
+import os
+from tornado.web import StaticFileHandler
 from traitlets import List, Unicode, Dict
 from traitlets.config import Config
 from typing import TYPE_CHECKING
@@ -15,6 +17,10 @@
 if TYPE_CHECKING:
     from asyncio import AbstractEventLoop
 
+JUPYTERNAUT_AVATAR_PATH = str(
+    os.path.join(os.path.dirname(__file__), "static", "jupyternaut.svg")
+)
+
 class JupyternautExtension(ExtensionApp):
     """
     The Jupyternaut server extension.
@@ -34,6 +40,11 @@ class JupyternautExtension(ExtensionApp):
         (r"api/jupyternaut/models/chat/?", ChatModelsRestAPI),
         (r"api/jupyternaut/model-parameters/?", ModelParametersRestAPI),
         (r"api/jupyternaut/secrets/?", SecretsRestAPI),
+        (
+            r"api/jupyternaut/static/jupyternaut.svg()/?",
+            StaticFileHandler,
+            {"path": JUPYTERNAUT_AVATAR_PATH},
+        ),
     ]
 
     allowed_providers = List(

jupyter_ai_jupyternaut/jupyternaut/jupyternaut.py

@@ -23,7 +23,7 @@ def __init__(self, *args, **kwargs):
     def defaults(self):
         return PersonaDefaults(
             name="Jupyternaut",
-            avatar_path="/api/ai/static/jupyternaut.svg",
+            avatar_path="/api/jupyternaut/static/jupyternaut.svg",
             description="The standard agent provided by JupyterLab. Currently has no tools.",
             system_prompt="...",
         )