Spaces:
Running
on
T4
Running
on
T4
ubi integration
Browse files
semantic_search/all_search_execute.py
CHANGED
@@ -485,7 +485,7 @@ def handler(input_,session_id):
|
|
485 |
|
486 |
arr = []
|
487 |
dup = []
|
488 |
-
|
489 |
for doc in docs:
|
490 |
if(doc['_source']['image_url'] not in dup):
|
491 |
res_ = {
|
@@ -509,7 +509,7 @@ def handler(input_,session_id):
|
|
509 |
# res_['rekog'] = {'color':doc['_source']['rekog_color'],'category': doc['_source']['rekog_categories'],'objects':doc['_source']['rekog_objects']}
|
510 |
|
511 |
res_['id'] = doc['_id']
|
512 |
-
|
513 |
res_['score'] = doc['_score']
|
514 |
res_['title'] = doc['_source']['product_description']
|
515 |
|
@@ -521,7 +521,7 @@ def handler(input_,session_id):
|
|
521 |
query_payload = {
|
522 |
"query_id": str(uuid.uuid4()),
|
523 |
"application": "Semantic Search",
|
524 |
-
"query_response_hit_ids":
|
525 |
"timestamp": datetime.utcnow().isoformat() + "Z",
|
526 |
"user_query": json.dumps(hybrid_payload),
|
527 |
|
|
|
485 |
|
486 |
arr = []
|
487 |
dup = []
|
488 |
+
doc_ids = []
|
489 |
for doc in docs:
|
490 |
if(doc['_source']['image_url'] not in dup):
|
491 |
res_ = {
|
|
|
509 |
# res_['rekog'] = {'color':doc['_source']['rekog_color'],'category': doc['_source']['rekog_categories'],'objects':doc['_source']['rekog_objects']}
|
510 |
|
511 |
res_['id'] = doc['_id']
|
512 |
+
doc_ids.append(doc['_id'])
|
513 |
res_['score'] = doc['_score']
|
514 |
res_['title'] = doc['_source']['product_description']
|
515 |
|
|
|
521 |
query_payload = {
|
522 |
"query_id": str(uuid.uuid4()),
|
523 |
"application": "Semantic Search",
|
524 |
+
"query_response_hit_ids": doc_ids,
|
525 |
"timestamp": datetime.utcnow().isoformat() + "Z",
|
526 |
"user_query": json.dumps(hybrid_payload),
|
527 |
|