检索器集成测试

class langchain_tests.integration_tests.retrievers.RetrieversIntegrationTests

检索器集成测试的基类。

属性

retriever_constructor	要测试的BaseRetriever子类。
retriever_constructor_params	返回一个字典，包含传递给检索器构造函数的参数。
retriever_query_example	返回一个表示示例检索器调用的“查询”的字符串。

方法

test_ainvoke_returns_documents(retriever)	如果使用示例参数调用，检索器应返回一个文档列表。
test_invoke_returns_documents(retriever)	如果使用示例参数调用，检索器应返回一个文档列表。
test_invoke_with_k_kwarg(retriever)	测试invoke方法是否接受一个k参数，表示要返回的文档数量。
test_k_constructor_param()	测试检索器构造函数是否接受一个k参数，表示要返回的文档数量。

async test_ainvoke_returns_documents(retriever: BaseRetriever) → None

如果使用示例参数调用，检索器应返回一个文档列表。

请参阅test_invoke_returns_documents()以获取更多关于故障排除的信息。

Parameters:

retriever (BaseRetriever)

Return type:

无

test_invoke_returns_documents(retriever: BaseRetriever) → None

如果使用示例参数调用，检索器应返回一个文档列表。

Troubleshooting

如果此测试失败，检索器的invoke方法不会返回langchain_core.document.Document对象的列表。请确认您的_get_relevant_documents方法返回的是Document对象的列表。

Parameters:

retriever (BaseRetriever)

Return type:

无

test_invoke_with_k_kwarg(retriever: BaseRetriever) → None

测试invoke方法是否接受一个k参数，表示要返回的文档数量。

Troubleshooting

如果此测试失败，检索器的invoke方法不接受k参数，或者当设置时检索器未返回正确数量的文档（k）。

例如，一个像

MyRetriever().invoke("query", k=3)

当使用查询调用时，应返回3个文档。

Parameters:

retriever (BaseRetriever)

Return type:

无

test_k_constructor_param() → None

测试检索器构造函数是否接受一个k参数，该参数表示要返回的文档数量。

Troubleshooting

如果此测试失败，可能是检索器构造函数不接受k参数，或者检索器在设置时未返回正确数量的文档（k）。

例如，一个像

MyRetriever(k=3).invoke("query")

当使用查询调用时，应返回3个文档。

Return type:

无