AI_customer_service/ds_r1.py

import os
import time
from openai import OpenAI

client = OpenAI(
    api_key = "817dff39-5586-4f9b-acba-55004167c0b1",
    base_url = "https://ark.cn-beijing.volces.com/api/v3",
)

start_time = time.time()
# Non-streaming:
print("----- standard request -----")
completion = client.chat.completions.create(
    model = "deepseek-r1-250120",  # your model endpoint ID
    messages = [
        {"role": "system", "content": "你是人工智能助手"},
        {"role": "user", "content": "常见的十字花科植物有哪些？"},
    ],
)
print(completion.choices[0].message.content)

end_time = time.time()
print(f"Time taken: {end_time - start_time} seconds")