Fimeg/letta-server
1
0
Fork 0
Code Issues 4 Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fix: Allow content to be None for role==tool (#971)

Browse Source
This commit is contained in:
Sarah Wooders
2024-02-08 12:44:48 -08:00
committed by GitHub
parent 5b25d9e1f2
commit 3c0dec5c4b
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
memgpt/data_types.py
View File

@@ -226,7 +226,7 @@ class Message(Record):
# TODO change to pydantic casting, eg `return SystemMessageModel(self)`
if self.role == "system":
assert all([v is not None for v in [self.text, self.role]]), vars(self)
assert all([v is not None for v in [self.role]]), vars(self)
openai_message = {
"content": self.text,
"role": self.role,
@@ -258,13 +258,12 @@ class Message(Record):
openai_message["tool_calls"] = [tool_call.to_dict() for tool_call in self.tool_calls]
elif self.role == "tool":
assert all([v is not None for v in [self.text, self.role, self.tool_call_id]]), vars(self)
assert all([v is not None for v in [self.role, self.tool_call_id]]), vars(self)
openai_message = {
"content": self.text,
"role": self.role,
"tool_call_id": self.tool_call_id,
}
else:
raise ValueError(self.role)