diff --git a/README.md b/README.md index 619c747842ffef39118d1382ec793ccde8658b85..26d09e660ff6b81787a8b1e76c26e16d8edf1719 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 1ce6a42790192d058c009eb9d6fd6da9be762022..7290a7f7364bf6c43678b08d8cdfec67e71b19db 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 = ''