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

tests/run_agent/test_copilot_native_vision_headers.py

@@ -0,0 +1,53 @@
+from unittest.mock import MagicMock, patch
+
+from run_agent import AIAgent
+
+
+def _make_copilot_agent():
+    with patch("run_agent.OpenAI") as mock_openai:
+        mock_openai.return_value = MagicMock()
+        agent = AIAgent(
+            api_key="gh-token",
+            base_url="https://api.githubcopilot.com",
+            provider="copilot",
+            model="gpt-5.4",
+            quiet_mode=True,
+            skip_context_files=True,
+            skip_memory=True,
+        )
+    return agent
+
+
+def test_request_client_adds_copilot_vision_header_for_native_image_payload():
+    agent = _make_copilot_agent()
+    built_kwargs = []
+
+    def fake_create(kwargs, *, reason, shared):
+        built_kwargs.append(dict(kwargs))
+        return MagicMock()
+
+    api_kwargs = {
+        "model": "gpt-5.4",
+        "messages": [
+            {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": "What is in this image?"},
+                    {"type": "image_url", "image_url": {"url": "data:image/png;base64,abc"}},
+                ],
+            }
+        ],
+    }
+
+    agent.client = object()
+    with patch.object(agent, "_is_openai_client_closed", return_value=False), patch.object(
+        agent, "_create_openai_client", side_effect=fake_create
+    ):
+        agent._create_request_openai_client(reason="test", api_kwargs=api_kwargs)
+
+    headers = built_kwargs[-1]["default_headers"]
+    assert headers["Copilot-Vision-Request"] == "true"
+
+
+
+