feat : example of output parser

chatmodel-springai/src/main/java/com/example/ai/controller/ChatController.java

@@ -2,6 +2,7 @@
 
 import com.example.ai.model.request.AIChatRequest;
 import com.example.ai.model.response.AIChatResponse;
+import com.example.ai.model.response.ActorsFilms;
 import java.util.List;
 import java.util.Map;
 import org.springframework.ai.chat.ChatClient;
@@ -12,9 +13,12 @@
 import org.springframework.ai.chat.prompt.Prompt;
 import org.springframework.ai.chat.prompt.PromptTemplate;
 import org.springframework.ai.embedding.EmbeddingClient;
+import org.springframework.ai.parser.BeanOutputParser;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
 @RestController
@@ -61,4 +65,21 @@ AIChatResponse chatWithEmbeddingClient(@RequestBody AIChatRequest aiChatRequest)
         List<Double> embed = embeddingClient.embed(aiChatRequest.query());
         return new AIChatResponse(embed.toString());
     }
+
+    @GetMapping("/output")
+    public ActorsFilms generate(@RequestParam(value = "actor", defaultValue = "Jr NTR") String actor) {
+        BeanOutputParser<ActorsFilms> outputParser = new BeanOutputParser<>(ActorsFilms.class);
+
+        String format = outputParser.getFormat();
+        String template = """
+				Generate the filmography for the actor {actor}.
+				{format}
+				""";
+        PromptTemplate promptTemplate = new PromptTemplate(template, Map.of("actor", actor, "format", format));
+        Prompt prompt = new Prompt(promptTemplate.createMessage());
+        ChatResponse response = chatClient.call(prompt);
+        Generation generation = response.getResult();
+
+        return outputParser.parse(generation.getOutput().getContent());
+    }
 }

chatmodel-springai/src/main/java/com/example/ai/model/response/ActorsFilms.java

@@ -0,0 +1,5 @@
+package com.example.ai.model.response;
+
+import java.util.List;
+
+public record ActorsFilms(String actor, List<String> movies) {}

chatmodel-springai/src/test/java/com/example/ai/controller/ChatControllerTest.java

@@ -1,11 +1,14 @@
 package com.example.ai.controller;
 
 import static io.restassured.RestAssured.given;
+import static org.hamcrest.Matchers.containsInAnyOrder;
 import static org.hamcrest.Matchers.containsString;
+import static org.hamcrest.Matchers.is;
 
 import com.example.ai.model.request.AIChatRequest;
 import io.restassured.RestAssured;
 import io.restassured.http.ContentType;
+import org.apache.http.HttpStatus;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.TestInstance;
@@ -56,4 +59,44 @@ void chatWithSystemPrompt() {
                 .statusCode(200)
                 .body("answer", containsString("cricket"));
     }
+
+    @Test
+    void outputParser() {
+        given().param("actor", "Jr NTR")
+                .when()
+                .get("/api/ai/output")
+                .then()
+                .statusCode(HttpStatus.SC_OK)
+                .contentType(ContentType.JSON)
+                .body("actor", is("Jr NTR"))
+                .body(
+                        "movies",
+                        containsInAnyOrder(
+                                "Ninnu Choodalani",
+                                "Student No.1",
+                                "Subbu",
+                                "Aadi",
+                                "Allari Ramudu",
+                                "Naaga",
+                                "Simhadri",
+                                "Samba",
+                                "Andhrawala",
+                                "Rakhi",
+                                "Yamadonga",
+                                "Kantri",
+                                "Adhurs",
+                                "Brindavanam",
+                                "Shakti",
+                                "Oosaravelli",
+                                "Dammu",
+                                "Baadshah",
+                                "Ramayya Vasthavayya",
+                                "Rabhasa",
+                                "Temper",
+                                "Nannaku Prematho",
+                                "Janatha Garage",
+                                "Jai Lava Kusa",
+                                "Aravinda Sametha Veera Raghava",
+                                "RRR"));
+    }
 }