From 949b8070230e496db1509c7b3a31939977de94e2 Mon Sep 17 00:00:00 2001
From: Paul Kompfner <paul@daily.co>
Date: Fri, 24 Oct 2025 11:36:25 -0400
Subject: [PATCH] Close genai client more gracefully to avoid printed warnings.
 We're now following the genai library guidance:
 https://github.com/googleapis/python-genai?tab=readme-ov-file#close-a-client

---
 src/pipecat/services/google/llm.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/pipecat/services/google/llm.py b/src/pipecat/services/google/llm.py
index b45c276d0..c59cb41ae 100644
--- a/src/pipecat/services/google/llm.py
+++ b/src/pipecat/services/google/llm.py
@@ -1034,6 +1034,23 @@ class GoogleLLMService(LLMService):
         if context:
             await self._process_context(context)
 
+    async def stop(self, frame):
+        """Override stop to gracefully close the client."""
+        await super().stop(frame)
+        await self._close_client()
+
+    async def cancel(self, frame):
+        """Override cancel to gracefully close the client."""
+        await super().cancel(frame)
+        await self._close_client()
+
+    async def _close_client(self):
+        try:
+            await self._client.aio.aclose()
+        except Exception:
+            # Do nothing - we're shutting down anyway
+            pass
+
     def create_context_aggregator(
         self,
         context: OpenAILLMContext,