FEAT: Simple Prompt Studio CRUD functionality

backend/backend/public_urls.py

@@ -22,6 +22,7 @@
 
 path_prefix = settings.PATH_PREFIX
 api_path_prefix = settings.API_DEPLOYMENT_PATH_PREFIX
+simple_prompt_studio_path_prefix = settings.SIMPLE_PROMPT_STUDIO_PATH_PREFIX
 
 urlpatterns = [
     path(f"{path_prefix}/", include("account.urls")),
@@ -47,10 +48,37 @@
 
 
 try:
-    import pluggable_apps.platform_admin.urls  # noqa: F401
+    import pluggable_apps.platform_admin.urls  # noqa # pylint: disable=unused-import
 
     urlpatterns += [
         path(f"{path_prefix}/", include("pluggable_apps.platform_admin.urls")),
     ]
 except ImportError:
     pass
+
+try:
+    import pluggable_apps.simple_prompt_studio.sps_document.urls  # noqa # pylint: disable=unused-import
+    import pluggable_apps.simple_prompt_studio.sps_project.urls  # noqa # pylint: disable=unused-import
+    import pluggable_apps.simple_prompt_studio.sps_prompt.urls  # noqa # pylint: disable=unused-import
+    import pluggable_apps.simple_prompt_studio.sps_prompt_output.urls  # noqa # pylint: disable=unused-import
+
+    urlpatterns += [
+        path(
+            f"{path_prefix}/{simple_prompt_studio_path_prefix}/",
+            include("pluggable_apps.simple_prompt_studio.sps_project.urls"),
+        ),
+        path(
+            f"{path_prefix}/{simple_prompt_studio_path_prefix}/",
+            include("pluggable_apps.simple_prompt_studio.sps_document.urls"),
+        ),
+        path(
+            f"{path_prefix}/{simple_prompt_studio_path_prefix}/",
+            include("pluggable_apps.simple_prompt_studio.sps_prompt.urls"),
+        ),
+        path(
+            f"{path_prefix}/{simple_prompt_studio_path_prefix}/",
+            include("pluggable_apps.simple_prompt_studio.sps_prompt_output.urls"),
+        ),
+    ]
+except ImportError:
+    pass

frontend/src/components/custom-tools/document-manager/DocumentManager.jsx

@@ -22,7 +22,7 @@ import { PdfViewer } from "../pdf-viewer/PdfViewer";
 import { TextViewerPre } from "../text-viewer-pre/TextViewerPre";
 import usePostHogEvents from "../../../hooks/usePostHogEvents";
 
-const items = [
+let items = [
   {
     key: "1",
     label: "PDF View",
@@ -92,22 +92,24 @@ function DocumentManager({ generateIndex, handleUpdateTool, handleDocChange }) {
 
   useEffect(() => {
     if (isSimplePromptStudio) {
-      items[0] = {
-        key: "1",
-        label: (
-          <Tooltip title="PDF View">
-            <FilePdfOutlined />
-          </Tooltip>
-        ),
-      };
-      items[1] = {
-        key: "2",
-        label: (
-          <Tooltip title="Raw View">
-            <FileTextOutlined />
-          </Tooltip>
-        ),
-      };
+      items = [
+        {
+          key: "1",
+          label: (
+            <Tooltip title="PDF View">
+              <FilePdfOutlined />
+            </Tooltip>
+          ),
+        },
+        {
+          key: "2",
+          label: (
+            <Tooltip title="Raw View">
+              <FileTextOutlined />
+            </Tooltip>
+          ),
+        },
+      ];
     }
   }, []);
 
@@ -172,7 +174,7 @@ function DocumentManager({ generateIndex, handleUpdateTool, handleDocChange }) {
   };
 
   const handleGetDocumentsReq = (getDocsFunc, viewType) => {
-    getDocsFunc(viewType)
+    getDocsFunc(details?.tool_id, selectedDoc?.document_id, viewType)
       .then((res) => {
         const data = res?.data?.data || "";
         processGetDocsResponse(data, viewType);
@@ -185,10 +187,10 @@ function DocumentManager({ generateIndex, handleUpdateTool, handleDocChange }) {
       });
   };
 
-  const getDocuments = async (viewType) => {
+  const getDocuments = async (toolId, docId, viewType) => {
     const requestOptions = {
       method: "GET",
-      url: `/api/v1/unstract/${sessionDetails?.orgId}/prompt-studio/file/${details?.tool_id}?document_id=${selectedDoc?.document_id}&view_type=${viewType}`,
+      url: `/api/v1/unstract/${sessionDetails?.orgId}/prompt-studio/file/${toolId}?document_id=${docId}&view_type=${viewType}`,
     };
 
     return axiosPrivate(requestOptions)

frontend/src/components/custom-tools/document-parser/DocumentParser.jsx

@@ -193,9 +193,13 @@ function DocumentParser({
   };
 
   const handleDelete = (promptId) => {
+    let url = promptUrl(promptId + "/");
+    if (isSimplePromptStudio) {
+      url = promptPatchApiSps(promptId);
+    }
     const requestOptions = {
       method: "DELETE",
-      url: promptUrl(promptId + "/"),
+      url,
       headers: {
         "X-CSRFToken": sessionDetails?.csrfToken,
       },

frontend/src/components/custom-tools/prompt-card/Header.jsx

@@ -41,6 +41,7 @@ function Header({
     singlePassExtractMode,
     isSinglePassExtractLoading,
     indexDocs,
+    isSimplePromptStudio,
   } = useCustomToolStore();
 
   const [isDisablePrompt, setIsDisablePrompt] = useState(promptDetails?.active);
@@ -157,30 +158,34 @@ function Header({
                 <PlayCircleOutlined className="prompt-card-actions-head" />
               </Button>
             </Tooltip>
-            <Tooltip title="Run All">
-              <Button
-                size="small"
-                type="text"
-                className="prompt-card-action-button"
-                onClick={() => handleRunBtnClick()}
-                disabled={
-                  (updateStatus?.promptId === promptDetails?.prompt_id &&
-                    updateStatus?.status ===
-                      promptStudioUpdateStatus?.isUpdating) ||
-                  disableLlmOrDocChange?.includes(promptDetails?.prompt_id) ||
-                  indexDocs?.includes(selectedDoc?.document_id)
-                }
-              >
-                <PlayCircleFilled className="prompt-card-actions-head" />
-              </Button>
-            </Tooltip>
+            {!isSimplePromptStudio && (
+              <Tooltip title="Run All">
+                <Button
+                  size="small"
+                  type="text"
+                  className="prompt-card-action-button"
+                  onClick={() => handleRunBtnClick()}
+                  disabled={
+                    (updateStatus?.promptId === promptDetails?.prompt_id &&
+                      updateStatus?.status ===
+                        promptStudioUpdateStatus?.isUpdating) ||
+                    disableLlmOrDocChange?.includes(promptDetails?.prompt_id) ||
+                    indexDocs?.includes(selectedDoc?.document_id)
+                  }
+                >
+                  <PlayCircleFilled className="prompt-card-actions-head" />
+                </Button>
+              </Tooltip>
+            )}
           </>
         )}
-        <Checkbox
-          checked={isDisablePrompt}
-          className="prompt-card-action-button"
-          onChange={handleDisablePrompt}
-        />
+        {!isSimplePromptStudio && (
+          <Checkbox
+            checked={isDisablePrompt}
+            className="prompt-card-action-button"
+            onChange={handleDisablePrompt}
+          />
+        )}
         <Divider type="vertical" className="header-delete-divider" />
         <ConfirmModal
           handleConfirm={() => handleDelete(promptDetails?.prompt_id)}

frontend/src/components/custom-tools/prompt-card/PromptCard.jsx

@@ -120,9 +120,6 @@ function PromptCard({
     resetInfoMsgs();
     handleGetOutput();
     handleGetCoverage();
-    if (isSinglePassExtractLoading) {
-      return;
-    }
   }, [
     selectedLlmProfileId,
     selectedDoc,

frontend/src/components/custom-tools/prompt-card/PromptCardItems.jsx

@@ -117,6 +117,11 @@ function PromptCardItems({
   };
 
   const getAdapterInfo = async (adapterData) => {
+    // If simple prompt studio, return early
+    if (isSimplePromptStudio) {
+      return;
+    }
+
     // Update llmProfiles with additional fields
     const updatedProfiles = llmProfiles?.map((profile) => {
       return { ...getModelOrAdapterId(profile, adapterData), ...profile };
@@ -233,6 +238,7 @@ function PromptCardItems({
   useEffect(() => {
     getAdapterInfo(adapters);
   }, [llmProfiles, selectedLlmProfileId, enabledProfiles]);
+
   return (
     <Card className="prompt-card">
       <div className="prompt-card-div prompt-card-bg-col1 prompt-card-rad">

frontend/src/setupProxy.js

@@ -8,11 +8,4 @@ module.exports = (app) => {
       changeOrigin: true,
     })
   );
-  app.use(
-    "/public",
-    createProxyMiddleware({
-      target: process.env.REACT_APP_BACKEND_URL,
-      changeOrigin: true,
-    })
-  );
 };