mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 05:50:44 +08:00
oliver
420abac9f1
- Rename platform/ to svc/ to avoid shadowing stdlib platform. - Replace from oclaw.* with from svc/runtime/interfaces; update -m CLI paths. - tests/conftest: prepend repo root to sys.path (no parent-folder package name). - CI: paths and offline_eval script under repo root. - Ops scripts: PYTHONPATH must be repo root for python -m runtime.* (fixes gateway/WhatsApp sidecar startup). - Fix default oclaw.json path in tabular/file attachment limits; stabilize attachment test config. Co-authored-by: Cursor <cursoragent@cursor.com>
66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
from svc.llm.transports.openai_chat_completions import (
|
|
_flatten_message_content_for_text_gateway,
|
|
_normalize_messages_for_text_only_gateway,
|
|
)
|
|
|
|
|
|
def test_text_only_downgrade_injects_ocr_when_vision_lane_ok(monkeypatch) -> None:
|
|
monkeypatch.setenv("AIA_MULTIMODAL_DOWNGRADE_OCR", "1")
|
|
monkeypatch.setenv("AIA_OCR_BASE_URL", "https://ocr.example/v1")
|
|
monkeypatch.setenv("AIA_OCR_API_KEY", "k")
|
|
monkeypatch.setenv("AIA_OCR_MODEL", "vl-test")
|
|
|
|
def fake_send(*, images, prompt, **kwargs):
|
|
_ = (images, prompt, kwargs)
|
|
return {"ok": True, "text": "HELLO_FROM_PIC"}
|
|
|
|
with patch("svc.llm.image_ocr_client.send_ocr_image_messages", fake_send):
|
|
content = [
|
|
{"type": "text", "text": "请看图"},
|
|
{"type": "image_url", "image_url": {"url": "data:image/png;base64,QUJD"}},
|
|
]
|
|
flat = _flatten_message_content_for_text_gateway(content)
|
|
assert "HELLO_FROM_PIC" in flat
|
|
assert "【图片内容·OCR】" in flat
|
|
assert "纯文本" in flat
|
|
assert "请看图" in flat
|
|
assert "无法直接读图" in flat
|
|
|
|
|
|
def test_text_only_downgrade_falls_back_when_ocr_disabled(monkeypatch) -> None:
|
|
monkeypatch.setenv("AIA_MULTIMODAL_DOWNGRADE_OCR", "0")
|
|
monkeypatch.setenv("AIA_OCR_BASE_URL", "https://ocr.example/v1")
|
|
monkeypatch.setenv("AIA_OCR_API_KEY", "k")
|
|
monkeypatch.setenv("AIA_OCR_MODEL", "vl-test")
|
|
|
|
content = [
|
|
{"type": "text", "text": "hi"},
|
|
{"type": "image_url", "image_url": {"url": "data:image/png;base64,QUJD"}},
|
|
]
|
|
flat = _flatten_message_content_for_text_gateway(content)
|
|
assert "【图片】本消息含图片" in flat
|
|
assert "hi" in flat
|
|
|
|
|
|
def test_normalize_messages_uses_ocr_flatten(monkeypatch) -> None:
|
|
monkeypatch.setenv("AIA_MULTIMODAL_DOWNGRADE_OCR", "1")
|
|
monkeypatch.setenv("AIA_OCR_BASE_URL", "https://ocr.example/v1")
|
|
monkeypatch.setenv("AIA_OCR_API_KEY", "k")
|
|
monkeypatch.setenv("AIA_OCR_MODEL", "vl-test")
|
|
|
|
def fake_send(*, images, prompt, **kwargs):
|
|
_ = (images, prompt, kwargs)
|
|
return {"ok": True, "text": "X"}
|
|
|
|
with patch("svc.llm.image_ocr_client.send_ocr_image_messages", fake_send):
|
|
out = _normalize_messages_for_text_only_gateway(
|
|
[{"role": "user", "content": [{"type": "image_url", "image_url": {"url": "https://x/y.png"}}]}]
|
|
)
|
|
assert len(out) == 1
|
|
assert out[0]["role"] == "user"
|
|
assert "X" in str(out[0]["content"])
|
|
assert isinstance(out[0]["content"], str)
|