github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-08 14:31:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[minor]: add BaseMessage.response_metadata (#18699)

Browse Source
This commit is contained in:
Bagatur
2024-03-08 09:35:56 -08:00
committed by GitHub
parent 67d48ea600
commit 3e29c04213
6 changed files with 2921 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -21,7 +21,12 @@ class BaseMessage(Serializable):
"""The string contents of the message."""
additional_kwargs: dict = Field(default_factory=dict)
"""Any additional information."""
"""Reserved for additional payload data associated with the message.
For example, for a message from an AI, this could include tool calls."""
response_metadata: dict = Field(default_factory=dict)
"""Response metadata. For example: response headers, logprobs, token counts."""
type: str

264
libs/core/tests/unit_tests/runnables/__snapshots__/test_fallbacks.ambr
View File

@@ -220,6 +220,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -276,6 +280,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -332,6 +340,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -388,6 +400,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -439,6 +455,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -491,6 +511,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -686,6 +710,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -742,6 +770,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -798,6 +830,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -854,6 +890,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -905,6 +945,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -957,6 +1001,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -1260,6 +1308,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -1316,6 +1368,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -1372,6 +1428,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -1428,6 +1488,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -1479,6 +1543,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -1531,6 +1599,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -1726,6 +1798,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -1782,6 +1858,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -1838,6 +1918,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -1894,6 +1978,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -1945,6 +2033,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -1997,6 +2089,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -2721,6 +2817,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -2777,6 +2877,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -2833,6 +2937,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -2889,6 +2997,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -2940,6 +3052,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -2992,6 +3108,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -3209,6 +3329,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -3265,6 +3389,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -3321,6 +3449,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -3377,6 +3509,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -3428,6 +3564,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -3480,6 +3620,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -3699,6 +3843,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -3755,6 +3903,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -3811,6 +3963,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -3867,6 +4023,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -3918,6 +4078,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -3970,6 +4134,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -4199,6 +4367,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -4255,6 +4427,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -4311,6 +4487,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -4367,6 +4547,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -4418,6 +4602,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -4470,6 +4658,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -4687,6 +4879,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -4743,6 +4939,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -4799,6 +4999,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -4855,6 +5059,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -4906,6 +5114,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -4958,6 +5170,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -5174,6 +5390,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -5230,6 +5450,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -5286,6 +5510,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -5342,6 +5570,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -5393,6 +5625,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -5445,6 +5681,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",
@@ -5664,6 +5904,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "ai",
@@ -5720,6 +5964,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "human",
@@ -5776,6 +6024,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "chat",
@@ -5832,6 +6084,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "system",
@@ -5883,6 +6139,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "function",
@@ -5935,6 +6195,10 @@
"title": "Additional Kwargs",
"type": "object"
},
"response_metadata": {
"title": "Response Metadata",
"type": "object"
},
"type": {
"title": "Type",
"default": "tool",

2398
libs/core/tests/unit_tests/runnables/__snapshots__/test_runnable.ambr
View File

File diff suppressed because one or more lines are too long

24
libs/core/tests/unit_tests/runnables/test_graph.py
View File

@@ -179,6 +179,10 @@ def test_graph_sequence_map(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "ai",
@@ -219,6 +223,10 @@ def test_graph_sequence_map(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "human",
@@ -259,6 +267,10 @@ def test_graph_sequence_map(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "chat",
@@ -295,6 +307,10 @@ def test_graph_sequence_map(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "system",
@@ -330,6 +346,10 @@ def test_graph_sequence_map(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "function",
@@ -365,6 +385,10 @@ def test_graph_sequence_map(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "tool",

24
libs/core/tests/unit_tests/runnables/test_runnable.py
View File

@@ -340,6 +340,10 @@ def test_schemas(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "ai",
@@ -377,6 +381,10 @@ def test_schemas(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "human",
@@ -414,6 +422,10 @@ def test_schemas(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "chat",
@@ -447,6 +459,10 @@ def test_schemas(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "system",
@@ -479,6 +495,10 @@ def test_schemas(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "function",
@@ -511,6 +531,10 @@ def test_schemas(snapshot: SnapshotAssertion) -> None:
"title": "Additional Kwargs",
"type": "object",
},
"response_metadata": {
"title": "Response Metadata",
"type": "object",
},
"type": {
"title": "Type",
"default": "tool",