From ccb0fd84172a51151876e3893ff373bc60f5db93 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 0777ee6e2..7a24c7369 100644
--- 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):