from openai import OpenAI
client = OpenAI(
base_url = "https://integrate.api.nvidia.com/v1",
api_key = "$NVIDIA_API_KEY"
)
completion = client.chat.completions.create(
model="bytedance/seed-oss-36b-instruct",
messages=[{"role":"user","content":""}],
temperature=1.1,
top_p=0.95,
max_tokens=4096,
frequency_penalty=0,
presence_penalty=0,
stream=False,
extra_body={
"thinking_budget": -1
}
)
reasoning = getattr(completion.choices[0].message, "reasoning_content", None)
if reasoning:
print(reasoning)
print(completion.choices[0].message.content)