github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-01 02:43:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

openai: add unit test (#20931)

Browse Source 
Test a helper function that was added earlier.
This commit is contained in:
ccurme 2024-04-26 11:02:19 -04:00 committed by GitHub
parent 6598757037
commit fe1304afc4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
libs/partners/openai/tests/unit_tests/chat_models/test_base.py
View File

@ -19,6 +19,7 @@ from langchain_openai import ChatOpenAI
from langchain_openai.chat_models.base import ( from langchain_openai.chat_models.base import (
_convert_dict_to_message, _convert_dict_to_message,
_convert_message_to_dict, _convert_message_to_dict,
_format_message_content,
) )
@ -287,3 +288,36 @@ def test_custom_token_counting() -> None:
llm = ChatOpenAI(custom_get_token_ids=token_encoder) llm = ChatOpenAI(custom_get_token_ids=token_encoder)
assert llm.get_token_ids("foo") == [1, 2, 3] assert llm.get_token_ids("foo") == [1, 2, 3]
def test_format_message_content() -> None:
content: Any = "hello"
assert content == _format_message_content(content)
content = None
assert content == _format_message_content(content)
content = []
assert content == _format_message_content(content)
content = [
{"type": "text", "text": "What is in this image?"},
{
"type": "image_url",
"image_url": {
"url": "url.com",
},
},
]
assert content == _format_message_content(content)
content = [
{"type": "text", "text": "hello"},
{
"type": "tool_use",
"id": "toolu_01A09q90qw90lq917835lq9",
"name": "get_weather",
"input": {"location": "San Francisco, CA", "unit": "celsius"},
},
]
assert [{"type": "text", "text": "hello"}] == _format_message_content(content)