1018 - SF

Author: ivicac

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

@@ -52,7 +52,7 @@ default Object getResponse(
             .messages(messages)
             .call();
 
-        return returnChatEntity(inputParameters, call, context);
+        return returnChatEntity(inputParameters, call, actionContext);
     }
 
     private List<org.springframework.ai.chat.messages.Message> getMessages(
@@ -74,7 +74,9 @@ private List<org.springframework.ai.chat.messages.Message> getMessages(
     }
 
     @SuppressFBWarnings("NP")
-    private Object returnChatEntity(Parameters parameters, ChatClient.CallResponseSpec call, Context context) {
+    private Object returnChatEntity(
+        Parameters parameters, ChatClient.CallResponseSpec call, ActionContext actionContext) {
+
         int responseFormat = parameters.getInteger(RESPONSE_FORMAT, 0);
 
         if (responseFormat == 0) {
@@ -84,7 +86,7 @@ private Object returnChatEntity(Parameters parameters, ChatClient.CallResponseSp
                 .getContent();
         } else {
             return call.entity(
-                new JsonSchemaStructuredOutputConverter(parameters.getRequiredString(RESPONSE_SCHEMA), context));
+                new JsonSchemaStructuredOutputConverter(parameters.getRequiredString(RESPONSE_SCHEMA), actionContext));
         }
     }
 

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

@@ -52,12 +52,12 @@ public static Message createMessage(Chat.Message message, ActionContext actionCo
                 if (fileEntry == null) {
                     yield new UserMessage(message.content());
                 } else {
-                    byte[] encodedImageBytes = actionContext.file(file -> file.readAllBytes(fileEntry));
                     String mimeType = fileEntry.getMimeType();
+                    byte[] encodedImageBytes = actionContext.file(file -> file.readAllBytes(fileEntry));
 
                     yield new UserMessage(
-                        message.content(), new Media(MimeTypeUtils.parseMimeType(mimeType),
-                            new ByteArrayResource(encodedImageBytes)));
+                        message.content(), new Media(
+                            MimeTypeUtils.parseMimeType(mimeType), new ByteArrayResource(encodedImageBytes)));
                 }
             }
             case "assistant" -> new AssistantMessage(message.content());