Reggie commited on
Commit
b6625a4
1 Parent(s): da50bf2
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -81,14 +81,21 @@ def search():
81
  else: results = client.search(collection_name=collection_name, query_vector=(qvector, sq), with_payload=True, limit=100)
82
  print('SEARCH TIME: ', time.time() - timh)
83
 
84
- print(results[0])
85
  # try:
86
- if collection_name == 'jks': results = [{"text": x.payload['text'], "date": str(int(x.payload['date'])), "id": x.id} for x in results] # Implement your Qdrant search here
 
 
 
 
87
  else:
88
- new_results = []
89
  for r in results:
90
- if 'context' in r.payload and r.payload['context'] != '': new_results.append({"text": r.payload['title'] + '\nContext: ' + r.payload['context'], "url": r.payload['url'], "date": r.payload['date'], "id": r.id})
91
- else: new_results.append({"text": r.payload['title'], "url": r.payload['url'], "date": r.payload['date'], "id": r.id})
 
 
 
 
92
  results = new_results
93
  return jsonify(results)
94
  # except:
 
81
  else: results = client.search(collection_name=collection_name, query_vector=(qvector, sq), with_payload=True, limit=100)
82
  print('SEARCH TIME: ', time.time() - timh)
83
 
84
+ #print(results[0])
85
  # try:
86
+ new_results = []
87
+ if collection_name == 'jks':
88
+ for r in results:
89
+ if 'date' not in r.payload: r.payload['date'] = '20200101'
90
+ new_results.append({"text": r.payload['text'], "date": str(int(r.payload['date'])), "id": r.id}) # Implement your Qdrant search here
91
  else:
 
92
  for r in results:
93
+ if 'context' in r.payload and r.payload['context'] != '':
94
+ if 'date' not in r.payload: r.payload['date'] = '20200101'
95
+ new_results.append({"text": r.payload['title'] + '\nContext: ' + r.payload['context'], "url": r.payload['url'], "date": r.payload['date'], "id": r.id})
96
+ else:
97
+ if 'date' not in r.payload: r.payload['date'] = '20200101'
98
+ new_results.append({"text": r.payload['title'], "url": r.payload['url'], "date": r.payload['date'], "id": r.id})
99
  results = new_results
100
  return jsonify(results)
101
  # except: