MlflowAIGatewayEmbeddings#

class langchain_community.embeddings.mlflow_gateway.MlflowAIGatewayEmbeddings[source]#

基础类: Embeddings, BaseModel

MLflow AI 网关嵌入。

要使用,您应该安装mlflow[gateway] python包。 更多信息,请参见https://mlflow.org/docs/latest/gateway/index.html

示例

from langchain_community.embeddings import MlflowAIGatewayEmbeddings

embeddings = MlflowAIGatewayEmbeddings(
    gateway_uri="<your-mlflow-ai-gateway-uri>",
    route="<your-mlflow-ai-gateway-embeddings-route>"
)
param gateway_uri: str | None = None#

MLflow AI Gateway API 的 URI。

param route: str [Required]#

用于MLflow AI Gateway API的路由。

async aembed_documents(texts: list[str]) list[list[float]]#

异步嵌入搜索文档。

Parameters:

文本 (列表[字符串]) – 要嵌入的文本列表。

Returns:

嵌入列表。

Return type:

列表[列表[浮点数]]

async aembed_query(text: str) list[float]#

异步嵌入查询文本。

Parameters:

文本 (str) – 要嵌入的文本。

Returns:

嵌入。

Return type:

列表[浮点数]

embed_documents(texts: List[str]) List[List[float]][source]#

嵌入搜索文档。

Parameters:

文本 (列表[字符串]) – 要嵌入的文本列表。

Returns:

嵌入列表。

Return type:

列表[列表[float]]

embed_query(text: str) List[float][source]#

嵌入查询文本。

Parameters:

文本 (str) – 要嵌入的文本。

Returns:

嵌入。

Return type:

列表[float]

使用MlflowAIGatewayEmbeddings的示例