Update to models handler path, added ai in the path.

Author: 3coins

README.md

@@ -16,7 +16,7 @@ This extension is composed of a Python package named `jupyter_ai_litellm` that e
 
 ### Chat Models
 
-- `GET /api/models/chat` - Returns a list of all available chat models
+- `GET /api/ai/models/chat` - Returns a list of all available chat models
 
 The response includes model IDs in LiteLLM format (e.g., `openai/gpt-4`, `anthropic/claude-3-sonnet`, etc.)
 

jupyter_ai_litellm/handlers.py

@@ -25,6 +25,6 @@ def setup_handlers(web_app):
     route_pattern = url_path_join(base_url, "jupyter-ai-litellm", "get-example")
     handlers = [
         (route_pattern, RouteHandler),
-        (url_path_join(base_url, "api/models/chat") + r"(?:\?.*)?", ChatModelsRestAPI)
+        (url_path_join(base_url, "api/ai/models/chat") + r"(?:\?.*)?", ChatModelsRestAPI)
     ]
     web_app.add_handlers(host_pattern, handlers)

jupyter_ai_litellm/tests/test_handlers.py

@@ -14,7 +14,7 @@ async def test_get_example(jp_fetch):
 
 async def test_get_chat_models(jp_fetch):
     # When
-    response = await jp_fetch("api", "models", "chat")
+    response = await jp_fetch("api", "ai", "models", "chat")
 
     # Then
     assert response.code == 200