feat: IA (#14)

Author: Matithieu

build.gradle

@@ -17,6 +17,8 @@ dependencies {
     implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
     implementation 'org.springframework.boot:spring-boot-starter-oauth2-resource-server'
     implementation 'org.springframework.boot:spring-boot-starter-web'
+    implementation platform("org.springframework.ai:spring-ai-bom:1.0.0")
+    implementation 'org.springframework.ai:spring-ai-starter-model-openai'
     implementation 'org.springframework.boot:spring-boot-starter-security'
     implementation 'org.springdoc:springdoc-openapi-starter-webmvc-ui:2.6.0'
     implementation 'org.springframework.boot:spring-boot-starter-validation'

src/main/java/com/example/spring/app/llm/LLMController.java

@@ -0,0 +1,24 @@
+package com.example.spring.app.llm;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.web.bind.annotation.*;
+
+@CrossOrigin
+@RestController
+@RequestMapping("/v1")
+public class LLMController {
+
+    private final ChatClient chatClient;
+
+    public LLMController(ChatClient.Builder chatClientBuilder) {
+        this.chatClient = chatClientBuilder.build();
+    }
+
+    @GetMapping("/ai")
+    String generation(String userInput) {
+        return this.chatClient.prompt()
+                .user(userInput)
+                .call()
+                .content();
+    }
+}

src/main/resources/application.properties

@@ -34,3 +34,8 @@ resilience4j.ratelimiter.instances.scrapService.timeout-duration=30s
 
 # Spring
 spring.profiles.active=${SPRING_PROFILES_ACTIVE}
+
+# Spring AI
+spring.ai.openai.api-key=ollama
+spring.ai.openai.base-url=http://localhost:11434/
+spring.ai.openai.chat.options.model=mistral:7b