wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(gemini): reposition WebSocket connection code

Browse Source 
Move WebSocket connection setup earlier in the function for better
organization and to prepare for subsequent configuration steps.
This commit is contained in:
Mert Sefa AKGUN
2025-01-08 18:53:21 +03:00
parent 94a6f1086e
commit a729834482
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/pipecat/services/gemini_multimodal_live/gemini.py
View File

@@ -382,6 +382,10 @@ class GeminiMultimodalLiveLLMService(LLMService):
# handle disconnections in the send/recv code paths. # handle disconnections in the send/recv code paths.
return 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( config = events.Config.model_validate(
{ {
"setup": { "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 "" system_instruction = self._system_instruction or ""
if self._context and hasattr(self._context, "extract_system_instructions"): if self._context and hasattr(self._context, "extract_system_instructions"):