ubi integration

semantic_search/all_search_execute.py

@@ -19,6 +19,7 @@ from datetime import datetime
 import boto3
 import streamlit as st
 import utilities.mvectors as cb
+import utilities.ubi_lambda as ubi
 current_date_time = (datetime.now()).isoformat()
 today_ = datetime.today().strftime('%Y-%m-%d')
 
@@ -514,6 +515,22 @@ def handler(input_,session_id):
             arr.append(res_)
             dup.append(doc['_source']['image_url'])
 
+    ########### ubi lambda call for query logging ###########
+    query_payload = {
+    "query_id": "abc123",
+    "user_query": json.dumps(hybrid_payload),
+    
+    }
+
+    status = ubi.send_to_lambda("ubi_queries", query_payload)
+
+    if status == 202:
+        st.success("Query sent to Lambda")
+    else:
+        st.warning("Lambda did not accept the request")
+
+    ########### end of ubi lambda call for query logging ###########
+
     return arr[0:k_]
     
     

utilities/ubi_lambda.py

@@ -0,0 +1,23 @@
+import boto3
+import json
+import os
+
+lambda_client = boto3.client(
+    'lambda',
+    aws_access_key_id=st.secrets['user_access_key_us_west_2'],
+    aws_secret_access_key=st.secrets['user_secret_key_us_west_2'], region_name = 'us-west-2'
+)
+
+def send_to_lambda(index, document):
+    payload = {
+        "target_index": index,
+        "document": document
+    }
+
+    response = lambda_client.invoke(
+        FunctionName='proxy-observability-opensearch-ubi',
+        InvocationType='Event',  # async!
+        Payload=json.dumps(payload)
+    )
+
+    return response['StatusCode']  # 202 if accepted