curo1305
11 lines
330 B
Python
11 lines
330 B
Python
from ai.openai_provider import OpenAIProvider
|
|
|
|
|
|
class OllamaProvider(OpenAIProvider):
|
|
def __init__(self, base_url: str = "http://host.docker.internal:11434", model: str = "llama3.2"):
|
|
super().__init__(
|
|
api_key="ollama",
|
|
model=model,
|
|
base_url=base_url.rstrip("/") + "/v1",
|
|
)
|