Allow overriding the Ollama base URL

jupyterlab · Jul 18, 2024 · 448ab41 · 448ab41
1 parent 0a60807
commit 448ab41
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -734,6 +734,9 @@ class OllamaProvider(BaseProvider, Ollama):
     )
     models = ["*"]
     registry = True
+    fields = [
+        TextField(key="base_url", label="Base API URL (optional)", format="text"),
+    ]
 
 
 class JsonContentHandler(LLMContentHandler):