livekit-plugins-google 1.1.7__py3-none-any.whl → 1.2.1__py3-none-any.whl

livekit/plugins/google/beta/realtime/realtime_api.py

@@ -52,7 +52,7 @@ class _RealtimeOptions:
     api_key: str | None
     voice: Voice | str
     language: NotGivenOr[str]
-    response_modalities: NotGivenOr[list[types.Modality]]
+    response_modalities: list[types.Modality]
     vertexai: bool
     project: str | None
     location: str | None
@@ -192,12 +192,15 @@ class RealtimeModel(llm.RealtimeModel):
         ):
             server_turn_detection = False
 
+        modalities = modalities if is_given(modalities) else [types.Modality.AUDIO]
+
         super().__init__(
             capabilities=llm.RealtimeCapabilities(
                 message_truncation=False,
                 turn_detection=server_turn_detection,
                 user_transcription=input_audio_transcription is not None,
                 auto_tool_reply_generation=True,
+                audio_output=types.Modality.AUDIO in modalities,
             )
         )
 
@@ -538,7 +541,9 @@ class RealtimeSession(llm.RealtimeSession):
             return
         self.start_user_activity()
 
-    def truncate(self, *, message_id: str, audio_end_ms: int) -> None:
+    def truncate(
+        self, *, message_id: str, audio_end_ms: int, audio_transcript: NotGivenOr[str] = NOT_GIVEN
+    ) -> None:
         logger.warning("truncate is not supported by the Google Realtime API.")
         pass
 
