最新快照版本请使用Spring AI 1.1.0spring-doc.cadn.net.cn

Azure OpenAI Embeddings

Azure 的 OpenAI 扩展了 OpenAI 的能力,提供安全的文本生成和嵌入计算模型,适用于多种任务:spring-doc.cadn.net.cn

Azure OpenAI 嵌入依赖于余弦相似性用于计算文档与查询之间的相似性。spring-doc.cadn.net.cn

前提条件

Azure OpenAI 客户端提供三种连接选项:使用 Azure API 密钥,或使用 OpenAI API 密钥,或使用 Microsoft Entra ID。spring-doc.cadn.net.cn

Azure API Key & Endpoint

获取你的Azure OpenAI端点API密钥来自Azure Portal的Azure OpenAI服务部分。spring-doc.cadn.net.cn

Spring AI 定义了两个配置属性:spring-doc.cadn.net.cn

  1. Spring.ai.Azure.Openai.API-Key:将此设为API 密钥摘自Azure。spring-doc.cadn.net.cn

  2. Spring.ai.azure.openai.endpoint:将此设置为在Azure中配置模型时获得的端点URL。spring-doc.cadn.net.cn

你可以在你的application.propertiesapplication.yml文件:spring-doc.cadn.net.cn

spring.ai.azure.openai.api-key=<your-azure-api-key>
spring.ai.azure.openai.endpoint=<your-azure-endpoint-url>

如果你更喜欢用环境变量处理敏感信息,比如 API 密钥,可以在配置中使用 Spring 表达式语言(SpEL):spring-doc.cadn.net.cn

# In application.yml
spring:
  ai:
    azure:
      openai:
        api-key: ${AZURE_OPENAI_API_KEY}
        endpoint: ${AZURE_OPENAI_ENDPOINT}
# In your environment or .env file
export AZURE_OPENAI_API_KEY=<your-azure-openai-api-key>
export AZURE_OPENAI_ENDPOINT=<your-azure-endpoint-url>

OpenAI 密钥

要通过OpenAI服务(非Azure)认证,请提供OpenAI API密钥。 这会自动将端点设置为 api.openai.com/v1spring-doc.cadn.net.cn

使用此方法时,设Spring.ai.azure.openai.chat.options.deployment-name属性与你希望使用的OpenAI模型名称相符。spring-doc.cadn.net.cn

在你的应用配置中:spring-doc.cadn.net.cn

spring.ai.azure.openai.openai-api-key=<your-azure-openai-key>
spring.ai.azure.openai.chat.options.deployment-name=<openai-model-name>

使用环境变量配合 SpEL:spring-doc.cadn.net.cn

# In application.yml
spring:
  ai:
    azure:
      openai:
        openai-api-key: ${AZURE_OPENAI_API_KEY}
        chat:
          options:
            deployment-name: ${OPENAI_MODEL_NAME}
# In your environment or .env file
export AZURE_OPENAI_API_KEY=<your-openai-key>
export OPENAI_MODEL_NAME=<openai-model-name>

Microsoft Entra ID

对于使用 Microsoft Entra ID(前称 Azure Active Directory)进行无密钥认证,Spring.ai.azure.openai.endpoint配置属性,而不是上述提到的API-key属性。spring-doc.cadn.net.cn

仅找到端点属性后,你的应用程序将评估多种获取凭证和OpenAIClient实例将使用Tokens凭证创建。spring-doc.cadn.net.cn

不再需要创建一个Tokens凭证豆;它是自动为你配置的。

添加仓库和物料清单

Spring AI 产物发布于 Maven Central 和 Spring Snapshot 仓库中。 请参阅神器仓库部分,将这些仓库添加到你的构建系统中。spring-doc.cadn.net.cn

为帮助依赖管理,Spring AI 提供了物料清单(BOM),确保整个项目中使用一致版本的 Spring AI。请参考依赖管理部分,将春季AI物料清单添加到你的构建系统中。spring-doc.cadn.net.cn

自动配置

春季AI自动配置、起始模块的工件名称发生了重大变化。 更多信息请参阅升级说明spring-doc.cadn.net.cn

Spring AI 为 Azure OpenAI 嵌入模型提供 Spring Boot 自动配置。 要启用它,请在项目的Maven中添加以下依赖pom.xml文件:spring-doc.cadn.net.cn

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-starter-model-azure-openai</artifactId>
</dependency>

或者去你的Gradlebuild.gradle构建文件。spring-doc.cadn.net.cn

dependencies {
    implementation 'org.springframework.ai:spring-ai-starter-model-azure-openai'
}
请参考依赖管理部分,将Spring AI的物料清单添加到你的构建文件中。

嵌入性质

前缀Spring.ai.azure.openai是配置与Azure OpenAI连接的属性前缀。spring-doc.cadn.net.cn

属性 描述 默认值

Spring.ai.Azure.Openai.API-Keyspring-doc.cadn.net.cn

Azure AI OpenAI 的密钥密钥与端点部分资源管理spring-doc.cadn.net.cn

-spring-doc.cadn.net.cn

Spring.ai.azure.openai.endpointspring-doc.cadn.net.cn

Azure AI OpenAI 的端点密钥与端点部分资源管理spring-doc.cadn.net.cn

-spring-doc.cadn.net.cn

Spring.ai.azure.openai.openai-api-keyspring-doc.cadn.net.cn

