regenerate

Author: Terry Zhao

test-proj/src/test_proj/export_types.py

@@ -65,20 +65,14 @@ def post_process_typescript_declarations(schema_dir: Path) -> None:
     index_signature_unknown_pattern = re.compile(r"\[k:\s*string\]:\s*unknown;?")
     index_signature_any_pattern = re.compile(r"\[k:\s*string\]:\s*any;?")
     import_statement = "import type { JSONObject } from '@llamaindex/ui';\n"
-    import_regex = re.compile(
-        r"import\s+type\s+\{\s*JSONObject\s*\}\s+from\s+'@llamaindex/ui';"
-    )
+    import_regex = re.compile(r"import\s+type\s+\{\s*JSONObject\s*\}\s+from\s+'@llamaindex/ui';")
 
     for dts_path in schema_dir.glob("*.d.ts"):
         content = dts_path.read_text(encoding="utf-8")
 
         # Replace index signature value types
-        new_content = index_signature_unknown_pattern.sub(
-            "[k: string]: JSONObject;", content
-        )
-        new_content = index_signature_any_pattern.sub(
-            "[k: string]: JSONObject;", new_content
-        )
+        new_content = index_signature_unknown_pattern.sub("[k: string]: JSONObject;", content)
+        new_content = index_signature_any_pattern.sub("[k: string]: JSONObject;", new_content)
 
         # Insert import if JSONObject is used and import not present
         if "JSONObject" in new_content and not import_regex.search(new_content):
@@ -88,14 +82,8 @@ def post_process_typescript_declarations(schema_dir: Path) -> None:
             if banner_end != -1:
                 # Move to the next newline after banner
                 next_newline = new_content.find("\n", banner_end + 2)
-                insertion_index = (
-                    next_newline + 1 if next_newline != -1 else banner_end + 2
-                )
-            new_content = (
-                new_content[:insertion_index]
-                + import_statement
-                + new_content[insertion_index:]
-            )
+                insertion_index = next_newline + 1 if next_newline != -1 else banner_end + 2
+            new_content = new_content[:insertion_index] + import_statement + new_content[insertion_index:]
 
         if new_content != content:
             dts_path.write_text(new_content, encoding="utf-8")