diff --git a/jupyter_ai_jupyternaut/extension_app.py b/jupyter_ai_jupyternaut/extension_app.py index 4df5bf3..dbf4db3 100644 --- a/jupyter_ai_jupyternaut/extension_app.py +++ b/jupyter_ai_jupyternaut/extension_app.py @@ -2,8 +2,7 @@ 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 @@ -17,10 +16,6 @@ if TYPE_CHECKING: from asyncio import AbstractEventLoop -JUPYTERNAUT_AVATAR_PATH = str( - os.path.join(os.path.dirname(__file__), "static", "jupyternaut.svg") -) - class JupyternautExtension(ExtensionApp): """ @@ -40,11 +35,6 @@ class JupyternautExtension(ExtensionApp): (r"api/jupyternaut/config/?", ConfigRestAPI), (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( diff --git a/jupyter_ai_jupyternaut/jupyternaut/jupyternaut.py b/jupyter_ai_jupyternaut/jupyternaut/jupyternaut.py index 2f8b6e7..02ab0c2 100644 --- a/jupyter_ai_jupyternaut/jupyternaut/jupyternaut.py +++ b/jupyter_ai_jupyternaut/jupyternaut/jupyternaut.py @@ -25,6 +25,10 @@ MEMORY_STORE_PATH = os.path.join(jupyter_data_dir(), "jupyter_ai", "memory.sqlite") +JUPYTERNAUT_AVATAR_PATH = str(os.path.abspath( + os.path.join(os.path.dirname(__file__), "../static", "jupyternaut.svg") +)) + def format_tool_args_compact(args_dict, threshold=25): """ @@ -105,7 +109,7 @@ def __init__(self, *args, **kwargs): def defaults(self): return PersonaDefaults( name="Jupyternaut", - avatar_path="/api/jupyternaut/static/jupyternaut.svg", + avatar_path=JUPYTERNAUT_AVATAR_PATH, description="The standard agent provided by JupyterLab. Currently has no tools.", system_prompt="...", )