mirror of
https://github.com/hwchase17/langchain
synced 2024-11-11 19:11:02 +00:00
35 lines
981 B
Python
35 lines
981 B
Python
|
"""Test EdenAI Chat API wrapper."""
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
|
||
|
|
|
||
|
|
from langchain_community.chat_models.dappier import _format_dappier_messages
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
("messages", "expected"),
|
||
|
|
[
|
||
|
|
(
|
||
|
|
[
|
||
|
|
SystemMessage(
|
||
|
|
content="You are a chat model with real time search tools"
|
||
|
|
),
|
||
|
|
HumanMessage(content="Hello how are you today?"),
|
||
|
|
],
|
||
|
|
[
|
||
|
|
{
|
||
|
|
"role": "system",
|
||
|
|
"content": "You are a chat model with real time search tools",
|
||
|
|
},
|
||
|
|
{"role": "user", "content": "Hello how are you today?"},
|
||
|
|
],
|
||
|
|
)
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_dappier_messages_formatting(
|
||
|
|
messages: List[BaseMessage], expected: str
|
||
|
|
) -> None:
|
||
|
|
result = _format_dappier_messages(messages)
|
||
|
|
assert result == expected
|