diff --git a/demo.py b/demo.py index 8c54128..b7f2682 100644 --- a/demo.py +++ b/demo.py @@ -2,7 +2,7 @@ client = OpenAI( # defaults to os.environ.get("OPENAI_API_KEY") - api_key="YOUR API KEY", + api_key="sk-QwhV8f1IXqSHH7u7QcnYAcEWEY8hPX52f5mbLifwpQ6IS8Xs", base_url="https://api.chatanywhere.tech/v1" ) @@ -39,3 +39,43 @@ 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): + """ردّ مباشر (streaming)""" + stream = client.chat.completions.create( + model='gpt-3.5-turbo', + messages=messages, + stream=True, + ) + print("\n🟢 الرد (بث مباشر):\n") + for chunk in stream: + if hasattr(chunk, "choices") and len(chunk.choices) > 0: + delta = chunk.choices[0].delta + if delta and delta.content: + print(delta.content, end="", flush=True) + print("\n\n✅ تمّ الانتهاء من البث.") + + +if __name__ == '__main__': + messages = [{'role': 'user','content': '鲁迅和周树人的关系'},] + # 非流式调用 + # gpt_35_api(messages) + # 流式调用 + gpt_35_api_stream(messages)