From fc10f1c4aedff9fceb5c5e4af48b8c98edd55465 Mon Sep 17 00:00:00 2001
From: sebastianMindee <sebastian.oliverasilvera@mindee.co>
Date: Wed, 16 Apr 2025 15:40:24 +0200
Subject: [PATCH] :sparkles: add support for rag param in workflow executions

---
 src/main/java/com/mindee/WorkflowOptions.java     | 9 ++++++++-
 src/main/java/com/mindee/http/MindeeHttpApi.java  | 3 +++
 src/test/java/com/mindee/workflow/WorkflowIT.java | 2 +-
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/mindee/WorkflowOptions.java b/src/main/java/com/mindee/WorkflowOptions.java
index 2ad7bbab9..2b5029adb 100644
--- a/src/main/java/com/mindee/WorkflowOptions.java
+++ b/src/main/java/com/mindee/WorkflowOptions.java
@@ -32,16 +32,23 @@ public class WorkflowOptions {
    */
   String publicUrl;
 
+  /**
+   * Whether to enable Retrieval-Augmented generation.
+   */
+  Boolean rag;
+
   @Builder
   private WorkflowOptions(
       String alias,
       ExecutionPriority priority,
       Boolean fullText,
-      String publicUrl
+      String publicUrl,
+      Boolean rag
   ) {
     this.alias = alias;
     this.priority = priority;
     this.fullText = fullText == null ? Boolean.FALSE : fullText;
     this.publicUrl = publicUrl;
+    this.rag = rag;
   }
 }
diff --git a/src/main/java/com/mindee/http/MindeeHttpApi.java b/src/main/java/com/mindee/http/MindeeHttpApi.java
index b65bc24ba..9f5b050fb 100644
--- a/src/main/java/com/mindee/http/MindeeHttpApi.java
+++ b/src/main/java/com/mindee/http/MindeeHttpApi.java
@@ -388,6 +388,9 @@ private List<NameValuePair> buildPostParams(
     if (Boolean.TRUE.equals(requestParameters.getPredictOptions().getFullText())) {
       params.add(new BasicNameValuePair("full_text_ocr", "true"));
     }
+    if (Boolean.TRUE.equals(requestParameters.getWorkflowOptions().getRag())) {
+      params.add(new BasicNameValuePair("rag", "true"));
+    }
     return params;
   }
 
diff --git a/src/test/java/com/mindee/workflow/WorkflowIT.java b/src/test/java/com/mindee/workflow/WorkflowIT.java
index 06c265820..72617bef7 100644
--- a/src/test/java/com/mindee/workflow/WorkflowIT.java
+++ b/src/test/java/com/mindee/workflow/WorkflowIT.java
@@ -35,7 +35,7 @@ protected Execution<GeneratedV1> getFinancialDocumentWorkflow(String workflowId)
       IOException, MindeeException {
 
     WorkflowOptions options = WorkflowOptions.builder().alias("java-" + currentDateTime).priority(
-        ExecutionPriority.LOW).build();
+        ExecutionPriority.LOW).rag(true).build();
     WorkflowResponse<GeneratedV1> response =
         client.executeWorkflow(workflowId, financialDocumentInputSource, options);
     return response.getExecution();