BTh>ddlmZddlmZddlmZGddeZy))List) Embeddings)EmbeddingsTestscPeZdZdZdeddfdZdeddfdZdeddfdZdeddfdZy) EmbeddingsIntegrationTestsaBase class for embeddings integration 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.integration_tests import EmbeddingsIntegrationTests from my_package.embeddings import MyEmbeddingsModel class TestMyEmbeddingsModelIntegration(EmbeddingsIntegrationTests): @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. modelreturnNc|jd}t|tsJt|dtsJ|jd}t |dkDsJt |t |k(sJy)a,Test embedding a string query. .. dropdown:: Troubleshooting If this test fails, check that: 1. The model will generate a list of floats when calling ``.embed_query`` on a string. 2. The length of the list is consistent across different inputs. foorbarN) embed_query isinstancerfloatlenselfr embedding_1 embedding_2s ~/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/langchain_tests/integration_tests/embeddings.pytest_embed_queryz+EmbeddingsIntegrationTests.test_embed_query(ss''. +t,,,+a.%000''. ;!###;3{#3333cgd}|j|tt|k(sJtdDsJtdDsJtddkDsJtfdDsJy)a,Test embedding a list of strings. .. dropdown:: Troubleshooting If this test fails, check that: 1. The model will generate a list of lists of floats when calling ``.embed_documents`` on a list of strings. 2. The length of each list is the same. r r bazc3<K|]}t|tywNrr.0 embeddings r zBEmbeddingsIntegrationTests.test_embed_documents..JK9:i.Kc3BK|]}t|dtywrNrrrs rr!zBEmbeddingsIntegrationTests.test_embed_documents..KOy:ilE2Orc3RK|]}t|tdk( ywr%rrr embeddingss rr!zBEmbeddingsIntegrationTests.test_embed_documents..M#TI3y>SA%77T$'N)embed_documentsrallrr documentsr,s @rtest_embed_documentsz/EmbeddingsIntegrationTests.test_embed_documents<s* **95 :#i.000K KKKKOJOOOO:a=!A%%%TTTTTrcK|jdd{}t|tsJt|dtsJ|jdd{}t |dkDsJt |t |k(sJy7n70w)a3Test embedding a string query async. .. dropdown:: Troubleshooting If this test fails, check that: 1. The model will generate a list of floats when calling ``.aembed_query`` on a string. 2. The length of the list is consistent across different inputs. r Nrr ) aembed_queryrrrrrs rtest_aembed_queryz,EmbeddingsIntegrationTests.test_aembed_queryOs"..u55 +t,,,+a.%000!..u55 ;!###;3{#33336 6s!B B?B B/B B c Kgd}|j|d{tt|k(sJtdDsJtdDsJtddkDsJtfdDsJy7ow)a3Test embedding a list of strings async. .. dropdown:: Troubleshooting If this test fails, check that: 1. The model will generate a list of lists of floats when calling ``.aembed_documents`` on a list of strings. 2. The length of each list is the same. rNc3<K|]}t|tywrrrs rr!zCEmbeddingsIntegrationTests.test_aembed_documents..qr"r#c3BK|]}t|dtywr%r&rs rr!zCEmbeddingsIntegrationTests.test_aembed_documents..rr'r(rc3RK|]}t|tdk( ywr%r*r+s rr!zCEmbeddingsIntegrationTests.test_aembed_documents..tr-r.)aembed_documentsrr0r1s @rtest_aembed_documentsz0EmbeddingsIntegrationTests.test_aembed_documentscs*  11)<< :#i.000K KKKKOJOOOO:a=!A%%%TTTTT =sBB A0B) __name__ __module__ __qualname____doc__rrr3r6r<rrrrsY>4j4T4(U*UU&4Z4D4(UUUrrN)typingrlangchain_core.embeddingsr%langchain_tests.unit_tests.embeddingsrrrArrrEs0AlUlUr