Simplify describe audio utility

Flagro · Dec 8, 2024 · c0fae6e · c0fae6e
1 parent 0bb2962
commit c0fae6e
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/bot/rp_bot/ai_agent/ai_utils/describe_audio.py b/bot/rp_bot/ai_agent/ai_utils/describe_audio.py
@@ -13,16 +13,13 @@ def __str__(self):
         return self.audio_description
 
 
-async def describe_audio(in_memory_image_stream: io.BytesIO) -> AudioInformation:
-    # Encode in base64:
-    parser = JsonOutputParser(pydantic_object=AudioInformation)
-
-    return AudioInformation(audio_description="an audio")
-
-
 class DescribeAudioUtililty(BaseUtility):
     def run(self, in_memory_image_stream: io.BytesIO) -> AudioInformation:
-        return describe_audio(in_memory_image_stream)
+        # Encode in base64:
+        parser = JsonOutputParser(pydantic_object=AudioInformation)
+
+        return AudioInformation(audio_description="an audio")
 
     async def arun(self, in_memory_image_stream: io.BytesIO) -> AudioInformation:
-        return await describe_audio(in_memory_image_stream)
+        # TODO: make it non-blocking
+        return self.run(in_memory_image_stream)