processors(rtvi): send initial empty metrics

pipecat-ai · Jul 23, 2024 · 05d4fba · 05d4fba
1 parent f41c2b3
commit 05d4fba
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/pipecat/processors/frameworks/rtvi.py b/src/pipecat/processors/frameworks/rtvi.py
@@ -19,6 +19,7 @@
     LLMMessagesAppendFrame,
     LLMMessagesUpdateFrame,
     LLMModelUpdateFrame,
+    MetricsFrame,
     StartFrame,
     SystemFrame,
     TTSSpeakFrame,
@@ -456,6 +457,13 @@ async def _handle_setup(self, setup: RTVISetup | None):
             start_frame = dataclasses.replace(self._start_frame)
             await self.push_frame(start_frame)
 
+            # Send new initial metrics with the new processors
+            processors = parent.processors_with_metrics()
+            processors.extend(self._pipeline.processors_with_metrics())
+            ttfb = [{"name": p.name, "time": 0.0} for p in processors]
+            processing = [{"name": p.name, "time": 0.0} for p in processors]
+            await self.push_frame(MetricsFrame(ttfb=ttfb, processing=processing))
+
         message = RTVIBotReady()
         frame = TransportMessageFrame(message=message.model_dump(exclude_none=True))
         await self.push_frame(frame)