name agnostic

dfusionai · Dec 14, 2024 · 2d29b7d · 2d29b7d
1 parent f189951
commit 2d29b7d
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/psl_proof/proof.py b/psl_proof/proof.py
@@ -27,14 +27,12 @@ def generate(self) -> ProofResponse:
 
         for input_filename in os.listdir(self.config['input_dir']):
             input_file = os.path.join(self.config['input_dir'], input_filename)
-            if os.path.splitext(input_file)[1].lower() == '.json':
-                with open(input_file, 'r') as f:
-                    input_data = json.load(f)
-                    if input_filename == 'chats.json':
-                        source_data = get_source_data(
-                            input_data
-                        )
-                        continue
+            with open(input_file, 'r') as f:
+                input_data = json.load(f)
+                source_data = get_source_data(
+                    input_data
+                )
+                break
 
         salt = self.config['salt']
         source_user_hash_64 = salted_data(