change synthesizer factory api

vocodedev · Jul 30, 2023 · 28b5587 · 28b5587
1 parent 4ce40fc
commit 28b5587
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 9 deletions.
diff --git a/vocode/streaming/agent/restful_user_implemented_agent.py b/vocode/streaming/agent/restful_user_implemented_agent.py
@@ -33,7 +33,7 @@ async def respond(
     ) -> Tuple[Optional[str], bool]:
         config = self.agent_config.respond
         try:
-            # todo: cache session
+            # TODO: cache session
             async with aiohttp.ClientSession() as session:
                 payload = RESTfulAgentInput(
                     human_input=human_input, conversation_id=conversation_id

diff --git a/vocode/streaming/synthesizer/factory.py b/vocode/streaming/synthesizer/factory.py
@@ -1,6 +1,7 @@
 import logging
 from typing import Optional
 import typing
+import aiohttp
 
 from vocode.streaming.models.synthesizer import (
     AzureSynthesizerConfig,
@@ -31,22 +32,39 @@ def create_synthesizer(
         self,
         synthesizer_config: SynthesizerConfig,
         logger: Optional[logging.Logger] = None,
+        aiohttp_session: Optional[aiohttp.ClientSession] = None,
     ):
         if isinstance(synthesizer_config, GoogleSynthesizerConfig):
-            return GoogleSynthesizer(synthesizer_config, logger=logger)
+            return GoogleSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         elif isinstance(synthesizer_config, AzureSynthesizerConfig):
-            return AzureSynthesizer(synthesizer_config, logger=logger)
+            return AzureSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         elif isinstance(synthesizer_config, ElevenLabsSynthesizerConfig):
-            return ElevenLabsSynthesizer(synthesizer_config, logger=logger)
+            return ElevenLabsSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         elif isinstance(synthesizer_config, PlayHtSynthesizerConfig):
-            return PlayHtSynthesizer(synthesizer_config, logger=logger)
+            return PlayHtSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         elif isinstance(synthesizer_config, RimeSynthesizerConfig):
-            return RimeSynthesizer(synthesizer_config, logger=logger)
+            return RimeSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         elif isinstance(synthesizer_config, GTTSSynthesizerConfig):
-            return GTTSSynthesizer(synthesizer_config, logger=logger)
+            return GTTSSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         elif isinstance(synthesizer_config, StreamElementsSynthesizerConfig):
-            return StreamElementsSynthesizer(synthesizer_config, logger=logger)
+            return StreamElementsSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         elif isinstance(synthesizer_config, CoquiTTSSynthesizerConfig):
-            return CoquiTTSSynthesizer(synthesizer_config, logger=logger)
+            return CoquiTTSSynthesizer(
+                synthesizer_config, logger=logger, aiohttp_session=aiohttp_session
+            )
         else:
             raise Exception("Invalid synthesizer config")