Python: Fix Http Schema (#2112)

* Rename to threadid

* Respond in plain text

* Make snake-case

* Add http prefix

* rename to wait-for-response

* Add query param check

* address comments

python/packages/azurefunctions/tests/integration_tests/test_01_single_agent.py

@@ -45,9 +45,9 @@ class TestSampleSingleAgent:
         """Test sending a simple message with JSON payload."""
         response = SampleTestHelper.post_json(
             f"{self.base_url}/run",
-            {"message": "Tell me a short joke about cloud computing.", "sessionId": "test-simple-json"},
+            {"message": "Tell me a short joke about cloud computing.", "thread_id": "test-simple-json"},
         )
-        # Agent can return 200 (immediate) or 202 (async with wait_for_completion=false)
+        # Agent can return 200 (immediate) or 202 (async with wait_for_response=false)
         assert response.status_code in [200, 202]
         data = response.json()
 
@@ -58,37 +58,35 @@ class TestSampleSingleAgent:
             assert data["message_count"] >= 1
         else:
             # Async response - check we got correlation info
-            assert "correlationId" in data or "sessionId" in data
+            assert "correlation_id" in data or "thread_id" in data
 
     def test_simple_message_plain_text(self) -> None:
         """Test sending a message with plain text payload."""
         response = SampleTestHelper.post_text(f"{self.base_url}/run", "Tell me a short joke about networking.")
         assert response.status_code in [200, 202]
-        data = response.json()
 
-        if response.status_code == 200:
-            assert data["status"] == "success"
-            assert "response" in data
+        # Agent responded with plain text when the request body was text/plain.
+        assert response.text.strip()
+        assert response.headers.get("x-ms-thread-id") is not None
 
-    def test_session_key_in_query(self) -> None:
-        """Test using sessionKey in query parameter."""
+    def test_thread_id_in_query(self) -> None:
+        """Test using thread_id in query parameter."""
         response = SampleTestHelper.post_text(
-            f"{self.base_url}/run?sessionKey=test-query-session", "Tell me a short joke about weather in Texas."
+            f"{self.base_url}/run?thread_id=test-query-thread", "Tell me a short joke about weather in Texas."
         )
         assert response.status_code in [200, 202]
-        data = response.json()
 
-        if response.status_code == 200:
-            assert data["status"] == "success"
+        assert response.text.strip()
+        assert response.headers.get("x-ms-thread-id") == "test-query-thread"
 
     def test_conversation_continuity(self) -> None:
         """Test conversation context is maintained across requests."""
-        session_id = "test-continuity"
+        thread_id = "test-continuity"
 
         # First message
         response1 = SampleTestHelper.post_json(
             f"{self.base_url}/run",
-            {"message": "Tell me a short joke about weather in Seattle.", "sessionId": session_id},
+            {"message": "Tell me a short joke about weather in Seattle.", "thread_id": thread_id},
         )
         assert response1.status_code in [200, 202]
 
@@ -98,7 +96,7 @@ class TestSampleSingleAgent:
 
             # Second message in same session
             response2 = SampleTestHelper.post_json(
-                f"{self.base_url}/run", {"message": "What about San Francisco?", "sessionId": session_id}
+                f"{self.base_url}/run", {"message": "What about San Francisco?", "thread_id": thread_id}
             )
             assert response2.status_code == 200
             data2 = response2.json()
@@ -107,7 +105,7 @@ class TestSampleSingleAgent:
             # In async mode, we can't easily test message count
             # Just verify we can make multiple calls
             response2 = SampleTestHelper.post_json(
-                f"{self.base_url}/run", {"message": "What about Texas?", "sessionId": session_id}
+                f"{self.base_url}/run", {"message": "What about Texas?", "thread_id": thread_id}
             )
             assert response2.status_code == 202
 

python/packages/azurefunctions/tests/integration_tests/test_02_multi_agent.py

@@ -38,21 +38,26 @@ class TestSampleMultiAgent:
     def test_weather_agent(self) -> None:
         """Test WeatherAgent endpoint."""
         response = SampleTestHelper.post_json(
-            f"{self.weather_base_url}/run", {"message": "What is the weather in Seattle?"}
+            f"{self.weather_base_url}/run",
+            {"message": "What is the weather in Seattle?"},
         )
-        assert response.status_code == 202
+        assert response.status_code == 200
         data = response.json()
-        assert data["status"] == "accepted"
+        assert data["status"] == "success"
+        assert "response" in data
 
     def test_math_agent(self) -> None:
         """Test MathAgent endpoint."""
         response = SampleTestHelper.post_json(
-            f"{self.math_base_url}/run", {"message": "Calculate a 20% tip on a $50 bill"}
+            f"{self.math_base_url}/run",
+            {"message": "Calculate a 20% tip on a $50 bill", "wait_for_response": False},
         )
         assert response.status_code == 202
         data = response.json()
+
         assert data["status"] == "accepted"
-        assert "response" in data
+        assert "correlation_id" in data
+        assert "thread_id" in data
 
 
 if __name__ == "__main__":

python/packages/azurefunctions/tests/integration_tests/test_03_callbacks.py

@@ -14,6 +14,8 @@ Usage:
     uv run pytest packages/azurefunctions/tests/integration_tests/test_03_callbacks.py -v
 """
 
+from typing import Any
+
 import pytest
 import requests
 
@@ -39,39 +41,60 @@ class TestSampleCallbacks:
         """Provide the callback agent base URL for each test."""
         self.base_url = f"{base_url}/api/agents/CallbackAgent"
 
+    @staticmethod
+    def _wait_for_callback_events(base_url: str, thread_id: str) -> list[dict[str, Any]]:
+        events: list[dict[str, Any]] = []
+        response = SampleTestHelper.get(f"{base_url}/callbacks/{thread_id}")
+        if response.status_code == 200:
+            events = response.json()
+        return events
+
     def test_agent_with_callbacks(self) -> None:
         """Test agent execution with callback tracking."""
-        conversation_id = "test-callback"
+        thread_id = "test-callback"
 
         response = SampleTestHelper.post_json(
-            f"{self.base_url}/run", {"message": "Tell me about Python", "conversationId": conversation_id}
+            f"{self.base_url}/run",
+            {"message": "Tell me about Python", "thread_id": thread_id},
         )
-        assert response.status_code == 202
+        assert response.status_code == 200
         data = response.json()
-        assert data["status"] == "accepted"
+
+        assert data["status"] == "success"
+
+        events = self._wait_for_callback_events(self.base_url, thread_id)
+
+        assert events
+        assert any(event.get("event_type") == "final" for event in events)
 
     def test_get_callbacks(self) -> None:
         """Test retrieving callback events."""
-        conversation_id = "test-callback-retrieve"
+        thread_id = "test-callback-retrieve"
 
         # Send a message first
-        SampleTestHelper.post_json(f"{self.base_url}/run", {"message": "Hello", "conversationId": conversation_id})
+        SampleTestHelper.post_json(
+            f"{self.base_url}/run",
+            {"message": "Hello", "thread_id": thread_id, "wait_for_response": False},
+        )
 
         # Get callbacks
-        response = SampleTestHelper.get(f"{self.base_url}/callbacks/{conversation_id}")
+        response = SampleTestHelper.get(f"{self.base_url}/callbacks/{thread_id}")
         assert response.status_code == 200
         data = response.json()
         assert isinstance(data, list)
 
     def test_delete_callbacks(self) -> None:
         """Test clearing callback events."""
-        conversation_id = "test-callback-delete"
+        thread_id = "test-callback-delete"
 
         # Send a message first
-        SampleTestHelper.post_json(f"{self.base_url}/run", {"message": "Test", "conversationId": conversation_id})
+        SampleTestHelper.post_json(
+            f"{self.base_url}/run",
+            {"message": "Test", "thread_id": thread_id, "wait_for_response": False},
+        )
 
         # Delete callbacks
-        response = requests.delete(f"{self.base_url}/callbacks/{conversation_id}", timeout=TIMEOUT)
+        response = requests.delete(f"{self.base_url}/callbacks/{thread_id}", timeout=TIMEOUT)
         assert response.status_code == 204