Merge pull request #2779 from LucasStringPay/patch-1

Ignore None value for 'completion_tokens' or similar for Gemini
2025-10-09 16:55:33 -07:00
parent 19ceb1a48f 607e3040d4
commit 9d2ad750b5
1 changed files with 3 additions and 3 deletions
--- a/src/pipecat/services/google/llm_openai.py
+++ b/src/pipecat/services/google/llm_openai.py
@@ -94,9 +94,9 @@ class GoogleLLMOpenAIBetaService(OpenAILLMService):
        async for chunk in chunk_stream:
            if chunk.usage:
                tokens = LLMTokenUsage(
-                    prompt_tokens=chunk.usage.prompt_tokens,
-                    completion_tokens=chunk.usage.completion_tokens,
-                    total_tokens=chunk.usage.total_tokens,
+                    prompt_tokens=chunk.usage.prompt_tokens or 0,
+                    completion_tokens=chunk.usage.completion_tokens or 0,
+                    total_tokens=chunk.usage.total_tokens or 0,
                )
                await self.start_llm_usage_metrics(tokens)