from openai import OpenAI

client = OpenAI(api_key="...", base_url="https://oneastronomy.zero2x.org/api/platform/v1", max_retries=0)

stream = client.chat.completions.create(
        model="astroone-chat",
        messages=[
            {'role':'system', 'content': 'You are a helpful assistant'},
            {'role':'user', 'content': 'Hi! Why is Mars red?'},
            ],
        stream=True,
        extra_headers={'Accept':'text/event-stream'},
        )

for chunk in stream:
    if chunk.choices[0].delta.content != None:
        print(chunk.choices[0].delta.content, end='')
print('')
