[Dify] 去除 Ollama Deepseek 中 <think> 内容,仅保留答案部分的实现方案

在使用 Ollama 的 Deepseek 模型时,我们经常会遇到这样的输出格式:

这是模型的思考过程
这是用户真正想要的回答。

虽然 部分对调试和理解模型推理过程有一定帮助,但在实际部署如 Dify 等前端平台中,我们通常只希望用户看到最终答案。本文将介绍如何通过一个简单的 Python 函数,从 Ollama 的输出中剥离掉 标签中的内容,仅保留“答案部分”。[Dify] 去除 Ollama Deepseek 中 <think> 内容,仅保留答案部分的实现方案_第1张图片


目标

我们的目标是:
将 Ollama 输出中 ... 标签包裹的内容去除,只返回标签后的答案文本。

最终返回结果应为一个标准的 Python 字典结构,方便与如 Dify 这样的接口系统对接。


实现代码

以下是核心代码的完整实现:

import json

def main(response: str) -> dict:
    # 解析

你可能感兴趣的:(AI实践-Dify专栏,Dify,人工智能,AI)