Spring AI教程（六）：集成第三方服务与扩展功能

在之前的文章中，我们已经介绍了Spring AI的基础和高级功能。这篇文章将重点探讨如何将Spring AI与第三方服务集成，以及如何扩展Spring AI的功能，以满足更复杂的需求。

集成第三方服务

集成OpenAI GPT模型

OpenAI的GPT模型是目前最流行的AI语言模型之一。Spring AI支持与OpenAI的集成，使你可以轻松使用GPT模型生成文本。以下是一个示例，展示如何集成OpenAI GPT模型并生成文本：

1. 配置OpenAI API密钥

在application.properties文件中配置你的OpenAI API密钥：

spring.ai.provider=openai
spring.ai.api-key=YOUR_OPENAI_API_KEY

2. 创建服务类

创建一个服务类，用于与OpenAI GPT模型进行交互：

import org.springframework.stereotype.Service;
import com.example.springai.OpenAiChatService;

@Service
public class GptService {

    private final OpenAiChatService openAiChatService;

    public GptService(OpenAiChatService openAiChatService) {
        this.openAiChatService = openAiChatService;
    }

    public String generateText(String prompt) {
        return openAiChatService.chat(prompt);
    }
}

3. 创建控制器

创建一个控制器，处理生成文本的HTTP请求：

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class GptController {

    private final GptService gptService;

    public GptController(GptService gptService) {
        this.gptService = gptService;
    }

    @GetMapping("/generate-text")
    public String generateText(@RequestParam String prompt) {
        return gptService.generateText(prompt);
    }
}

扩展功能

自定义模型提供商

除了内置支持的模型提供商，Spring AI还允许你集成自定义的AI模型提供商。以下是一个示例，展示如何创建和集成自定义模型提供商：

1. 实现自定义模型提供商

创建一个实现了AiModelProvider接口的自定义模型提供商类：

import com.example.springai.AiModelProvider;
import org.springframework.stereotype.Component;

@Component
public class CustomModelProvider implements AiModelProvider {

    @Override
    public String generateResponse(String input) {
        // 实现自定义AI模型的交互逻辑
        return "这是自定义模型的响应：" + input;
    }
}

2. 配置自定义模型提供商

在application.properties文件中配置自定义模型提供商：

spring.ai.provider=custom

3. 创建服务类

创建一个服务类，用于与自定义模型提供商进行交互：

import org.springframework.stereotype.Service;

@Service
public class CustomAiService {

    private final CustomModelProvider customModelProvider;

    public CustomAiService(CustomModelProvider customModelProvider) {
        this.customModelProvider = customModelProvider;
    }

    public String generateResponse(String input) {
        return customModelProvider.generateResponse(input);
    }
}

4. 创建控制器

创建一个控制器，处理生成响应的HTTP请求：

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class CustomAiController {

    private final CustomAiService customAiService;

    public CustomAiController(CustomAiService customAiService) {
        this.customAiService = customAiService;
    }

    @GetMapping("/custom-response")
    public String generateResponse(@RequestParam String input) {
        return customAiService.generateResponse(input);
    }
}

使用多个模型提供商

在实际应用中，你可能需要同时使用多个模型提供商。Spring AI允许你灵活配置和切换不同的模型提供商。

1. 配置多个模型提供商

在application.properties文件中配置多个模型提供商：

spring.ai.provider=openai
spring.ai.secondary-provider=custom
spring.ai.api-key=YOUR_OPENAI_API_KEY

2. 创建服务类

创建一个服务类，根据需要切换不同的模型提供商：

import org.springframework.stereotype.Service;

@Service
public class MultiProviderService {

    private final OpenAiChatService openAiChatService;
    private final CustomModelProvider customModelProvider;

    public MultiProviderService(OpenAiChatService openAiChatService, CustomModelProvider customModelProvider) {
        this.openAiChatService = openAiChatService;
        this.customModelProvider = customModelProvider;
    }

    public String generateResponse(String input, boolean useCustom) {
        if (useCustom) {
            return customModelProvider.generateResponse(input);
        } else {
            return openAiChatService.chat(input);
        }
    }
}

3. 创建控制器

创建一个控制器，处理使用不同模型提供商生成响应的HTTP请求：

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class MultiProviderController {

    private final MultiProviderService multiProviderService;

    public MultiProviderController(MultiProviderService multiProviderService) {
        this.multiProviderService = multiProviderService;
    }

    @GetMapping("/multi-response")
    public String generateResponse(@RequestParam String input, @RequestParam boolean useCustom) {
        return multiProviderService.generateResponse(input, useCustom);
    }
}

结论

通过集成第三方服务和扩展Spring AI的功能，你可以构建更加灵活和强大的AI应用程序。无论是使用OpenAI的GPT模型，还是自定义模型提供商，Spring AI都为你提供了丰富的工具和配置选项，满足不同应用场景的需求。

希望这篇文章能帮助你在实际项目中充分利用Spring AI的强大功能。下一篇文章中，我们将继续探讨更多实际应用场景和高级功能，帮助你进一步掌握这一强大的工具。