BTh^ddlmZddlmZddlZddlmZddlmZddl m Z Gdde Z y) )abstractmethod)TypeN)Document) BaseRetriever)BaseStandardTestsceZdZdZeedeefdZede fdZ eede fdZ e jdefdZd dZd eddfd Zd eddfd Zd eddfd Zy)RetrieversIntegrationTestsz6 Base class for retrievers integration tests. returncy)z8 A BaseRetriever subclass to be tested. Nselfs ~/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/langchain_tests/integration_tests/retrievers.pyretriever_constructorz0RetrieversIntegrationTests.retriever_constructor ciS)zZ Returns a dictionary of parameters to pass to the retriever constructor. r r s rretriever_constructor_paramsz7RetrieversIntegrationTests.retriever_constructor_paramss  rcy)zV Returns a str representing the "query" of an example retriever call. Nr r s rretriever_query_examplez2RetrieversIntegrationTests.retriever_query_examplerrc:|jdi|jS)z :private: r )rrr s r retrieverz$RetrieversIntegrationTests.retriever's *t))ND,M,MNNrNc|jjDcic]\}}|dk7s ||}}}i|ddi}|jdi|}|j|j}t |dk(sJt d|DsJi|ddi}|jdi|}|j|j} t | dk(sJt d| DsJycc}}w)aB Test that the retriever constructor accepts a k parameter, representing the number of documents to return. .. dropdown:: Troubleshooting If this test fails, either the retriever constructor does not accept a k parameter, or the retriever does not return the correct number of documents (`k`) when it is set. For example, a retriever like .. code-block:: python MyRetriever(k=3).invoke("query") should return 3 documents when invoked with a query. kc3<K|]}t|tywN isinstancer.0docs r zFRetrieversIntegrationTests.test_k_constructor_param..HA:c8,Ac3<K|]}t|tywrrr s rr#zFRetrieversIntegrationTests.test_k_constructor_param..Nr$r%Nr )ritemsrinvokerlenall) rrvparamsparams_3 retriever_3result_3params_1 retriever_1result_1s rtest_k_constructor_paramz3RetrieversIntegrationTests.test_k_constructor_param.s(">>DDF Q!s(AqD  &f%c1%0d00<8< %%d&B&BC8}!!!AAAAA%f%c1%0d00<8< %%d&B&BC8}!!!AAAAA s C'C'rc|j|jd}t|dk(sJtd|DsJ|j|jd}t|dk(sJtd|DsJy)a5 Test that the invoke method accepts a k parameter, representing the number of documents to return. .. dropdown:: Troubleshooting If this test fails, the retriever's invoke method does not accept a k parameter, or the retriever does not return the correct number of documents (`k`) when it is set. For example, a retriever like .. code-block:: python MyRetriever().invoke("query", k=3) should return 3 documents when invoked with a query. r&)rc3<K|]}t|tywrrr s rr#zFRetrieversIntegrationTests.test_invoke_with_k_kwarg..er$r%rc3<K|]}t|tywrrr s rr#zFRetrieversIntegrationTests.test_invoke_with_k_kwarg..ir$r%N)r)rr*r+)rrr3r0s rtest_invoke_with_k_kwargz3RetrieversIntegrationTests.test_invoke_with_k_kwargPs&##D$@$@A#F8}!!!AAAAA##D$@$@A#F8}!!!AAAAArc|j|j}t|tsJt d|DsJy)a If invoked with the example params, the retriever should return a list of Documents. .. dropdown:: Troubleshooting If this test fails, the retriever's invoke method does not return a list of `langchain_core.document.Document` objects. Please confirm that your `_get_relevant_documents` method returns a list of `Document` objects. c3<K|]}t|tywrrr s rr#zKRetrieversIntegrationTests.test_invoke_returns_documents..y?:c8,?r%N)r)rrlistr+rrresults rtest_invoke_returns_documentsz8RetrieversIntegrationTests.test_invoke_returns_documentsks>!!$">">?&$'''?????rcK|j|jd{}t|tsJt d|DsJy7+w)z If ainvoked with the example params, the retriever should return a list of Documents. See :meth:`test_invoke_returns_documents` for more information on troubleshooting. Nc3<K|]}t|tywrrr s rr#zLRetrieversIntegrationTests.test_ainvoke_returns_documents..r;r%)ainvokerrr<r+r=s rtest_ainvoke_returns_documentsz9RetrieversIntegrationTests.test_ainvoke_returns_documents{sK!(()E)EFF&$'''?????GsAA ,A)r N)__name__ __module__ __qualname____doc__propertyrrrrdictrstrrpytestfixturerr4r8r?rCr rrr r s tM':  d     ^^O=OO BDB-BDB6@}@@ @m @PT @rr ) abcrtypingrrKlangchain_core.documentsrlangchain_core.retrieversrlangchain_tests.baserr r rrrRs( -32{@!2{@r