Allow overriding the Ollama base URL (#904)

jupyterlab · Jul 22, 2024 · 69f1d5d · 69f1d5d
1 parent c047e5c
commit 69f1d5d
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
@@ -705,6 +705,9 @@ class OllamaProvider(BaseProvider, Ollama):
     )
     models = ["*"]
     registry = True
+    fields = [
+        TextField(key="base_url", label="Base API URL (optional)", format="text"),
+    ]
 
 
 class JsonContentHandler(LLMContentHandler):