1  import os
 2  import requests
 3  import logging
 4  
 5  # Set up logging
 6  logging.basicConfig(level=logging.INFO)
 7  logger = logging.getLogger(__name__)
 8  
 9  
10  # def get_token():
11  #     access_key = os.getenv("RAGA-TRACER_ACCESS_KEY")
12  #     secret_key = os.getenv("RAGA-TRACER_SECRET_KEY")
13  #     headers = {"Content-Type": "application/json"}
14  #     json_data = {
15  #         "accessKey": access_key,
16  #         "secretKey": secret_key,
17  #     }
18  #     response = requests.post(
19  #         "https://backend.dev3.ragaai.ai/api/token", headers=headers, json=json_data
20  #     )
21  #     token_response = response.json()
22  #     token = token_response.get("data", {}).get("token", None)
23  #     if token is not None:
24  #         os.environ["RAGAAI_CATALYST_TOKEN"] = token
25  #     return token_response
26  
27  
28  def response_checker(response, context=""):
29      """
30      Checks the response status code and logs the appropriate message.
31  
32      Args:
33          response (requests.Response): The response object.
34          context (str, optional): The context in which the response is being checked. Defaults to "".
35  
36      Returns:
37          int: The status code of the response.
38      """
39      logger.debug(f" Response : {response}")
40      if response.status_code == 200:
41          logger.debug(
42              f"{context} - Successful Request. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
43          )
44          return response.status_code
45      elif response.status_code == 400:
46          logger.debug(
47              f"{context} - Bad Request. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
48          )
49          return response.status_code
50      elif response.status_code == 401:
51          logger.debug(
52              f"{context} - Unauthorized. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
53          )
54          return response.status_code
55      elif response.status_code == 403:
56          logger.debug(
57              f"{context} - Forbidden. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
58          )
59          return response.status_code
60      elif response.status_code == 404:
61          logger.debug(
62              f"{context} - Not Found. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
63          )
64          return response.status_code
65      elif response.status_code == 408:
66          logger.debug(
67              f"{context} - Request Timeout. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
68          )
69          return response.status_code
70      elif response.status_code == 500:
71          logger.debug(
72              f"{context} - Internal Server Error. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
73          )
74          return response.status_code
75      elif response.status_code == 502:
76          logger.debug(
77              f"{context} - Bad Gateway. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
78          )
79          return response.status_code
80      elif response.status_code == 503:
81          logger.debug(
82              f"{context} - Service Unavailable. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
83          )
84          return response.status_code
85      elif response.status_code == 504:
86          logger.debug(
87              f"{context} - Gateway Timeout. Response Code: {response.status_code}, Response Text: {(response.json()['message'])}"
88          )
89          return response.status_code
90      else:
91          error_message = response.json().get("message", "No message returned.")
92          logger.debug(
93              f"{context}{response.reason}. Response Code: {response.status_code}, Response Text: {error_message}"
94          )
95          return response.status_code