github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-25 04:49:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: Fixes for convert_messages (#21207)

Browse Source 
- support two-tuples of any sequence type (eg. json.loads never produces
tuples)
- support type alias for role key
- if id is passed in in dict form use it
- if tool_calls passed in in dict form use them

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
Nuno Campos
2024-05-02 09:55:42 -07:00
committed by GitHub
parent df49404794
commit 663747b730
3 changed files with 49 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -545,8 +545,8 @@ def test_convert_to_messages() -> None:
[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Hello!"},
{"role": "ai", "content": "Hi!"},
{"role": "human", "content": "Hello!", "name": "Jane"},
{"role": "ai", "content": "Hi!", "id": "ai1"},
{"type": "human", "content": "Hello!", "name": "Jane", "id": "human1"},
{
"role": "assistant",
"content": "Hi!",
@@ -554,13 +554,20 @@ def test_convert_to_messages() -> None:
"function_call": {"name": "greet", "arguments": '{"name": "Jane"}'},
},
{"role": "function", "name": "greet", "content": "Hi!"},
{
"role": "assistant",
"content": "",
"tool_calls": [
{"name": "greet", "args": {"name": "Jane"}, "id": "tool_id"}
],
},
{"role": "tool", "tool_call_id": "tool_id", "content": "Hi!"},
]
) == [
SystemMessage(content="You are a helpful assistant."),
HumanMessage(content="Hello!"),
AIMessage(content="Hi!"),
HumanMessage(content="Hello!", name="Jane"),
AIMessage(content="Hi!", id="ai1"),
HumanMessage(content="Hello!", name="Jane", id="human1"),
AIMessage(
content="Hi!",
name="JaneBot",
@@ -569,6 +576,10 @@ def test_convert_to_messages() -> None:
},
),
FunctionMessage(name="greet", content="Hi!"),
AIMessage(
content="",
tool_calls=[ToolCall(name="greet", args={"name": "Jane"}, id="tool_id")],
),
ToolMessage(tool_call_id="tool_id", content="Hi!"),
]
@@ -579,7 +590,7 @@ def test_convert_to_messages() -> None:
"hello!",
("ai", "Hi!"),
("human", "Hello!"),
("assistant", "Hi!"),
["assistant", "Hi!"],
]
) == [
SystemMessage(content="You are a helpful assistant."),