from openai import OpenAI

if __name__ == '__main__':
    #DeepSeek-R1-32K
    base_url = "https://maas-cn-southwest-2.modelarts-maas.com/v1/infers/8a062fd4-7367-4ab4-a936-5eeb8fb821c4/v1"
    #DeepSeek-R1-671B-32K
    #base_url = "https://maas-cn-southwest-2.modelarts-maas.com/v1/infers/952e4f88-ef93-4398-ae8d-af37f63f0d8e/v1"
    
    api_key = "Y****"  # 把<your_apiKey>替換成已獲取的API Key。

    client = OpenAI(api_key=api_key, base_url=base_url)

    response = client.chat.completions.create(
        model="DeepSeek-R1",
        messages=[
            {"role": "system", "content": "You are an assistant"},
            {"role": "user", "content": "Your task is to help me complete some simple word processing tasks"},

        ],
        max_tokens=1024,
        temperature=0.6,
        stream=False
    )
    # Print result.     
print(response.choices[0].message.content)