feat: optimize ping event

5ae263bb · John Wang · b63f1fa5 · 5ae263bb
Commit 5ae263bb authored Jul 27, 2023 by John Wang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

completion_service.py api/services/completion_service.py +2 -0

No files found.
--- a/api/services/completion_service.py
+++ b/api/services/completion_service.py
@@ -424,6 +424,8 @@ class CompletionService:
                                yield "data: " + json.dumps(cls.get_chain_response_data(result.get('data'))) + "\n\n"
                            elif event == 'agent_thought':
                                yield "data: " + json.dumps(cls.get_agent_thought_response_data(result.get('data'))) + "\n\n"
+                            elif event == 'ping':
+                                yield "event: ping\n\n"
                            else:
                                yield "data: " + json.dumps(result) + "\n\n"
                except ValueError as e: