Generate chat completions

curl --location 'https://api.writer.com/v1/chat' \
--header 'Content-Type: application/json' \
--header "Authorization: Bearer $WRITER_API_KEY" \
--data '{
    "model": "palmyra-x5",
    "messages": [
        {
            "role": "user",
            "content": "You are an expert at writing concise product descriptions for an E-Commerce Retailer"
        },
        {
            "role": "assistant",
            "content": "Okay, great I can help write these descriptions. Do you have a specific product in mind?"
        },
        {
            "role": "user",
            "content": "Please write a one sentence product description for a cozy, stylish sweater suitable for both casual and formal occasions"
        }
    ]
}'

system_prompt = "You are a helpful assistant that responds in a casual, friendly tone and uses emojis in your responses."

messages = [
  {"role": "system", "content": system_prompt},
]

# Stream the chat response to the user using the `stream` helper method
with client.chat.stream(messages=messages, model="palmyra-x5") as stream:
    for event in stream:
        # Check if the event is a content delta, which contains this chunk of the chat response
        if event.type == "content.delta":
            print(event.delta, end="", flush=True)
# Collect the full response from the stream
completion = stream.get_final_completion().choices[0].message.content

from writerai import Writer

# Initialize the client. If you don't pass the `api_key` parameter,
# the client looks for the `WRITER_API_KEY` environment variable.
client = Writer()

end = False
system_prompt = "You are a helpful assistant that responds in a casual, friendly tone and uses emojis in your responses."

# Ask the user for an initial message
input_message = "\nEnter a message for the assistant. Type 'exit' to end the conversation. > "
initial_message = input(input_message)

# Add the user's message to the conversation history
messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": initial_message}]

# Main loop to handle the conversation.
while not end:
    # Stream the chat response to the user using the `stream` helper method
    with client.chat.stream(messages=messages, model="palmyra-x5") as stream:
        for event in stream:
            # Check if the event is a content delta, which contains this chunk of the chat response
            if event.type == "content.delta":
                print(event.delta, end="", flush=True)
    # Collect the full response from the stream and add it to the conversation history
    full_message = stream.get_final_completion().choices[0].message.content
    messages.append({"role": "assistant", "content": full_message})
    new_message = input(input_message)
    messages.append({"role": "user", "content": new_message})
    if new_message == "exit":
        end = True