Spring AI 指南

发布时间:2024年01月05日

近年来,人工智能技术的迅猛发展改变了我们对科技的看法,并在各个领域引发了巨大的变革。每个人都希望在自己的项目上能够使用人工智能。Spring 框架提供了一个名为 “Spring AI” 的项目,Spring AI 项目旨在简化包含人工智能功能的应用程序的开发,避免不必要的复杂性。目前还处于预测试版,也就意味着将来可能发生变化,不过事实上现在已经可以和OpenAI、Azure OpenAI、Hugging Face、Ollama进行API交互。

让我们看看如何在 Spring 应用程序中实现生成式AI吧。

创建一个Spirng项目

首先我们创建一个全新的Spring项目,引入Spring AI依赖。

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
        <version>3.2.0</version>
    </dependency>

    <dependency>
        <groupId>org.springframework.ai</groupId>
        <artifactId>spring-ai-openai-spring-boot-starter</artifactId>
        <version>0.8.0-SNAPSHOT</version>
    </dependency>
</dependencies>

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

在 application.yml 中设置OpenAI的API Key。具体如何获取可以参考如何获取Api Key

spring:
  ai:
    openai:
      api-key: {OPENAI_API_KEY}

添加完成OpenAI API Key 我们就可以使用ChatClient与AI模型进行交互了,ChatClient简化了与各种人工智能模型的连接,针对各模型有自己的实现类。我们先看下基础用法,这里我创建一个接口,要求OpenAI生成一个答案给我,此时ChatClient将调用OpenAI独有的API接口向OpenAI发起请求,我们可以通过测试来看看OpenAI给我们的答案是什么吧。

@GetMapping("/top/code/language")
public String topCodeLanguage() {
    String message = "2023最受欢迎编程语言是什么";
    return chatClient.generate(message);
}

提示模板

我们稍微对上面的接口做些更改,我首先将一个路径变量放入请求,这样我们就可以将年份模板化。PromptTemplate旨在帮助创建结构化提示,我们通过add()方法插入动态内容,render()会渲染为最终字符串格式。

@GetMapping("/top/code/language/{year}")
public String topCodeLanguageByYear(@PathVariable("year") Integer year) {
    String message = "{year}最受欢迎编程语言是什么";
    PromptTemplate promptTemplate = new PromptTemplate(message);
    promptTemplate.add("year", year);
    return chatClient.generate(promptTemplate.render());
}

输出解析器

也许我们不希望它返回的是一个字符串,我们希望它作为一个对象,所以让我们看看如何做到这一点。首先我们创建一个TopLanguage类。

public class TopLanguage {
    private Integer ranking;
    private String languageName;
    private String introduction;
}

然后我们先创建一个输出解析器,BeanOutputParser是指定 Java 类转换为 JSON 模式,然后,AI 模型的 JSON 输出被反序列化为 Java 对象。同样我们通过占位符{format}插入动态内容。

@GetMapping("/top/code/language/{year}/json")
public TopLanguage topCodeLanguageByYearForFormat(@PathVariable("year") Integer year) {
    BeanOutputParser<TopLanguage> parser = new BeanOutputParser<>(TopLanguage.class);
    String message = "What is the most popular programming language in {year}? {format}";
    PromptTemplate promptTemplate = new PromptTemplate(message);
    promptTemplate.add("year", year);
    promptTemplate.add("format", parser.getFormat());
    promptTemplate.setOutputParser(parser);

    log.info("FORMAT STRING: " + parser.getFormat());

    Prompt prompt = promptTemplate.create();
    ChatResponse chatResponse = chatClient.generate(prompt);
    String content = chatResponse.getGeneration().getContent();
    return parser.parse(content);
}

结论

我们已经完成了Spring AI的基本示例,与往常一样,本文中使用的源代码可在 GitHub 上获得。

一起学习的小伙伴可以关注下我的公众号,我会经常更新技术实践类文章,所有文章都会有相应源码供大家学习交流!
在这里插入图片描述

文章来源:https://blog.csdn.net/new_ord/article/details/135407910
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。