mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-14 04:02:26 +00:00
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
import pytest
|
|
from acp.schema import (
|
|
EmbeddedResourceContentBlock,
|
|
ImageContentBlock,
|
|
ResourceContentBlock,
|
|
TextContentBlock,
|
|
TextResourceContents,
|
|
)
|
|
|
|
from acp_adapter.server import HermesACPAgent, _content_blocks_to_openai_user_content
|
|
|
|
|
|
def test_acp_image_blocks_convert_to_openai_multimodal_content():
|
|
content = _content_blocks_to_openai_user_content([
|
|
TextContentBlock(type="text", text="What is in this image?"),
|
|
ImageContentBlock(type="image", data="aGVsbG8=", mimeType="image/png"),
|
|
])
|
|
|
|
assert content == [
|
|
{"type": "text", "text": "What is in this image?"},
|
|
{
|
|
"type": "image_url",
|
|
"image_url": {"url": "data:image/png;base64,aGVsbG8="},
|
|
},
|
|
]
|
|
|
|
|
|
def test_text_only_acp_blocks_stay_string_for_legacy_prompt_path():
|
|
content = _content_blocks_to_openai_user_content([
|
|
TextContentBlock(type="text", text="/help"),
|
|
])
|
|
|
|
assert content == "/help"
|
|
|
|
|
|
def test_acp_resource_link_file_is_inlined_as_text(tmp_path):
|
|
attached = tmp_path / "notes.md"
|
|
attached.write_text("# Notes\n\nAttached file body", encoding="utf-8")
|
|
|
|
content = _content_blocks_to_openai_user_content([
|
|
TextContentBlock(type="text", text="Please read this file"),
|
|
ResourceContentBlock(
|
|
type="resource_link",
|
|
name="notes.md",
|
|
title="Project notes",
|
|
uri=attached.as_uri(),
|
|
mimeType="text/markdown",
|
|
),
|
|
])
|
|
|
|
assert content == (
|
|
"Please read this file\n"
|
|
"[Attached file: Project notes (notes.md)]\n"
|
|
f"URI: {attached.as_uri()}\n\n"
|
|
"# Notes\n\nAttached file body"
|
|
)
|
|
|
|
|
|
def test_acp_embedded_text_resource_is_inlined_as_text():
|
|
content = _content_blocks_to_openai_user_content([
|
|
EmbeddedResourceContentBlock(
|
|
type="resource",
|
|
resource=TextResourceContents(
|
|
uri="file:///workspace/todo.txt",
|
|
mimeType="text/plain",
|
|
text="first\nsecond",
|
|
),
|
|
),
|
|
])
|
|
|
|
assert content == (
|
|
"[Attached file: todo.txt]\n"
|
|
"URI: file:///workspace/todo.txt\n\n"
|
|
"first\nsecond"
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_initialize_advertises_image_prompt_capability():
|
|
response = await HermesACPAgent().initialize()
|
|
|
|
assert response.agent_capabilities is not None
|
|
assert response.agent_capabilities.prompt_capabilities is not None
|
|
assert response.agent_capabilities.prompt_capabilities.image is True
|