diff --git a/code/ai/demo.py b/code/ai/demo.py new file mode 100644 index 00000000..f8940362 --- /dev/null +++ b/code/ai/demo.py @@ -0,0 +1,20 @@ +# Please install OpenAI SDK first: `pip3 install openai` +import os +from openai import OpenAI + +client = OpenAI( + api_key=os.environ.get('DEEPSEEK_API_KEY'), + base_url="https://api.deepseek.com") + +response = client.chat.completions.create( + model="deepseek-v4-pro", + messages=[ + {"role": "system", "content": "You are a helpful assistant"}, + {"role": "user", "content": "Hello"}, + ], + stream=False, + reasoning_effort="high", + extra_body={"thinking": {"type": "enabled"}} +) + +print(response.choices[0].message.content) \ No newline at end of file