import json
import openai
client = openai.OpenAI(
base_url='https://api.inference.wandb.ai/v1',
api_key="<your-api-key>", # Available from https://wandb.ai/authorize
)
response = client.chat.completions.create(
model="openai/gpt-oss-20b",
messages=[
{"role": "system", "content": "You are a helpful assistant that outputs JSON."},
{"role": "user", "content": "Give me a list of three fruits with their colors."},
],
response_format={"type": "json_object"} # This enables JSON mode
)
content = response.choices[0].message.content
parsed = json.loads(content)
print(parsed)