Export semantic router sample app HTTP request data (#22)

* chore(samples): format and remove unused imports

* refactor(samples): export http request data

Author: arpad-csepi

samples/autogen/semantic-router/tests/test_api.py

@@ -2,12 +2,13 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import json
-import os
-import sys
 import time
 
 import urllib3
 
+payloads = []
+response_data_array = []
+
 
 def wait_for_service(url, retries=5, delay=2):
     """Wait for the service to be available by polling the health endpoint."""
@@ -47,9 +48,9 @@ def test_api_post_request():
     health_url = "http://localhost:8000/healthz"
 
     # Wait for the service to be ready
-    assert wait_for_service(
-        health_url, retries=10
-    ), "Service did not become ready in time."
+    assert wait_for_service(health_url, retries=10), (
+        "Service did not become ready in time."
+    )
 
     # Sleep for a few seconds to ensure that the service is ready
     time.sleep(2)
@@ -66,6 +67,7 @@ def test_api_post_request():
         "context": "ctx",
         "intent": "asd",
     }
+    payloads.append(payload)
 
     # Make a POST request to the API
     response = http.request(
@@ -77,13 +79,14 @@ def test_api_post_request():
 
     # Assert that the status code is 404 (OK)
     # as there is no agent to handle the request for the given intent
-    assert (
-        response.status == 404
-    ), f"Expected status code 404, but got {response.status}: {response.data}"
+    assert response.status == 404, (
+        f"Expected status code 404, but got {response.status}: {response.data}"
+    )
 
     # Now let's make a valid request for the intent "hr"
     payload["intent"] = "hr"
     payload["message"] = "My name is Python"
+    payloads.append(payload)
 
     # Make a POST request to the API
     response = http.request(
@@ -94,23 +97,23 @@ def test_api_post_request():
     )
 
     # Assert that the status code is 200 (OK)
-    assert (
-        response.status == 200
-    ), f"Expected status code 200, but got {response.status}"
+    assert response.status == 200, (
+        f"Expected status code 200, but got {response.status}"
+    )
 
     # Decode the response body
     response_data = json.loads(response.data.decode("utf-8"))
+    response_data_array.append(response_data)
 
     # Assert that the response contains the expected keys
-    assert (
-        "agent_id" in response_data
-    ), "Response does not contain 'agent_id' key"
+    assert "agent_id" in response_data, "Response does not contain 'agent_id' key"
 
     # Save agent_id
     agent_id = response_data["agent_id"]
 
     # Send another request with a different context
     payload["context"] = "ctx2"
+    payloads.append(payload)
 
     # Make a POST request to the API
     response = http.request(
@@ -121,22 +124,21 @@ def test_api_post_request():
     )
 
     # Assert that the status code is 200 (OK)
-    assert (
-        response.status == 200
-    ), f"Expected status code 200, but got {response.status}"
+    assert response.status == 200, (
+        f"Expected status code 200, but got {response.status}"
+    )
 
     # Decode the response body
     response_data = json.loads(response.data.decode("utf-8"))
+    response_data_array.append(response_data)
 
     # Assert that the response contains the expected keys
-    assert (
-        "agent_id" in response_data
-    ), "Response does not contain 'agent_id' key"
+    assert "agent_id" in response_data, "Response does not contain 'agent_id' key"
 
     # Assert that the agent_id is different from the previous response
-    assert (
-        response_data["agent_id"] != agent_id
-    ), "Agent ID should be different for different contexts"
+    assert response_data["agent_id"] != agent_id, (
+        "Agent ID should be different for different contexts"
+    )
 
     # Optionally, release the connection back to the pool
     response.release_conn()