侧边栏壁纸
博主头像
鱼箴日记,AI,Java,Liunx,Spring,Spring AI博主等级

行动起来,活在当下

  • 累计撰写 10 篇文章
  • 累计创建 8 个标签
  • 累计收到 0 条评论

目 录CONTENT

文章目录

Spring AI框架使用Open AI入门案例

Administrator
2024-06-07 / 0 评论 / 0 点赞 / 71 阅读 / 4592 字 / 正在检测是否收录...

1. 引言

Spring AI 是 Spring 生态系统中的一个新兴框架,旨在简化人工智能和机器学习模型的集成和部署。本文将介绍如何在 Spring 项目中集成和使用 Spring AI,通过一个简单的案例展示其基本用法。

2. 环境准备

在开始之前,确保你已经安装了以下工具:

  • JDK 11 或更高版本

  • Maven 3.6.3 或更高版本

  • 一个集成开发环境(IDE),如 IntelliJ IDEA 或 Eclipse

3. 创建 Spring Boot 项目

使用 Spring Initializr 创建一个新的 Spring Boot 项目。在 IDE 中打开项目并添加必要的依赖项。

添加依赖项

pom.xml 中添加 Spring AI 及其相关依赖项:

<dependencies>
    <!-- Spring Boot Starter -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter</artifactId>
    </dependency>

    <!-- Spring Boot Starter Web -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>

    <!-- Spring AI Dependency -->
	<dependency>
	    <groupId>org.springframework.ai</groupId>
	    <artifactId>spring-ai-openai-spring-boot-starter</artifactId>
		<version>1.0.0-SNAPSHOT</version>
	</dependency>

    <!-- Gson for JSON Parsing -->
    <dependency>
        <groupId>com.google.code.gson</groupId>
        <artifactId>gson</artifactId>
        <version>2.8.6</version>
    </dependency>
</dependencies>

4. 编写示例代码

配置 Spring AI

application.properties 中配置你的 AI 服务,例如 OpenAI 的 API 密钥:

spring.ai.openai.api-key=your_openai_api_key

创建服务类

创建一个服务类,用于与 AI 模型交互:

package com.example.springai.service;

import org.springframework.ai.openai.OpenAIService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service
public class AIService {

    private final OpenAIService openAIService;

    @Autowired
    public AIService(OpenAIService openAIService) {
        this.openAIService = openAIService;
    }

    public String generateText(String prompt) {
        // 调用 OpenAI API 生成文本
        String response = openAIService.createCompletion(prompt);
        return response;
    }
}

创建控制器

创建一个控制器类,用于处理 HTTP 请求:

package com.example.springai.controller;

import com.example.springai.service.AIService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping("/api/ai")
public class AIController {

    private final AIService aiService;

    @Autowired
    public AIController(AIService aiService) {
        this.aiService = aiService;
    }

    @PostMapping("/generate")
    public String generateText(@RequestBody String prompt) {
        return aiService.generateText(prompt);
    }
}

5. 测试

启动 Spring Boot 应用程序,并使用以下命令测试 API:

curl -X POST http://localhost:8080/api/ai/generate -H "Content-Type: application/json" -d '{"prompt":"Tell me a joke"}'

你应该会得到一个由 AI 模型生成的文本响应。

6. 结论

通过本文,你学会了如何在 Spring Boot 项目中集成和使用 Spring AI。这个示例只是一个入门案例,Spring AI 提供了更多高级功能,如处理不同类型的模型、优化模型推理等。希望这篇文章能帮助你更好地理解和使用 Spring AI,为你的应用程序添加智能化功能。

7. 参考资料

你可以根据需要进一步扩展和优化示例代码,为你的实际应用场景提供支持。

0

评论区