addressed PR comment

langchain-ai · Mar 12, 2024 · 5e58df8 · 5e58df8
1 parent bdcb3ab
commit 5e58df8
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/libs/community/langchain_community/utilities/__init__.py b/libs/community/langchain_community/utilities/__init__.py
@@ -31,6 +31,9 @@
     "MaxComputeAPIWrapper": "langchain_community.utilities.max_compute",
     "MerriamWebsterAPIWrapper": "langchain_community.utilities.merriam_webster",
     "MetaphorSearchAPIWrapper": "langchain_community.utilities.metaphor_search",
+    "NVIDIARivaASR": "langchain_community.utilities.nvidia_riva",
+    "NVIDIARivaStream": "langchain_community.utilities.nvidia_riva",
+    "NVIDIARivaTTS": "langchain_community.utilities.nvidia_riva",
     "NasaAPIWrapper": "langchain_community.utilities.nasa",
     "NutritionAIAPI": "langchain_community.utilities.passio_nutrition_ai",
     "OpenWeatherMapAPIWrapper": "langchain_community.utilities.openweathermap",

diff --git a/libs/community/langchain_community/utilities/nvidia_riva.py b/libs/community/langchain_community/utilities/nvidia_riva.py
@@ -665,3 +665,9 @@ async def _consumer_coro() -> None:
 
         await producer
         await consumer
+
+
+# Backwards compatibility:
+NVIDIARivaASR = RivaASR
+NVIDIARivaTTS = RivaTTS
+NVIDIARivaStream = AudioStream
diff --git a/libs/community/tests/unit_tests/utilities/test_imports.py b/libs/community/tests/unit_tests/utilities/test_imports.py
@@ -27,6 +27,9 @@
     "RivaASR",
     "RivaTTS",
     "AudioStream",
+    "NVIDIARivaASR",
+    "NVIDIARivaTTS",
+    "NVIDIARivaStream",
     "OpenWeatherMapAPIWrapper",
     "OutlineAPIWrapper",
     "NutritionAIAPI",