Reverted back

query.py

@@ -13,7 +13,7 @@ class VectaraQuery():
         self.customer_id = customer_id
         self.corpus_ids = corpus_ids
         self.api_key = api_key
-        self.conv_id = "1dc2c542-925f-4b48-aab5-e3df6f2a3a64"
+        self.conv_id = None
 
     def submit_query(self, query_str: str):
         corpora_key_list = [{
@@ -80,18 +80,18 @@ class VectaraQuery():
         docs = res['responseSet'][0]['document']
         chat = res['responseSet'][0]['summary'][0]['chat']
 
-#        if chat['status'] != None:
-#  #          chat['status'] =  # I have no idea what to put here hahaha
- #           st_code = chat['status']
-  #          print(f"Chat query failed with code {st_code}")
-   #         if st_code == 'RESOURCE_EXHAUSTED':
-    #            self.conv_id = None
-     #              return 'Sorry, chat turns exceeds plan limit.'
-      #      return 'Sorry, something went wrong in my brain. Please try again later.'
-
-        self.conv_id = res['responseSet'][0]['summary'][0]['chat']["1dc2c542-925f-4b48-aab5-e3df6f2a3a64"]
-        print("BIG CHANGE")
-
+        if chat['status'] != None:
+            st_code = chat['status']
+            print(f"Chat query failed with code {st_code}")
+            if st_code == 'RESOURCE_EXHAUSTED':
+                self.conv_id = None
+                return 'Sorry, Vectara chat turns exceeds plan limit.'
+            return 'Sorry, something went wrong in my brain. Please try again later.'
+        
+        try:
+            self.conv_id = res['responseSet'][0]['summary'][0]['chat']['conversationId']
+        else:
+            print("I'm sorry. I am experiencing an error in Vectara API conversationId assignment")
         
         pattern = r'\[\d{1,2}\]'
         matches = [match.span() for match in re.finditer(pattern, summary)]