获取每次与 AI 交互时消耗的 token 数量（★☆）

rayaaa

哈哈很实用的教程！

Jiawe1

成功复刻有道理

douzisss

一定要用本地部署才可以吗

lovecarbon

学习了

Imnwiw

好难好难好难好难

wu同学学编程

真难啊，慢慢学吧我

SiegfriedStyx

学习一下，刚好已经有本地部署的API了

owenfu

学习！！！

Esleepsheep

学习

python_LeVis

晚上去复刻一下

吸油烟机长

学学

code3

真是棒棒哒 又学到了

gaming1990

学习

萧随风

打卡学习

pyzyd

1. stream = client.chat.completions.create(
   
2.                 model="deepseek-ai/DeepSeek-R1",
   
3.                 messages=messages,
   
4.                 stream=True,
   
5.                 stream_options={"include_usage": True}
   
6.             )
   
7. 
   
8.             print("\nDeepSeek：", end="", flush=True)
   
9.             assistant_response = ""
   
10.             has_printed_thinking = False  # 控制是否显示思考过程分隔线
    
11.             has_printed_answer = False    # 控制正式回答分隔线
    
12. 
    
13.             for chunk in stream:
    
14.                 if chunk.choices:
    
15.                     # 如果需要显示思考过程，可根据 delta 内容判断
    
16.                     if hasattr(chunk.choices[0].delta, 'reasoning_content'):
    
17.                         reasoning = chunk.choices[0].delta.reasoning_content or ""
    
18.                         if not has_printed_thinking and reasoning.strip():
    
19.                             print("\n-------- Think -------->\n")
    
20.                             has_printed_thinking = True
    
21.                         # 如果不希望实时打印思考过程，可以注释下面这一行
    
22.                         print(reasoning, end="", flush=True)
    
23. 
    
24.                     if hasattr(chunk.choices[0].delta, 'content'):
    
25.                         content = chunk.choices[0].delta.content or ""
    
26.                         if not has_printed_answer and content.strip():
    
27.                             print("\n<------- Answer ---------\n")
    
28.                             has_printed_answer = True
    
29.                         assistant_response += content
    
30.                         print(content, end="", flush=True)
    
31.                 if chunk.usage:
    
32.                     # 打印消耗的 tokens 信息
    
33.                     print("\n======= 消耗信息 =======")
    
34.                     print(f"本次消耗：prompt_tokens -> {chunk.usage.prompt_tokens}, completion_tokens -> {chunk.usage.completion_tokens}, total_tokens -> {chunk.usage.total_tokens}")
    
35.                     print("========================")

复制代码