เพิ่มฟังก์ชันการใช้งาน OpenAI API

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)