Changed script code to run under main.

Codium-ai · Nov 8, 2024 · 1685631 · 1685631
1 parent d537b18
commit 1685631
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/scripts/build_docs_context.py b/scripts/build_docs_context.py
@@ -38,7 +38,8 @@ def get_markdown_content(directory):
     markdown_content = re.sub(r'\n{3,}', '\n', markdown_content)
     return markdown_content
 
-docs_path = Path(__file__).parent.parent / "docs"
-markdown_content = get_markdown_content(docs_path)
-with open(".llm_context.txt", "w") as f:
-    f.write(markdown_content)
+if __name__ == "__main__":
+    docs_path = Path(__file__).parent.parent / "docs"
+    markdown_content = get_markdown_content(docs_path)
+    with open(".llm_context.txt", "w") as f:
+        f.write(markdown_content)