initial commit

threat_extraction.py

@@ -23,11 +23,11 @@ class ThreatIntelExtractorTool(Tool):
     inputs = ["text"]
     outputs = ["text"]
 
-    def __init__(self, firewall_manager: FirewallManager, packet_filter: PacketFilter):
+    def __init__(self):
         self.openai_api_key = env.get("OPENAI_API_KEY")
         openai.api_key = self.openai_api_key
-        self.firewall_manager = firewall_manager
-        self.packet_filter = packet_filter
+        self.firewall_manager = FirewallManager
+        self.packet_filter = PacketFilter
 
     def __call__(self, threat_intel_feed_url: str):
         # Scrape threat intelligence feed