Build zipline-ai with Commands "zipline compile" and "zipline run"

api/py/ai/chronon/repo/compile.py

@@ -44,7 +44,7 @@ def get_folder_name_from_class_name(class_name):
     return {v.__name__: k for k, v in FOLDER_NAME_TO_CLASS.items()}[class_name]
 
 
-@click.command()
+@click.command(name='compile')
 @click.option(
     "--chronon_root",
     envvar="CHRONON_ROOT",

api/py/ai/chronon/repo/zipline.py

@@ -0,0 +1,13 @@
+import click
+
+from ai.chronon.repo.compile import extract_and_convert
+from ai.chronon.repo.run import main as run_main
+
+
+@click.group()
+def zipline():
+    pass
+
+
+zipline.add_command(extract_and_convert)
+zipline.add_command(run_main)

api/py/requirements/base.in

@@ -1,2 +1,3 @@
 click
 thrift==0.21.0
+google-cloud-storage==2.19.0

api/py/requirements/base.txt

@@ -1,13 +1,63 @@
-# SHA1:e6acf05ccde0547fea15f1185172abe12ea24af2
+# SHA1:fe8b0a1dc101ff0b0ffa9a959160d459b7f7d0e3
 #
 # This file is autogenerated by pip-compile-multi
 # To update, run:
 #
 #    pip-compile-multi
 #
+cachetools==5.5.0
+    # via google-auth
+certifi==2024.12.14
+    # via requests
+charset-normalizer==3.4.1
+    # via requests
 click==8.1.8
     # via -r requirements/base.in
+google-api-core==2.24.0
+    # via
+    #   google-cloud-core
+    #   google-cloud-storage
+google-auth==2.37.0
+    # via
+    #   google-api-core
+    #   google-cloud-core
+    #   google-cloud-storage
+google-cloud-core==2.4.1
+    # via google-cloud-storage
+google-cloud-storage==2.19.0
+    # via -r requirements/base.in
+google-crc32c==1.6.0
+    # via
+    #   google-cloud-storage
+    #   google-resumable-media
+google-resumable-media==2.7.2
+    # via google-cloud-storage
+googleapis-common-protos==1.66.0
+    # via google-api-core
+idna==3.10
+    # via requests
+proto-plus==1.25.0
+    # via google-api-core
+protobuf==5.29.3
+    # via
+    #   google-api-core
+    #   googleapis-common-protos
+    #   proto-plus
+pyasn1==0.6.1
+    # via
+    #   pyasn1-modules
+    #   rsa
+pyasn1-modules==0.4.1
+    # via google-auth
+requests==2.32.3
+    # via
+    #   google-api-core
+    #   google-cloud-storage
+rsa==4.9
+    # via google-auth
 six==1.17.0
     # via thrift
 thrift==0.21.0
     # via -r requirements/base.in
+urllib3==2.3.0
+    # via requests

api/py/requirements/dev.in

@@ -6,6 +6,4 @@ pre-commit
 isort
 autoflake
 zipp==3.19.1
-importlib-metadata==8.4.0
-google-cloud-storage==2.19.0
-
+importlib-metadata==8.4.0

api/py/requirements/dev.txt

@@ -1,4 +1,4 @@
-# SHA1:06b28bca353cb7f65a2a3712ef7c6deaeb35f0bb
+# SHA1:fb02b7333620d08e6d72ffb575b99015aee7c274
 #
 # This file is autogenerated by pip-compile-multi
 # To update, run:
@@ -10,16 +10,10 @@ autoflake==2.3.1
     # via -r requirements/dev.in
 black==24.10.0
     # via -r requirements/dev.in
-cachetools==5.5.0
-    # via
-    #   google-auth
-    #   tox
 cfgv==3.4.0
     # via pre-commit
 chardet==5.2.0
     # via tox
-charset-normalizer==3.4.1
-    # via requests
 colorama==0.4.6
     # via tox
 coverage[toml]==7.6.10
@@ -30,31 +24,8 @@ filelock==3.16.1
     # via
     #   tox
     #   virtualenv
-google-api-core==2.24.0
-    # via
-    #   google-cloud-core
-    #   google-cloud-storage
-google-auth==2.37.0
-    # via
-    #   google-api-core
-    #   google-cloud-core
-    #   google-cloud-storage
-google-cloud-core==2.4.1
-    # via google-cloud-storage
-google-cloud-storage==2.19.0
-    # via -r requirements/dev.in
-google-crc32c==1.6.0
-    # via
-    #   google-cloud-storage
-    #   google-resumable-media
-google-resumable-media==2.7.2
-    # via google-cloud-storage
-googleapis-common-protos==1.66.0
-    # via google-api-core
 identify==2.6.5
     # via pre-commit
-idna==3.10
-    # via requests
 importlib-metadata==8.4.0
     # via -r requirements/dev.in
 iniconfig==2.0.0
@@ -84,19 +55,6 @@ pluggy==1.5.0
     #   tox
 pre-commit==4.0.1
     # via -r requirements/dev.in
-proto-plus==1.25.0
-    # via google-api-core
-protobuf==5.29.3
-    # via
-    #   google-api-core
-    #   googleapis-common-protos
-    #   proto-plus
-pyasn1==0.6.1
-    # via
-    #   pyasn1-modules
-    #   rsa
-pyasn1-modules==0.4.1
-    # via google-auth
 pyflakes==3.2.0
     # via autoflake
 pyproject-api==1.8.0
@@ -107,16 +65,8 @@ pytest-cov==6.0.0
     # via -r requirements/dev.in
 pyyaml==6.0.2
     # via pre-commit
-requests==2.32.3
-    # via
-    #   google-api-core
-    #   google-cloud-storage
-rsa==4.9
-    # via google-auth
 tox==4.23.2
     # via -r requirements/dev.in
-urllib3==2.3.0
-    # via requests
 virtualenv==20.28.1
     # via
     #   pre-commit

api/py/setup.py

@@ -52,11 +52,15 @@ def get_version():
     ],
     long_description=long_description,
     long_description_content_type="text/markdown",
-    scripts=['ai/chronon/repo/explore.py', 'ai/chronon/repo/compile.py', 'ai/chronon/repo/run.py'],
-    description="Chronon python API library",
+    entry_points={
+        "console_scripts": [
+            "zipline=ai.chronon.repo.zipline:zipline",
+        ]
+    },
+    description="Zipline python API library",
     include_package_data=True,
     install_requires=basic_requirements,
-    name="chronon-ai",
+    name="zipline-ai",
     packages=find_packages(),
     extras_require={
         # Extra requirement to have access to cli commands in python2 environments.