From 8ed065836a19dd65340a890a14dfbb8fa154ab84 Mon Sep 17 00:00:00 2001
From: Kurt Heiden <kurt.w.heiden@gmail.com>
Date: Tue, 29 Oct 2024 08:44:19 -0600
Subject: [PATCH] Update ollama.py

---
 simplemind/providers/ollama.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/simplemind/providers/ollama.py b/simplemind/providers/ollama.py
index 2373e06..e4b45fd 100644
--- a/simplemind/providers/ollama.py
+++ b/simplemind/providers/ollama.py
@@ -1,5 +1,6 @@
 import ollama as ol
 
+from ._base import BaseProvider
 from ..settings import settings
 
 PROVIDER_NAME = "ollama"
@@ -9,7 +10,8 @@ NOT_IMPLEMENTED_REASON = """
 # TODO: instructor does not natively support ollama.
 # Alternate python dependency may be required
 """
-class Ollama:
+
+class Ollama(BaseProvider):
     NAME = PROVIDER_NAME
     DEFAULT_MODEL = DEFAULT_MODEL