langchain_community.embeddings.mlflow_gateway.MlflowAIGatewayEmbeddings

class langchain_community.embeddings.mlflow_gateway.MlflowAIGatewayEmbeddings

基础：Embeddings，BaseModel

MLflow AI Gateway 嵌入。

要使用，您需要安装 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: Optional[str] = None

MLflow AI Gateway API 的 URI。

param route: str [Required]

用于 MLflow AI Gateway API 的路由。

async aembed_documents(texts: List[str]) → List[List[float]]

异步嵌入文档搜索。

参数

texts (列表[str]) – 要嵌入的文本列表。

返回

嵌入列表。

返回类型

列表[列表[float]]

async aembed_query(text: str) → List[float]

异步嵌入查询文本。

参数

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

返回

嵌入。

返回类型

列表[float]

embed_documents(texts: List[str]) → List[List[float]]

嵌入文档搜索。

参数

texts (列表[str]) – 要嵌入的文本列表。

返回

嵌入列表。

返回类型

列表[列表[float]]

embed_query(text: str) → List[float]

嵌入查询文本。

参数

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

返回

嵌入。

返回类型

列表[float]

使用MlflowAIGatewayEmbeddings的示例

• MLflow AI网关