From a729834482bc7cc34a58094bc882ab0c93ae1272 Mon Sep 17 00:00:00 2001 From: Mert Sefa AKGUN Date: Wed, 8 Jan 2025 18:53:21 +0300 Subject: [PATCH] refactor(gemini): reposition WebSocket connection code Move WebSocket connection setup earlier in the function for better organization and to prepare for subsequent configuration steps. --- src/pipecat/services/gemini_multimodal_live/gemini.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/pipecat/services/gemini_multimodal_live/gemini.py b/src/pipecat/services/gemini_multimodal_live/gemini.py index 1387ffac8..a94399a1d 100644 --- a/src/pipecat/services/gemini_multimodal_live/gemini.py +++ b/src/pipecat/services/gemini_multimodal_live/gemini.py @@ -382,6 +382,10 @@ class GeminiMultimodalLiveLLMService(LLMService): # handle disconnections in the send/recv code paths. return + uri = f"wss://{self.base_url}/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key={self.api_key}" + logger.info(f"Connecting to {uri}") + self._websocket = await websockets.connect(uri=uri) + self._receive_task = self.get_event_loop().create_task(self._receive_task_handler()) config = events.Config.model_validate( { "setup": { @@ -403,10 +407,6 @@ class GeminiMultimodalLiveLLMService(LLMService): }, } ) - uri = f"wss://{self.base_url}/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key={self.api_key}" - logger.info(f"Connecting to {uri}") - self._websocket = await websockets.connect(uri=uri) - self._receive_task = self.get_event_loop().create_task(self._receive_task_handler()) system_instruction = self._system_instruction or "" if self._context and hasattr(self._context, "extract_system_instructions"):