github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-25 16:13:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Eugene Yurtsev 2024-12-18 11:11:42 -05:00
parent d1f49fe44a
commit e956663b94
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -699,6 +699,7 @@ def test_convert_to_messages() -> None:
actual = convert_to_messages( actual = convert_to_messages(
[ [
{"role": "system", "content": "You are a helpful assistant."}, {"role": "system", "content": "You are a helpful assistant."},
{"role": "developer", "content": "You are a helpful assistant 2."},
{"role": "user", "content": "Hello!"}, {"role": "user", "content": "Hello!"},
{"role": "ai", "content": "Hi!", "id": "ai1"}, {"role": "ai", "content": "Hi!", "id": "ai1"},
{"type": "human", "content": "Hello!", "name": "Jane", "id": "human1"}, {"type": "human", "content": "Hello!", "name": "Jane", "id": "human1"},
@ -733,6 +734,7 @@ def test_convert_to_messages() -> None:
) )
expected = [ expected = [
SystemMessage(content="You are a helpful assistant."), SystemMessage(content="You are a helpful assistant."),
SystemMessage(content="You are a helpful assistant 2."),
HumanMessage(content="Hello!"), HumanMessage(content="Hello!"),
AIMessage(content="Hi!", id="ai1"), AIMessage(content="Hi!", id="ai1"),
HumanMessage(content="Hello!", name="Jane", id="human1"), HumanMessage(content="Hello!", name="Jane", id="human1"),