added console_script for vectore store creation

Author: XThomasBU

edubotics_core/vectorstore/store_manager.py

@@ -7,6 +7,8 @@
 import os
 import time
 import asyncio
+import yaml
+import argparse
 
 
 class VectorStoreManager:
@@ -162,10 +164,7 @@ def __len__(self):
         return len(self.vector_db)
 
 
-if __name__ == "__main__":
-    import yaml
-    import argparse
-
+def main():
     # Add argument parsing for config files
     parser = argparse.ArgumentParser(description="Load configuration files.")
     parser.add_argument(
@@ -216,3 +215,7 @@ def __len__(self):
     print("Loaded database")
 
     print(f"View the logs at {config['log_dir']}/vector_db.log")
+
+
+if __name__ == "__main__":
+    main()

setup.py

@@ -20,4 +20,9 @@
     license="MIT",
     long_description=readme,
     long_description_content_type="text/markdown",
+    entry_points={
+        "console_scripts": [
+            "vectorstore_creator=edubotics_core.vectorstore.store_manager:main",
+        ],
+    },
 )