Spaces:
Runtime error
Runtime error
File size: 679 Bytes
e72aedf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import os
from fastchat.model import get_conversation_template
def chatgpt():
import openai
model = "gpt-3.5-turbo"
conv = get_conversation_template(model)
conv.append_message(conv.roles[0], "Hello!")
conv.append_message(conv.roles[1], None)
messages = conv.to_openai_api_messages()
print(messages)
res = openai.ChatCompletion.create(model=model, messages=messages)
msg = res["choices"][0]["message"]["content"]
print(msg)
res = openai.ChatCompletion.create(model=model, messages=messages, stream=True)
msg = ""
for chunk in res:
msg += chunk["choices"][0]["delta"].get("content", "")
print(msg)
chatgpt()
|