Agentic RAG技術(shù)是一種新興的智能體驅(qū)動解決方案,能夠應對復雜的多文檔問答挑戰(zhàn)。這項技術(shù)通過智能化手段,不僅提升了我們處理文本數(shù)據(jù)的效率,還極大地增強了問答系統(tǒng)的準確性和深度。本文帶大家來了解這項技術(shù),闡述其如何提升信息檢索和分析的效率與準確性。

1 Agentic RAG簡介

Agentic RAG是由智能體驅(qū)動的技術(shù),能夠靈活處理多文檔問答任務。這項技術(shù)不僅能比較文檔、總結(jié)內(nèi)容,還能對多個摘要進行對比分析。

正因如此,我們不完全依賴于大型語言模型,而是通過智能體來承擔那些需要深思熟慮、邏輯推理、工具運用以及持續(xù)學習的任務。Agentic RAG通過智能化手段,提高了解決復雜問題的能力,讓問答更加高效和精確。

2 基本架構(gòu)

基本架構(gòu)是為每個文檔設(shè)置一個文檔智能體,每個文檔智能體都能在其自己的文檔內(nèi)執(zhí)行問答和摘要。

然后,設(shè)置一個頂級智能體(元智能體)來管理所有這些較低級別的文檔智能體。

3 技術(shù)棧

技術(shù)棧包括以下幾個核心組件:

  1. Langchain:特別是LCEL,這是用于構(gòu)建大型語言模型(LLM)應用的編排框架
  2. OpenAI:提供我們需要的大型語言模型(LLM)。
  3. FAISS-cpu:用作向量存儲解決方案。
  4. 數(shù)據(jù)源:我們使用ArxivLoader來檢索arXiv上發(fā)表文章的元數(shù)據(jù)

4 代碼實現(xiàn)

安裝所需的依賴項:

!pip install -qU langchain langchain_openai langgraph arxiv duckduckgo-search
!pip install -qU faiss-cpu pymupdf

設(shè)置環(huán)境變量:

from google.colab import userdata
from uuid import uuid4
import os
#
os.environ['OPENAI_API_KEY'] = userdata.get('OPENAI_API_KEY')
os.environ["LANGCHAIN_TRACING_V2"] = "true"
os.environ["LANGCHAIN_PROJECT"] = f"AIE1 - LangGraph - {uuid4().hex[0:8]}"
os.environ["LANGCHAIN_API_KEY"] = userdata.get('LANGCHAIN_API_KEY')

使用LCEL實例化一個簡單的檢索鏈。

from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain_community.document_loaders import ArxivLoader
from langchain_community.vectorstores import FAISS
from langchain_openai import OpenAIEmbeddings

# 加載特定主題的文檔
docs = ArxivLoader(query="Retrieval Augmented Generation", load_max_docs=5).load()

# 將文檔分割成更小的塊
text_splitter = RecursiveCharacterTextSplitter.from_tiktoken_encoder(
chunk_size=350, chunk_overlap=50
)

chunked_documents = text_splitter.split_documents(docs)
#
# 實例化嵌入模型
embeddings = OpenAIEmbeddings(model="text-embedding-3-small",openai_api_key=os.environ['OPENAI_API_KEY'])
# 創(chuàng)建索引-將文檔塊加載到向量存儲中
faiss_vectorstore = FAISS.from_documents(
documents=chunked_documents,
embedding=embeddings,
)
# 創(chuàng)建檢索器
retriever = faiss_vectorstore.as_retriever()

生成RAG提示:

from langchain_core.prompts import ChatPromptTemplate

RAG_PROMPT = """\
使用以下上下文回答用戶的查詢。如果你不能回答問題,請回答“我不知道”。

問題:
{question}

上下文:
{context}
"""

rag_prompt = ChatPromptTemplate.from_template(RAG_PROMPT)

實例化LLM:

from langchain_openai import ChatOpenAI

openai_chat_model = ChatOpenAI(model="gpt-3.5-turbo")

構(gòu)建LCEL RAG鏈:

from operator import itemgetter
from langchain.schema.output_parser import StrOutputParser
from langchain.schema.runnable import RunnablePassthrough

retrieval_augmented_generation_chain = (
{"context": itemgetter("question")
| retriever, "question": itemgetter("question")}
| RunnablePassthrough.assign(context=itemgetter("context"))
| {"response": rag_prompt | openai_chat_model, "context": itemgetter("context")}
)
#
retrieval_augmented_generation_chain

響應:

