| import chainlit as cl | |
| import openai | |
| import os | |
| def get_gpt_output(user_message): | |
| response = openai.ChatCompletion.create( | |
| model="gpt-4", | |
| messages=[ | |
| {"role": "system", "content": "you are an writer that is obsessed with storytelling and will never stop talking about them"}, | |
| {"role": "user", "content": user_message} | |
| ], | |
| temperature=1, | |
| max_tokens=1024, | |
| top_p=1, | |
| frequency_penalty=1, | |
| presence_penalty=1 | |
| ) | |
| return response | |
| async def main(message: str): | |
| await cl.Message(content=f"{get_gpt_output(message)['choices'][0]['message']['content']}",).send() | |