(非青色)OpenAI API 密钥。用来通过OpenAI服务认证,而不是Azure OpenAI。 这会自动将端点设置为 api.openai.com/v1。使用任一API密钥OpenAI-API-Key财产。 在这种配置下Spring.ai.azure.openai.embedding.options.deployment-name被视为 OpenAi 模型名称。spring-doc.cadn.net.cn

-spring-doc.cadn.net.cn

嵌入自动配置的启用和禁用现在通过带有前缀的顶层属性配置spring.ai.model.embedding.spring-doc.cadn.net.cn

启用 spring.ai.model.embedding=azure-openai(默认启用)spring-doc.cadn.net.cn

要禁用,请选择 spring.ai.model.embedding=none(或任何不匹配 azure-openai 的值)spring-doc.cadn.net.cn

此改动旨在允许配置多个模型。spring-doc.cadn.net.cn

前缀spring.ai.azure.openai.embedding是配置嵌入模型implementation for Azure OpenAIspring-doc.cadn.net.cn

属性 描述 默认值

spring.ai.azure.openai.embedding.enabled(已移除且不再有效)spring-doc.cadn.net.cn

启用Azure OpenAI嵌入模型。spring-doc.cadn.net.cn

truespring-doc.cadn.net.cn

spring.ai.model.embeddingspring-doc.cadn.net.cn

启用Azure OpenAI嵌入模型。spring-doc.cadn.net.cn

azure-openaispring-doc.cadn.net.cn

Spring.ai.azure.openai.embedding.metadata-modespring-doc.cadn.net.cn

文档内容提取模式spring-doc.cadn.net.cn

嵌入spring-doc.cadn.net.cn

Spring.ai.azure.openai.embedding.options.deployment-namespring-doc.cadn.net.cn

这是Azure AI门户中显示的“部署名称”的数值spring-doc.cadn.net.cn

文本嵌入-ada-002spring-doc.cadn.net.cn

spring.ai.azure.openai.embedding.options.userspring-doc.cadn.net.cn

作的调用者或终端用户的标识符。这可能用于跟踪或限速目的。spring-doc.cadn.net.cn

-spring-doc.cadn.net.cn

所有以spring.ai.azure.openai.embedding.options可以通过在运行时为 在嵌入请求叫。

运行时选项

AzureOpenAiEmbeddingOptions提供嵌入请求的配置信息。 这AzureOpenAiEmbeddingOptions提供建造者来创建选项。spring-doc.cadn.net.cn

开始时使用该AzureOpenAiEmbeddingModel构建器(constructor)来设置所有嵌入请求的默认选项。 运行时你可以通过传递AzureOpenAiEmbeddingOptions实例中,你对嵌入请求请求。spring-doc.cadn.net.cn

例如,要覆盖特定请求的默认型号名称:spring-doc.cadn.net.cn

EmbeddingResponse embeddingResponse = embeddingModel.call(
    new EmbeddingRequest(List.of("Hello World", "World is big and salvation is near"),
        AzureOpenAiEmbeddingOptions.builder()
        .model("Different-Embedding-Model-Deployment-Name")
        .build()));

示例代码

这将产生嵌入模型你可以把这些实现注入到你的类里。 这里有一个简单的例子@Controller使用嵌入模型实现。spring-doc.cadn.net.cn

spring.ai.azure.openai.api-key=YOUR_API_KEY
spring.ai.azure.openai.endpoint=YOUR_ENDPOINT
spring.ai.azure.openai.embedding.options.model=text-embedding-ada-002
@RestController
public class EmbeddingController {

    private final EmbeddingModel embeddingModel;

    @Autowired
    public EmbeddingController(EmbeddingModel embeddingModel) {
        this.embeddingModel = embeddingModel;
    }

    @GetMapping("/ai/embedding")
    public Map embed(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        EmbeddingResponse embeddingResponse = this.embeddingModel.embedForResponse(List.of(message));
        return Map.of("embedding", embeddingResponse);
    }
}

手动配置

如果你不想用 Spring Boot 的自动配置,可以手动配置AzureOpenAiEmbeddingModel在你的申请中。 为此添加Spring-ai-azure-openai对你项目Maven的依赖pom.xml文件:spring-doc.cadn.net.cn

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-azure-openai</artifactId>
</dependency>

或者去你的Gradlebuild.gradle构建文件。spring-doc.cadn.net.cn

dependencies {
    implementation 'org.springframework.ai:spring-ai-azure-openai'
}
请参考依赖管理部分,将Spring AI的物料清单添加到你的构建文件中。
Spring-ai-azure-openai依赖关系还提供了访问AzureOpenAiEmbeddingModel.欲了解更多关于AzureOpenAiChatModel请参阅Azure OpenAI嵌入部分。

接下来,创建一个AzureOpenAiEmbeddingModel实例并用它来计算两个输入文本之间的相似性:spring-doc.cadn.net.cn

var openAIClient = OpenAIClientBuilder()
        .credential(new AzureKeyCredential(System.getenv("AZURE_OPENAI_API_KEY")))
		.endpoint(System.getenv("AZURE_OPENAI_ENDPOINT"))
		.buildClient();

var embeddingModel = new AzureOpenAiEmbeddingModel(this.openAIClient)
    .withDefaultOptions(AzureOpenAiEmbeddingOptions.builder()
        .model("text-embedding-ada-002")
        .user("user-6")
        .build());

EmbeddingResponse embeddingResponse = this.embeddingModel
	.embedForResponse(List.of("Hello World", "World is big and salvation is near"));
文本嵌入-ada-002实际上是部署名称如Azure AI门户所示。