reait 1.2.1__tar.gz → 1.2.2__tar.gz

{reait-1.2.1 → reait-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reait
-Version: 1.2.1
+Version: 1.2.2
 Summary: RevEng.AI Toolkit and Python API
 Home-page: https://github.com/RevEng-AI/reait
 Author: James Patrick-Evans

{reait-1.2.1 → reait-1.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "reait"
-version = "1.2.1"
+version = "1.2.2"
 readme = "README.md"
 classifiers=[
   "Programming Language :: Python :: 3",

{reait-1.2.1 → reait-1.2.2}/src/reait/api.py

@@ -15,7 +15,7 @@ from pandas import DataFrame
 from requests import request, Response, HTTPError
 from sklearn.metrics.pairwise import cosine_similarity
 
-__version__ = "1.2.1"
+__version__ = "1.2.2"
 
 re_conf = {
     "apikey": environ.get("REAI_API_KEY", ""),
@@ -335,43 +335,25 @@ def RE_upload(fpath: str) -> Response:
     :param fpath: File path for binary to analyse
     """
     bin_id = re_binary_id(fpath)
-    result = re_hash_check(bin_id)
 
-    if result:
+    with open(fpath, "rb") as fd:
+        res: Response = reveng_req(
+            requests.post, "v1/upload", files={"file": fd})
+
+    if res.ok:
         logger.info(
-            "File %s - %s already uploaded. Skipping upload...", fpath, bin_id)
-
-        res = Response()
-        res.status_code = 200
-        res.url = f"{re_conf['host']}/v1/upload"
-        res._content = (
-            (
-                '{0}"success": true,'
-                '"message": "File already uploaded!",'
-                '"sha_256_hash": "{1}"{2}'
-            )
-            .format("{", bin_id, "}")
-            .encode()
+            "Successfully uploaded binary to your account. %s - %s", fpath, bin_id
+        )
+    elif res.status_code == 400:
+        if "error" in res.json().keys():
+            logger.warning("Error uploading %s - %s",
+                           fpath, res.json()["error"])
+    elif res.status_code == 413:
+        logger.warning("File too large. Please upload files under 10MB.")
+    elif res.status_code == 500:
+        logger.error(
+            "Internal Server Error. Please contact support. Skipping upload..."
         )
-    else:
-        with open(fpath, "rb") as fd:
-            res: Response = reveng_req(
-                requests.post, "v1/upload", files={"file": fd})
-
-        if res.ok:
-            logger.info(
-                "Successfully uploaded binary to your account. %s - %s", fpath, bin_id
-            )
-        elif res.status_code == 400:
-            if "error" in res.json().keys():
-                logger.warning("Error uploading %s - %s",
-                               fpath, res.json()["error"])
-        elif res.status_code == 413:
-            logger.warning("File too large. Please upload files under 10MB.")
-        elif res.status_code == 500:
-            logger.error(
-                "Internal Server Error. Please contact support. Skipping upload..."
-            )
 
     res.raise_for_status()
     return res
@@ -1301,3 +1283,31 @@ def RE_similar_functions(
 
     res.raise_for_status()
     return res
+
+
+def RE_binary_ann(
+    analysis_id: int,
+    confidence: float = 0.0,
+    nns: int = 1,
+    collection_ids: list[int] = None,
+    binary_ids: list[int] = None,
+) -> Response:
+    """
+    Perform binary ANN (Approximate Nearest Neighbor) search.
+    :param analysis_id: Analysis ID
+    :param confidence: Confidence threshold for the search
+    :param nns: Number of nearest neighbors to retrieve
+    :param collection_ids: List of collection IDs to search within
+    :param binary_ids: List of binary IDs to search within
+    """
+    end_point = f"v2/binary_ann/{analysis_id}"
+    json_data = {
+        "confidence": confidence,
+        "nns": nns,
+        "collection_ids": collection_ids or [],
+        "binary_ids": binary_ids or [],
+    }
+
+    res: Response = reveng_req(requests.post, end_point, json_data=json_data)
+    res.raise_for_status()
+    return res

{reait-1.2.1 → reait-1.2.2}/src/reait.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reait
-Version: 1.2.1
+Version: 1.2.2
 Summary: RevEng.AI Toolkit and Python API
 Home-page: https://github.com/RevEng-AI/reait
 Author: James Patrick-Evans