client = OpenAI(
<<<<<<< HEAD
api_key="您的 APIKEY", # 从https://cloud.siliconflow.com/account/ak获取
base_url="https://api.siliconflow.com/v1"
=======
api_key="您的 APIKEY", # 从https://cloud.siliconflow.com/account/ak获取
base_url="https://api.siliconflow.com/v1"
>>>>>>> upstream/main
)
messages = [
{"role": "user", "content": "Please write quick sort code"},
]
response = client.chat.completions.create(
model="deepseek-ai/DeepSeek-V2.5",
messages=messages,
extra_body={
"prefix": f"""
def quick_sort(arr):
# 基本情况,如果数组长度小于等于 1,则返回数组
if len(arr) <= 1:
return arr
else:
""",
"suffix": f"""
# 测试 quick_sort 函数
arr = [3, 6, 8, 10, 1, 2, 1]
sorted_arr = quick_sort(arr)
print("Sorted array:", sorted_arr)
"""
},
stream=True,
max_tokens=4096
)
for chunk in response:
print(chunk.choices[0].delta.content, end='')