Update query.py

query.py

@@ -1,6 +1,10 @@
 import requests
 import json
 
+import logging
+
+logging.basicConfig(level=logging.DEBUG)
+
 
 class VectaraQuery():
     def __init__(self, api_key: str, customer_id: str, corpus_ids: list[str], prompt_name: str = None):
@@ -68,6 +72,7 @@ class VectaraQuery():
 
         response = requests.post(endpoint, data=json.dumps(body), verify=True, headers=self.get_headers())    
         if response.status_code != 200:
+            logging.debug(f"Query failed with code {response.status_code}, reason {response.reason}, text {response.text}")
             print(f"Query failed with code {response.status_code}, reason {response.reason}, text {response.text}")
             return "Sorry, something went wrong in my brain. Please try again later."
 
@@ -85,6 +90,7 @@ class VectaraQuery():
             return 'Sorry, something went wrong in my brain. Please try again later.'
         
         self.conv_id = chat['conversationId'] if chat else None
+        logging.debug(summary)
         return summary
 
     def submit_query_streaming(self, query_str: str):
@@ -94,6 +100,7 @@ class VectaraQuery():
 
         response = requests.post(endpoint, data=json.dumps(body), verify=True, headers=self.get_headers(), stream=True) 
         if response.status_code != 200:
+            logging.debug(f"Query failed with code {response.status_code}, reason {response.reason}, text {response.text}")
             print(f"Query failed with code {response.status_code}, reason {response.reason}, text {response.text}")
             return "Sorry, something went wrong in my brain. Please try again later."