mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-08-04 16:38:49 +00:00
0e754d2591
Re-implements PR #2908's generalized multimodal support on the post-FSM main branch: - Audio input: detects WAV/MP3/OGG/FLAC via magic bytes, sends input_audio blocks to compatible providers, falls back to [audio: path] placeholder when unsupported. - Video input: sends video_url data-URI blocks to compatible providers, falls back to [video: path] placeholder. - InputLimitsConfig: count limits (images/audios/videos) and byte limits per media type. - AgentDefaults: pattern-matched vision_models, audio_models, video_models with supports_*() helpers. - Provider retry: strips all media types (image_url, input_audio, video_url) on non-transient errors and retries once. - Feishu: extracts media tags from post messages. - Anthropic & OpenAI Responses converters handle audio/video. Tests: 17 new multimodal tests + existing suite passes.
97 lines
2.6 KiB
Python
97 lines
2.6 KiB
Python
# Check optional Feishu dependencies before running tests
|
|
try:
|
|
from nanobot.channels import feishu
|
|
FEISHU_AVAILABLE = getattr(feishu, "FEISHU_AVAILABLE", False)
|
|
except ImportError:
|
|
FEISHU_AVAILABLE = False
|
|
|
|
if not FEISHU_AVAILABLE:
|
|
import pytest
|
|
pytest.skip("Feishu dependencies not installed (lark-oapi)", allow_module_level=True)
|
|
|
|
from nanobot.channels.feishu import FeishuChannel, _extract_post_content
|
|
|
|
|
|
def test_extract_post_content_supports_post_wrapper_shape() -> None:
|
|
payload = {
|
|
"post": {
|
|
"zh_cn": {
|
|
"title": "日报",
|
|
"content": [
|
|
[
|
|
{"tag": "text", "text": "完成"},
|
|
{"tag": "img", "image_key": "img_1"},
|
|
]
|
|
],
|
|
}
|
|
}
|
|
}
|
|
|
|
text, image_keys, media_items = _extract_post_content(payload)
|
|
|
|
assert text == "日报 完成"
|
|
assert image_keys == ["img_1"]
|
|
assert media_items == []
|
|
|
|
|
|
def test_extract_post_content_keeps_direct_shape_behavior() -> None:
|
|
payload = {
|
|
"title": "Daily",
|
|
"content": [
|
|
[
|
|
{"tag": "text", "text": "report"},
|
|
{"tag": "img", "image_key": "img_a"},
|
|
{"tag": "img", "image_key": "img_b"},
|
|
]
|
|
],
|
|
}
|
|
|
|
text, image_keys, media_items = _extract_post_content(payload)
|
|
|
|
assert text == "Daily report"
|
|
assert image_keys == ["img_a", "img_b"]
|
|
assert media_items == []
|
|
|
|
|
|
def test_extract_post_content_extracts_media_tags() -> None:
|
|
payload = {
|
|
"title": "Video",
|
|
"content": [
|
|
[
|
|
{"tag": "text", "text": "see this"},
|
|
{"tag": "media", "file_key": "vid_1"},
|
|
]
|
|
],
|
|
}
|
|
|
|
text, image_keys, media_items = _extract_post_content(payload)
|
|
|
|
assert text == "Video see this"
|
|
assert image_keys == []
|
|
assert media_items == [{"tag": "media", "file_key": "vid_1"}]
|
|
|
|
|
|
def test_register_optional_event_keeps_builder_when_method_missing() -> None:
|
|
class Builder:
|
|
pass
|
|
|
|
builder = Builder()
|
|
same = FeishuChannel._register_optional_event(builder, "missing", object())
|
|
assert same is builder
|
|
|
|
|
|
def test_register_optional_event_calls_supported_method() -> None:
|
|
called = []
|
|
|
|
class Builder:
|
|
def register_event(self, handler):
|
|
called.append(handler)
|
|
return self
|
|
|
|
builder = Builder()
|
|
handler = object()
|
|
same = FeishuChannel._register_optional_event(builder, "register_event", handler)
|
|
|
|
assert same is builder
|
|
assert called == [handler]
|