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

(standard tests): add test for basic conversation sequence (#21213)

Browse Source
This commit is contained in:
ccurme 2024-05-02 13:47:10 -04:00 committed by GitHub
parent 1ebb5a70ad
commit 2d77e5e3a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py
View File

@ -117,6 +117,21 @@ class ChatModelIntegrationTests(ABC):
assert isinstance(result.content, str)
assert len(result.content) > 0
def test_conversation(
self, chat_model_class: Type[BaseChatModel], chat_model_params: dict
) -> None:
model = chat_model_class(**chat_model_params)
messages = [
HumanMessage(content="hello"),
AIMessage(content="hello"),
HumanMessage(content="how are you"),
]
result = model.invoke(messages)
assert result is not None
assert isinstance(result, AIMessage)
assert isinstance(result.content, str)
assert len(result.content) > 0
def test_tool_message_histories_string_content(
self,
chat_model_class: Type[BaseChatModel],