txt2detection 1.0.12__tar.gz → 1.0.13__tar.gz

{txt2detection-1.0.12 → txt2detection-1.0.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: txt2detection
-Version: 1.0.12
+Version: 1.0.13
 Summary: A command line tool that takes a txt file containing threat intelligence and turns it into a detection rule.
 Project-URL: Homepage, https://github.com/muchdogesec/txt2detection
 Project-URL: Issues, https://github.com/muchdogesec/txt2detection/issues
@@ -12,6 +12,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.11
+Requires-Dist: json-repair
 Requires-Dist: jsonschema>=4.22.0; python_version >= '3.8'
 Requires-Dist: llama-index-core>=0.12.42
 Requires-Dist: llama-index-llms-openai>=0.4.5

{txt2detection-1.0.12 → txt2detection-1.0.13}/pyproject.toml

@@ -4,35 +4,32 @@ build-backend = "hatchling.build"
 
 [project]
 name = "txt2detection"
-version = "1.0.12"
-authors = [
-  { name = "dogesec" }
-]
-maintainers = [
-  { name = "dogesec" }
-]
+version = "1.0.13"
+authors = [{ name = "dogesec" }]
+maintainers = [{ name = "dogesec" }]
 description = "A command line tool that takes a txt file containing threat intelligence and turns it into a detection rule."
 readme = "README.md"
 requires-python = ">=3.11"
 classifiers = [
-    "Programming Language :: Python :: 3",
-    "License :: OSI Approved :: Apache Software License",
-    "Operating System :: OS Independent",
+  "Programming Language :: Python :: 3",
+  "License :: OSI Approved :: Apache Software License",
+  "Operating System :: OS Independent",
 ]
 dependencies = [
-    "stix2",
-    "python-arango>=8.1.3; python_version >= '3.8'",
-    "tqdm>=4.66.4; python_version >= '3.7'",
-    "jsonschema>=4.22.0; python_version >= '3.8'",
-    "requests>=2.31.0; python_version >= '3.7'",
-    "python-dotenv>=1.0.1",
-    "pyyaml",
-    "python-slugify",
-    "validators>=0.34.0",
-    "llama-index-core>=0.12.42",
-    'llama-index-llms-openai>=0.4.5',
-    'stix2extensions',
-  ]
+  "stix2",
+  "python-arango>=8.1.3; python_version >= '3.8'",
+  "tqdm>=4.66.4; python_version >= '3.7'",
+  "jsonschema>=4.22.0; python_version >= '3.8'",
+  "requests>=2.31.0; python_version >= '3.7'",
+  "python-dotenv>=1.0.1",
+  "pyyaml",
+  "python-slugify",
+  "validators>=0.34.0",
+  "llama-index-core>=0.12.42",
+  'llama-index-llms-openai>=0.4.5',
+  'stix2extensions',
+  'json-repair',
+]
 [project.urls]
 Homepage = "https://github.com/muchdogesec/txt2detection"
 Issues = "https://github.com/muchdogesec/txt2detection/issues"
@@ -50,4 +47,4 @@ llms = ['txt2detection[anthropic,gemini,deepseek,openrouter]']
 
 
 [tool.hatch.build.targets.wheel.force-include]
-"config" = "txt2detection/config"
+"config" = "txt2detection/config"

{txt2detection-1.0.12 → txt2detection-1.0.13}/txt2detection/__main__.py

@@ -265,8 +265,9 @@ def run_txt2detection(
                 created=detection.date,
                 modified=detection.modified,
             )
-        detection.level = kwargs.get("level", detection.level)
-        detection.status = kwargs.get("status", detection.status)
+        kwargs['license'] = kwargs.get('license') or detection.license
+        detection.level = kwargs.get("level") or detection.level
+        detection.status = kwargs.get("status") or detection.status
         detection.date = as_date(kwargs.get("created"))
         detection.modified = as_date(kwargs.get("modified"))
         detection.references = kwargs["reference_urls"]

{txt2detection-1.0.12 → txt2detection-1.0.13}/txt2detection/ai_extractor/utils.py

@@ -1,6 +1,7 @@
 import io
 import logging
 import typing
+import json_repair
 
 from llama_index.core.output_parsers import PydanticOutputParser
 
@@ -16,4 +17,5 @@ class ParserWithLogging(PydanticOutputParser):
         print(text, file=f)
         print("=================close=================" + "\n" * 5, file=f)
         logging.debug(f.getvalue())
-        return super().parse(text)
+        repaired_json = json_repair.repair_json(text)
+        return super().parse(repaired_json)