This commit is contained in:
Timothy Jaeryang Baek 2025-05-28 01:34:53 +04:00
parent 100a764293
commit e4a53e0a3c

View file

@ -74,12 +74,12 @@ def get_last_user_message_item(messages: list[dict]) -> Optional[dict]:
def get_content_from_message(message: dict) -> Optional[str]: def get_content_from_message(message: dict) -> Optional[str]:
if isinstance(message["content"], list): if isinstance(message.get("content"), list):
for item in message["content"]: for item in message["content"]:
if item["type"] == "text": if item["type"] == "text":
return item["text"] return item["text"]
else: else:
return message["content"] return message.get("content")
return None return None