Updating tests (#11)

* Updating tests

---------

Co-authored-by: Preetam Joshi <info@aimon.ai>

Author: pjoshi30

aimon/client.py

@@ -186,7 +186,7 @@ def detect(self, data_to_send: List[Dict[str, Any]], config=Config()):
             if 'config' not in item:
                 item['config'] = config.detectors
             payload.append(item)
-        response = requests.post(self.DETECTION_API_URL, json=payload, headers=headers, timeout=30)
+        response = requests.post(self.DETECTION_API_URL, json=payload, headers=headers, timeout=90)
         if response.status_code in [503, 504]:
             raise RetryableError("Status code: {} received".format(response.status_code))
         if response.status_code == 401:

test/client_test.py

@@ -123,7 +123,7 @@ def test_short_text_valid_response_hallucination_v0_2(self):
         config = Config({'hallucination_v0.2': 'default'})
         client = Client(api_key=API_KEY, email="preetam@aimon.ai")
         short_text = "Yes"
-        data_to_send = [{"context": "This is the context", "generated_text": short_text}]
+        data_to_send = [{"context": "This is the context", "user_query": "summarization", "generated_text": short_text}]
         response = client.detect(data_to_send, config=config)[0]
         assert "hallucination_v0.2" in response
         assert "is_hallucinated" in response['hallucination_v0.2']
@@ -138,7 +138,7 @@ def test_special_characters_valid_response_hallucination_v0_2(self):
         config = Config({'hallucination_v0.2': 'default'})
         client = Client(api_key=API_KEY, email="preetam@aimon.ai")
         special_text = "!@#$%^&*()_+"
-        data_to_send = [{"context": "This is the context", "generated_text": special_text}]
+        data_to_send = [{"context": "This is the context", "user_query": "summarization", "generated_text": special_text}]
         response = client.detect(data_to_send, config=config)[0]
         assert "hallucination_v0.2" in response
         assert "is_hallucinated" in response['hallucination_v0.2']
@@ -185,7 +185,6 @@ def test_valid_data_valid_response_instruction_adherence(self):
         assert "results" in response["instruction_adherence"]
         assert len(response["instruction_adherence"]["results"]) == 3
         assert "instruction" in response["instruction_adherence"]["results"][2]
-        assert "examples" in response["instruction_adherence"]["results"][2]
         assert "detailed_explanation" in response["instruction_adherence"]["results"][2]
         assert "adherence" in response["instruction_adherence"]["results"][2]
         # print(response)