Spring AI 介绍以及与 Spring Boot 项目整合

news2024/10/23 16:17:31

Spring AI 项目旨在简化使用 Spring Boot 开发包含人工智能功能的应用程序，提供抽象和支持多种模型提供商及矢量数据库提供商。

Spring AI 的功能特点

支持主流模型提供商：如 OpenAI、Microsoft、Amazon、Google 和 Huggingface 等。
支持多种模型类型：包括聊天、文本到图像、音频转录、文本到语音等。
提供可移植 API：支持同步和流 API 选项，还支持访问模型特定功能。
实现 AI 模型输出到 POJO 的映射。
支持主流矢量数据库提供商：如 Apache Cassandra、Azure Vector Search、Chroma、Milvus、Neo4j、PostgreSQL/PGVector、PineCone、Qdrant、Redis 和 Weaviate 等。
提供跨 Vector Store 提供程序的可移植 API，包括类似 SQL 的元数据过滤器 API。
支持 Function calling（函数调用）。
提供 Spring Boot 自动配置和启动器的 AI 模型。
包含数据工程的 ETL 框架。

Spring Boot 项目整合 Spring AI 的步骤

初始化 Spring Boot 项目：
- 使用 Spring Initializr：访问 https://start.spring.io/，选择 Maven 或 Gradle 作为构建工具，选择 Java 作为语言，勾选“Web”依赖以启用 Spring Web 支持，选择 OpenAI 依赖，然后生成项目。

在这里插入图片描述
手动添加依赖

添加仓库依赖：在 pom.xml 文件中添加以下仓库依赖：

<repositories>
    <repository>
        <id>spring-milestones</id>
        <name>Spring Milestones</name>
        <url>https://repo.spring.io/milestone</url>
        <snapshots>
            <enabled>false</enabled>
        </snapshots>
    </repository>
    <repository>
        <id>spring-snapshots</id>
        <name>Spring Snapshots</name>
        <url>https://repo.spring.io/snapshot</url>
        <releases>
            <enabled>false</enabled>
        </releases>
    </repository>
</repositories>

使用 BOM：通过将以下内容添加到 pom.xml 文件中来使用 BOM，使用的 BOM 版本决定了使用的依赖项版本，它还可确保默认使用受支持和测试的依赖项版本。

<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>org.springframework.ai</groupId>
            <artifactId>spring-ai-bom</artifactId>
            <version>1.0.0-SNAPSHOT</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
    </dependencies>
</dependencyManagement>

添加 OpenAI 的 maven 依赖：在 pom.xml 文件中添加以下依赖：

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

配置 OpenAi API KEY：请加入你生成的 OpenAI 的 API key，可以通过直连或者中转的方式进行访问，具体可以百度和淘宝。在 application.properties 文件中添加以下配置：

spring.application.name=spring-ai
spring.ai.openai.api-key=sk-8*******************
spring.ai.openai.base-ur=h*******************

在控制器中调用 Spring API 的方法进行访问和输出：

@RestController
@RequestMapping("/openai")
public class OpenAiChatController {

    @Resource
    private OpenAiChatModel openAiChatModel;

    @GetMapping("/chat/generate")
    public String chat(@RequestParam(value = "message", defaultValue = "你是谁?") String message) {
        String result = openAiChatModel.call(message);
        System.out.println(result);
        return result;
    }
}