Deepseek R1在cmd中实现推理过程和回答的流式输出

superchaosir

已于 2025-02-10 00:54:27 修改

阅读量1.5k

点赞数 10

文章标签：数据库自然语言处理 python 人机交互学习方法

于 2025-02-09 23:36:24 首次发布

本文链接：https://blog.csdn.net/superchaosir/article/details/145539071

版权

本文代码基于下文大佬代码进行小小修改

DeepSeek-V3 API 调用+代码（流式输出切换+持续交互chat）_deepseek源代码-CSDN博客

由于deepseek r1在返回时候，会有context 和reasoning_context两个参数，所以直接套用v3的话会出现下面的问题：能用，但不美观，所以进行了个小小的补丁，实现推理过程的流式输出，主要修改部分是chat()部分的流式输出，代码如下：

from openai import OpenAI
 
# 初始化 OpenAI 客户端
client = OpenAI(
    api_key="api_key",  # 替换自己的 API 密钥
    base_url="https://api.siliconflow.cn/v1",
)
messages = [{"role": "system", "content": "You are a helpful assistant."}]
stream_mode = True  #True：启用流式输出，False：非流式输出
 

def get_res