From 5c2809a22feb63efe72d6c8bc21657a030c1da37 Mon Sep 17 00:00:00 2001
From: zengrr <47846202+zeng-rr@users.noreply.github.com>
Date: Tue, 22 Aug 2023 14:59:58 +0800
Subject: set encoding and temperature

---
 g4f/Provider/Aichat.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'g4f/Provider/Aichat.py')

diff --git a/g4f/Provider/Aichat.py b/g4f/Provider/Aichat.py
index 6992d071..a1d90db7 100644
--- a/g4f/Provider/Aichat.py
+++ b/g4f/Provider/Aichat.py
@@ -40,9 +40,9 @@ class Aichat(BaseProvider):
 
         json_data = {
             "message": base,
-            "temperature": 1,
+            "temperature": kwargs.get('temperature', 0.5),
             "presence_penalty": 0,
-            "top_p": 1,
+            "top_p": kwargs.get('top_p', 1),
             "frequency_penalty": 0,
         }
 
@@ -52,4 +52,6 @@ class Aichat(BaseProvider):
             json=json_data,
         )
         response.raise_for_status()
+        if not response.json()['response']:
+            raise Exception("Error Response: " + response.json())
         yield response.json()["message"]
-- 
cgit v1.2.3