返回顶部
热门问答 更多热门问答
技术文章 更多技术文章

r1-reasoning-rag:一种新的 RAG 思路

[复制链接]
链载Ai 显示全部楼层 发表于 半小时前 |阅读模式 打印 上一主题 下一主题
Reasoning & Recursive Retrieval With Deepseek-r1, Tavily, and LangGraph
最近发现了一个开源项目,它提供了一种很好的 RAG 思路,它将DeepSeek-R1的推理能力结合Agentic Workflow应用于 RAG 检索

项目地址
https://github.com/deansaco/r1-reasoning-rag.git

项目通过结合DeepSeek-R1TavilyLangGraph,实现了由 AI 主导的动态信息检索与回答机制,利用deepseekr1推理来主动地从知识库中检索、丢弃和综合信息,以完整回答一个复杂问题


新旧 RAG 对比

传统的 RAG(检索增强生成)做法有点死板,通常是在处理完搜索后,通过相似性搜索找到一些内容,再按匹配度重新排个序,选出看起来靠谱的信息片段给到大型语言模型(LLM)去生成答案。但这么做特别依赖于那个重排序模型的质量,要是这模型不给力,就容易漏掉重要信息或者把错的东西传给 LLM,结果出来的答案就不靠谱了。

现在LangGraph团队对这个过程做了大升级,用上了DeepSeek-R1的强大推理能力,把以前那种固定不动的筛选方式变成了一个更灵活、能根据情况调整的动态过程。他们把这个叫做“代理检索”,这种方式让 AI 不仅能主动发现缺少的信息,还能在找资料的过程中不断优化自己的策略,形成一种循环优化的效果,这样交给 LLM 的内容就更加准确了。

这种改进实际上是把测试时扩展的概念从模型内部推理应用到了 RAG 检索中,大大提高了检索的准确性和效率。对于搞 RAG 检索技术的人来说,这个新方法绝对值得好好研究一下!


核心技术与亮点


DeepSeek-R1 推理能力

最新的DeepSeek-R1是一款强大的推理模型

  • 深度思考分析资讯内容
  • 对现有内容进行评估
  • 通过多轮推理辨别缺失的内容,以提高检索结果的准确性


Tavily 即时资讯搜索

Tavily提供即时的资讯搜索,能使大模型过去最新的资讯,扩展模型的知识范围

  • 可动态检索来铺冲缺失的资讯内容,而非仅仅依靠静态数据


LangGraph 递归检索(Recursive Retrieval)

透过Agentic AI机制,让大模型在多轮检索与推理后形成闭环学习,大致流程如下:

  • 第一步检索问题相关的资讯
  • 第二步分析资讯内容是否足够以回答问题
  • 第三步如果资讯不足,则进行进一步查询
  • 第四步过滤不相关的内容,只保留有效的资讯

这样的递归式检索机制,确保大模型能够不断优化查询结果,使得过滤后的资讯更加完整与准确


源码分析

从源码上来看,就很简单的三个文件:agentllmprompts


Agent

这部分的核心思路在于create_workflow这个函数

它定义了这个workflow的节点,其中add_conditional_edges部分定义的是条件边,整个处理思路就是一开始看到的那张图的递归逻辑

如果不熟悉LangGraph的话,可以查看一下相关的资料。
LangGraph构造的是个图的数据结构,有节点(node) 和边(edge),那它的边也可以是带条件的。


每次检索后,都会通过大模型进行筛查,过滤掉没用的信息(Filter Out Irrelevant Information),保留有用的信息(Retain Useful Information),对于缺失的信息(Identify Missing Information)就再进行一次检索,然后重复这个过程直到找到想要的答案

prompts

这里主要定义了两个提示词。
VALIDATE_RETRIEVAL:它用于验证检索到的信息是否能够回答给定的问题。该模板有两个输入变量:retrieved_context(检索到的上下文)和 question(问题)。其主要目的是生成一个JSON格式的响应,根据提供的文本块来判断它们是否包含能够回答问题的信息。

ANSWER_QUESTION:用于指导一个问答代理(question answering agent)根据提供的文本块来回答问题。该模板同样有两个输入变量:retrieved_context(检索到的上下文)和 question(问题)。其主要目的是基于给定的上下文信息提供一个直接且简洁的答案。


llm

这里很简单就是定义使用的r1模型

可以改用其他厂商提供的模型,例如openrouter的免费r1模型


测试效果

我这里单独写了一个脚本,没有使用项目中的,问一下《哪吒2》中哪吒的师傅的师傅是谁

首先它会先调用搜索去查资料,然后开始验证
接着它会开始分析,并得到哪吒的师父是太乙真人这个有效信息,而且也发现了缺失的信息太乙真人的师父(即哪吒的师祖)的具体身份或名字
接着它就会去继续搜索缺失的信息,并继续对搜索回来的信息进行分析校验
后面因为我这边网络断了,就报错了,但是从上图可以看到,它应该是可以找到这个关键的信息了

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

链载AI是专业的生成式人工智能教程平台。提供Stable Diffusion、Midjourney AI绘画教程,Suno AI音乐生成指南,以及Runway、Pika等AI视频制作与动画生成实战案例。从提示词编写到参数调整,手把手助您从入门到精通。
  • 官方手机版

  • 微信公众号

  • 商务合作

  • Powered by Discuz! X3.5 | Copyright © 2025-2025. | 链载Ai
  • 桂ICP备2024021734号 | 营业执照 | |广西笔趣文化传媒有限公司|| QQ