BThddlZddlmZddlmZmZddlmZddlZddl m Z ddl m Z ddl mZGdd eZGd d eZy) N)abstractmethod)TupleType)mock) Embeddings) SecretStr)BaseStandardTestscveZdZdZeedeefdZede fdZ e jdefdZ y)EmbeddingsTestsz :private: returncyNselfs w/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/langchain_tests/unit_tests/embeddings.pyembeddings_classz EmbeddingsTests.embeddings_classs47ciSrrrs rembedding_model_paramsz&EmbeddingsTests.embedding_model_paramss rc:|jdi|jS)Nrrrrs rmodelzEmbeddingsTests.models$t$$Ct'B'BCCrN)__name__ __module__ __qualname____doc__propertyrrrrdictrpytestfixturerrrrr r sd7$z"277  ^^DzDDrr cBeZdZdZddZedeeeeffdZddZ y)EmbeddingsUnitTestsaBase class for embeddings unit tests. Test subclasses must implement the ``embeddings_class`` property to specify the embeddings model to be tested. You can also override the ``embedding_model_params`` property to specify initialization parameters. Example: .. code-block:: python from typing import Type from langchain_tests.unit_tests import EmbeddingsUnitTests from my_package.embeddings import MyEmbeddingsModel class TestMyEmbeddingsModelUnit(EmbeddingsUnitTests): @property def embeddings_class(self) -> Type[MyEmbeddingsModel]: # Return the embeddings model class to test here return MyEmbeddingsModel @property def embedding_model_params(self) -> dict: # Return initialization parameters for the model. return {"model": "model-001"} .. note:: API references for individual test methods include troubleshooting tips. Testing initialization from environment variables Overriding the ``init_from_env_params`` property will enable additional tests for initialization from environment variables. See below for details. .. dropdown:: init_from_env_params This property is used in unit tests to test initialization from environment variables. It should return a tuple of three dictionaries that specify the environment variables, additional initialization args, and expected instance attributes to check. Defaults to empty dicts. If not overridden, the test is skipped. Example: .. code-block:: python @property def init_from_env_params(self) -> Tuple[dict, dict, dict]: return ( { "MY_API_KEY": "api_key", }, { "model": "model-001", }, { "my_api_key": "api_key", }, ) r NcD|jdi|j}|Jy)zTest model initialization. .. dropdown:: Troubleshooting If this test fails, ensure that ``embedding_model_params`` is specified and the model can be initialized from those params. Nrr)rrs r test_initzEmbeddingsUnitTests.test_init^s+&%%D(C(CD   rc iiifS)aThis property is used in unit tests to test initialization from environment variables. It should return a tuple of three dictionaries that specify the environment variables, additional initialization args, and expected instance attributes to check.rrs rinit_from_env_paramsz(EmbeddingsUnitTests.init_from_env_paramsis 2rzrcr|j\}}}|rtjjtj |5|j di|}dddJ|jD]8\}}t||}t|tr|j}||k(r8Jyy#1swYZxYw)aTest initialization from environment variables. Relies on the ``init_from_env_params`` property. Test is skipped if that property is not set. .. dropdown:: Troubleshooting If this test fails, ensure that ``init_from_env_params`` is specified correctly and that model parameters are properly set from environment variables during initialization. Nr) r'rpatchrosenvironritemsgetattr isinstancerget_secret_value)r env_paramsembeddings_paramsexpected_attrsrkexpectedactuals rtest_init_from_envz&EmbeddingsUnitTests.test_init_from_envqs9=8Q8Q5 %~ Z8 C---B0AB C$ $$-335 * 8 *fi0#446F)))  *  C Cs B--B6)r N) rrrrr%rrrr'r6rrrr#r#s9<| !eD$,<&=*rr#)r*abcrtypingrrunittestrr langchain_core.embeddingsrpydanticrlangchain_tests.baser r r#rrrr=s:  02D'D$f*/f*r