Added jupyter-ai-litellm pkg, removed related code

Author: 3coins

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

@@ -9,15 +9,17 @@ const API_NAMESPACE = 'api/jupyternaut';
  *
  * @param endPoint API REST end point for the extension
  * @param init Initial values for the request
+ * @param api_namespace Optional API namespace (defaults to API_NAMESPACE)
  * @returns The response body interpreted as JSON
  */
 export async function requestAPI<T>(
   endPoint = '',
-  init: RequestInit = {}
+  init: RequestInit = {},
+  api_namespace: string = API_NAMESPACE
 ): 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, api_namespace, endPoint);
 
   let response: Response;
   try {
@@ -169,7 +171,7 @@ export namespace AiService {
   export async function listChatModels(): Promise<string[]> {
     const response = await requestAPI<ListChatModelsResponse>('models/chat/', {
       method: 'GET'
-    });
+    }, 'api/ai');
     return response.chat_models;
   }