Add BaseUtility class

bot/rp_bot/ai_agent/ai_utils/base_utility.py

@@ -0,0 +1,25 @@
+from abc import ABC, abstractmethod
+
+from ....models.handlers_input import Person, Context, Message
+from ..models_toolkit import ModelsToolkit
+
+
+class BaseUtility(ABC):
+    name: str
+    description: str
+
+    def __init__(
+        self,
+        person: Person,
+        context: Context,
+        message: Message,
+        models_toolkit: ModelsToolkit,
+    ):
+        self.person = person
+        self.context = context
+        self.message = message
+        self.models_toolkit = models_toolkit
+
+    @abstractmethod
+    async def run(self, *args, **kwargs):
+        pass

bot/rp_bot/ai_agent/ai_utils/describe_audio.py

@@ -2,6 +2,7 @@
 from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.output_parsers import JsonOutputParser
 
+from .base_utility import BaseUtility
 from ..models_toolkit import ModelsToolkit
 from ....models.handlers_input import Person, Context, Message
 
@@ -25,18 +26,6 @@ async def describe_audio(in_memory_image_stream: io.BytesIO) -> AudioInformation
     )
 
 
-class DescribeAudioUtililty:
-    def __init__(
-        self,
-        person: Person,
-        context: Context,
-        message: Message,
-        models_toolkit: ModelsToolkit,
-    ):
-        self.person = person
-        self.context = context
-        self.message = message
-        self.models_toolkit = models_toolkit
-
+class DescribeAudioUtililty(BaseUtility):
     def run(self, in_memory_image_stream: io.BytesIO) -> AudioInformation:
         return describe_audio(in_memory_image_stream)

bot/rp_bot/ai_agent/ai_utils/describe_image.py

@@ -4,6 +4,7 @@
 from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.output_parsers import JsonOutputParser
 
+from .base_utility import BaseUtility
 from ..models_toolkit import ModelsToolkit
 from ....models.handlers_input import Person, Context, Message
 
@@ -40,18 +41,6 @@ async def describe_image(in_memory_image_stream: io.BytesIO) -> ImageInformation
     )
 
 
-class DescribeImageUtililty:
-    def __init__(
-        self,
-        person: Person,
-        context: Context,
-        message: Message,
-        models_toolkit: ModelsToolkit,
-    ):
-        self.person = person
-        self.context = context
-        self.message = message
-        self.models_toolkit = models_toolkit
-
+class DescribeImageUtililty(BaseUtility):
     def run(self, in_memory_image_stream: io.BytesIO) -> ImageInformation:
         return describe_image(in_memory_image_stream)