import json
import openai
client = openai.OpenAI(
base_url='https://api.inference.wandb.ai/v1',
api_key="<your-api-key>", # https://wandb.ai/settings 에서 API 키를 생성하세요
)
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"} # JSON 모드를 활성화합니다
)
content = response.choices[0].message.content
parsed = json.loads(content)
print(parsed)