3233 - ClassifyTextAction done

Author: marko-kriskovic

server/libs/modules/components/ai/llm/src/main/java/com/bytechef/component/ai/llm/ChatModel.java

@@ -61,7 +61,7 @@ default Object getResponse(
             inputParameters, connectionParameters, true);
 
         List<org.springframework.ai.chat.messages.Message> messages = ModelUtils.getMessages(
-            inputParameters, actionContext);
+            inputParameters, actionContext, false);
 
         ChatClient.ChatClientRequestSpec chatClientRequestSpec = createPrompt(
             inputParameters, actionContext, chatModel);

server/libs/modules/components/ai/llm/src/main/java/com/bytechef/component/ai/llm/util/ModelUtils.java

@@ -120,27 +120,31 @@ public static <R> List<Option<R>> getEnumOptions(Map<String, R> map) {
             .toList();
     }
 
-    public static List<Message> getMessages(Parameters inputParameters, ActionContext actionContext) {
+    public static List<Message> getMessages(Parameters inputParameters, ActionContext actionContext, boolean responseFormatRequired) {
         List<ChatModel.Message> chatModelMessages = new ArrayList<>();
 
-        String format = inputParameters.getRequiredString(FORMAT);
+        if (responseFormatRequired) {
+            String format = inputParameters.getRequiredString(FORMAT);
+            if (format.equals(Format.SIMPLE.name())) {
+                String userPrompt = inputParameters.getRequiredString(USER_PROMPT);
 
-        if (format.equals(Format.SIMPLE.name())) {
-            String userPrompt = inputParameters.getRequiredString(USER_PROMPT);
+                ChatModel.Message userMessage = new ChatModel.Message(
+                    userPrompt, inputParameters.getList(ATTACHMENTS, FileEntry.class), Role.USER);
 
-            ChatModel.Message userMessage = new ChatModel.Message(
-                userPrompt, inputParameters.getList(ATTACHMENTS, FileEntry.class), Role.USER);
+                chatModelMessages.add(userMessage);
 
-            chatModelMessages.add(userMessage);
+                String systemPrompt = inputParameters.getString(SYSTEM_PROMPT);
 
-            String systemPrompt = inputParameters.getString(SYSTEM_PROMPT);
+                if (systemPrompt != null && !systemPrompt.isEmpty()) {
+                    ChatModel.Message systeMMessage = new ChatModel.Message(systemPrompt, null, Role.SYSTEM);
 
-            if (systemPrompt != null && !systemPrompt.isEmpty()) {
-                ChatModel.Message systeMMessage = new ChatModel.Message(systemPrompt, null, Role.SYSTEM);
-
-                chatModelMessages.add(systeMMessage);
+                    chatModelMessages.add(systeMMessage);
+                }
+            } else {
+                chatModelMessages = inputParameters.getList(MESSAGES, new TypeReference<>() {});
             }
-        } else {
+        }
+        else {
             chatModelMessages = inputParameters.getList(MESSAGES, new TypeReference<>() {});
         }
 

server/libs/modules/components/ai/universal/universal-text/src/main/java/com/bytechef/component/ai/universal/text/action/ClassifyTextAction.java

@@ -34,6 +34,7 @@
 import static com.bytechef.component.definition.ComponentDsl.object;
 import static com.bytechef.component.definition.ComponentDsl.string;
 
+import com.bytechef.component.ai.llm.ChatModel;
 import com.bytechef.component.ai.universal.text.action.definition.AiTextActionDefinition;
 import com.bytechef.component.ai.universal.text.constant.AiTextConstants;
 import com.bytechef.component.definition.Parameters;
@@ -113,6 +114,8 @@ public Parameters createParameters(Parameters inputParameters) {
                 Map.of("content", systemPrompt, ROLE, SYSTEM.name()),
                 Map.of("content", userBuilder, ROLE, USER.name())));
         modelInputParametersMap.put("model", inputParameters.getString(MODEL));
+        modelInputParametersMap.put("response", Map.of(
+            "responseFormat", ChatModel.ResponseFormat.TEXT));
 
         return ParametersFactory.createParameters(modelInputParametersMap);
     }