summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/local/Ollama.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/Provider/local/Ollama.py (renamed from g4f/Provider/Ollama.py)8
1 files changed, 4 insertions, 4 deletions
diff --git a/g4f/Provider/Ollama.py b/g4f/Provider/local/Ollama.py
index f9116541..de68a218 100644
--- a/g4f/Provider/Ollama.py
+++ b/g4f/Provider/local/Ollama.py
@@ -3,10 +3,10 @@ from __future__ import annotations
import requests
import os
-from .needs_auth.Openai import Openai
-from ..typing import AsyncResult, Messages
+from ..needs_auth.OpenaiAPI import OpenaiAPI
+from ...typing import AsyncResult, Messages
-class Ollama(Openai):
+class Ollama(OpenaiAPI):
label = "Ollama"
url = "https://ollama.com"
needs_auth = False
@@ -37,4 +37,4 @@ class Ollama(Openai):
api_base: str = f"http://{host}:{port}/v1"
return super().create_async_generator(
model, messages, api_base=api_base, **kwargs
- ) \ No newline at end of file
+ )