Google VertexAI Multimodal Embeddings

Vertex AI 支持两种类型的嵌入模型:文本嵌入多模态嵌入。本文档将介绍如何通过 Vertex AI 的 多模态嵌入 API 创建多模态嵌入。

多模态嵌入模型根据您提供的输入生成 1408 维向量,这些输入可以包括图像、文本和视频数据的组合。生成的嵌入向量随后可用于图像分类或视频内容审核等后续任务。

图像嵌入向量和文本嵌入向量位于同一语义空间,且具有相同的维度。因此,这些向量可以在诸如通过文本搜索图像或通过图像搜索视频等应用场景中互换使用。

前提条件(Prerequisites)

  • 安装适用于您操作系统的 gcloud CLI。
  • 通过运行以下命令进行身份验证。将 PROJECT_ID 替换为您的 Google Cloud 项目 ID,将 ACCOUNT 替换为您的 Google Cloud 用户名
gcloud config set project <PROJECT_ID> &&
gcloud auth application-default login <ACCOUNT>

添加存储库和 BOM

Spring AI 工件发布在 Spring MilestoneSnapshot 存储库中。请参阅存储库部分将这些存储库添加到您的构建系统中。

为了帮助进行依赖管理,Spring AI 提供了 BOM(物料清单),以确保在整个项目中使用一致的 Spring AI 版本。请参阅依赖管理部分将 Spring AI BOM 添加到您的构建系统。

自动配置(Auto-configuration)

Spring AI 为 VertexAI 嵌入模型提供了 Spring Boot 自动配置。要启用此功能,请将以下依赖项添加到项目的 Maven pom.xml文件中:

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-starter-model-vertex-ai-embedding</artifactId>
</dependency>

或者,在你的 Gradle 构建文件 build.gradle 中添加:

dependencies {
    implementation 'org.springframework.ai:spring-ai-starter-model-vertex-ai-embedding'
}

嵌入属性(Embedding Properties)

前缀 spring.ai.vertex.ai.embedding 用作属性前缀,使您能够连接到 VertexAI 嵌入 API。

属性 描述 默认值
spring.ai.vertex.ai.embedding.project-id Google 云平台项目ID -
spring.ai.vertex.ai.embedding.location 区域 -
spring.ai.vertex.ai.embedding.apiEndpoint Vertex AI 嵌入API端点 -

前缀 spring.ai.vertex.ai.embedding.multimodal 是用于配置 VertexAI 多模态嵌入模型实现的属性前缀。

属性 描述 默认值
spring.ai.vertex.ai.embedding.multimodal.enabled (已移除且不再有效) 启用Vertex AI多模态嵌入API模型 true
spring.ai.model.embedding.multimodal=vertexai 启用Vertex AI多模态嵌入API模型 vertexai
spring.ai.vertex.ai.embedding.multimodal.options.model 用于获取多模态嵌入的模型 multimodalembedding@001
spring.ai.vertex.ai.embedding.multimodal.options.dimensions 指定低维度嵌入向量。默认返回1408维浮点向量,也可为文本和图像数据指定更低维度(128/256/512维) 1408
spring.ai.vertex.ai.embedding.multimodal.options.video-start-offset-sec 视频片段的起始偏移时间(秒)。未指定时按max(0, endOffsetSec - 120)计算 -
spring.ai.vertex.ai.embedding.multimodal.options.video-end-offset-sec 视频片段的结束偏移时间(秒)。未指定时按min(视频长度, startOffSec + 120)计算。若同时指定起止时间,则调整为min(startOffsetSec+120, endOffsetSec) -
spring.ai.vertex.ai.embedding.multimodal.options.video-interval-sec 生成嵌入的视频采样间隔(秒)。最小间隔4秒,小于4秒将返回InvalidArgumentError。最大间隔无限制,但若超过min(视频长度,120秒)会影响嵌入质量 16

手动配置(Manual Configuration)

VertexAiMultimodalEmbeddingModel 实现了 DocumentEmbeddingModel

要启用它,添加 spring-ai-vertex-ai-embedding 依赖到你的项目 Maven pom.xml 文件:

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-vertex-ai-embedding</artifactId>
</dependency>

或者,在你的 Gradle 构建文件 build.gradle 中添加:

dependencies {
    implementation 'org.springframework.ai:spring-ai-vertex-ai-embedding'
}

接下来,创建一个 VertexAiMultimodalEmbeddingModel 并用于生成嵌入:

VertexAiEmbeddingConnectionDetails connectionDetails =
    VertexAiEmbeddingConnectionDetails.builder()
        .projectId(System.getenv(<VERTEX_AI_GEMINI_PROJECT_ID>))
        .location(System.getenv(<VERTEX_AI_GEMINI_LOCATION>))
        .build();

VertexAiMultimodalEmbeddingOptions options = VertexAiMultimodalEmbeddingOptions.builder()
    .model(VertexAiMultimodalEmbeddingOptions.DEFAULT_MODEL_NAME)
    .build();

var embeddingModel = new VertexAiMultimodalEmbeddingModel(this.connectionDetails, this.options);

Media imageMedial = new Media(MimeTypeUtils.IMAGE_PNG, new ClassPathResource("/test.image.png"));
Media videoMedial = new Media(new MimeType("video", "mp4"), new ClassPathResource("/test.video.mp4"));

var document = new Document("Explain what do you see on this video?", List.of(this.imageMedial, this.videoMedial), Map.of());

EmbeddingResponse embeddingResponse = this.embeddingModel
    .embedForResponse(List.of("Hello World", "World is big and salvation is near"));

DocumentEmbeddingRequest embeddingRequest = new DocumentEmbeddingRequest(List.of(this.document),
        EmbeddingOptions.EMPTY);

EmbeddingResponse embeddingResponse = multiModelEmbeddingModel.call(this.embeddingRequest);

assertThat(embeddingResponse.getResults()).hasSize(3);
作者:Jeebiz  创建时间:2025-08-08 00:50
最后编辑:Jeebiz  更新时间:2025-08-31 23:07