From 8fc5c10e4b24c351c70c16d0d7405e2a5cc0f4f8 Mon Sep 17 00:00:00 2001
From: Moritz Althaus <moritz.althaus@aleph-alpha.com>
Date: Thu, 24 Oct 2024 10:54:32 +0200
Subject: [PATCH] feat: add with_messages method to construct TaskChat

---
 src/chat.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/chat.rs b/src/chat.rs
index 15450d3..b4a29fd 100644
--- a/src/chat.rs
+++ b/src/chat.rs
@@ -61,6 +61,17 @@ impl<'a> TaskChat<'a> {
         }
     }
 
+    /// Creates a new TaskChat containing the given messages.
+    /// All optional TaskChat attributes are left unset.
+    pub fn with_messages(messages: Vec<Message<'a>>) -> Self {
+        TaskChat {
+            messages,
+            maximum_tokens: None,
+            temperature: None,
+            top_p: None,
+        }
+    }
+
     /// Pushes a new Message to this TaskChat.
     pub fn push_message(mut self, message: Message<'a>) -> Self {
         self.messages.push(message);