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

small fix to wss base_url

Browse Source
This commit is contained in:
Kwindla Hultman Kramer
2025-04-08 20:22:26 -07:00
parent 77ab841cab
commit 1828127f41
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -164,7 +164,7 @@ class GeminiMultimodalLiveLLMService(LLMService):
self, self,
*, *,
api_key: str, api_key: str,
base_url="generativelanguage.googleapis.com", base_url: str = "",
model="models/gemini-2.0-flash-exp", model="models/gemini-2.0-flash-exp",
voice_id: str = "Charon", voice_id: str = "Charon",
start_audio_paused: bool = False, start_audio_paused: bool = False,
@@ -180,7 +180,10 @@ class GeminiMultimodalLiveLLMService(LLMService):
super().__init__(base_url=base_url, **kwargs) super().__init__(base_url=base_url, **kwargs)
self._last_sent_time = 0 self._last_sent_time = 0
self.api_key = api_key self.api_key = api_key
self.base_url = base_url if base_url:
self.base_url = base_url
else:
self.base_url = "generativelanguage.googleapis.com/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent"
self.set_model_name(model) self.set_model_name(model)
self._voice_id = voice_id self._voice_id = voice_id
@@ -407,8 +410,8 @@ class GeminiMultimodalLiveLLMService(LLMService):
logger.info("Connecting to Gemini service") logger.info("Connecting to Gemini service")
try: try:
uri = f"wss://{self.base_url}/ws/google.ai.generativelanguage.v1alpha.GenerativeService.BidiGenerateContent?key={self.api_key}" logger.info(f"Connecting to wss://{self.base_url}")
logger.info(f"Connecting to {uri}") uri = f"wss://{self.base_url}?key={self.api_key}"
self._websocket = await websockets.connect(uri=uri) self._websocket = await websockets.connect(uri=uri)
self._receive_task = self.create_task(self._receive_task_handler()) self._receive_task = self.create_task(self._receive_task_handler())
self._transcribe_audio_task = self.create_task(self._transcribe_audio_handler()) self._transcribe_audio_task = self.create_task(self._transcribe_audio_handler())