From ff13f4f5008cbf56a33ca3248b58cc223b45f4a9 Mon Sep 17 00:00:00 2001 From: H Lohaus Date: Sat, 6 Apr 2024 23:37:41 +0200 Subject: Update Aura.py --- g4f/Provider/Aura.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'g4f') diff --git a/g4f/Provider/Aura.py b/g4f/Provider/Aura.py index 877b7fef..4501df2c 100644 --- a/g4f/Provider/Aura.py +++ b/g4f/Provider/Aura.py @@ -17,6 +17,8 @@ class Aura(AsyncGeneratorProvider): model: str, messages: Messages, proxy: str = None, + temperature: float = 0.5, + max_tokens: int = 8192. webdriver: WebDriver = None, **kwargs ) -> AsyncResult: @@ -34,14 +36,14 @@ class Aura(AsyncGeneratorProvider): "id": "openchat_v3.2_mistral", "name": "OpenChat Aura", "maxLength": 24576, - "tokenLimit": 8192 + "tokenLimit": max_tokens }, "messages": new_messages, "key": "", "prompt": "\n".join(system_message), - "temperature": 0.5 + "temperature": temperature } async with session.post(f"{cls.url}/api/chat", json=data, proxy=proxy) as response: response.raise_for_status() async for chunk in response.content.iter_any(): - yield chunk.decode() \ No newline at end of file + yield chunk.decode(error="ignore") \ No newline at end of file -- cgit v1.2.3