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

Pull _connection_established_threshold and _max_consecutive_failures into file-level constants

Browse Source
This commit is contained in:
Paul Kompfner
2025-10-08 17:04:05 -04:00
parent 7efb22a323
commit 7da37a0d1f
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -112,6 +112,11 @@ except ModuleNotFoundError as e:
raise Exception(f"Missing module: {e}") raise Exception(f"Missing module: {e}")
# Connection management constants
MAX_CONSECUTIVE_FAILURES = 3
CONNECTION_ESTABLISHED_THRESHOLD = 10.0 # seconds
def language_to_gemini_language(language: Language) -> Optional[str]: def language_to_gemini_language(language: Language) -> Optional[str]:
"""Maps a Language enum value to a Gemini Live supported language code. """Maps a Language enum value to a Gemini Live supported language code.
@@ -622,8 +627,6 @@ class GeminiMultimodalLiveLLMService(LLMService):
# Reconnection tracking # Reconnection tracking
self._consecutive_failures = 0 self._consecutive_failures = 0
self._max_consecutive_failures = 3
self._connection_established_threshold = 10.0 # seconds
self._connection_start_time = None self._connection_start_time = None
self._settings = { self._settings = {
@@ -1030,10 +1033,10 @@ class GeminiMultimodalLiveLLMService(LLMService):
if ( if (
self._connection_start_time self._connection_start_time
and self._consecutive_failures > 0 and self._consecutive_failures > 0
and time.time() - self._connection_start_time >= self._connection_established_threshold and time.time() - self._connection_start_time >= CONNECTION_ESTABLISHED_THRESHOLD
): ):
logger.info( logger.info(
f"Connection stable for {self._connection_established_threshold}s, " f"Connection stable for {CONNECTION_ESTABLISHED_THRESHOLD}s, "
f"resetting failure counter from {self._consecutive_failures} to 0" f"resetting failure counter from {self._consecutive_failures} to 0"
) )
self._consecutive_failures = 0 self._consecutive_failures = 0
@@ -1049,12 +1052,12 @@ class GeminiMultimodalLiveLLMService(LLMService):
""" """
self._consecutive_failures += 1 self._consecutive_failures += 1
logger.warning( logger.warning(
f"Connection error (failure {self._consecutive_failures}/{self._max_consecutive_failures}): {error}" f"Connection error (failure {self._consecutive_failures}/{MAX_CONSECUTIVE_FAILURES}): {error}"
) )
if self._consecutive_failures >= self._max_consecutive_failures: if self._consecutive_failures >= MAX_CONSECUTIVE_FAILURES:
logger.error( logger.error(
f"Max consecutive failures ({self._max_consecutive_failures}) reached, " f"Max consecutive failures ({MAX_CONSECUTIVE_FAILURES}) reached, "
"treating as fatal error" "treating as fatal error"
) )
await self.push_error( await self.push_error(
@@ -1063,7 +1066,7 @@ class GeminiMultimodalLiveLLMService(LLMService):
return False return False
else: else:
logger.info( logger.info(
f"Attempting reconnection ({self._consecutive_failures}/{self._max_consecutive_failures})" f"Attempting reconnection ({self._consecutive_failures}/{MAX_CONSECUTIVE_FAILURES})"
) )
return True return True