fix annotation reply (#2127)

Co-authored-by: jyong <jyong@dify.ai>
2 years ago · 741079f317
parent 0f5d4fd11b
commit 741079f317
1 changed files with 4 additions and 0 deletions
--- a/api/core/app_runner/generate_task_pipeline.py
+++ b/api/core/app_runner/generate_task_pipeline.py
@ -493,6 +493,10 @@ class GenerateTaskPipeline:
                        'score': resource['score'],
                        'content': resource['content'],
                    })
+        # show annotation reply
+        if 'annotation_reply' in self._task_state.metadata:
+            if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API]:
+                metadata['annotation_reply'] = self._task_state.metadata['annotation_reply']

        # show usage
        if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API]: