fix: assert 200 status on AgentCore invocation responses

Author: mesutoezdil

tests/integration/test_live_agents.py

@@ -406,7 +406,8 @@ def test_session_created_spans_only(self, live_servers):
         main_port, otlp_http_port, mgr = live_servers
         session_name = "e2e-agentcore"
         with _agentcore_server(otlp_http_port, session_name, extra_env=_AGENTCORE_ENV):
-            httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Roll a 20-sided die"}, timeout=60)
+            r = httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Roll a 20-sided die"}, timeout=60)
+            assert r.status_code == 200
         wait_for_session_complete_sync(mgr, session_name, timeout=60)
         s = mgr.sessions[session_name]
         assert s.is_complete and s.source == "otlp" and len(s.spans) > 0
@@ -415,15 +416,17 @@ def test_invocations_extracted(self, live_servers):
         main_port, otlp_http_port, mgr = live_servers
         session_name = "e2e-agentcore-inv"
         with _agentcore_server(otlp_http_port, session_name, extra_env=_AGENTCORE_ENV):
-            httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Is 17 prime?"}, timeout=60)
+            r = httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Is 17 prime?"}, timeout=60)
+            assert r.status_code == 200
         wait_for_session_complete_sync(mgr, session_name, timeout=60)
         assert len(mgr.sessions[session_name].invocations) > 0
 
     def test_session_visible_via_api(self, live_servers):
         main_port, otlp_http_port, mgr = live_servers
         session_name = "e2e-agentcore-api"
         with _agentcore_server(otlp_http_port, session_name, extra_env=_AGENTCORE_ENV):
-            httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Hello!"}, timeout=60)
+            r = httpx.post(f"http://127.0.0.1:{_AGENTCORE_PORT}/invocations", json={"prompt": "Hello!"}, timeout=60)
+            assert r.status_code == 200
         wait_for_session_complete_sync(mgr, session_name, timeout=60)
         data = httpx.get(f"http://127.0.0.1:{main_port}/api/streaming/sessions").json()["data"]
         assert session_name in [s["sessionId"] for s in data]