Update app.py

app.py

@@ -14,6 +14,7 @@ from datetime import timedelta
 from datetime import timezone
 from faster_whisper import WhisperModel
 import io
+import requests
 
 app = Flask(__name__)
 
@@ -79,7 +80,11 @@ def search():
     print('EMBEDDING TIME: ', time.time() - timh)
 
     timh = time.time()
-    if collection_name == "jks": results = client.search(collection_name=collection_name, query_vector=sq, with_payload=True, limit=topN)
+    if collection_name == "jks":
+        data = {"vector": sq, "with_payload": True, "limit": topN}
+        response = requests.post(qdrant_url+f'jks/points/search', json=data, headers={'Content-Type': 'application/json'})
+        
+        # results = client.search(collection_name=collection_name, query_vector=sq, with_payload=True, limit=topN)
     else: results = client.search(collection_name=collection_name, query_vector=(qvector, sq), with_payload=True, limit=100)
     print('SEARCH TIME: ', time.time() - timh)
     
@@ -88,8 +93,8 @@ def search():
     new_results = []
     if collection_name == 'jks': 
         for r in results:
-            if 'date' not in r.payload: r.payload['date'] = '20200101'
-            new_results.append({"text": r.payload['text'], "date": str(int(r.payload['date'])), "id": r.id})  # Implement your Qdrant search here     
+            if 'date' not in r['payload']: r['payload']['date'] = '20200101'
+            new_results.append({"text": r['payload']['text'], "date": str(int(r['payload']['date'])), "id": r['id']})  # Implement your Qdrant search here     
     else:
         for r in results:
             if 'context' in r.payload and r.payload['context'] != '':