Added jupyter-ai-litellm pkg, removed related code (#5)

* Added jupyter-ai-litellm pkg, removed related code

* lint

* Removed api namespace

* Updated endpoints

* lint

Author: 3coins

CLAUDE.md

@@ -13,34 +13,40 @@ This is **jupyter_ai_jupyternaut**, a JupyterLab extension that provides the def
 ## Development Commands
 
 ### Setup
+
 ```bash
 micromamba install uv jupyterlab nodejs=22
 jlpm
 jlpm dev:install
 ```
 
 ### Build Commands
+
 - `jlpm build` - Build TypeScript sources with source maps
 - `jlpm build:prod` - Production build (clean + lib + labextension)
 - `jlpm build:lib` - Build TypeScript library only
 - `jlpm build:labextension` - Build JupyterLab extension
 
 ### Development
+
 - `jlpm watch` - Watch source files and auto-rebuild
 - `jlpm dev:install` - Full development installation
 - `jlpm dev:uninstall` - Remove development installation
 
 ### Testing
+
 - **Frontend**: `jlpm test` (Jest)
 - **Backend**: `pytest -vv -r ap --cov jupyter_ai_jupyternaut`
 - **UI Tests**: See `ui-tests/README.md` (Playwright/Galata)
 
 ### Code Quality
+
 - `jlpm lint` - Run all linters (stylelint + prettier + eslint)
 - `jlpm lint:check` - Check without fixing
 - Individual tools: `jlpm eslint`, `jlpm prettier`, `jlpm stylelint`
 
 ### Clean Commands
+
 - `jlpm clean` - Clean lib directory
 - `jlpm clean:all` - Clean everything including labextension
 
@@ -71,6 +77,7 @@ jlpm dev:install
 - **Frontend**: `@jupyter/chat`, `@mui/material`, `@emotion/react`, JupyterLab 4.x
 
 ### File Structure
+
 - `jupyter_ai_jupyternaut/` - Python package (server extension)
   - `jupyternaut/` - Core persona implementation
   - `config/` - Configuration management
@@ -89,7 +96,8 @@ jlpm dev:install
 - Entry point registration allows automatic persona discovery by Jupyter AI
 
 ### Code Style
+
 - TypeScript: ESLint with Prettier, single quotes, no trailing commas
 - Python: Standard formatting (details in pyproject.toml)
 - Interface naming: Must start with `I` and use PascalCase
-- CSS: Stylelint with standard config
+- CSS: Stylelint with standard config

jupyter_ai_jupyternaut/extension_app.py

@@ -11,7 +11,7 @@
 from .config import ConfigManager, ConfigRestAPI
 from .handlers import RouteHandler
 from .jupyternaut import JupyternautPersona
-from .models import ChatModelsRestAPI, ModelParametersRestAPI
+from .models import ModelParametersRestAPI
 from .secrets import EnvSecretsManager, SecretsRestAPI
 
 if TYPE_CHECKING:
@@ -38,7 +38,6 @@ class JupyternautExtension(ExtensionApp):
     handlers = [
         (r"api/jupyternaut/get-example/?", RouteHandler),
         (r"api/jupyternaut/config/?", ConfigRestAPI),
-        (r"api/jupyternaut/models/chat/?", ChatModelsRestAPI),
         (r"api/jupyternaut/model-parameters/?", ModelParametersRestAPI),
         (r"api/jupyternaut/secrets/?", SecretsRestAPI),
         (

jupyter_ai_jupyternaut/models/__init__.py

@@ -1,3 +1 @@
-from .model_list import CHAT_MODELS, EMBEDDING_MODELS
-from .chat_models_rest_api import ChatModelsRestAPI
 from .parameters_rest_api import ModelParametersRestAPI

jupyter_ai_jupyternaut/models/chat_models_rest_api.py

@@ -38,7 +38,7 @@ dependencies = [
     # NOTE: Make sure to update the corresponding dependency in
     # `packages/jupyter-ai/package.json` to match the version range below
     "jupyterlab-chat>=0.17.0,<0.18.0",
-    "litellm>=1.73,<2",
+    "jupyter_ai_litellm>=0.0.1",
     "jinja2>=3.0,<4",
     "python_dotenv>=1,<2",
     "jupyter_ai_persona_manager>=0.0.1",

jupyter_ai_jupyternaut/models/model_list.py

@@ -43,7 +43,7 @@ export function StopButton(props: MessageFooterSectionProps): JSX.Element {
 
   const onClick = () => {
     // Post request to the stop streaming handler.
-    requestAPI('chats/stop_streaming', {
+    requestAPI('api/jupyternaut/chats/stop_streaming', {
       method: 'POST',
       body: JSON.stringify({
         message_id: message.id

pyproject.toml

@@ -2,12 +2,10 @@ import { URLExt } from '@jupyterlab/coreutils';
 
 import { ServerConnection } from '@jupyterlab/services';
 
-const API_NAMESPACE = 'api/jupyternaut';
-
 /**
  * Call the API extension
  *
- * @param endPoint API REST end point for the extension
+ * @param endPoint Full API REST end point path for the extension
  * @param init Initial values for the request
  * @returns The response body interpreted as JSON
  */
@@ -17,7 +15,7 @@ export async function requestAPI<T>(
 ): Promise<T> {
   // Make request to Jupyter API
   const settings = ServerConnection.makeSettings();
-  const requestUrl = URLExt.join(settings.baseUrl, API_NAMESPACE, endPoint);
+  const requestUrl = URLExt.join(settings.baseUrl, endPoint);
 
   let response: Response;
   try {
@@ -76,7 +74,7 @@ export namespace AiService {
   };
 
   export async function getConfig(): Promise<DescribeConfigResponse> {
-    return requestAPI<DescribeConfigResponse>('config');
+    return requestAPI<DescribeConfigResponse>('api/jupyternaut/config');
   }
 
   export type EnvAuthStrategy = {
@@ -120,17 +118,19 @@ export namespace AiService {
   };
 
   export async function listLmProviders(): Promise<ListProvidersResponse> {
-    return requestAPI<ListProvidersResponse>('providers');
+    return requestAPI<ListProvidersResponse>('api/jupyternaut/providers');
   }
 
   export async function listEmProviders(): Promise<ListProvidersResponse> {
-    return requestAPI<ListProvidersResponse>('providers/embeddings');
+    return requestAPI<ListProvidersResponse>(
+      'api/jupyternaut/providers/embeddings'
+    );
   }
 
   export async function updateConfig(
     config: UpdateConfigRequest
   ): Promise<void> {
-    return requestAPI<void>('config', {
+    return requestAPI<void>('api/jupyternaut/config', {
       method: 'POST',
       body: JSON.stringify(config)
     });
@@ -142,7 +142,7 @@ export namespace AiService {
   };
 
   export async function listSecrets(): Promise<SecretsList> {
-    return requestAPI<SecretsList>('secrets/', {
+    return requestAPI<SecretsList>('api/jupyternaut/secrets/', {
       method: 'GET'
     });
   }
@@ -154,7 +154,7 @@ export namespace AiService {
   export async function updateSecrets(
     updatedSecrets: Record<string, string | null>
   ): Promise<void> {
-    return requestAPI<void>('secrets/', {
+    return requestAPI<void>('api/jupyternaut/secrets/', {
       method: 'PUT',
       body: JSON.stringify({
         updated_secrets: updatedSecrets
@@ -167,14 +167,19 @@ export namespace AiService {
   }
 
   export async function listChatModels(): Promise<string[]> {
-    const response = await requestAPI<ListChatModelsResponse>('models/chat/', {
-      method: 'GET'
-    });
+    const response = await requestAPI<ListChatModelsResponse>(
+      'api/ai/models/chat/',
+      {
+        method: 'GET'
+      }
+    );
     return response.chat_models;
   }
 
   export async function getChatModel(): Promise<string | null> {
-    const response = await requestAPI<DescribeConfigResponse>('config/');
+    const response = await requestAPI<DescribeConfigResponse>(
+      'api/jupyternaut/config/'
+    );
     return response.model_provider_id;
   }
 
@@ -185,7 +190,9 @@ export namespace AiService {
   }
 
   export async function getCompletionModel(): Promise<string | null> {
-    const response = await requestAPI<DescribeConfigResponse>('config/');
+    const response = await requestAPI<DescribeConfigResponse>(
+      'api/jupyternaut/config/'
+    );
     return response.completions_model_provider_id;
   }
 
@@ -228,22 +235,27 @@ export namespace AiService {
     const endpoint = `model-parameters${
       params.toString() ? `?${params.toString()}` : ''
     }`;
-    return await requestAPI<GetModelParametersResponse>(endpoint);
+    return await requestAPI<GetModelParametersResponse>(
+      `api/jupyternaut/${endpoint}`
+    );
   }
 
   export async function saveModelParameters(
     modelId: string,
     parameters: Record<string, any>
   ): Promise<UpdateModelParametersResponse> {
-    return await requestAPI<UpdateModelParametersResponse>('model-parameters', {
-      method: 'PUT',
-      headers: {
-        'Content-Type': 'application/json'
-      },
-      body: JSON.stringify({
-        model_id: modelId,
-        parameters: parameters
-      })
-    });
+    return await requestAPI<UpdateModelParametersResponse>(
+      'api/jupyternaut/model-parameters',
+      {
+        method: 'PUT',
+        headers: {
+          'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({
+          model_id: modelId,
+          parameters: parameters
+        })
+      }
+    );
   }
 }

src/components/message-footer/stop-button.tsx

@@ -42,7 +42,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
   activate: (app: JupyterFrontEnd) => {
     console.log('JupyterLab extension @jupyter-ai/jupyternaut is activated!');
 
-    requestAPI<any>('get-example')
+    requestAPI<any>('api/jupyternaut/get-example')
       .then(data => {
         console.log(data);
       })