Update query.py

query.py

@@ -9,7 +9,7 @@ def extract_between_tags(text, start_tag, end_tag):
     return text[start_index+len(start_tag):end_index-len(end_tag)]
 
 class VectaraQuery():
-    def __init__(self, api_key: str, customer_id: int, corpus_ids: list):
+    def __init__(self, api_key: str, customer_id: str, corpus_ids: list[str]):
         self.customer_id = customer_id
         self.corpus_ids = corpus_ids
         self.api_key = api_key
@@ -17,7 +17,7 @@ class VectaraQuery():
 
     def submit_query(self, query_str: str):
         corpora_key_list = [{
-                'customer_id': str(self.customer_id), 'corpus_id': str(corpus_id), 'lexical_interpolation_config': {'lambda': 0.025}
+                'customer_id': self.customer_id, 'corpus_id': corpus_id, 'lexical_interpolation_config': {'lambda': 0.025}
             } for corpus_id in self.corpus_ids
         ]
 
@@ -27,7 +27,7 @@ class VectaraQuery():
         headers = {
             "Content-Type": "application/json",
             "Accept": "application/json",
-            "customer-id": str(self.customer_id),
+            "customer-id": self.customer_id,
             "x-api-key": self.api_key,
             "grpc-timeout": "60S"
         }