diff options
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 + ) |