github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-15 15:46:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: make BaseMessage subscriptable

Browse Source
This commit is contained in:
Bagatur 2025-01-15 13:04:06 -08:00
parent 1051fa5729
commit cde3b1eb48
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libs/core/langchain_core/messages/base.py
View File

@ -118,6 +118,9 @@ class BaseMessage(Serializable):
def pretty_print(self) -> None:
print(self.pretty_repr(html=is_interactive_env())) # noqa: T201
def __getitem__(self, item: str) -> Any:
return self.model_dump()[item]
def merge_content(
first_content: Union[str, list[Union[str, dict]]],

6
libs/core/tests/unit_tests/test_messages.py
View File

@ -1008,3 +1008,9 @@ def test_tool_message_tool_call_id() -> None:
ToolMessage("foo", tool_call_id=uuid.uuid4())
ToolMessage("foo", tool_call_id=1)
ToolMessage("foo", tool_call_id=1.0)
def test_message_getitem() -> None:
msg = BaseMessage(content="foo", role="bar", id=1, type="baz")
for k in msg.model_fields:
assert msg[k] == getattr(msg, k)