PostgresML Embeddings
Spring AI 支持 PostgresML 文本嵌入模型。
嵌入是文本的数字表示。它们用于将单词和句子表示为向量(数字数组)。嵌入可以通过使用距离度量比较数字向量的相似性来查找相似的文本片段,或者它们可以用作其他机器学习模型的输入特征,因为大多数算法不能直接使用文本。
许多预先训练的 LLMs 可用于从 PostgresML 中的文本生成嵌入。您可以通过浏览在 Hugging Face 上的 所有可用模型,找到最佳解决方案。
添加存储库和 BOM
Spring AI 工件发布在 Spring Milestone
和 Snapshot
存储库中。请参阅存储库部分将这些存储库添加到您的构建系统中。
为了帮助进行依赖管理,Spring AI 提供了 BOM(物料清单),以确保在整个项目中使用一致的 Spring AI 版本。请参阅依赖管理部分将 Spring AI BOM 添加到您的构建系统。
自动配置(Auto-configuration)
Spring AI 为 PostgresML 嵌入客户端提供 Spring Boot 自动配置。要启用它,请将以下依赖项添加到项目的 Maven pom.xml
文件中:
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-starter-model-postgresml-embedding</artifactId>
</dependency>
或者,在你的 Gradle 构建文件 build.gradle
中添加:
dependencies {
implementation 'org.springframework.ai:spring-ai-starter-model-postgresml-embedding'
}
嵌入属性(Embedding Properties)
前缀 spring.ai.postgresml.embedding
是用于配置 PostgresML 嵌入的 EmbeddingModel
实现的属性前缀。
属性 | 描述 | 默认值 |
---|---|---|
spring.ai.postgresml.embedding.enabled (已移除且不再有效) | 启用 PostgresML 嵌入模型 | true |
spring.ai.model.embedding | 启用 PostgresML 嵌入模型 | postgresml |
spring.ai.postgresml.embedding.create-extension | 执行 SQL ‘CREATE EXTENSION IF NOT EXISTS pgml’ 来启用扩展 | false |
spring.ai.postgresml.embedding.options.transformer | 用于嵌入的 Hugging Face 转换器模型 | distilbert-base-uncased |
spring.ai.postgresml.embedding.options.kwargs | 转换器特定的额外选项 | 空 map |
spring.ai.postgresml.embedding.options.vectorType | 用于嵌入的 PostgresML 向量类型,支持两种选项:PG_ARRAY 和 PG_VECTOR |
PG_ARRAY |
spring.ai.postgresml.embedding.options.metadataMode | 文档元数据聚合模式 | EMBED |
运行时选项(Runtime Options )
PostgresMlEmbeddingOptions.java 提供了嵌入请求的配置信息,并提供了一个构建器来创建这些选项。
默认选项也可以通过配置 spring.ai.postgresml.embedding.options
属性来进行设置。
启动时,您可以将 PostgresMlEmbeddingOptions
传递给 PostgresMlEmbeddingModel
构造函数,以配置所有嵌入请求使用的默认选项。
在运行时,您可以通过在 EmbeddingRequest
中使用 PostgresMlEmbeddingOptions
实例来覆盖这些默认选项。
例如,要为特定请求覆盖默认模型名称:
EmbeddingResponse embeddingResponse = embeddingModel.call(
new EmbeddingRequest(List.of("Hello World", "World is big and salvation is near"),
PostgresMlEmbeddingOptions.builder()
.transformer("intfloat/e5-small")
.vectorType(VectorType.PG_ARRAY)
.kwargs(Map.of("device", "gpu"))
.build()));
示例控制器(Sample Controller)
创建一个新的 Spring Boot 项目,并将 spring-ai-starter-model-postgresml-embedding
添加到您的 pom(或 gradle)依赖项中。
在 src/main/resources
目录下添加一个 application.properties
文件,以启用和配置 VertexAi 聊天客户端:
spring.ai.postgresml.embedding.options.transformer=distilbert-base-uncased
spring.ai.postgresml.embedding.options.vectorType=PG_ARRAY
spring.ai.postgresml.embedding.options.metadataMode=EMBED
spring.ai.postgresml.embedding.options.kwargs.device=cpu
这将创建一个可以注入到您的类中的 PostgresMlEmbeddingModel
实现。下面是一个@Controller
使用 PostgresMlEmbeddingModel
的简单示例。
@RestController
public class EmbeddingController {
private final PostgresMlEmbeddingModel embeddingModel;
@Autowired
public EmbeddingController(PostgresMlEmbeddingModel embeddingModel) {
this.embeddingModel = embeddingModel;
}
@GetMapping("/v1/embedding")
public Map<String, Object> embedding(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
return Map.of("embeddings", embeddingModel.embed(message));
}
}
手动配置(Manual Configuration)
PostgresMlEmbeddingModel
实现了 EmbeddingModel
, 并使用轻量级 OpenAiApi 客户端连接到 OpenAI 服务。
要启用它,添加 spring-ai-postgresml
依赖到你的项目 Maven pom.xml
文件:
<dependency>
<groupId>org.springframework.ai</groupId>
<artifactId>spring-ai-postgresml</artifactId>
</dependency>
或者,在你的 Gradle 构建文件 build.gradle
中添加:
dependencies {
implementation 'org.springframework.ai:spring-ai-postgresml'
}
接下来,创建一个 PostgresMlEmbeddingModel
实例,并使用它来计算两个输入文本之间的相似度:
var jdbcTemplate = new JdbcTemplate(dataSource); // your posgresml data source
PostgresMlEmbeddingModel embeddingModel = new PostgresMlEmbeddingModel(this.jdbcTemplate,
PostgresMlEmbeddingOptions.builder()
.transformer("distilbert-base-uncased") // huggingface transformer model name.
.vectorType(VectorType.PG_VECTOR) //vector type in PostgreSQL.
.kwargs(Map.of("device", "cpu")) // optional arguments.
.metadataMode(MetadataMode.EMBED) // Document metadata mode.
.build());
embeddingModel.afterPropertiesSet(); // initialize the jdbc template and database.
EmbeddingResponse embeddingResponse = this.embeddingModel
.embedForResponse(List.of("Hello World", "World is big and salvation is near"));
注意:当手动创建时,您必须在设置属性之后、使用客户端之前调用 ‘afterPropertiesSet ()‘ 方法。更方便(且推荐)的做法是将 ‘PostgresMlEmbeddingModel‘
创建为一个 ‘@Bean‘。这样,您就不需要手动调用 ‘afterPropertiesSet ()‘ 方法了:
@Bean
public EmbeddingModel embeddingModel(JdbcTemplate jdbcTemplate) {
return new PostgresMlEmbeddingModel(jdbcTemplate,
PostgresMlEmbeddingOptions.builder()
....
.build());
}
最后编辑:Jeebiz 更新时间:2025-08-31 23:07