make OpenAI.build_model_input return all arguments as dict

evoprompt/models.py

@@ -328,8 +328,19 @@ class OpenAI(LLMModel):
         usage = ModelUsage(**response.usage.__dict__)
         return response_text, usage
 
-    def build_model_input(self, **kwargs):
-        return kwargs
+    def build_model_input(
+        self,
+        prompt: str,
+        system_message: str | None,
+        messages: list[dict[str, str]],
+        history: list[dict[str, str]] | None = None,
+    ):
+        return {
+            "prompt": prompt,
+            "system_message": system_message,
+            "messages": messages,
+            "history": history,
+        }
 
     @classmethod
     def register_arguments(cls, parser: ArgumentParser):