Ft/jupyter server (#35)

* chore: add server handlers

* bump: package.json

* chore: handlers

* fix

* dep: add jupyter-server

* Automatic application of license header

---------

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

Author: echarles

Makefile

@@ -32,13 +32,13 @@ jupyterlab: ## jupyterlab
 	jupyter lab \
 		--port 8888 \
 		--ServerApp.root_dir ./dev/content \
-		--IdentityProvider.token MY_TOKEN
+		--IdentityProvider.token=
 
 server: ## server
 	@exec echo
 	@exec echo open http://localhost:4400/api/ai-agents/v1/ping
 	@exec echo
-	python -m uvicorn jupyter_ai_agents.datalayer.server:app --reload --port 4400
+	python -m uvicorn jupyter_ai_agents.server.main:main --reload --port 4400
 
 prompt:
 	jupyter-ai-agents prompt \

README.md

@@ -143,10 +143,10 @@ pip uninstall jupyter_ai_agents
 
 ## Deploy in a Server
 
-You can start a Jupyter AI Agents server to be used in combination with the [Datalayer service](https://datalayer.io).
+You can start a Jupyter AI Agents server to be used in combination with the [Datalayer online services](https://datalayer.app).
 
 ```bash
-make start
+make server
 ```
 
 ## Contributing

dev/content/test.ipynb

@@ -3,7 +3,32 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "id": "fac937b5-6be5-415f-9c4f-d0511508955a",
+   "metadata": {
+    "datalayer": {
+     "ai": {
+      "prompts": [
+       {
+        "id": "01JRJ8Y2KF9QCQZTXQT8FJH",
+        "prompt": "create a plot",
+        "timestamp": 1744370076271,
+        "username": null
+       },
+       {
+        "id": "01JRJ8Y2KF9QCQZTXJH",
+        "prompt": "create a matplot",
+        "timestamp": 1744370076271,
+        "username": null
+       }
+      ]
+     }
+    },
+    "editable": true,
+    "slideshow": {
+     "slide_type": ""
+    },
+    "tags": []
+   },
    "outputs": [],
    "source": []
   }
@@ -28,5 +53,5 @@
   }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 5
 }

docs/package.json

@@ -15,18 +15,18 @@
   },
   "dependencies": {
     "@datalayer/icons-react": "^1.0.0",
-    "@datalayer/primer-addons": "^0.3.1",
-    "@docusaurus/core": "3.5.2",
-    "@docusaurus/preset-classic": "3.5.2",
-    "@docusaurus/theme-live-codeblock": "3.5.2",
-    "@docusaurus/theme-mermaid": "3.5.2",
+    "@datalayer/primer-addons": "^1.0.0",
+    "@docusaurus/core": "^3.5.2",
+    "@docusaurus/preset-classic": "^3.5.2",
+    "@docusaurus/theme-live-codeblock": "^3.5.2",
+    "@docusaurus/theme-mermaid": "^3.5.2",
     "@mdx-js/react": "^3.0.1",
     "clsx": "^2.1.1",
-    "docusaurus-lunr-search": "3.5.0",
-    "react": "18.2.0",
-    "react-calendly": "4.1.0",
-    "react-dom": "18.2.0",
-    "react-modal-image": "2.6.0"
+    "docusaurus-lunr-search": "^3.5.0",
+    "react": "18.3.1",
+    "react-calendly": "^4.1.0",
+    "react-dom": "18.3.1",
+    "react-modal-image": "^2.6.0"
   },
   "browserslist": {
     "production": [

jupyter-config/nb-config/jupyter_ai_agents.json

@@ -0,0 +1,7 @@
+{
+  "NotebookApp": {
+    "nbserver_extensions": {
+      "jupyter_ai_agents": true
+    }
+  }
+}

jupyter-config/server-config/jupyter_ai_agents.json

@@ -0,0 +1,7 @@
+{
+  "ServerApp": {
+    "jpserver_extensions": {
+      "jupyter_ai_agents": true
+    }
+  }
+}

jupyter_ai_agents/__init__.py

@@ -4,4 +4,17 @@
 
 """Jupyter AI Agents."""
 
+from typing import Any, Dict, List
+
+from jupyter_ai_agents.__version__ import __version__
+from jupyter_ai_agents.serverapplication import JupyterAIAgentsExtensionApp
+
+
 __all__ = []
+
+
+def _jupyter_server_extension_points() -> List[Dict[str, Any]]:
+    return [{
+        "module": "jupyter_ai_agents",
+        "app": JupyterAIAgentsExtensionApp,
+    }]

jupyter_ai_agents/__version__.py

@@ -4,4 +4,4 @@
 
 """Jupyter AI Agents."""
 
-__version__ = "0.9.1"
+__version__ = "0.10.1"

jupyter_ai_agents/agents/base.py

@@ -12,6 +12,7 @@
 from jupyter_nbmodel_client import BaseNbAgent
 from jupyter_nbmodel_client.constants import REQUEST_TIMEOUT
 
+
 logger = logging.getLogger(__name__)
 
 
@@ -45,5 +46,3 @@ async def stop(self) -> None:
         await super().stop()
         if self._runtime_client:
             self._runtime_client.stop()
-
-

jupyter_ai_agents/agents/prompt.py

@@ -115,9 +115,7 @@ async def _on_user_prompt(
         """
         document_client = self
         runtime_client = self.runtime_client
-
         current_cell_index = self.get_cell_index(cell_id)
-
         agent_executor = _create_agent(
             document_client,
             runtime_client,
@@ -126,9 +124,7 @@ async def _on_user_prompt(
             self.full_context,
             current_cell_index,
         )
-
         output = None
-
         try:
             await self.notify("Thinking…", cell_id=cell_id)
             async for reply in agent_executor.astream({"input": prompt}):