From 970989f97a42957d1cfdfc1a52a9534cd5ec405b Mon Sep 17 00:00:00 2001 From: GuYaiizza Date: Tue, 11 Feb 2025 22:50:25 +0700 Subject: [PATCH] =?UTF-8?q?=E0=B9=80=E0=B8=9E=E0=B8=B4=E0=B9=88=E0=B8=A1?= =?UTF-8?q?=E0=B8=9F=E0=B8=B1=E0=B8=87=E0=B8=81=E0=B9=8C=E0=B8=8A=E0=B8=B1?= =?UTF-8?q?=E0=B8=99=E0=B8=81=E0=B8=B2=E0=B8=A3=E0=B9=83=E0=B8=8A=E0=B9=89?= =?UTF-8?q?=E0=B8=87=E0=B8=B2=E0=B8=99=20OpenAI=20API?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- demo.py | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) diff --git a/demo.py b/demo.py index 8c54128..e487137 100644 --- a/demo.py +++ b/demo.py @@ -39,3 +39,41 @@ def gpt_35_api_stream(messages: list): # gpt_35_api(messages) # 流式调用 gpt_35_api_stream(messages) + +from openai import OpenAI + +client = OpenAI( + # defaults to os.environ.get("OPENAI_API_KEY") + api_key="YOUR API KEY", + base_url="https://api.chatanywhere.tech/v1" +) + +# 非流式响应 +def gpt_35_api(messages: list): + """为提供的对话消息创建新的回答 + Args: + messages (list): 完整的对话消息 + """ + completion = client.chat.completions.create(model="gpt-3.5-turbo", messages=messages) + print(completion.choices[0].message.content) + +def gpt_35_api_stream(messages: list): + """为提供的对话消息创建新的回答 (流式传输) + Args: + messages (list): 完整的对话消息 + """ + stream = client.chat.completions.create( + model='gpt-3.5-turbo', + messages=messages, + stream=True, + ) + for chunk in stream: + if chunk.choices[0].delta.content is not None: + print(chunk.choices[0].delta.content, end="") + +if __name__ == '__main__': + messages = [{'role': 'user','content': '鲁迅和周树人的关系'},] + # 非流式调用 + # gpt_35_api(messages) + # 流式调用 + gpt_35_api_stream(messages)