github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-07 22:11:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: rm image loading (#27797)

Browse Source
This commit is contained in:
Bagatur
2024-10-31 10:34:51 -07:00
committed by GitHub
parent ad0387ac97
commit c1e742347f
3 changed files with 35 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
libs/core/tests/unit_tests/prompts/test_chat.py
View File

@@ -719,7 +719,7 @@ async def test_chat_tmpl_from_messages_multipart_image() -> None:
async def test_chat_tmpl_from_messages_multipart_formatting_with_path() -> None:
"""Verify that we can pass `path` for an image as a variable."""
"""Verify that we cannot pass `path` for an image as a variable."""
in_mem = "base64mem"
in_file_data = "base64file01"
@@ -746,35 +746,19 @@ async def test_chat_tmpl_from_messages_multipart_formatting_with_path() -> None:
),
]
)
expected = [
SystemMessage(content="You are an AI assistant named R2D2."),
HumanMessage(
content=[
{"type": "text", "text": "What's in this image?"},
{
"type": "image_url",
"image_url": {"url": f"data:image/jpeg;base64,{in_mem}"},
},
{
"type": "image_url",
"image_url": {"url": f"data:image/jpeg;base64,{in_file_data}"},
},
]
),
]
messages = template.format_messages(
name="R2D2",
in_mem=in_mem,
file_path=temp_file.name,
)
assert messages == expected
with pytest.raises(ValueError):
template.format_messages(
name="R2D2",
in_mem=in_mem,
file_path=temp_file.name,
)
messages = await template.aformat_messages(
name="R2D2",
in_mem=in_mem,
file_path=temp_file.name,
)
assert messages == expected
with pytest.raises(ValueError):
await template.aformat_messages(
name="R2D2",
in_mem=in_mem,
file_path=temp_file.name,
)
def test_messages_placeholder() -> None: