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

feat: extend message model to support more content types (#756)

Browse Source
This commit is contained in:
cthomas
2025-01-23 17:24:52 -08:00
committed by GitHub
parent d4c0264e95
commit c6293f2ac9
9 changed files with 99 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
letta/orm/message.py
View File

@@ -8,6 +8,7 @@ from letta.orm.custom_columns import ToolCallColumn
from letta.orm.mixins import AgentMixin, OrganizationMixin
from letta.orm.sqlalchemy_base import SqlalchemyBase
from letta.schemas.message import Message as PydanticMessage
from letta.schemas.message import TextContent as PydanticTextContent
class Message(SqlalchemyBase, OrganizationMixin, AgentMixin):
@@ -45,3 +46,10 @@ class Message(SqlalchemyBase, OrganizationMixin, AgentMixin):
def job(self) -> Optional["Job"]:
"""Get the job associated with this message, if any."""
return self.job_message.job if self.job_message else None
def to_pydantic(self) -> PydanticMessage:
"""custom pydantic conversion for message content mapping"""
model = self.__pydantic_model__.model_validate(self)
if self.text:
model.content = [PydanticTextContent(text=self.text)]
return model