diff --git a/backend/api_v2/api_deployment_views.py b/backend/api_v2/api_deployment_views.py
index 5750f6b7f..c3265e75d 100644
--- a/backend/api_v2/api_deployment_views.py
+++ b/backend/api_v2/api_deployment_views.py
@@ -49,10 +49,9 @@ def post(
         file_objs = request.FILES.getlist(ApiExecution.FILES_FORM_DATA)
         serializer = ExecutionRequestSerializer(data=request.data)
         serializer.is_valid(raise_exception=True)
-        timeout = serializer.get_timeout(serializer.validated_data)
-        include_metadata = (
-            request.data.get(ApiExecution.INCLUDE_METADATA, "false").lower() == "true"
-        )
+        timeout = serializer.validated_data.get(ApiExecution.TIMEOUT_FORM_DATA)
+        include_metadata = serializer.validated_data.get(ApiExecution.INCLUDE_METADATA)
+        use_file_history = serializer.validated_data.get(ApiExecution.USE_FILE_HISTORY)
         if not file_objs or len(file_objs) == 0:
             raise InvalidAPIRequest("File shouldn't be empty")
         response = DeploymentHelper.execute_workflow(
@@ -61,6 +60,7 @@ def post(
             file_objs=file_objs,
             timeout=timeout,
             include_metadata=include_metadata,
+            use_file_history=use_file_history,
         )
         if "error" in response and response["error"]:
             return Response(
diff --git a/backend/api_v2/constants.py b/backend/api_v2/constants.py
index 0ec324cc9..2de41efb2 100644
--- a/backend/api_v2/constants.py
+++ b/backend/api_v2/constants.py
@@ -4,3 +4,4 @@ class ApiExecution:
     FILES_FORM_DATA: str = "files"
     TIMEOUT_FORM_DATA: str = "timeout"
     INCLUDE_METADATA: str = "include_metadata"
+    USE_FILE_HISTORY: str = "use_file_history"  # Undocumented parameter
diff --git a/backend/api_v2/deployment_helper.py b/backend/api_v2/deployment_helper.py
index 47ec0615e..b35bb0a3a 100644
--- a/backend/api_v2/deployment_helper.py
+++ b/backend/api_v2/deployment_helper.py
@@ -131,6 +131,7 @@ def execute_workflow(
         file_objs: list[UploadedFile],
         timeout: int,
         include_metadata: bool = False,
+        use_file_history: bool = False,
     ) -> ReturnDict:
         """Execute workflow by api.
 
@@ -138,6 +139,8 @@ def execute_workflow(
             organization_name (str): organization name
             api (APIDeployment): api model object
             file_obj (UploadedFile): input file
+            use_file_history (bool): Use FileHistory table to return results on already
+                processed files. Defaults to False
 
         Returns:
             ReturnDict: execution status/ result
@@ -150,6 +153,7 @@ def execute_workflow(
             workflow_id=workflow_id,
             execution_id=execution_id,
             file_objs=file_objs,
+            use_file_history=use_file_history,
         )
         try:
             result = WorkflowHelper.execute_workflow_async(
@@ -159,6 +163,7 @@ def execute_workflow(
                 timeout=timeout,
                 execution_id=execution_id,
                 include_metadata=include_metadata,
+                use_file_history=use_file_history,
             )
             result.status_api = DeploymentHelper.construct_status_endpoint(
                 api_endpoint=api.api_endpoint, execution_id=execution_id
diff --git a/backend/api_v2/serializers.py b/backend/api_v2/serializers.py
index 4273b6093..88cc22dfb 100644
--- a/backend/api_v2/serializers.py
+++ b/backend/api_v2/serializers.py
@@ -6,6 +6,7 @@
 from django.core.validators import RegexValidator
 from pipeline_v2.models import Pipeline
 from rest_framework.serializers import (
+    BooleanField,
     CharField,
     IntegerField,
     JSONField,
@@ -80,26 +81,22 @@ def to_representation(self, instance: APIKey) -> OrderedDict[str, Any]:
 
 
 class ExecutionRequestSerializer(Serializer):
-    """Execution request serializer
-    timeout: 0: maximum value of timeout, -1: async execution
+    """Execution request serializer.
+
+    Attributes:
+        timeout (int): Timeout for the API deployment, maximum value can be 300s.
+            If -1 it corresponds to async execution. Defaults to -1
+        include_metadata (bool): Flag to include metadata in API response
+        use_file_history (bool): Flag to use FileHistory to save and retrieve
+            responses quickly. This is undocumented to the user and can be
+            helpful for demos.
     """
 
     timeout = IntegerField(
         min_value=-1, max_value=ApiExecution.MAXIMUM_TIMEOUT_IN_SEC, default=-1
     )
-
-    def validate_timeout(self, value: Any) -> int:
-        if not isinstance(value, int):
-            raise ValidationError("timeout must be a integer.")
-        if value == 0:
-            value = ApiExecution.MAXIMUM_TIMEOUT_IN_SEC
-        return value
-
-    def get_timeout(self, validated_data: dict[str, Union[int, None]]) -> int:
-        value = validated_data.get(ApiExecution.TIMEOUT_FORM_DATA, -1)
-        if not isinstance(value, int):
-            raise ValidationError("timeout must be a integer.")
-        return value
+    include_metadata = BooleanField(default=False)
+    use_file_history = BooleanField(default=False)
 
 
 class APIDeploymentListSerializer(ModelSerializer):
diff --git a/backend/backend/custom_db/__init__.py b/backend/backend/custom_db/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/backend/backend/custom_db/base.py b/backend/backend/custom_db/base.py
new file mode 100644
index 000000000..5ff93aad9
--- /dev/null
+++ b/backend/backend/custom_db/base.py
@@ -0,0 +1,52 @@
+import logging
+
+from django.conf import settings
+from django.db.backends.postgresql.base import (
+    DatabaseWrapper as PostgresDatabaseWrapper,
+)
+
+logger = logging.getLogger(__name__)
+
+
+class DatabaseWrapper(PostgresDatabaseWrapper):
+    """Custom DatabaseWrapper to manage PostgreSQL connections and set the
+    search path."""
+
+    def get_new_connection(self, conn_params):
+        """Establish a new database connection or reuse an existing one, and
+        set the search path.
+
+        Args:
+            conn_params: Parameters for the new database connection.
+
+        Returns:
+            connection: The database connection
+        """
+        connection = super().get_new_connection(conn_params)
+        logger.info(f"DB connection (ID: {id(connection)}) is established or reused.")
+        self.set_search_path(connection)
+        return connection
+
+    def set_search_path(self, connection):
+        """Set the search path for the given database connection.
+
+        This ensures that the database queries will look in the specified schema.
+
+        Args:
+            connection: The database connection for which to set the search path.
+        """
+        conn_id = id(connection)
+        original_autocommit = connection.autocommit
+        try:
+            connection.autocommit = True
+            logger.debug(
+                f"Setting search_path to {settings.DB_SCHEMA} for DB connection ID "
+                f"{conn_id}."
+            )
+            with connection.cursor() as cursor:
+                cursor.execute(f"SET search_path TO {settings.DB_SCHEMA}")
+            logger.debug(
+                f"Successfully set search_path for DB connection ID {conn_id}."
+            )
+        finally:
+            connection.autocommit = original_autocommit
diff --git a/backend/backend/settings/base.py b/backend/backend/settings/base.py
index b78d8d2fe..f45dbf46d 100644
--- a/backend/backend/settings/base.py
+++ b/backend/backend/settings/base.py
@@ -375,7 +375,7 @@ def get_required_setting(
     ROOT_URLCONF = "backend.base_urls"
 
     # DB Configuration
-    DB_ENGINE = "django.db.backends.postgresql"
+    DB_ENGINE = "backend.custom_db"
 
     # Models
     AUTH_USER_MODEL = "account_v2.User"
@@ -397,7 +397,6 @@ def get_required_setting(
             "PORT": f"{DB_PORT}",
             "ATOMIC_REQUESTS": ATOMIC_REQUESTS,
             "OPTIONS": {
-                "options": f"-c search_path={DB_SCHEMA}",
                 "application_name": os.environ.get("APPLICATION_NAME", ""),
             },
         }
diff --git a/backend/entrypoint.sh b/backend/entrypoint.sh
index ebbb38c3f..686755124 100755
--- a/backend/entrypoint.sh
+++ b/backend/entrypoint.sh
@@ -16,5 +16,4 @@ fi
     --log-level debug \
     --timeout 600 \
     --access-logfile - \
-    --reload \
     backend.wsgi:application
diff --git a/backend/prompt_studio/prompt_studio_core_v2/views.py b/backend/prompt_studio/prompt_studio_core_v2/views.py
index 2b36f3f53..f5437a304 100644
--- a/backend/prompt_studio/prompt_studio_core_v2/views.py
+++ b/backend/prompt_studio/prompt_studio_core_v2/views.py
@@ -529,12 +529,15 @@ def export_tool(self, request: Request, pk: Any = None) -> Response:
         serializer.is_valid(raise_exception=True)
         is_shared_with_org: bool = serializer.validated_data.get("is_shared_with_org")
         user_ids = set(serializer.validated_data.get("user_id"))
+        force_export = serializer.validated_data.get("force_export")
 
         PromptStudioRegistryHelper.update_or_create_psr_tool(
             custom_tool=custom_tool,
             shared_with_org=is_shared_with_org,
             user_ids=user_ids,
+            force_export=force_export,
         )
+
         return Response(
             {"message": "Custom tool exported sucessfully."},
             status=status.HTTP_200_OK,
diff --git a/backend/prompt_studio/prompt_studio_registry/exceptions.py b/backend/prompt_studio/prompt_studio_registry/exceptions.py
index 93086d815..cb0e3c71b 100644
--- a/backend/prompt_studio/prompt_studio_registry/exceptions.py
+++ b/backend/prompt_studio/prompt_studio_registry/exceptions.py
@@ -20,8 +20,8 @@ class EmptyToolExportError(APIException):
     status_code = 500
     default_detail = (
         "Prompt Studio project without prompts cannot be exported. "
-        "Please ensure there is at least one prompt and "
-        "it is active before exporting."
+        "Please ensure there is at least one active prompt "
+        "that has been run before exporting."
     )
 
 
diff --git a/backend/prompt_studio/prompt_studio_registry/prompt_studio_registry_helper.py b/backend/prompt_studio/prompt_studio_registry/prompt_studio_registry_helper.py
index af00cffa6..941fd70e9 100644
--- a/backend/prompt_studio/prompt_studio_registry/prompt_studio_registry_helper.py
+++ b/backend/prompt_studio/prompt_studio_registry/prompt_studio_registry_helper.py
@@ -292,6 +292,9 @@ def frame_export_json(
                 invalidated_prompts.append(prompt.prompt_key)
                 continue
 
+            if not prompt.profile_manager:
+                prompt.profile_manager = default_llm_profile
+
             if not force_export:
                 prompt_output = PromptStudioOutputManager.objects.filter(
                     tool_id=tool.tool_id,
@@ -302,9 +305,6 @@ def frame_export_json(
                     invalidated_outputs.append(prompt.prompt_key)
                     continue
 
-            if not prompt.profile_manager:
-                prompt.profile_manager = default_llm_profile
-
             vector_db = str(prompt.profile_manager.vector_store.id)
             embedding_model = str(prompt.profile_manager.embedding_model.id)
             llm = str(prompt.profile_manager.llm.id)
diff --git a/backend/prompt_studio/prompt_studio_registry_v2/exceptions.py b/backend/prompt_studio/prompt_studio_registry_v2/exceptions.py
index 93086d815..cb0e3c71b 100644
--- a/backend/prompt_studio/prompt_studio_registry_v2/exceptions.py
+++ b/backend/prompt_studio/prompt_studio_registry_v2/exceptions.py
@@ -20,8 +20,8 @@ class EmptyToolExportError(APIException):
     status_code = 500
     default_detail = (
         "Prompt Studio project without prompts cannot be exported. "
-        "Please ensure there is at least one prompt and "
-        "it is active before exporting."
+        "Please ensure there is at least one active prompt "
+        "that has been run before exporting."
     )
 
 
diff --git a/backend/prompt_studio/prompt_studio_registry_v2/prompt_studio_registry_helper.py b/backend/prompt_studio/prompt_studio_registry_v2/prompt_studio_registry_helper.py
index 3854e5874..c9de231cd 100644
--- a/backend/prompt_studio/prompt_studio_registry_v2/prompt_studio_registry_helper.py
+++ b/backend/prompt_studio/prompt_studio_registry_v2/prompt_studio_registry_helper.py
@@ -136,7 +136,10 @@ def get_tool_by_prompt_registry_id(
 
     @staticmethod
     def update_or_create_psr_tool(
-        custom_tool: CustomTool, shared_with_org: bool, user_ids: set[int]
+        custom_tool: CustomTool,
+        shared_with_org: bool,
+        user_ids: set[int],
+        force_export: bool,
     ) -> PromptStudioRegistry:
         """Updates or creates the PromptStudioRegistry record.
 
@@ -144,7 +147,13 @@ def update_or_create_psr_tool(
         1:1 with the `CustomTool`.
 
         Args:
-            tool_id (str): ID of the custom tool.
+            custom_tool (CustomTool): The instance of the custom tool to be updated
+                or created.
+            shared_with_org (bool): Flag indicating whether the tool is shared with
+                the organization.
+            user_ids (set[int]): A set of user IDs to whom the tool is shared.
+            force_export (bool): Indicates if the export is being forced.
+
 
         Raises:
             ToolSaveError
@@ -162,7 +171,7 @@ def update_or_create_psr_tool(
                 tool_id=custom_tool.tool_id
             )
             metadata = PromptStudioRegistryHelper.frame_export_json(
-                tool=custom_tool, prompts=prompts
+                tool=custom_tool, prompts=prompts, force_export=force_export
             )
 
             obj: PromptStudioRegistry
@@ -208,7 +217,9 @@ def update_or_create_psr_tool(
 
     @staticmethod
     def frame_export_json(
-        tool: CustomTool, prompts: list[ToolStudioPrompt]
+        tool: CustomTool,
+        prompts: list[ToolStudioPrompt],
+        force_export: bool,
     ) -> dict[str, Any]:
         export_metadata = {}
 
@@ -283,19 +294,19 @@ def frame_export_json(
                 invalidated_prompts.append(prompt.prompt_key)
                 continue
 
-            prompt_output = PromptStudioOutputManager.objects.filter(
-                tool_id=tool.tool_id,
-                prompt_id=prompt.prompt_id,
-                profile_manager=prompt.profile_manager,
-            ).all()
-
-            if not prompt_output:
-                invalidated_outputs.append(prompt.prompt_key)
-                continue
-
             if not prompt.profile_manager:
                 prompt.profile_manager = default_llm_profile
 
+            if not force_export:
+                prompt_output = PromptStudioOutputManager.objects.filter(
+                    tool_id=tool.tool_id,
+                    prompt_id=prompt.prompt_id,
+                    profile_manager=prompt.profile_manager,
+                ).all()
+                if not prompt_output:
+                    invalidated_outputs.append(prompt.prompt_key)
+                    continue
+
             vector_db = str(prompt.profile_manager.vector_store.id)
             embedding_model = str(prompt.profile_manager.embedding_model.id)
             llm = str(prompt.profile_manager.llm.id)
@@ -354,10 +365,12 @@ def frame_export_json(
                 f"Cannot export tool. Prompt(s): {', '.join(invalidated_prompts)} "
                 "are empty. Please enter a valid prompt."
             )
-        if invalidated_outputs:
+        if not force_export and invalidated_outputs:
             raise InValidCustomToolError(
-                f"Cannot export tool. Prompt(s): {', '.join(invalidated_outputs)} "
-                "were not run. Please run them before exporting."
+                detail="Cannot export tool. Prompt(s):"
+                f" {', '.join(invalidated_outputs)}"
+                " were not run. Please run them before exporting.",
+                code="warning",
             )
         export_metadata[JsonSchemaKey.TOOL_SETTINGS] = tool_settings
         export_metadata[JsonSchemaKey.OUTPUTS] = outputs
diff --git a/backend/prompt_studio/prompt_studio_registry_v2/serializers.py b/backend/prompt_studio/prompt_studio_registry_v2/serializers.py
index e995bb266..bbabe78a2 100644
--- a/backend/prompt_studio/prompt_studio_registry_v2/serializers.py
+++ b/backend/prompt_studio/prompt_studio_registry_v2/serializers.py
@@ -36,3 +36,4 @@ def get_prompt_studio_users(self, obj: PromptStudioRegistry) -> Any:
 class ExportToolRequestSerializer(serializers.Serializer):
     is_shared_with_org = serializers.BooleanField(default=False)
     user_id = serializers.ListField(child=serializers.IntegerField(), required=False)
+    force_export = serializers.BooleanField(default=False)
diff --git a/backend/usage_v2/helper.py b/backend/usage_v2/helper.py
index 20182fc3d..fd217cd6f 100644
--- a/backend/usage_v2/helper.py
+++ b/backend/usage_v2/helper.py
@@ -61,4 +61,4 @@ def get_aggregated_token_count(run_id: str) -> dict:
         except Exception as e:
             # Handle any other exceptions that might occur during the execution
             logger.error(f"An unexpected error occurred for run_id {run_id}: {str(e)}")
-            raise APIException("An unexpected error occurred")
+            raise APIException("Error while aggregating token counts")
diff --git a/backend/workflow_manager/endpoint_v2/destination.py b/backend/workflow_manager/endpoint_v2/destination.py
index 6f3ec1d75..39a989d70 100644
--- a/backend/workflow_manager/endpoint_v2/destination.py
+++ b/backend/workflow_manager/endpoint_v2/destination.py
@@ -152,8 +152,9 @@ def handle_output(
         file_name: str,
         file_hash: FileHash,
         workflow: Workflow,
+        input_file_path: str,
         error: Optional[str] = None,
-        input_file_path: Optional[str] = None,
+        use_file_history: bool = True,
     ) -> None:
         """Handle the output based on the connection type."""
         connection_type = self.endpoint.connection_type
@@ -163,9 +164,12 @@ def handle_output(
             if connection_type == WorkflowEndpoint.ConnectionType.API:
                 self._handle_api_result(file_name=file_name, error=error, result=result)
             return
-        file_history = FileHistoryHelper.get_file_history(
-            workflow=workflow, cache_key=file_hash.file_hash
-        )
+
+        file_history = None
+        if use_file_history:
+            file_history = FileHistoryHelper.get_file_history(
+                workflow=workflow, cache_key=file_hash.file_hash
+            )
         if connection_type == WorkflowEndpoint.ConnectionType.FILESYSTEM:
             self.copy_output_to_output_directory()
         elif connection_type == WorkflowEndpoint.ConnectionType.DATABASE:
@@ -188,7 +192,7 @@ def handle_output(
             self.execution_service.publish_log(
                 message=f"File '{file_name}' processed successfully"
             )
-        if not file_history:
+        if use_file_history and not file_history:
             FileHistoryHelper.create_file_history(
                 cache_key=file_hash.file_hash,
                 workflow=workflow,
diff --git a/backend/workflow_manager/endpoint_v2/source.py b/backend/workflow_manager/endpoint_v2/source.py
index 4ba76a488..e7808b607 100644
--- a/backend/workflow_manager/endpoint_v2/source.py
+++ b/backend/workflow_manager/endpoint_v2/source.py
@@ -485,9 +485,7 @@ def add_input_from_api_storage_to_volume(self, input_file_path: str) -> None:
         shutil.copyfile(input_file_path, infile_path)
         shutil.copyfile(input_file_path, source_path)
 
-    def add_file_to_volume(
-        self, input_file_path: str, file_hash: FileHash
-    ) -> tuple[str, str]:
+    def add_file_to_volume(self, input_file_path: str, file_hash: FileHash) -> str:
         """Add input file to execution directory.
 
         Args:
@@ -555,7 +553,11 @@ def load_file(self, input_file_path: str) -> tuple[str, BytesIO]:
 
     @classmethod
     def add_input_file_to_api_storage(
-        cls, workflow_id: str, execution_id: str, file_objs: list[UploadedFile]
+        cls,
+        workflow_id: str,
+        execution_id: str,
+        file_objs: list[UploadedFile],
+        use_file_history: bool = False,
     ) -> dict[str, FileHash]:
         """Add input file to api storage.
 
@@ -563,6 +565,8 @@ def add_input_file_to_api_storage(
             workflow_id (str): UUID of the worklfow
             execution_id (str): UUID of the execution
             file_objs (list[UploadedFile]): List of uploaded files
+            use_file_history (bool): Use FileHistory table to return results on already
+                processed files. Defaults to False
         Returns:
             dict[str, FileHash]: Dict containing file name and its corresponding hash
         """
@@ -582,9 +586,12 @@ def add_input_file_to_api_storage(
                 f.write(buffer)
             file_hash = cls.hash_str(buffer)
             connection_type = WorkflowEndpoint.ConnectionType.API
-            file_history = FileHistoryHelper.get_file_history(
-                workflow=workflow, cache_key=file_hash
-            )
+
+            file_history = None
+            if use_file_history:
+                file_history = FileHistoryHelper.get_file_history(
+                    workflow=workflow, cache_key=file_hash
+                )
             is_executed = (
                 True if file_history and file_history.is_completed() else False
             )
diff --git a/backend/workflow_manager/workflow_v2/execution.py b/backend/workflow_manager/workflow_v2/execution.py
index 3af07d748..e71b198f4 100644
--- a/backend/workflow_manager/workflow_v2/execution.py
+++ b/backend/workflow_manager/workflow_v2/execution.py
@@ -34,6 +34,7 @@ def __init__(
         scheduled: bool = False,
         mode: tuple[str, str] = WorkflowExecution.Mode.INSTANT,
         workflow_execution: Optional[WorkflowExecution] = None,
+        use_file_history: bool = True,
     ) -> None:
         tool_instances_as_dto = []
         for tool_instance in tool_instances:
@@ -95,6 +96,7 @@ def __init__(
         self.project_settings = project_settings
         self.pipeline_id = pipeline_id
         self.execution_id = str(workflow_execution.id)
+        self.use_file_history = use_file_history
         logger.info(
             f"Executing for Pipeline ID: {pipeline_id}, "
             f"workflow ID: {self.workflow_id}, execution ID: {self.execution_id}, "
diff --git a/backend/workflow_manager/workflow_v2/workflow_helper.py b/backend/workflow_manager/workflow_v2/workflow_helper.py
index 0ac9b79ec..ca41c9f95 100644
--- a/backend/workflow_manager/workflow_v2/workflow_helper.py
+++ b/backend/workflow_manager/workflow_v2/workflow_helper.py
@@ -92,6 +92,7 @@ def build_workflow_execution_service(
         scheduled: bool,
         execution_mode: tuple[str, str],
         workflow_execution: Optional[WorkflowExecution],
+        use_file_history: bool = True,  # Will be False for API deployment alone
     ) -> WorkflowExecutionServiceHelper:
         workflow_execution_service = WorkflowExecutionServiceHelper(
             organization_id=organization_id,
@@ -102,6 +103,7 @@ def build_workflow_execution_service(
             scheduled=scheduled,
             mode=execution_mode,
             workflow_execution=workflow_execution,
+            use_file_history=use_file_history,
         )
         workflow_execution_service.build()
         return workflow_execution_service
@@ -218,8 +220,9 @@ def process_file(
             file_name=file_name,
             file_hash=file_hash,
             workflow=workflow,
-            error=error,
             input_file_path=input_file,
+            error=error,
+            use_file_history=execution_service.use_file_history,
         )
         execution_service.publish_update_log(
             LogState.SUCCESS,
@@ -249,6 +252,7 @@ def run_workflow(
         single_step: bool = False,
         workflow_execution: Optional[WorkflowExecution] = None,
         execution_mode: Optional[tuple[str, str]] = None,
+        use_file_history: bool = True,
     ) -> ExecutionResponse:
         tool_instances: list[ToolInstance] = (
             ToolInstanceHelper.get_tool_instances_by_workflow(
@@ -267,6 +271,7 @@ def run_workflow(
             scheduled=scheduled,
             execution_mode=execution_mode,
             workflow_execution=workflow_execution,
+            use_file_history=use_file_history,
         )
         execution_id = execution_service.execution_id
         source = SourceConnector(
@@ -391,6 +396,7 @@ def execute_workflow_async(
         timeout: int = -1,
         pipeline_id: Optional[str] = None,
         queue: Optional[str] = None,
+        use_file_history: bool = True,
     ) -> ExecutionResponse:
         """Adding a workflow to the queue for execution.
 
@@ -399,6 +405,9 @@ def execute_workflow_async(
             execution_id (str): Execution ID
             timeout (int):  Celery timeout (timeout -1 : async execution)
             pipeline_id (Optional[str], optional): Optional pipeline. Defaults to None.
+            queue (Optional[str]): Name of the celery queue to push into
+            use_file_history (bool): Use FileHistory table to return results on already
+                processed files. Defaults to True
 
         Returns:
             ExecutionResponse: Existing status of execution
@@ -421,6 +430,7 @@ def execute_workflow_async(
                     "execution_mode": None,
                     "pipeline_id": pipeline_id,
                     "log_events_id": log_events_id,
+                    "use_file_history": use_file_history,
                 },
                 queue=queue,
             )
@@ -482,6 +492,7 @@ def execute_bin(
         scheduled: bool = False,
         execution_mode: Optional[tuple[str, str]] = None,
         pipeline_id: Optional[str] = None,
+        use_file_history: bool = True,
         **kwargs: dict[str, Any],
     ) -> Optional[list[Any]]:
         """Asynchronous Execution By celery.
@@ -495,6 +506,8 @@ def execute_bin(
             execution_mode (Optional[WorkflowExecution.Mode]): WorkflowExecution Mode
                 Defaults to None
             pipeline_id (Optional[str], optional): Id of pipeline. Defaults to None
+            use_file_history (bool): Use FileHistory table to return results on already
+                processed files. Defaults to True
 
         Kwargs:
             log_events_id (str): Session ID of the user,
@@ -515,6 +528,7 @@ def execute_bin(
             scheduled=scheduled,
             execution_mode=execution_mode,
             pipeline_id=pipeline_id,
+            use_file_history=use_file_history,
             **kwargs,
         )
 
diff --git a/frontend/src/components/custom-tools/prompt-card/Header.jsx b/frontend/src/components/custom-tools/prompt-card/Header.jsx
index 0397d9019..906434ce1 100644
--- a/frontend/src/components/custom-tools/prompt-card/Header.jsx
+++ b/frontend/src/components/custom-tools/prompt-card/Header.jsx
@@ -56,7 +56,7 @@ function Header({
   } = useCustomToolStore();
   const [items, setItems] = useState([]);
 
-  const [isDisablePrompt, setIsDisablePrompt] = useState(promptDetails?.active);
+  const [isDisablePrompt, setIsDisablePrompt] = useState(null);
 
   const handleRunBtnClick = (profileManager = null, coverAllDoc = true) => {
     setExpandCard(true);
@@ -72,6 +72,9 @@ function Header({
       }
     );
   };
+  useEffect(() => {
+    setIsDisablePrompt(promptDetails?.active);
+  }, [promptDetails, details]);
 
   useEffect(() => {
     const dropdownItems = [
@@ -105,7 +108,7 @@ function Header({
     }
 
     setItems(dropdownItems);
-  }, [promptDetails, details]);
+  }, [isDisablePrompt]);
 
   return (
     <Row>
diff --git a/platform-service/src/unstract/platform_service/controller/platform.py b/platform-service/src/unstract/platform_service/controller/platform.py
index 232bf2c2d..fb00d4b2d 100644
--- a/platform-service/src/unstract/platform_service/controller/platform.py
+++ b/platform-service/src/unstract/platform_service/controller/platform.py
@@ -499,7 +499,6 @@ def custom_tool_instance() -> Any:
 
         try:
             data_dict = PromptStudioRequestHelper.get_prompt_instance_from_db(
-                db_instance=db,
                 organization_id=organization_id,
                 prompt_registry_id=prompt_registry_id,
             )
diff --git a/prompt-service/entrypoint.sh b/prompt-service/entrypoint.sh
index 9a928875d..9c7e79fd3 100755
--- a/prompt-service/entrypoint.sh
+++ b/prompt-service/entrypoint.sh
@@ -8,5 +8,4 @@
     --log-level debug \
     --timeout 900 \
     --access-logfile - \
-    --reload \
     unstract.prompt_service.main:app