当前位置：首页 > news >正文

【Spring Boot】在项目中使用Spring AI

news 来源：原创 2024/9/29 7:24:03

Spring AI是Spring框架中用于集成和使用人工智能和机器学习功能的组件。它提供了一种简化的方式来与AI模型进行交互。下面是一个简单的示例，展示了如何在Spring Boot项目中使用Spring AI。

步骤 1: 添加依赖

首先，在pom.xml文件中添加Spring AI的依赖：

<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter</artifactId>
</dependency>
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-jpa</artifactId>
</dependency>
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency><groupId>org.springframework.cloud</groupId><artifactId>spring-cloud-starter-openfeign</artifactId>
</dependency>
<dependency><groupId>org.springframework.cloud</groupId><artifactId>spring-cloud-starter-ai</artifactId><version>0.1.0</version>
</dependency>

确保配置了Spring Cloud的版本管理，例如：

<dependencyManagement><dependencies><dependency><groupId>org.springframework.cloud</groupId><artifactId>spring-cloud-dependencies</artifactId><version>Hoxton.SR9</version><type>pom</type><scope>import</scope></dependency></dependencies>
</dependencyManagement>

步骤 2: 创建一个AI模型服务

创建一个服务来使用AI模型。这可以是一个简单的Spring服务类。以下是一个示例，展示了如何使用Spring AI来预测数据：

创建一个AI模型配置类

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.cloud.ai.annotation.EnableAi;@Configuration
@EnableAi
public class AiModelConfig {@Beanpublic AiModel myAiModel() {return new AiModel("my-model");}
}

创建一个AI服务类

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cloud.ai.annotation.AiModel;
import org.springframework.stereotype.Service;@Service
public class AiService {@Autowiredprivate AiModel aiModel;public String predict(String input) {return aiModel.predict(input);}
}

步骤 3: 创建一个控制器来使用AI服务

创建一个Spring MVC控制器，来调用AI服务：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;@RestController
public class AiController {@Autowiredprivate AiService aiService;@GetMapping("/predict")public String predict(@RequestParam String input) {return aiService.predict(input);}
}

步骤 4: 启动应用程序

确保启动类已经配置：

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;@SpringBootApplication
public class AiApplication {public static void main(String[] args) {SpringApplication.run(AiApplication.class, args);}
}

全部代码示例

整合以上所有部分，完整的代码示例如下：

`pom.xml`

<dependencies><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter</artifactId></dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-jpa</artifactId></dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId></dependency><dependency><groupId>org.springframework.cloud</groupId><artifactId>spring-cloud-starter-openfeign</artifactId></dependency><dependency><groupId>org.springframework.cloud</groupId><artifactId>spring-cloud-starter-ai</artifactId><version>0.1.0</version></dependency>
</dependencies><dependencyManagement><dependencies><dependency><groupId>org.springframework.cloud</groupId><artifactId>spring-cloud-dependencies</artifactId><version>Hoxton.SR9</version><type>pom</type><scope>import</scope></dependency></dependencies>
</dependencyManagement>

`AiApplication.java`

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;@SpringBootApplication
public class AiApplication {public static void main(String[] args) {SpringApplication.run(AiApplication.class, args);}
}

`AiModelConfig.java`

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.cloud.ai.annotation.EnableAi;@Configuration
@EnableAi
public class AiModelConfig {@Beanpublic AiModel myAiModel() {return new AiModel("my-model");}
}

`AiService.java`

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cloud.ai.annotation.AiModel;
import org.springframework.stereotype.Service;@Service
public class AiService {@Autowiredprivate AiModel aiModel;public String predict(String input) {return aiModel.predict(input);}
}

`AiController.java`

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;@RestController
public class AiController {@Autowiredprivate AiService aiService;@GetMapping("/predict")public String predict(@RequestParam String input) {return aiService.predict(input);}
}