From 448ab419cfd7e66b026c9a83b8da85f0bb97ca64 Mon Sep 17 00:00:00 2001
From: Jeremy Tuloup <jeremy.tuloup@gmail.com>
Date: Thu, 18 Jul 2024 20:35:33 +0200
Subject: [PATCH] Allow overriding the Ollama base URL

---
 packages/jupyter-ai-magics/jupyter_ai_magics/providers.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
index e017125ba..54c2fe62b 100644
--- 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):