Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/gateway/test_vision_preprocess.py

@@ -0,0 +1,34 @@
+"""Gateway vision pre-process prompt should stay concise."""
+
+import json
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+
+@pytest.mark.asyncio
+async def test_enrich_message_with_vision_uses_concise_prompt():
+    from gateway.run import GatewayRunner
+
+    runner = GatewayRunner.__new__(GatewayRunner)
+
+    with patch(
+        "tools.vision_tools.vision_analyze_tool",
+        new_callable=AsyncMock,
+        return_value=json.dumps({"success": True, "analysis": "A cat on a chair."}),
+    ) as mock_vision:
+        result = await runner._enrich_message_with_vision(
+            user_text="What is happening here?",
+            image_paths=["/tmp/cat.png"],
+        )
+
+    assert "A cat on a chair." in result
+    assert "What is happening here?" in result
+    assert (
+        "Concisely describe this image in 2-4 sentences"
+        in mock_vision.await_args.kwargs["user_prompt"]
+    )
+    assert "Skip decorative details." in mock_vision.await_args.kwargs["user_prompt"]
+    # No output cap is forwarded: per the max-tokens-knob policy the aux
+    # client decides token handling; conciseness comes from the prompt.
+    assert "max_tokens" not in mock_vision.await_args.kwargs