昇腾社区首页
中文
注册

类功能

功能描述

知识图谱Chain,继承自SingleText2TextChain类

函数原型

from mx_rag.chain.single_text_to_text import GraphRagText2TextChain
GraphRagText2TextChain(llm, retriever, reranker)

输入参数说明

参数名

数据类型

是否必选

说明

llm

Text2TextLLM

必选

大模型对象,具体可参见Text2TextLLM类

retriever

GraphRetriever

必选

GraphRetriever对象,由GraphRAGPipelineas_retriever方法返回。

reranker

Reranker

可选

Reranker对象,实现对检索的文档进行精排,默认为“None”,具体可参见Reranker