Adding Virus_total_key argument

Ostorlab · Dec 2, 2024 · 091a3d4 · 091a3d4
1 parent 7f2634f
commit 091a3d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agent/subfinder_agent.py b/agent/subfinder_agent.py
@@ -25,7 +25,7 @@
 STORAGE_NAME = "agent_subfinder_storage"
 
 
-def update_provider_config(
+def _update_provider_config(
     virustotal_key: str,
     config_path: str = "/root/.config/subfinder/provider-config.yaml",
 ) -> None:
@@ -65,7 +65,7 @@ def __init__(
         virustotal_key = self.args.get("virustotal_key")
         if virustotal_key is not None:
             logger.info("Updating configuration with VirusTotal API key.")
-            update_provider_config(virustotal_key)
+            _update_provider_config(virustotal_key)
 
         agent_persist_mixin.AgentPersistMixin.__init__(self, agent_settings)