From c645c5ceea407e2849a566bed65cff675eec881d Mon Sep 17 00:00:00 2001
From: Pawan Kumar <pawan@appsmith.com>
Date: Thu, 6 Feb 2025 21:52:07 +0530
Subject: [PATCH] add ce files

---
 .../components/formControls/BaseControl.tsx   |  1 +
 .../formControls/Rag/RagDocumentsSelector.tsx | 11 +++++++++++
 .../ee/components/formControls/Rag/index.ts   |  1 +
 app/client/src/pages/Editor/FormControl.tsx   |  1 +
 .../utils/formControl/FormControlRegistry.tsx | 19 ++++++++++++++++++-
 .../src/utils/formControl/formControlTypes.ts |  1 +
 6 files changed, 33 insertions(+), 1 deletion(-)
 create mode 100644 app/client/src/ee/components/formControls/Rag/RagDocumentsSelector.tsx

diff --git a/app/client/src/components/formControls/BaseControl.tsx b/app/client/src/components/formControls/BaseControl.tsx
index 50189b9e6a65..874c4d9a3d55 100644
--- a/app/client/src/components/formControls/BaseControl.tsx
+++ b/app/client/src/components/formControls/BaseControl.tsx
@@ -111,6 +111,7 @@ export interface ControlData {
   addMoreButtonLabel?: string;
   datasourceId?: string;
   workspaceId?: string;
+  actionId?: string;
 }
 export type FormConfigType = Omit<ControlData, "configProperty"> & {
   configProperty?: string;
diff --git a/app/client/src/ee/components/formControls/Rag/RagDocumentsSelector.tsx b/app/client/src/ee/components/formControls/Rag/RagDocumentsSelector.tsx
new file mode 100644
index 000000000000..d8966699fe81
--- /dev/null
+++ b/app/client/src/ee/components/formControls/Rag/RagDocumentsSelector.tsx
@@ -0,0 +1,11 @@
+interface RagDocumentsSelectorProps {
+  workspaceId?: string;
+  actionId?: string;
+  datasourceId?: string;
+}
+
+// Used in EE
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+export const RagDocumentsSelector = (props: RagDocumentsSelectorProps) => {
+  return null;
+};
diff --git a/app/client/src/ee/components/formControls/Rag/index.ts b/app/client/src/ee/components/formControls/Rag/index.ts
index 9200f00520eb..645877874f24 100644
--- a/app/client/src/ee/components/formControls/Rag/index.ts
+++ b/app/client/src/ee/components/formControls/Rag/index.ts
@@ -1,2 +1,3 @@
 export { RagDocuments } from "./RagDocuments";
 export { RagIntegrations } from "./RagIntegrations";
+export { RagDocumentsSelector } from "./RagDocumentsSelector";
diff --git a/app/client/src/pages/Editor/FormControl.tsx b/app/client/src/pages/Editor/FormControl.tsx
index f849891e2169..de05b6a37365 100644
--- a/app/client/src/pages/Editor/FormControl.tsx
+++ b/app/client/src/pages/Editor/FormControl.tsx
@@ -184,6 +184,7 @@ function FormControl(props: FormControlProps) {
         ...config,
         datasourceId: dsId,
         workspaceId,
+        actionId: actionValues?.id,
       },
       props.formName,
       props?.multipleConfig,
diff --git a/app/client/src/utils/formControl/FormControlRegistry.tsx b/app/client/src/utils/formControl/FormControlRegistry.tsx
index 391e64f159c7..de045982861a 100644
--- a/app/client/src/utils/formControl/FormControlRegistry.tsx
+++ b/app/client/src/utils/formControl/FormControlRegistry.tsx
@@ -37,7 +37,10 @@ import MultiFilePickerControl from "components/formControls/MultiFilePickerContr
 import type { MultipleFilePickerControlProps } from "components/formControls/MultiFilePickerControl";
 import type { RadioButtonControlProps } from "components/formControls/RadioButtonControl";
 import RadioButtonControl from "components/formControls/RadioButtonControl";
-import { RagIntegrations } from "ee/components/formControls/Rag";
+import {
+  RagIntegrations,
+  RagDocumentsSelector,
+} from "ee/components/formControls/Rag";
 import {
   SliderControl,
   type SliderControlProps,
@@ -208,6 +211,20 @@ class FormControlRegistry {
         return <SliderControl {...controlProps} />;
       },
     });
+    FormControlFactory.registerControlBuilder(
+      formControlTypes.RAG_DOCUMENTS_SELECTOR,
+      {
+        buildPropertyControl(controlProps): JSX.Element {
+          return (
+            <RagDocumentsSelector
+              actionId={controlProps.actionId}
+              datasourceId={controlProps.datasourceId}
+              workspaceId={controlProps.workspaceId}
+            />
+          );
+        },
+      },
+    );
   }
 }
 
diff --git a/app/client/src/utils/formControl/formControlTypes.ts b/app/client/src/utils/formControl/formControlTypes.ts
index ee81c6d24c74..830495d431d2 100644
--- a/app/client/src/utils/formControl/formControlTypes.ts
+++ b/app/client/src/utils/formControl/formControlTypes.ts
@@ -21,4 +21,5 @@ export default {
   RADIO_BUTTON: "RADIO_BUTTON",
   RAG_INTEGRATIONS: "RAG_INTEGRATIONS",
   SLIDER: "SLIDER",
+  RAG_DOCUMENTS_SELECTOR: "RAG_DOCUMENTS_SELECTOR",
 };