Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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 |
|