txt2stix 1.2.1__tar.gz → 1.2.3__tar.gz

{txt2stix-1.2.1 → txt2stix-1.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: txt2stix
-Version: 1.2.1
+Version: 1.2.3
 Summary: txt2stix is a Python script that is designed to identify and extract IoCs and TTPs from text files, identify the relationships between them, convert them to STIX 2.1 objects, and output as a STIX 2.1 bundle.
 Project-URL: Homepage, https://github.com/muchdogesec/txt2stix
 Project-URL: Issues, https://github.com/muchdogesec/txt2stix/issues

{txt2stix-1.2.1 → txt2stix-1.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "txt2stix"
-version = "1.2.1"
+version = "1.2.3"
 authors = [{ name = "dogesec" }]
 maintainers = [{ name = "dogesec" }]
 description = "txt2stix is a Python script that is designed to identify and extract IoCs and TTPs from text files, identify the relationships between them, convert them to STIX 2.1 objects, and output as a STIX 2.1 bundle."

{txt2stix-1.2.1 → txt2stix-1.2.3}/tests/src/test_retriever.py

@@ -105,12 +105,12 @@ def f():
         (
             "ctibutler-mitre-atlas-id",
             "AML.T0050",
-            ["attack-pattern--eda10125-6f7d-479f-857a-19ef5d86a961"],
+            ["attack-pattern--07421f1a-a5ae-5936-9713-c77e4758177c"],
         ),
         (
             "ctibutler-mitre-atlas-name",
             "Defense Evasion",
-            ["x-mitre-tactic--c9cf0175-296e-4439-852d-afb870ed5e0c"],
+            ["x-mitre-tactic--22a483dc-1102-5fd0-94bd-b4259c537274"],
         ),
         (
             "ctibutler-disarm-id",

{txt2stix-1.2.1 → txt2stix-1.2.3}/txt2stix/txt2stix.py

@@ -461,17 +461,21 @@ def run_txt2stix(
     # First, perform extraction-phase (LLM and extractor calls). This does not
     # modify the provided bundler so the results can be saved and replayed.
     # skip extraction phase if txt2stix_data is passed
-    txt2stix_data = txt2stix_data or extraction_phase(
-        preprocessed_text,
-        extractors_map,
-        ai_content_check_provider=ai_content_check_provider,
-        input_token_limit=input_token_limit,
-        ai_settings_extractions=ai_settings_extractions,
-        ai_settings_relationships=ai_settings_relationships,
-        relationship_mode=relationship_mode,
-        ignore_extraction_boundary=ignore_extraction_boundary,
-        ai_extract_if_no_incidence=ai_extract_if_no_incidence,
-    )
+    if not txt2stix_data:
+        logging.info("=== Extraction Phase ===")
+        txt2stix_data = extraction_phase(
+            preprocessed_text,
+            extractors_map,
+            ai_content_check_provider=ai_content_check_provider,
+            input_token_limit=input_token_limit,
+            ai_settings_extractions=ai_settings_extractions,
+            ai_settings_relationships=ai_settings_relationships,
+            relationship_mode=relationship_mode,
+            ignore_extraction_boundary=ignore_extraction_boundary,
+            ai_extract_if_no_incidence=ai_extract_if_no_incidence,
+        )
+    else:
+        logging.info("=== Skipping Extraction Phase (replaying saved data) ===")
 
     # Then, process the extracted data into the bundler (no LLM calls).
     processing_phase(