From e246ea6aaed0a49c2ce3b01a01d6acd2c0c61a8f Mon Sep 17 00:00:00 2001
From: Sarah Wooders <sarahwooders@gmail.com>
Date: Thu, 8 Feb 2024 20:00:40 -0800
Subject: [PATCH] fix: Require `tool_calls` or `content` to be set for
 assistant role (#976)

---
 memgpt/data_types.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/memgpt/data_types.py b/memgpt/data_types.py
index aeaf159a..dfa48dda 100644
--- a/memgpt/data_types.py
+++ b/memgpt/data_types.py
@@ -246,7 +246,7 @@ class Message(Record):
                 openai_message["name"] = self.name
 
         elif self.role == "assistant":
-            assert all([v is not None for v in [self.text, self.role]]), vars(self)
+            assert self.tool_calls is not None or self.text is not None
             openai_message = {
                 "content": self.text,
                 "role": self.role,