fix: upload tags + scope

Author: vgoat21

reai_toolkit/features/configuration/config.py

@@ -106,6 +106,9 @@ def get_all_analyses(self):
 
 
     def set_current_info(self, binary_id, analysis_id):
+
+        log_info(f"SET_CURRENT_INFO: {binary_id}, {analysis_id}")
+
         try:
             binary_id = int(binary_id)
             analysis_id = int(analysis_id)

reai_toolkit/features/upload/upload.py

@@ -74,15 +74,22 @@ def upload_binary(self, bv: BinaryView, options: dict):
 
                 analyses_client = revengai.AnalysesCoreApi(api_client)
 
+                tags = []
+
+                for tag in options["tags"]:
+                    tags.append(revengai.Tag(name=tag))
+
+                analysis_create_request=revengai.AnalysisCreateRequest(
+                    filename=basename(bv.file.filename),
+                    sha_256_hash=sha_256_hash,
+                    debug_hash=debug_info_hash if options["debug_info"] else None,
+                    tags=tags,
+                    analysis_scope=revengai.AnalysisScope.PRIVATE if options["is_private"] else revengai.AnalysisScope.PUBLIC,
+                    symbols=symbols
+                )
+                
                 analysis_result = analyses_client.create_analysis(
-                    analysis_create_request=revengai.AnalysisCreateRequest(
-                        filename=basename(bv.file.filename),
-                        sha_256_hash=sha_256_hash,
-                        debug_hash=debug_info_hash if options["debug_info"] else None,
-                        tags=options["tags"] or [],
-                        analysis_scope=revengai.AnalysisScope.PUBLIC if options["is_private"] else revengai.AnalysisScope.PRIVATE,
-                        symbols=symbols
-                    )
+                    analysis_create_request=analysis_create_request
                 )
 
                 log_info(f"RevEng.AI | Analysis started successfully. Analysis ID: {analysis_result.data.analysis_id}, Binary ID: {analysis_result.data.binary_id}")

reai_toolkit/utils/monitoring/process_binary_monitor.py

@@ -53,6 +53,7 @@ def _worker(bv: BinaryView, bid: int, aid: int):
                             f"RevEng.AI | Scheduled next status check for: {basename(bv.file.filename)} [Binary ID: {bid}] [Analysis ID: {aid}]"
                         )
                 else:
+                    log_info("CALLBACK MADE")
                     callback(bid, aid)
                     log_info(f"RevEng.AI | Analysis completed with status: {status} for Binary ID: {bid} and Analysis ID: {aid}")
             except RequestException as ex: