嵌入集成测试#

class langchain_tests.integration_tests.embeddings.EmbeddingsIntegrationTests[source]#

嵌入集成测试的基类。

测试子类必须实现embeddings_class属性以指定要测试的嵌入模型。你也可以重写embedding_model_params属性以指定初始化参数。

示例：

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"}

注意

个别测试方法的API参考包括故障排除提示。

属性

embedding_model_params

embeddings_class

方法

model()

test_aembed_documents(model)

测试异步嵌入字符串列表。

test_aembed_query(model)

测试异步嵌入字符串查询。

test_embed_documents(model)

测试嵌入字符串列表。

test_embed_query(model)

测试嵌入字符串查询。
model() → Embeddings#
Return type:

Embeddings

async test_aembed_documents(model: Embeddings) → None[source]#

测试异步嵌入字符串列表。

Troubleshooting

如果此测试失败，请检查：

  1. 当在字符串列表上调用 .aembed_documents 时，模型将生成一个浮点数列表的列表。

  2. 每个列表的长度相同。

Parameters:

模型 (Embeddings)

Return type:

无

async test_aembed_query(model: Embeddings) → None[source]#

测试嵌入一个字符串查询异步。

Troubleshooting

如果此测试失败，请检查：

  1. 模型将在字符串上调用.aembed_query时生成一个浮点数列表。

  2. 列表的长度在不同的输入中是一致的。

Parameters:

模型 (Embeddings)

Return type:

无

test_embed_documents(model: Embeddings) → None[source]#

测试嵌入字符串列表。

Troubleshooting

如果此测试失败，请检查：

  1. 模型将在字符串列表上调用.embed_documents时生成一个浮点数列表的列表。

  2. 每个列表的长度相同。

Parameters:

模型 (Embeddings)

Return type:

无

test_embed_query(model: Embeddings) → None[source]#

测试嵌入字符串查询。

Troubleshooting

如果此测试失败，请检查：

  1. 模型在调用字符串上的.embed_query时，将生成一个浮点数列表。

  2. 列表的长度在不同的输入中是一致的。

Parameters:

模型 (Embeddings)

Return type:

无