Merge pull request #423 from TomasBack2Future/dev/deepgram_refactor

refact into async extension
TEN-framework · Nov 19, 2024 · 8a3b2ea · 8a3b2ea
2 parents b7049f7 + 4a9c6f5
commit 8a3b2ea
Show file tree

Hide file tree

Showing 7 changed files with 153 additions and 264 deletions.
diff --git a/agents/ten_packages/extension/deepgram_asr_python/__init__.py b/agents/ten_packages/extension/deepgram_asr_python/__init__.py
@@ -1,5 +1 @@
-from . import deepgram_asr_addon
-from .extension import EXTENSION_NAME
-from .log import logger
-
-logger.info(f"{EXTENSION_NAME} extension loaded")
+from . import addon
diff --git a/...deepgram_asr_python/deepgram_asr_addon.py → ...es/extension/deepgram_asr_python/addon.py b/...deepgram_asr_python/deepgram_asr_addon.py → ...es/extension/deepgram_asr_python/addon.py
@@ -3,12 +3,10 @@
     register_addon_as_extension,
     TenEnv,
 )
-from .extension import EXTENSION_NAME
 
-@register_addon_as_extension(EXTENSION_NAME)
+@register_addon_as_extension("deepgram_asr_python")
 class DeepgramASRExtensionAddon(Addon):
     def on_create_instance(self, ten: TenEnv, addon_name: str, context) -> None:
-        from .log import logger
-        from .deepgram_asr_extension import DeepgramASRExtension
-        logger.info("on_create_instance")
+        from .extension import DeepgramASRExtension
+        ten.log_info("on_create_instance")
         ten.on_create_instance_done(DeepgramASRExtension(addon_name), context)
diff --git a/...on/deepgram_asr_python/deepgram_config.py → ...s/extension/deepgram_asr_python/config.py b/...on/deepgram_asr_python/deepgram_config.py → ...s/extension/deepgram_asr_python/config.py
diff --git a/agents/ten_packages/extension/deepgram_asr_python/deepgram_asr_extension.py b/agents/ten_packages/extension/deepgram_asr_python/deepgram_asr_extension.py
diff --git a/agents/ten_packages/extension/deepgram_asr_python/deepgram_wrapper.py b/agents/ten_packages/extension/deepgram_asr_python/deepgram_wrapper.py