test_init_embedding_model() — langchain Function Reference

Architecture documentation for the test_init_embedding_model() function in test_base.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_init_embedding_model() — langchain Function Reference

Dependency Diagram

graph TD
  dcc06579_4baa_f4cb_fea9_72f949182b01["test_init_embedding_model()"]
  87acdd67_cae1_c571_1563_8fd925f72551["test_base.py"]
  dcc06579_4baa_f4cb_fea9_72f949182b01 -->|defined in| 87acdd67_cae1_c571_1563_8fd925f72551
  style dcc06579_4baa_f4cb_fea9_72f949182b01 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/tests/integration_tests/embeddings/test_base.py lines 20–44

async def test_init_embedding_model(provider: str, model: str) -> None:
    package = _SUPPORTED_PROVIDERS[provider]
    try:
        importlib.import_module(package)
    except ImportError:
        pytest.skip(f"Package {package} is not installed")

    model_colon = init_embeddings(f"{provider}:{model}")
    assert isinstance(model_colon, Embeddings)

    model_explicit = init_embeddings(
        model=model,
        provider=provider,
    )
    assert isinstance(model_explicit, Embeddings)

    text = "Hello world"

    embedding_colon = await model_colon.aembed_query(text)
    assert isinstance(embedding_colon, list)
    assert all(isinstance(x, float) for x in embedding_colon)

    embedding_explicit = await model_explicit.aembed_query(text)
    assert isinstance(embedding_explicit, list)
    assert all(isinstance(x, float) for x in embedding_explicit)