Implement async run methods

Flagro · Dec 1, 2024 · 35737e2 · 35737e2
1 parent 87c8818
commit 35737e2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 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
@@ -21,11 +21,12 @@ async def describe_audio(in_memory_image_stream: io.BytesIO) -> AudioInformation
 
     # TODO: implement moderation
 
-    return AudioInformation(
-        audio_description="an audio"
-    )
+    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)
+
+    async def arun(self, in_memory_image_stream: io.BytesIO) -> AudioInformation:
+        return await describe_audio(in_memory_image_stream)
diff --git a/bot/rp_bot/ai_agent/ai_utils/describe_image.py b/bot/rp_bot/ai_agent/ai_utils/describe_image.py
@@ -44,3 +44,6 @@ async def describe_image(in_memory_image_stream: io.BytesIO) -> ImageInformation
 class DescribeImageUtililty(BaseUtility):
     def run(self, in_memory_image_stream: io.BytesIO) -> ImageInformation:
         return describe_image(in_memory_image_stream)
+
+    async def arun(self, in_memory_image_stream: io.BytesIO) -> ImageInformation:
+        return await describe_image(in_memory_image_stream)