mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-10 05:20:39 +00:00
change behavior for converting a string to openai messages (#29446)
This commit is contained in:
parent
b1fdac726b
commit
2bb2c9bfe8
@ -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):
|
||||
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user