Spring AI 使用 OCI GenAI Cohere

OCI 生成式 AI 服务提供支持按需模型或专用 AI 集群的生成式 AI 对话能力。

OCI 聊天模型页面和 OCI 生成式 AI 实验场提供了在 OCI 上使用和托管聊天模型的详细信息。

前提条件

使用 OCI GenAI Cohere 聊天客户端需要一个有效的甲骨文云基础设施（OCI）账户。客户端提供四种不同的连接方式，包括使用用户和私钥的简易认证、工作负载身份、实例主体或 OCI 配置文件认证。

添加仓库与物料清单（BOM）

Spring AI 构件发布在 Maven 中央仓库和 Spring 快照仓库。参考构件仓库章节将这些仓库添加到你的构建系统中。

为了帮助进行依赖管理，Spring AI 提供了 BOM（物料清单），确保整个项目使用一致版本的 Spring AI。参考依赖管理章节将 Spring AI BOM 添加到你的构建系统中。

自动配置

Spring AI 自动配置、启动器模块的构件名称发生了重大变更。更多信息请参考升级说明。

Spring AI 为 OCI GenAI Cohere 聊天客户端提供 Spring Boot 自动配置。要启用它，请在项目的 Maven pom.xml 文件中添加以下依赖：

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-starter-model-oci-genai</artifactId>
</dependency>

或者在 Gradle build.gradle 构建文件中添加：

dependencies {
    implementation 'org.springframework.ai:spring-ai-starter-model-oci-genai'
}

请参考依赖管理章节将 Spring AI BOM 添加到你的构建文件中。

聊天配置属性

连接属性

前缀 spring.ai.oci.genai 是用于配置 OCI GenAI 连接的属性前缀。

配置属性

聊天自动配置的启用与禁用现在通过前缀为 spring.ai.model.chat 的顶级属性配置。 启用：spring.ai.model.chat=oci-genai（默认启用） 禁用：spring.ai.model.chat=none（或任何不等于 oci-genai 的值） 此变更用于支持多模型配置。

前缀 spring.ai.oci.genai.cohere.chat 是用于配置 OCI GenAI Cohere Chat 的 ChatModel 实现的属性前缀。

所有以 spring.ai.oci.genai.cohere.chat.options 为前缀的属性都可以在运行时通过向 Prompt 调用添加请求特定的运行时选项来覆盖。

运行时选项

OCICohereChatOptions.java 提供模型配置，例如要使用的模型、温度、频率惩罚等。

启动时，可以通过 OCICohereChatModel(api, options) 构造函数或 spring.ai.oci.genai.cohere.chat.options.* 属性配置默认选项。

运行时，你可以通过向 Prompt 调用添加新的、请求特定的选项来覆盖默认选项。例如，为特定请求覆盖默认模型和温度：

ChatResponse response = chatModel.call(
    new Prompt(
        "Generate the names of 5 famous pirates.",
        OCICohereChatOptions.builder()
            .model("my-model-ocid")
            .compartment("my-compartment-ocid")
            .temperature(0.5)
        .build()
    ));

示例控制器

创建一个新的 Spring Boot 项目，并将 spring-ai-starter-model-oci-genai 添加到你的 pom 或 gradle 依赖中。

在 src/main/resources 目录下添加 application.properties 文件，启用并配置 OCI GenAI Cohere 聊天模型：

spring.ai.oci.genai.authenticationType=file
spring.ai.oci.genai.file=/path/to/oci/config/file
spring.ai.oci.genai.cohere.chat.options.compartment=my-compartment-ocid
spring.ai.oci.genai.cohere.chat.options.servingMode=on-demand
spring.ai.oci.genai.cohere.chat.options.model=my-chat-model-ocid

将文件路径、分区和模型替换为你 OCI 账户中的对应值。

这将创建一个 OCICohereChatModel 实现，你可以将其注入到类中。下面是一个简单的 @RestController 类示例，使用聊天模型进行文本生成：

@RestController
public class ChatController {

    private final OCICohereChatModel chatModel;

    @Autowired
    public ChatController(OCICohereChatModel chatModel) {
        this.chatModel = chatModel;
    }

    @GetMapping("/ai/generate")
    public Map generate(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        return Map.of("generation", chatModel.call(message));
    }

    @GetMapping("/ai/generateStream")
	public FluxgenerateStream(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        var prompt = new Prompt(new UserMessage(message));
        return chatModel.stream(prompt);
    }
}

手动配置

OCICohereChatModel 实现了 ChatModel 接口，并使用 OCI Java SDK 连接到 OCI GenAI 服务。

将 spring-ai-oci-genai 依赖添加到项目的 Maven pom.xml 文件：

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-oci-genai</artifactId>
</dependency>

或者添加到 Gradle build.gradle 文件：

dependencies {
    implementation 'org.springframework.ai:spring-ai-oci-genai'
}

请参考依赖管理章节将 Spring AI BOM 添加到你的构建文件中。

接下来，创建 OCICohereChatModel 并用于文本生成：

var CONFIG_FILE = Paths.get(System.getProperty("user.home"), ".oci", "config").toString();
var COMPARTMENT_ID = System.getenv("OCI_COMPARTMENT_ID");
var MODEL_ID = System.getenv("OCI_CHAT_MODEL_ID");

ConfigFileAuthenticationDetailsProvider authProvider = new ConfigFileAuthenticationDetailsProvider(
        CONFIG_FILE,
        "DEFAULT"
);
var genAi = GenerativeAiInferenceClient.builder()
        .region(Region.valueOf("us-chicago-1"))
        .build(authProvider);

var chatModel = new OCICohereChatModel(genAi, OCICohereChatOptions.builder()
        .model(MODEL_ID)
        .compartment(COMPARTMENT_ID)
        .servingMode("on-demand")
        .build());

ChatResponse response = chatModel.call(
        new Prompt("Generate the names of 5 famous pirates."));

OCICohereChatOptions 提供聊天请求的配置信息。OCICohereChatOptions.Builder 是流式选项构建器。