类功能
功能描述
知识图谱Chain,继承自SingleText2TextChain类
函数原型
from mx_rag.chain.single_text_to_text import GraphRagText2TextChain GraphRagText2TextChain(llm, retriever, reranker)
输入参数说明
参数名 |
数据类型 |
是否必选 |
说明 |
---|---|---|---|
llm |
Text2TextLLM |
必选 |
大模型对象,具体可参见Text2TextLLM类。 |
retriever |
GraphRetriever |
必选 |
GraphRetriever对象,由GraphRAGPipeline的as_retriever方法返回。 |
reranker |
Reranker |
可选 |
Reranker对象,实现对检索的文档进行精排,默认为“None”,具体可参见Reranker。 |