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/Acytoo.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

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

diff --git a/g4f/Provider/Acytoo.py b/g4f/Provider/Acytoo.py
index 32c67c0c..2edd9efd 100644
--- a/g4f/Provider/Acytoo.py
+++ b/g4f/Provider/Acytoo.py
@@ -19,11 +19,12 @@ class Acytoo(BaseProvider):
         **kwargs: Any,
     ) -> CreateResult:
         headers = _create_header()
-        payload = _create_payload(messages)
+        payload = _create_payload(messages, kwargs.get('temperature', 0.5))
 
         url = "https://chat.acytoo.com/api/completions"
         response = requests.post(url=url, headers=headers, json=payload)
         response.raise_for_status()
+        response.encoding = "utf-8"
         yield response.text
 
 
@@ -34,7 +35,7 @@ def _create_header():
     }
 
 
-def _create_payload(messages: list[dict[str, str]]):
+def _create_payload(messages: list[dict[str, str]], temperature):
     payload_messages = [
         message | {"createdAt": int(time.time()) * 1000} for message in messages
     ]
@@ -42,6 +43,6 @@ def _create_payload(messages: list[dict[str, str]]):
         "key": "",
         "model": "gpt-3.5-turbo",
         "messages": payload_messages,
-        "temperature": 1,
+        "temperature": temperature,
         "password": "",
     }
-- 
cgit v1.2.3