upgini 1.1.291a3232.post4__tar.gz → 1.1.292__tar.gz

{upgini-1.1.291a3232.post4 → upgini-1.1.292}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: upgini
-Version: 1.1.291a3232.post4
+Version: 1.1.292
 Summary: Intelligent data search & enrichment for Machine Learning
 Project-URL: Bug Reports, https://github.com/upgini/upgini/issues
 Project-URL: Homepage, https://upgini.com/

upgini-1.1.292/src/upgini/__about__.py

@@ -0,0 +1 @@
+__version__ = "1.1.292"

{upgini-1.1.291a3232.post4 → upgini-1.1.292}/src/upgini/data_source/data_source_publisher.py

@@ -347,3 +347,29 @@ class DataSourcePublisher:
             self.upload_online(search_keys=keys)
 
         print("All ADS-es successfully uploaded")
+
+    def union_search_tasks(
+        self,
+        search_ids: List[str],
+        target_user_email: str,
+        selected_features: Optional[List[str]] = None,
+        exclude_features: Optional[List[str]] = None,
+    ) -> str:
+        if not search_ids:
+            raise Exception("Empty search ids list")
+
+        if not target_user_email:
+            raise Exception("Empty target user email")
+
+        request = {
+            "search_task_ids": search_ids,
+            "target_user_email": target_user_email,
+        }
+        if selected_features:
+            request["selected_features"] = selected_features
+        if exclude_features:
+            request["exclude_features"] = exclude_features
+
+        response = self._rest_client.union_search_tasks(request, "trace_id")
+        print(response)
+        return response

{upgini-1.1.291a3232.post4 → upgini-1.1.292}/src/upgini/http.py

@@ -291,6 +291,7 @@ class _RestClient:
     GET_ACTIVE_ADS_DEFINITIONS_URI = "private/api/v2/ads/definitions"
     UPLOAD_ONLINE_URI = "private/api/v2/ads/upload-online"
     STOP_ADS_MANAGEMENT_TASK_URI_FMT = "private/api/v2/ads/management-task/{0}/stop"
+    UNION_SEARCH_TASKS_URI_FMT = SERVICE_ROOT_V2 + "search/merge"
 
     ACCESS_TOKEN_HEADER_NAME = "Authorization"
     CONTENT_TYPE_HEADER_NAME = "Content-Type"
@@ -760,6 +761,10 @@ class _RestClient:
         api_path = self.STOP_ADS_MANAGEMENT_TASK_URI_FMT.format(ads_management_task_id)
         self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id))
 
+    def union_search_tasks(self, request: dict, trace_id: str):
+        api_path = self.UNION_SEARCH_TASKS_URI_FMT
+        return self._with_unauth_retry(lambda: self._send_post_req(api_path, trace_id, request, result_format=None))
+
     # ---
 
     def _send_get_req(self, api_path: str, trace_id: Optional[str], additional_headers: Optional[dict] = None):

upgini-1.1.291a3232.post4/src/upgini/__about__.py

@@ -1 +0,0 @@
-__version__ = "1.1.291a3232-4"