fix(factory.py): handle assistant null content

This commit is contained in:
Krrish Dholakia
2024-08-14 15:44:38 -07:00
parent 75ed22d391
commit 6333b04be3
+3 -1
View File
@@ -1705,7 +1705,9 @@ def cohere_messages_pt_v2(
for m in messages[msg_i]["content"]:
if m.get("type", "") == "text":
assistant_content += m["text"]
else:
elif messages[msg_i].get("content") is not None and isinstance(
messages[msg_i]["content"], str
):
assistant_content += messages[msg_i]["content"]
if messages[msg_i].get(
"tool_calls", []