Spring AI Tutorial: How to Integrate OpenAI and Hugging Face with Spring Boot

Unlock the Power of AI: Spring Boot, OpenAI & Hugging Face Integration

Dive into Spring AI and learn how to harness the power of OpenAI and Hugging Face within your Spring Boot applications. Master integration techniques and build intelligent features effortlessly.

Read More

Introduction to Spring AI

Spring AI simplifies the integration of Artificial Intelligence models into Spring Boot applications. It provides a consistent and streamlined approach to working with various AI providers, such as OpenAI and Hugging Face.

Setting Up Your Spring Boot Project

First, you'll need a Spring Boot project. You can create one using Spring Initializr. Add the necessary dependencies to your pom.xml file.

 <dependencies>
  <dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-starter-web</artifactId>
  </dependency>
  <dependency>
   <groupId>org.springframework.ai</groupId>
   <artifactId>spring-ai-openai</artifactId>
   <version>latest.release</version> <!-- Replace with the latest version -->
  </dependency>
  <dependency>
   <groupId>org.springframework.ai</groupId>
   <artifactId>spring-ai-transformers</artifactId>
   <version>latest.release</version> <!-- Replace with the latest version -->
  </dependency>
  <dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-starter-test</artifactId>
   <scope>test</scope>
  </dependency>
 </dependencies>
 

Configuring OpenAI

To use OpenAI, you need an API key. Once you have it, configure it in your application.properties or application.yml file:

 spring:
  ai:
   openai:
    api-key: YOUR_OPENAI_API_KEY
 

Using the OpenAI Chat API

Here's how you can use the OpenAI Chat API to generate responses based on a prompt:

 import org.springframework.ai.chat.ChatClient;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;

 @Service
 public class OpenAIService {

  @Autowired
  private ChatClient chatClient;

  public String generateResponse(String prompt) {
   return chatClient.call(prompt);
  }
 }
 

Integrating Hugging Face

Spring AI also supports integration with Hugging Face Transformers. You'll need to configure your Hugging Face API token:

 spring:
  ai:
   huggingface:
    api-key: YOUR_HUGGINGFACE_API_KEY
 

Example with Hugging Face

Here's a simple example of using Hugging Face for text generation:

 import org.springframework.ai.transformers.TransformersChatClient;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;

 @Service
 public class HuggingFaceService {

  @Autowired
  private TransformersChatClient transformersChatClient;

  public String generateText(String prompt) {
   return transformersChatClient.call(prompt);
  }
 }
 

Further Exploration

• Explore different models offered by OpenAI and Hugging Face.
• Implement more complex prompts and fine-tune the AI responses.
• Integrate Spring AI with other Spring projects, such as Spring Data and Spring Cloud.

Conclusion

By following this guide, you’ve successfully integrated OpenAI and Hugging Face with your Spring Boot application using Spring AI. Happy coding!

Show your love, follow us javaoneworld