|
|
|
@@ -853,7 +853,7 @@ class OllamaChat(Base): |
|
|
|
def __init__(self, key, model_name, base_url=None, **kwargs): |
|
|
|
super().__init__(key, model_name, base_url=base_url, **kwargs) |
|
|
|
|
|
|
|
self.client = Client(host=kwargs["base_url"]) if not key or key == "x" else Client(host=kwargs["base_url"], headers={"Authorization": f"Bearer {key}"}) |
|
|
|
self.client = Client(host=base_url) if not key or key == "x" else Client(host=base_url, headers={"Authorization": f"Bearer {key}"}) |
|
|
|
self.model_name = model_name |
|
|
|
|
|
|
|
def _clean_conf(self, gen_conf): |