Fixed the simple aimon client's handling of batch requests.

Author: Preetam Joshi

src/aimon_rely_client/simple_client.py

@@ -163,4 +163,4 @@ def detect(self, data_to_send: List[Dict[str, Any]]):
         if len(response.json()) == 0 or 'error' in response.json() or 'error' in response.json()[0]:
             raise Exception(
                 f"Received an error in the response: {response if len(response.json()) == 0 else response.json()}")
-        return response.json()[0]
+        return response.json()

src/aimon_rely_client/simple_client_test.py

@@ -13,48 +13,66 @@ def test_valid_data_valid_response(self):
         config = Config({'hallucination': 'default'})
         client = SimpleAimonRelyClient(api_key=API_KEY, config=config)
         data_to_send = [{"context": "This is the context", "generated_text": "This is the context"}]
-        response = client.detect(data_to_send)
+        response = client.detect(data_to_send)[0]
         assert "hallucination" in response
         assert "is_hallucinated" in response['hallucination']
         assert response['hallucination']["is_hallucinated"] == "False"
         assert "score" in response['hallucination']
         assert "sentences" in response['hallucination']
         assert len(response['hallucination']["sentences"]) == 1
+        assert "This is the context" in response['hallucination']["sentences"][0]["text"]
+
+    def test_valid_batch_data_valid_response(self):
+        config = Config({'hallucination': 'default'})
+        client = SimpleAimonRelyClient(api_key=API_KEY, config=config)
+        data_to_send = [{"context": "This is the context", "generated_text": "This is the context"}, {"context": "This is the second context", "generated_text": "This is the second context"}]
+        response = client.detect(data_to_send)
+        for i in range(2):
+            assert "hallucination" in response[i]
+            assert "is_hallucinated" in response[i]['hallucination']
+            assert response[i]['hallucination']["is_hallucinated"] == "False"
+            assert "score" in response[i]['hallucination']
+            assert "sentences" in response[i]['hallucination']
+            assert len(response[i]['hallucination']["sentences"]) == 1
+        assert "This is the context" in response[0]['hallucination']["sentences"][0]["text"]
+        assert "This is the second context" in response[1]['hallucination']["sentences"][0]["text"]
 
     #  Sends an HTTP POST request to the Aimon Rely Hallucination Detection API with a single dict object containing a valid "context" and "generated_text" but with a very short text and receives a valid response
     def test_short_text_valid_response(self):
         config = Config({'hallucination': 'default'})
         client = SimpleAimonRelyClient(api_key=API_KEY, config=config)
-        short_text = "a"
+        short_text = "Yes"
         data_to_send = [{"context": "This is the context", "generated_text": short_text}]
-        response = client.detect(data_to_send)
+        response = client.detect(data_to_send)[0]
         assert "hallucination" in response
         assert "is_hallucinated" in response['hallucination']
-        assert response['hallucination']["is_hallucinated"] == "True"
         assert "score" in response['hallucination']
+        assert response['hallucination']["score"] >= 0.48
         assert "sentences" in response['hallucination']
         assert len(response['hallucination']["sentences"]) == 1
+        assert "Yes" in response['hallucination']["sentences"][0]["text"]
 
     #  Sends an HTTP POST request to the Aimon Rely Hallucination Detection API with a single dict object containing a valid "context" and "generated_text" but with a text containing special characters and receives a valid response
     def test_special_characters_valid_response(self):
         client = SimpleAimonRelyClient(api_key=API_KEY)
         special_text = "!@#$%^&*()_+"
         data_to_send = [{"context": "This is the context", "generated_text": special_text}]
-        response = client.detect(data_to_send)
+        response = client.detect(data_to_send)[0]
         assert "hallucination" in response
         assert "is_hallucinated" in response['hallucination']
         assert response['hallucination']["is_hallucinated"] == "True"
         assert "score" in response['hallucination']
+        assert response['hallucination']["score"] >= 0.5
         assert "sentences" in response['hallucination']
-        assert len(response['hallucination']["sentences"]) == 12
+        assert len(response['hallucination']["sentences"]) == 1
 
     def test_valid_data_valid_response_conciseness(self):
         config = Config({'conciseness': 'default'})
         client = SimpleAimonRelyClient(api_key=API_KEY, config=config)
         data_to_send = [{
                             "context": "the abc have reported that those who receive centrelink payments made up half of radio rental's income last year. Centrelink payments themselves were up 20%.",
                             "generated_text": "those who receive centrelink payments made up half of radio rental's income last year. The Centrelink payments were 20% up."}]
-        response = client.detect(data_to_send)
+        response = client.detect(data_to_send)[0]
         print(response["quality_metrics"])
         assert "quality_metrics" in response
         assert "results" in response["quality_metrics"]
@@ -69,7 +87,7 @@ def test_valid_data_valid_response_completeness(self):
         data_to_send = [{
                             "context": "the abc have reported that those who receive centrelink payments made up half of radio rental's income last year. Centrelink payments themselves were up 20%.",
                             "generated_text": "those who receive centrelink payments made up half of radio rental's income last year. The Centrelink payments were 20% up."}]
-        response = client.detect(data_to_send)
+        response = client.detect(data_to_send)[0]
         assert "quality_metrics" in response
         print(response["quality_metrics"])
         assert "results" in response["quality_metrics"]
@@ -83,7 +101,7 @@ def test_valid_data_valid_response_conciseness_completeness(self):
         client = SimpleAimonRelyClient(api_key=API_KEY, config=config)
         data_to_send = [{"context": "the abc have reported that those who receive centrelink payments made up half of radio rental's income last year. Centrelink payments themselves were up 20%.",
                          "generated_text": "those who receive centrelink payments made up half of radio rental's income last year. The Centrelink payments were 20% up."}]
-        response = client.detect(data_to_send)
+        response = client.detect(data_to_send)[0]
         assert "quality_metrics" in response
         print(response["quality_metrics"])
         assert "results" in response["quality_metrics"]