Update app.py

app.py

@@ -9,29 +9,46 @@ def query_vectara(question):
     user_message = question
 
     # Read authentication parameters from the .env file
-    customer_id = config('CUSTOMER_ID')
-    corpus_id = config('CORPUS_ID')
-    api_key = config('API_KEY')
-
-    # Define the query URL
-    query_url = f"https://api.vectara.io:443/v1/query"
-
-    headers = {
-        "x-api-key": api_key,  # Use the x-api-key header for authentication
+    CUSTOMER_ID = config('CUSTOMER_ID')
+    CORPUS_ID = config('CORPUS_ID')
+    API_KEY = config('API_KEY')
+
+    # Define the data dictionary
+    data_dict = {
+        "query": [
+            {
+                "query": user_message,
+                "num_results": 10,
+                "corpus_key": [
+                    {
+                        "customer_id": CUSTOMER_ID,
+                        "corpus_id": CORPUS_ID
+                    }
+                ]
+            }
+        ]
     }
 
-    query_body = {
-        "query": user_message,
-        "num_results": 10
+    # Define the headers
+    api_key_header = {
+        "customer-id": CUSTOMER_ID,
+        "x-api-key": API_KEY
     }
 
-    query_response = requests.post(query_url, json=query_body, headers=headers)
-
-    if query_response.status_code == 200:
-        query_data = query_response.json()
+    # Make the API request
+    response = requests.post(
+        "https://api.vectara.io/v1/query",
+        data=json.dumps(data_dict),
+        verify=True,
+        headers=api_key_header,
+        headers={"Content-Type": "application/json"}  # Set Content-Type header
+    )
+
+    if response.status_code == 200:
+        query_data = response.json()
         response_message = f"Response from Vectara API: {json.dumps(query_data, indent=2)}"
     else:
-        response_message = f"Error: {query_response.status_code}"
+        response_message = f"Error: {response.status_code}"
 
     return response_message