From 7420ba003d8b5df5559076628d8d50863e537b33 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Grie=C3=9Fhaber?= <griesshaber@hdm-stuttgart.de>
Date: Mon, 19 Aug 2024 13:55:03 +0200
Subject: [PATCH] make OpenAI.build_model_input return all arguments as dict
---
evoprompt/models.py | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/evoprompt/models.py b/evoprompt/models.py
index 2d65147..78dd7ed 100644
--- a/evoprompt/models.py
+++ b/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):
--
GitLab