Added kgwizard script

Author: spgarcica

pyproject.toml

@@ -30,13 +30,14 @@ dataraider = [
     "requests", "opencv-python", "numpy", "regex",
     "pubchempy", "openai", "huggingface_hub"
 ]
-kgwizard = ["gremlin", "openai"]
+kgwizard = ["gremlinpython", "openai"]
 full = ["visualheist", "dataraider", "kgwizard"]
 
 [project.scripts]
 dataraider   = "scripts.run_dataraider:main"
 visualheist  = "scripts.run_visualheist:main"
 mermaid      = "scripts.run_mermaid:main"
+kgwizard     = "src.kgwizard.__main__:main"
 
 [tool.setuptools]
 # Tell setuptools to look in BOTH src/ and the current directory (where "scripts" is).

src/kgwizard/__main__.py

@@ -23,9 +23,9 @@
 from typing import Any, Callable, NewType, Sequence, TypeAlias, TypeVar
 
 import numpy as np
-from graphdb import janus
+from .graphdb import janus
 from gremlin_python.structure.graph import GraphTraversalSource
-from prompt import build_prompt, build_prompt_from_react_file, get_response
+from .prompt import build_prompt, build_prompt_from_react_file, get_response
 
 # This is the only way I've found to execute the transformation using multiprocessing
 global schema
@@ -867,12 +867,15 @@ def exec_transform(
             , start=args.dynamic_start
         )
 
-    
-if __name__ == "__main__":
+def main() -> None:
     parser = build_main_argparser()
     args = parser.parse_args()
     match args.command:
         case Commands.PARSE:
             exec_parser(args)
         case Commands.TRANSFORM:
             exec_transform(args)
+
+
+if __name__ == "__main__":
+    main()