curo1305
11 lines
340 B
Python
11 lines
340 B
Python
from ai.openai_provider import OpenAIProvider
|
|
|
|
|
|
class LMStudioProvider(OpenAIProvider):
|
|
def __init__(self, base_url: str = "http://host.docker.internal:1234", model: str = "gemma-4-e4b-it"):
|
|
super().__init__(
|
|
api_key="lm-studio",
|
|
model=model,
|
|
base_url=base_url.rstrip("/") + "/v1",
|
|
)
|