Fix Citation Format for Prompt Flow Endpoint Responses (#1092)

Co-authored-by: Anushree Kaipa Ramanath <[email protected]>
microsoft · Sep 11, 2024 · a11773a · a11773a
1 parent ba64e7c
commit a11773a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backend/utils.py b/backend/utils.py
@@ -166,10 +166,12 @@ def format_pf_non_streaming_response(
                 "content": chatCompletion[response_field_name] 
             })
         if citations_field_name in chatCompletion:
+            citation_content= {"citations": chatCompletion[citations_field_name]}
             messages.append({ 
                 "role": "tool",
-                "content": chatCompletion[citations_field_name]
+                "content": json.dumps(citation_content)
             })
+
         response_obj = {
             "id": chatCompletion["id"],
             "model": "",