Build a Chatbot

from openai import OpenAI

client = OpenAI(
    base_url="https://kymaapi.com/v1",
    api_key="ky-your-api-key"
)

messages = [{"role": "system", "content": "You are a helpful assistant."}]

def chat(user_input: str) -> str:
    messages.append({"role": "user", "content": user_input})
    stream = client.chat.completions.create(
        model="qwen-3.6-plus",
        messages=messages,
        stream=True
    )
    response = ""
    for chunk in stream:
        content = chunk.choices[0].delta.content
        if content:
            print(content, end="", flush=True)
            response += content
    print()
    messages.append({"role": "assistant", "content": response})
    return response

chat("What is machine learning?")
chat("Can you give me a simple example?")