bring back user transcript logs (#1014)

livekit · Oct 30, 2024 · ee0ca5a · ee0ca5a
1 parent 5376f08
commit ee0ca5a
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 8 deletions.
diff --git a/examples/voice-pipeline-agent/cost_metrics.py b/examples/voice-pipeline-agent/cost_metrics.py
@@ -51,9 +51,9 @@ async def entrypoint(ctx: JobContext):
     usage_collector = metrics.UsageCollector()
 
     @agent.on("metrics_collected")
-    def _on_metrics_collected(metrics: metrics.AgentMetrics):
-        metrics.log_metrics(metrics)
-        usage_collector.add_usage(metrics)
+    def _on_metrics_collected(mtrcs: metrics.AgentMetrics):
+        metrics.log_metrics(mtrcs)
+        usage_collector.collect(mtrcs)
 
     async def log_session_cost():
         summary = usage_collector.get_summary()

diff --git a/examples/voice-pipeline-agent/minimal_assistant.py b/examples/voice-pipeline-agent/minimal_assistant.py
@@ -57,9 +57,9 @@ async def entrypoint(ctx: JobContext):
     usage_collector = metrics.UsageCollector()
 
     @agent.on("metrics_collected")
-    def _on_metrics_collected(metrics: metrics.AgentMetrics):
-        metrics.log_metrics(metrics)
-        usage_collector.add_usage(metrics)
+    def _on_metrics_collected(mtrcs: metrics.AgentMetrics):
+        metrics.log_metrics(mtrcs)
+        usage_collector.collect(mtrcs)
 
     async def log_usage():
         summary = usage_collector.get_summary()

diff --git a/livekit-agents/livekit/agents/metrics/usage_collector.py b/livekit-agents/livekit/agents/metrics/usage_collector.py
@@ -17,9 +17,9 @@ def __init__(self) -> None:
         self._summary = UsageSummary(0, 0, 0, 0.0)
 
     def __call__(self, metrics: AgentMetrics) -> None:
-        self.add_usage(metrics)
+        self.collect(metrics)
 
-    def add_usage(self, metrics: AgentMetrics) -> None:
+    def collect(self, metrics: AgentMetrics) -> None:
         if isinstance(metrics, LLMMetrics):
             self._summary.llm_prompt_tokens += metrics.prompt_tokens
             self._summary.llm_completion_tokens += metrics.completion_tokens

diff --git a/livekit-agents/livekit/agents/pipeline/pipeline_agent.py b/livekit-agents/livekit/agents/pipeline/pipeline_agent.py
@@ -512,6 +512,11 @@ def _on_final_transcript(ev: stt.SpeechEvent) -> None:
             if not new_transcript:
                 return
 
+            logger.debug(
+                "received user transcript",
+                extra={"user_transcript": new_transcript},
+            )
+
             self._last_final_transcript_time = time.perf_counter()
 
             self._transcribed_text += (