Merge pull request #6 from Zsailer/jupyter-ai-personas

Migrate to jupyter-ai-persona-manager and add avatar support

Author: Zsailer

pyproject.toml

@@ -13,8 +13,7 @@ readme = "README.md"
 license = {text = "BSD-3-Clause"}
 requires-python = ">=3.8"
 dependencies = [
-    "jupyter-ai>=3.0.0b5",
-    "jupyter-ai-magics>=3.0.0b5", 
+    "jupyter-ai-persona-manager",
     "claude-code-sdk",
     "jupyter_server_documents",
 ]

src/jupyter_ai_claude_code/persona.py

@@ -1,6 +1,7 @@
+import os
 from typing import Dict, Any, List, Optional, AsyncIterator
 
-from jupyter_ai.personas.base_persona import BasePersona, PersonaDefaults
+from jupyter_ai_persona_manager import BasePersona, PersonaDefaults
 from jupyterlab_chat.models import Message
 
 from claude_code_sdk import (
@@ -28,6 +29,9 @@
     'WebSearch': 'query',
 }
 
+# Path to avatar file in this package
+AVATAR_PATH = os.path.join(os.path.dirname(__file__), "static", "claude.svg")
+
 PROMPT_TEMPLATE = """
 {{body}}
 
@@ -101,7 +105,7 @@ def defaults(self) -> PersonaDefaults:
         """Return default configuration for the Claude Code persona."""
         return PersonaDefaults(
             name="Claude",
-            avatar_path="/files/.jupyter/claude.svg",
+            avatar_path=AVATAR_PATH,
             description="Claude Code",
             system_prompt="...",
         )