Allow overriding the Ollama base URL (#904)

jtpio · web-flow · commit 69f1d5d5bc4e · 2024-07-22T16:13:12.000Z
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):

Original file line number	Diff line number	Diff line change
`@@ -705,6 +705,9 @@ class OllamaProvider(BaseProvider, Ollama):`
`705`	`705`	`)`
`706`	`706`	`models = ["*"]`
`707`	`707`	`registry = True`
	`708`	`+ fields = [`
	`709`	`+ TextField(key="base_url", label="Base API URL (optional)", format="text"),`
	`710`	`+ ]`
`708`	`711`
`709`	`712`
`710`	`713`	`class JsonContentHandler(LLMContentHandler):`