[FIX] Adding feature flag condition & fixing context parsing (#868)

* Fix: Context parsing with No chunks * Fix:Toggling feature flag for FS API --------- Co-authored-by: Gayathri <[email protected]>
Zipstack · Dec 4, 2024 · 7b6b64b · 7b6b64b
1 parent 159a9da
commit 7b6b64b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/backend/prompt_studio/prompt_studio_core_v2/views.py b/backend/prompt_studio/prompt_studio_core_v2/views.py
@@ -459,7 +459,7 @@ def upload_for_ide(self, request: HttpRequest, pk: Any = None) -> Response:
             logger.info(
                 f"Uploading file: {file_name}" if file_name else "Uploading file"
             )
-            if check_feature_flag_status(flag_key=FeatureFlag.REMOTE_FILE_STORAGE):
+            if not check_feature_flag_status(flag_key=FeatureFlag.REMOTE_FILE_STORAGE):
                 file_path = FileManagerHelper.handle_sub_directory_for_tenants(
                     UserSessionUtils.get_organization_id(request),
                     is_create=True,

diff --git a/backend/prompt_studio/prompt_studio_output_manager_v2/serializers.py b/backend/prompt_studio/prompt_studio_output_manager_v2/serializers.py
@@ -47,7 +47,9 @@ def to_representation(self, instance):
             data["coverage"] = {}
         # Convert string to list
         try:
-            data["context"] = json.loads(data["context"])
+            context = data["context"]
+            if context:
+                data["context"] = json.loads(data["context"])
         except json.JSONDecodeError:
             # Convert the old value of data["context"] to a list
             data["context"] = [data["context"]]