From 577b2710da5c582a94596fe569793d4993ac2a95 Mon Sep 17 00:00:00 2001 From: William Jin Date: Fri, 3 Nov 2023 10:27:10 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0temperature=E5=8F=82=E6=95=B0?= =?UTF-8?q?=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 3 ++- sparkai/api_resources/chat_completion.py | 13 +++++++++++-- 2 files changed, 13 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 619c747..26d09e6 100644 --- a/README.md +++ b/README.md @@ -21,7 +21,8 @@ pip install --upgrade spark_ai_sdk Install from source with: ```sh -python setup.py install +#python setup.py install +poerty build ``` ## 如何使用 diff --git a/sparkai/api_resources/chat_completion.py b/sparkai/api_resources/chat_completion.py index 1ce6a42..7290a7f 100644 --- a/sparkai/api_resources/chat_completion.py +++ b/sparkai/api_resources/chat_completion.py @@ -66,7 +66,7 @@ class SparkOnceWebsocket(): app_id=None, api_secret=None, messages=None, - temperature=None, + temperature=0.4, max_tokens=2048, top_p=None, request_id=None, @@ -77,6 +77,7 @@ class SparkOnceWebsocket(): self.api_key = api_key self.api_secret = api_secret self.max_token = max_tokens + self.temperature = temperature self.stopping = False self.ws = websocket.WebSocket() @@ -89,7 +90,15 @@ class SparkOnceWebsocket(): self.connect() domain = os.environ.get("SPARK_DOMAIN", "general") - req_data = ChatBody(self.app_id, messages, domain=domain, max_tokens=self.max_token).json() + + req_data = ChatBody( + self.app_id, + messages, + domain=domain, + max_tokens=self.max_token, + temperature=self.temperature + ).json() + self.ws.send(req_data) lastFrame = False full_msg_response = '' -- Gitee