diff --git a/backend/prompt_studio/prompt_studio_core/prompt_studio_helper.py b/backend/prompt_studio/prompt_studio_core/prompt_studio_helper.py index c0f7b73d7..e1e8e76b5 100644 --- 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