@@ -732,9 +737,7 @@ class RealtimeSession(llm.RealtimeSession):
             gemini_tools=self._opts.gemini_tools if is_given(self._opts.gemini_tools) else None,
         )
         conf = types.LiveConnectConfig(
-            response_modalities=self._opts.response_modalities
-            if is_given(self._opts.response_modalities)
-            else [types.Modality.AUDIO],
+            response_modalities=self._opts.response_modalities,
             generation_config=types.GenerationConfig(
                 candidate_count=self._opts.candidate_count,
                 temperature=temp,
@@ -793,6 +796,8 @@ class RealtimeSession(llm.RealtimeSession):
             audio_ch=utils.aio.Chan[rtc.AudioFrame](),
             _created_timestamp=time.time(),
         )
+        if not self._realtime_model.capabilities.audio_output:
+            self._current_generation.audio_ch.close()
 
         self._current_generation.message_ch.send_nowait(
             llm.MessageGeneration(

livekit/plugins/google/llm.py

@@ -187,6 +187,10 @@ class LLM(llm.LLM):
             location=gcp_location,
         )
 
+    @property
+    def model(self) -> str:
+        return self._opts.model
+
     def chat(
         self,
         *,

livekit/plugins/google/tts.py

@@ -34,7 +34,6 @@ from livekit.agents.utils import is_given
 from .log import logger
 from .models import Gender, SpeechLanguages
 
-BUFFERED_WORDS_COUNT = 8
 NUM_CHANNELS = 1
 DEFAULT_VOICE_NAME = "en-US-Chirp3-HD-Charon"
 DEFAULT_LANGUAGE = "en-US"
@@ -124,7 +123,7 @@ class TTS(tts.TTS):
             ssml_gender=ssml_gender,
         )
         if not is_given(tokenizer):
-            tokenizer = tokenize.basic.SentenceTokenizer(min_sentence_len=BUFFERED_WORDS_COUNT)
+            tokenizer = tokenize.blingfire.SentenceTokenizer()
 
         pronunciations = None if not is_given(custom_pronunciations) else custom_pronunciations
 

livekit/plugins/google/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.1.7"
+__version__ = "1.2.1"

{livekit_plugins_google-1.1.7.dist-info → livekit_plugins_google-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-google
-Version: 1.1.7
+Version: 1.2.1
 Summary: Agent Framework plugin for services from Google Cloud
 Project-URL: Documentation, https://docs.livekit.io
 Project-URL: Website, https://livekit.io/
@@ -22,7 +22,7 @@ Requires-Dist: google-auth<3,>=2
 Requires-Dist: google-cloud-speech<3,>=2
 Requires-Dist: google-cloud-texttospeech<3,>=2.27
 Requires-Dist: google-genai>=v1.23.0
-Requires-Dist: livekit-agents>=1.1.7
+Requires-Dist: livekit-agents>=1.2.1
 Description-Content-Type: text/markdown
 
 # Google AI plugin for LiveKit Agents

{livekit_plugins_google-1.1.7.dist-info → livekit_plugins_google-1.2.1.dist-info}/RECORD

@@ -1,18 +1,18 @@
 livekit/plugins/google/__init__.py,sha256=XIyZ-iFnRBpaLtOJgVwojlB-a8GjdDugVFcjBpMEww8,1412
-livekit/plugins/google/llm.py,sha256=PqRQk6E2XfWkTdDrtEtcHjknGZMGbkQgVLr8uTg7F-s,18960
+livekit/plugins/google/llm.py,sha256=VmM-OEDRplHEYEVHh9rq9rI180yP7xvu_JTI2zFolbY,19035
 livekit/plugins/google/log.py,sha256=GI3YWN5YzrafnUccljzPRS_ZALkMNk1i21IRnTl2vNA,69
 livekit/plugins/google/models.py,sha256=hOpfbN_qdQ1ZTpCN9m9dvG2eb6WgQ3KE3WRpIeeM_T0,1569
 livekit/plugins/google/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 livekit/plugins/google/stt.py,sha256=ssDMH5U1vQOLA44XMlovYWIR4UqVtZSge3YFN-zZ7Iw,24696
 livekit/plugins/google/tools.py,sha256=tD5HVDHO5JfUF029Cx3axHMJec0Gxalkl7s1FDgxLzI,259
-livekit/plugins/google/tts.py,sha256=QVM4xcF7WHpbQOZDAhRJrz481iMhO9ACjjqPEdTT4Lw,16277
+livekit/plugins/google/tts.py,sha256=SODcGwteJIpGmuFArVRwuwy49k8-uQXJAIK5wBNiMC8,16219
 livekit/plugins/google/utils.py,sha256=6iihkKx76DDtLiHOoTU2ZXqzupBRY_gN3njpnwdmeqY,8829
-livekit/plugins/google/version.py,sha256=EcBB23XE8aEiF7xHMivcb9wptFeYkGB1WNGSn1bIV3A,600
+livekit/plugins/google/version.py,sha256=iPQIVYvAlr-64s2ztWSUDdom8KYqdQVSojiZcxUdAdM,600
 livekit/plugins/google/beta/__init__.py,sha256=RvAUdvEiRN-fe4JrgPcN0Jkw1kZR9wPerGMFVjS1Cc0,270
 livekit/plugins/google/beta/gemini_tts.py,sha256=esWjr0Xf95tl0_AB7MXiFZ_VCORWgcWjzvLvRa3t0FQ,8515
 livekit/plugins/google/beta/realtime/__init__.py,sha256=_fW2NMN22F-hnQ4xAJ_g5lPbR7CvM_xXzSWlUQY-E-U,188
 livekit/plugins/google/beta/realtime/api_proto.py,sha256=cbKmpX32G4gPjF6cxFNzGEDfYX19SK-vWi4Myxb8Yks,777
-livekit/plugins/google/beta/realtime/realtime_api.py,sha256=nqiDiAtyHYFRd_Or1Y_95syjHyAVFjaEYTka0qPfXdE,45853
-livekit_plugins_google-1.1.7.dist-info/METADATA,sha256=yG5QbYo-vfSQQ4oyHXrima24mYz9K1sFfT8Bkx6Yh2A,1907
-livekit_plugins_google-1.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-livekit_plugins_google-1.1.7.dist-info/RECORD,,
+livekit/plugins/google/beta/realtime/realtime_api.py,sha256=LMUOhmv4KWTI6mrLAKnnRdZ72ouug69a_oULGKvoMvw,46073
+livekit_plugins_google-1.2.1.dist-info/METADATA,sha256=z6fi69F8Gjpn9XhCQlC8aZRWZeZ5T7YvzK7XkHzZ9MY,1907
+livekit_plugins_google-1.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livekit_plugins_google-1.2.1.dist-info/RECORD,,