Merge branch 'main' into fix/enable-handle-fs-connector

Zipstack · Jul 11, 2024 · abeb28c · abeb28c
2 parents 510b652 + f66c8ff
commit abeb28c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/backend/prompt_studio/prompt_studio_core/constants.py b/backend/prompt_studio/prompt_studio_core/constants.py
@@ -88,6 +88,7 @@ class ToolStudioPromptKeys:
     PROFILE_MANAGER_ID = "profile_manager"
     CONTEXT = "context"
     METADATA = "metadata"
+    INCLUDE_METADATA = "include_metadata"
 
 
 class FileViewTypes:

diff --git a/backend/prompt_studio/prompt_studio_core/prompt_studio_helper.py b/backend/prompt_studio/prompt_studio_core/prompt_studio_helper.py
@@ -731,8 +731,9 @@ def _fetch_response(
             prompt_host=settings.PROMPT_HOST,
             prompt_port=settings.PROMPT_PORT,
         )
+        include_metadata = {TSPKeys.INCLUDE_METADATA: True}
 
-        answer = responder.answer_prompt(payload)
+        answer = responder.answer_prompt(payload, include_metadata)
         # TODO: Make use of dataclasses
         if answer["status"] == "ERROR":
             # TODO: Publish to FE logs from here