core[patch]: concrete prompt value test (#26128)

This commit is contained in:
Bagatur
2024-09-05 20:49:05 -04:00
committed by GitHub
parent 776d01db49
commit 8a3a9c8968

View File

@@ -0,0 +1,28 @@
import pytest
from langchain_core.messages import (
AIMessage,
AIMessageChunk,
HumanMessage,
HumanMessageChunk,
SystemMessage,
SystemMessageChunk,
ToolMessage,
ToolMessageChunk,
)
from langchain_core.prompt_values import ChatPromptValueConcrete
@pytest.mark.xfail(reason="Broken union type.")
def test_chat_prompt_value_concrete() -> None:
messages: list = [
AIMessage("foo"),
HumanMessage("foo"),
SystemMessage("foo"),
ToolMessage("foo", tool_call_id="foo"),
AIMessageChunk(content="foo"),
HumanMessageChunk(content="foo"),
SystemMessageChunk(content="foo"),
ToolMessageChunk(content="foo", tool_call_id="foo"),
]
assert ChatPromptValueConcrete(messages=messages).messages == messages