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

core[patch]: fix chat prompt partial messages placeholder var (#16918)

Browse Source
This commit is contained in:
Bagatur
2024-02-02 10:23:37 -08:00
committed by GitHub
parent 3b0fa9079d
commit c29e9b6412
5 changed files with 38 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
libs/core/tests/unit_tests/prompts/test_chat.py
View File

@@ -503,9 +503,27 @@ def test_messages_placeholder() -> None:
prompt.format_messages()
prompt = MessagesPlaceholder("history", optional=True)
assert prompt.format_messages() == []
prompt.format_messages(
assert prompt.format_messages(
history=[("system", "You are an AI assistant."), "Hello!"]
) == [
SystemMessage(content="You are an AI assistant."),
HumanMessage(content="Hello!"),
]
def test_chat_prompt_message_placeholder_partial() -> None:
prompt = ChatPromptTemplate.from_messages([MessagesPlaceholder("history")])
prompt = prompt.partial(history=[("system", "foo")])
assert prompt.format_messages() == [SystemMessage(content="foo")]
assert prompt.format_messages(history=[("system", "bar")]) == [
SystemMessage(content="bar")
]
prompt = ChatPromptTemplate.from_messages(
[
MessagesPlaceholder("history", optional=True),
]
)
assert prompt.format_messages() == []
prompt = prompt.partial(history=[("system", "foo")])
assert prompt.format_messages() == [SystemMessage(content="foo")]

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

@@ -1544,7 +1544,8 @@
}
}
}
]
],
"partial_variables": {}
}
},
"middle": [],
@@ -1617,7 +1618,8 @@
}
}
}
]
],
"partial_variables": {}
}
},
"middle": [],