{
"response": "AIMessage(content='Retrieval Augmented Generation 是一種結(jié)合了深度學習技術(shù)和傳統(tǒng)檢索技術(shù)的文本生成范式。它通過顯式獲取知識以插件方式,引領(lǐng)了許多NLP任務的可擴展性和可能緩解文本生成的難度。它涉及從檢索到的人類編寫的參考資料中生成文本,而不是從頭開始生成。', response_metadata={'token_usage': {'completion_tokens': 73, 'prompt_tokens': 2186, 'total_tokens': 2259}, 'model_name': 'gpt-3.5-turbo', 'system_fingerprint': 'fp_b28b39ffa8', 'finish_reason': 'stop', 'logprobs': None})",
"context": "[Document(page_content='...'), Document(page_content='...'), Document(page_content='...')]"
}

構(gòu)建工具箱:

為了使智能體能夠回答各種問題并引入外部知識,通常會給其配備一個工具箱。

LangChain社區(qū)的存儲庫(https://github.com/langchain-ai/langchain/tree/master/libs/community/langchain_community/tools)中提供了眾多工具,這里選擇了幾種來使用,這樣可以更直觀地展示LangGraph的循環(huán)特性。

具體來說,將整合以下工具:

from langchain_community.tools.ddg_search import DuckDuckGoSearchRun
from langchain_community.tools.arxiv.tool import ArxivQueryRun
from langgraph.prebuilt import ToolExecutor
tool_belt = [
DuckDuckGoSearchRun(),
ArxivQueryRun()
]

tool_executor = ToolExecutor(tool_belt)

實例化OpenAI函數(shù)調(diào)用:

from langchain_core.utils.function_calling import convert_to_openai_function
#
model = ChatOpenAI(temperature=0)
#
functions = [convert_to_openai_function(t) for t in tool_belt]
model = model.bind_functions(functions)

使用LangGraph:

LangGraph通過StatefulGraph來實現(xiàn),這個結(jié)構(gòu)使用AgentState對象在圖的不同節(jié)點間傳遞信息。

雖然有很多配置選項,但核心是AgentState對象,它被存儲在一個TypedDict中。這個對象的鍵是“messages”,對應的值是一個BaseMessages的序列,每當狀態(tài)發(fā)生變化時,新的消息就會被添加到這個序列中。

from typing import TypedDict, Annotated, Sequence
import operator
from langchain_core.messages import BaseMessage

class AgentState(TypedDict):
messages: Annotated[Sequence[BaseMessage], operator.add]

構(gòu)建節(jié)點

call_model是一個節(jié)點,將...嗯...調(diào)用模型

call_tool是一個節(jié)點,將調(diào)用工具

```python
from langgraph.prebuilt import ToolInvocation
import json
from langchain_core.messages import FunctionMessage

def call_model(state):
messages = state["messages"]
response = model.invoke(messages)
return {"messages" : [response]}

def call_tool(state):
last_message = state["messages"][-1]

action = ToolInvocation(
tool=last_message.additional_kwargs["function_call"]["name"],
tool_input=json.loads(
last_message.additional_kwargs["function_call"]["arguments"]
)
)

response = tool_executor.invoke(action)

function_message = FunctionMessage(content=str(response), name=action.tool)

return {"messages" : [function_message]}

構(gòu)建工作流:

from langgraph.graph import StateGraph, END

workflow = StateGraph(AgentState)

workflow.add_node("agent", call_model)
workflow.add_node("action", call_tool)
workflow.nodes

設(shè)置入口點:

workflow.set_entry_point("agent")

構(gòu)建條件邊進行路由:

def should_continue(state):
last_message = state["messages"][-1]

if"function_call"notin last_message.additional_kwargs:
return"end"

return"continue"

workflow.add_conditional_edges(
"agent",
should_continue,
{
"continue" : "action",
"end" : END
}
)

最后將條件邊連接到智能體節(jié)點和動作節(jié)點:

workflow.add_edge("action", "agent")

編譯工作流:

app = workflow.compile()
#
app

調(diào)用LangGraph – 提問:

from langchain_core.messages import HumanMessage

inputs = {"messages" : [HumanMessage(content="在大型語言模型的背景下,RAG是什么?它是什么時候出現(xiàn)的?")]}
response = app.invoke(inputs)
print(response)

響應:

{
"messages": [
HumanMessage(content="在大型語言模型的背景下,RAG是什么?它是什么時候出現(xiàn)的?"),
AIMessage(content='', additional_kwargs={'function_call': {'arguments': '{"query":"RAG in the context of Large Language Models"}', 'name': 'duckduckgo_search'}}, response_metadata={'token_usage': {'completion_tokens': 25, 'prompt_tokens': 171, 'total_tokens': 196}, 'model_name': 'gpt-3.5-turbo', 'system_fingerprint': 'fp_b28b39ffa8', 'finish_reason': 'function_call', 'logprobs': None}),

FunctionMessage(content="大型語言模型(LLMs)是處理和生成文本的非常強大的工具。然而,它們天生難以理解更廣泛的信息背景,特別是在處理長篇對話或復雜任務時。這就是大型上下文窗口和檢索增強生成(RAG)發(fā)揮作用的地方。這些先進的、泛化的語言模型是在龐大的數(shù)據(jù)集上訓練的,使它們能夠理解和生成類似人類的文本。在RAG的背景下,LLMs被用來基于用戶查詢和從向量數(shù)據(jù)庫檢索的上下文信息生成完整的響應。在語言模型快速發(fā)展的背景下,檢索增強生成(RAG)和長上下文大型語言模型(LLMs)之間的辯論引起了廣泛關(guān)注。檢索增強生成(RAG)是一個AI框架,通過將模型基于外部知識源來補充LLM的內(nèi)部信息表示,從而提高LLM生成響應的質(zhì)量。在基于LLM的問答系統(tǒng)中實施RAG有兩個主要好處:它確保模型有... RAG代表檢索增強生成。RAG使大型語言模型(LLM)能夠訪問和利用最新的信息。因此,它提高了LLM的響應質(zhì)量和相關(guān)性。下面是一個簡單的RAG實現(xiàn)圖。", name='duckduckgo_search'),
AIMessage(content="在大型語言模型(LLMs)的背景下,RAG代表檢索增強生成。它是一個AI框架,通過將模型基于外部知識源來補充LLM的內(nèi)部信息表示,從而提高LLM生成響應的質(zhì)量。RAG使LLM能夠訪問和利用最新的信息,從而提高模型生成的響應的相關(guān)性和質(zhì)量。RAG作為增強LLM在理解和生成類似人類文本的能力的方式,在語言模型快速發(fā)展的領(lǐng)域中出現(xiàn)。", response_metadata={'token_usage': {'completion_tokens': 117, 'prompt_tokens': 491, 'total_tokens': 608}, 'model_name': 'gpt-3.5-turbo', 'system_fingerprint': 'fp_3bc1b5746c', 'finish_reason': 'stop', 'logprobs': None})
]
}

響應內(nèi)容:

RAG代表檢索增強生成,在大型語言模型(LLMs)的背景下。它是一個AI框架,通過將模型基于外部知識源來補充LLM的內(nèi)部信息表示,從而提高LLM生成響應的質(zhì)量。RAG使LLM能夠訪問和利用最新的信息,從而提高模型生成的響應的相關(guān)性和質(zhì)量。RAG作為增強LLM在理解和生成類似人類文本的能力的方式,在語言模型快速發(fā)展的領(lǐng)域中出現(xiàn)。

提問:

問題 = “檢索增強生成論文的主要作者是誰 – 他們參加了哪所大學?”

inputs = {“messages” : [HumanMessage(content=問題)]}

response = app.invoke(inputs) print(response[‘messages’][-1].content)

**響應:**

關(guān)于“檢索增強生成”論文的主要作者是Huayang Li。不幸的是,提供的摘要中沒有提到他們參加的大學。

**提問:**

問題 = "檢索增強生成論文的主要作者是誰?"

inputs = {"messages" : [HumanMessage(content=問題)]}

response = app.invoke(inputs)
print(response['messages'][-1].content)

響應:論文“檢索增強文本生成”的主要作者是Huayang Li、Yixuan Su、Deng Cai、Yan Wang和Lemao Liu。

本文章轉(zhuǎn)載微信公眾號@AI科技論談

上一篇:

LangChain的問題所在

下一篇:

使用LangChain和LangGraph大幅提升RAG效果
#你可能也喜歡這些API文章!

我們有何不同?

API服務商零注冊

多API并行試用

數(shù)據(jù)驅(qū)動選型,提升決策效率

查看全部API→
??

熱門場景實測,選對API

#AI文本生成大模型API

對比大模型API的內(nèi)容創(chuàng)意新穎性、情感共鳴力、商業(yè)轉(zhuǎn)化潛力

25個渠道
一鍵對比試用API 限時免費

#AI深度推理大模型API

對比大模型API的邏輯推理準確性、分析深度、可視化建議合理性

10個渠道
一鍵對比試用API 限時免費