import os

from xai_sdk import Client
from xai_sdk.chat import user, system

client = Client(base_url='https://api.zhizengzeng.com/xai',
        api_key='sk-xxxxxxxxxxxxxxxx')

chat = client.chat.create(model="grok-3-mini")
chat.append(system("You are a PhD-level mathematician."))
chat.append(user("What is 2 + 2?"))

response = chat.sample()
print(response.content)