reginaldCog/discord_bot/llm_interface.py

@@ -0,0 +1,9 @@
+from ..openai_client.models import Message, Prompt
+
+
+class LLMInterface:
+    pass
+
+
+if __name__ == '__main__':
+    pass

reginaldCog/messenger_clients/_common.py

@@ -1,7 +0,0 @@
-from abc import ABC, abstractmethod
-
-
-class MessageFactory(ABC):
-    @abstractmethod
-    def new(self, raw_message: dict) -> object:
-        pass

reginaldCog/openai_client/models.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass, field, asdict
+from dataclasses import dataclass, field, asdict, fields
 from abc import ABC
 
 
@@ -71,6 +71,14 @@ class Response:
     output: list[Message]
 
 
+def filter_fields(data: dict, dataclass_type: dataclass) -> dict:
+    return {
+        i_key: i_value
+        for i_key, i_value in data.items()
+        if i_key in tuple(i_field.name for i_field in fields(dataclass_type))
+    }
+
+
 if __name__ == '__main__':
     from openai import OpenAI
     client = OpenAI()