change behavior for converting a string to openai messages (#29446)

This commit is contained in:
Isaac Francisco 2025-01-27 18:18:54 -08:00 committed by GitHub
parent b1fdac726b
commit 2bb2c9bfe8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 1 deletions

View File

@ -948,8 +948,9 @@ def convert_to_openai_messages(
oai_messages: list = []
if is_single := isinstance(messages, (BaseMessage, dict)):
if is_single := isinstance(messages, (BaseMessage, dict, str)):
messages = [messages]
messages = convert_to_messages(messages)
for i, message in enumerate(messages):

View File

@ -666,6 +666,12 @@ def create_base64_image(format: str = "jpeg") -> str:
return f"data:image/{format};base64,{data}"
def test_convert_to_openai_messages_string() -> None:
message = "Hello"
result = convert_to_openai_messages(message)
assert result == {"role": "user", "content": "Hello"}
def test_convert_to_openai_messages_single_message() -> None:
message = HumanMessage(content="Hello")
result = convert_to_openai_messages(message)