awx-zipline-ai 0.3.4__tar.gz

awx_zipline_ai-0.3.4/PKG-INFO

@@ -0,0 +1,196 @@
+Metadata-Version: 2.4
+Name: awx-zipline-ai
+Version: 0.3.4
+Summary: CLI tool for the Zipline AI platform
+Author-email: Zipline AI <hello@zipline.ai>
+License: Apache License 2.0
+Project-URL: homepage, https://zipline.ai
+Project-URL: documentation, https://docs.zipline.ai
+Project-URL: github, https://github.com/zipline-ai/chronon/
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: boto3==1.40.26
+Requires-Dist: botocore==1.40.26
+Requires-Dist: cachetools==5.5.2
+Requires-Dist: certifi==2025.8.3
+Requires-Dist: charset-normalizer==3.4.3
+Requires-Dist: click==8.2.1
+Requires-Dist: crcmod==1.7
+Requires-Dist: gitdb==4.0.12
+Requires-Dist: gitpython==3.1.45
+Requires-Dist: google-api-core[grpc]==2.25.1
+Requires-Dist: google-auth==2.40.3
+Requires-Dist: google-cloud-bigquery-storage==2.33.0
+Requires-Dist: google-cloud-core==2.4.3
+Requires-Dist: google-cloud-iam==2.19.1
+Requires-Dist: google-cloud-storage==2.19.0
+Requires-Dist: google-crc32c==1.7.1
+Requires-Dist: google-resumable-media==2.7.2
+Requires-Dist: googleapis-common-protos[grpc]==1.70.0
+Requires-Dist: grpc-google-iam-v1==0.14.2
+Requires-Dist: grpcio==1.74.0
+Requires-Dist: grpcio-status==1.74.0
+Requires-Dist: idna==3.10
+Requires-Dist: importlib-resources==6.5.2
+Requires-Dist: jmespath==1.0.1
+Requires-Dist: markdown-it-py==4.0.0
+Requires-Dist: mdurl==0.1.2
+Requires-Dist: proto-plus==1.26.1
+Requires-Dist: protobuf==6.32.0
+Requires-Dist: py4j==0.10.9.7
+Requires-Dist: pyasn1==0.6.1
+Requires-Dist: pyasn1-modules==0.4.2
+Requires-Dist: pygments==2.19.2
+Requires-Dist: pyspark==3.5.4
+Requires-Dist: python-dateutil==2.9.0.post0
+Requires-Dist: requests==2.32.5
+Requires-Dist: rich==14.1.0
+Requires-Dist: rsa==4.9.1
+Requires-Dist: s3transfer==0.13.1
+Requires-Dist: six==1.17.0
+Requires-Dist: smmap==5.0.2
+Requires-Dist: thrift==0.21.0
+Requires-Dist: urllib3==2.5.0
+
+### Chronon Python API
+
+
+#### Overview
+
+Chronon Python API for materializing configs to be run by the Chronon Engine. Contains python helpers to help managed a repo of feature and join definitions to be executed by the chronon scala engine.
+
+
+#### User API Overview
+
+##### Sources
+
+Most fields are self explanatory. Time columns are expected to be in milliseconds (unixtime).
+
+```python
+# File <repo>/sources/test_sources.py
+from ai.chronon.query import (
+  Query,
+  select,
+)
+from ai.chronon.api.ttypes import Source, EventSource, EntitySource
+
+# Sample query
+Query(
+  selects=select(
+      user="user_id",
+      created_at="created_at",
+  ),
+  wheres=["has_availability = 1"],
+  start_partition="2021-01-01",  # Defines the beginning of time for computations related to the source.
+  setups=["...UDF..."],
+  time_column="ts",
+  end_partition=None,
+  mutation_time_column="mutation_timestamp",
+  reversal_column="CASE WHEN mutation_type IN ('DELETE', 'UPDATE_BEFORE') THEN true ELSE false END"
+)
+
+user_activity = Source(entities=EntitySource(
+  snapshotTable="db_exports.table",
+  mutationTable="mutations_namespace.table_mutations",
+  mutationTopic="mutationsKafkaTopic",
+  query=Query(...)
+)
+
+website__views = Source(events=EventSource(
+  table="namespace.table",
+  topic="kafkaTopicForEvents",
+)
+```
+
+
+##### Group By (Features)
+
+Group Bys are aggregations over sources that define features. For example:
+
+```python
+# File <repo>/group_bys/example_team/example_group_by.py
+from ai.chronon.group_by import (
+  GroupBy,
+  Window,
+  TimeUnit,
+  Accuracy,
+  Operation,
+  Aggregations,
+  Aggregation,
+  DefaultAggregation,
+)
+from sources import test_sources
+
+sum_cols = [f"active_{x}_days" for x in [30, 90, 120]]
+
+
+v0 = GroupBy(
+  sources=test_source.user_activity,
+  keys=["user"],
+  aggregations=Aggregations(
+    user_active_1_day=Aggregation(operation=Operation.LAST),
+    second_feature=Aggregation(
+      input_column="active_7_days",
+      operation=Operation.SUM,
+      windows=[
+        Window(n, TimeUnit.DAYS) for n in [3, 5, 9]
+      ]
+    ),
+  ) + [
+    Aggregation(
+      input_column=col,
+      operation=Operation.SUM
+    ) for col in sum_columns           # Alternative syntax for defining aggregations.
+  ] + [
+    Aggregation(
+      input_column="device",
+      operation=LAST_K(10)
+    )
+  ],
+  dependencies=[
+    "db_exports.table/ds={{ ds }}"      # If not defined will be derived from the Source info.
+  ],
+  accuracy=Accuracy.SNAPSHOT,          # This could be TEMPORAL for point in time correctness.
+  env={
+    "backfill": {                      # Execution environment variables for each of the modes for `run.py`
+      "EXECUTOR_MEMORY": "4G"
+     },
+  },
+  online=True,                         # True if this group by needs to be uploaded to a KV Store.
+  production=False                     # True if this group by is production level.
+)
+```
+
+##### Join
+
+A Join is a collection of feature values for the keys and (times if applicable) defined on the left (source). Example:
+
+```python
+# File <repo>/joins/example_team/example_join.py
+from ai.chronon.join import Join, JoinPart
+from sources import test_sources
+from group_bys.example_team import example_group_by
+
+v1 = Join(
+    left=test_sources.website__views,
+    right_parts=[
+        JoinPart(group_by=example_group_by.v0),
+    ],
+    online=True,       # True if this join will be fetched in production.
+    production=False,  # True if this join should not use non-production group bys.
+    env={"backfill": {"PARALLELISM": "10"}, "streaming": {"STREAMING_ENV_VAR": "VALUE"}},
+)
+```
+
+##### Pre-commit Setup
+
+1. Install pre-commit and other dev libraries:
+```
+pip install -r requirements/dev.txt
+```
+2. Run the following command under `api/python` to install the git hook scripts:
+```
+pre-commit install
+```
+
+To support more pre-commit hooks, add them to the `.pre-commit-config.yaml` file.

awx_zipline_ai-0.3.4/README.md

@@ -0,0 +1,142 @@
+### Chronon Python API
+
+
+#### Overview
+
+Chronon Python API for materializing configs to be run by the Chronon Engine. Contains python helpers to help managed a repo of feature and join definitions to be executed by the chronon scala engine.
+
+
+#### User API Overview
+
+##### Sources
+
+Most fields are self explanatory. Time columns are expected to be in milliseconds (unixtime).
+
+```python
+# File <repo>/sources/test_sources.py
+from ai.chronon.query import (
+  Query,
+  select,
+)
+from ai.chronon.api.ttypes import Source, EventSource, EntitySource
+
+# Sample query
+Query(
+  selects=select(
+      user="user_id",
+      created_at="created_at",
+  ),
+  wheres=["has_availability = 1"],
+  start_partition="2021-01-01",  # Defines the beginning of time for computations related to the source.
+  setups=["...UDF..."],
+  time_column="ts",
+  end_partition=None,
+  mutation_time_column="mutation_timestamp",
+  reversal_column="CASE WHEN mutation_type IN ('DELETE', 'UPDATE_BEFORE') THEN true ELSE false END"
+)
+
+user_activity = Source(entities=EntitySource(
+  snapshotTable="db_exports.table",
+  mutationTable="mutations_namespace.table_mutations",
+  mutationTopic="mutationsKafkaTopic",
+  query=Query(...)
+)
+
+website__views = Source(events=EventSource(
+  table="namespace.table",
+  topic="kafkaTopicForEvents",
+)
+```
+
+
+##### Group By (Features)
+
+Group Bys are aggregations over sources that define features. For example:
+
+```python
+# File <repo>/group_bys/example_team/example_group_by.py
+from ai.chronon.group_by import (
+  GroupBy,
+  Window,
+  TimeUnit,
+  Accuracy,
+  Operation,
+  Aggregations,
+  Aggregation,
+  DefaultAggregation,
+)
+from sources import test_sources
+
+sum_cols = [f"active_{x}_days" for x in [30, 90, 120]]
+
+
+v0 = GroupBy(
+  sources=test_source.user_activity,
+  keys=["user"],
+  aggregations=Aggregations(
+    user_active_1_day=Aggregation(operation=Operation.LAST),
+    second_feature=Aggregation(
+      input_column="active_7_days",
+      operation=Operation.SUM,
+      windows=[
+        Window(n, TimeUnit.DAYS) for n in [3, 5, 9]
+      ]
+    ),
+  ) + [
+    Aggregation(
+      input_column=col,
+      operation=Operation.SUM
+    ) for col in sum_columns           # Alternative syntax for defining aggregations.
+  ] + [
+    Aggregation(
+      input_column="device",
+      operation=LAST_K(10)
+    )
+  ],
+  dependencies=[
+    "db_exports.table/ds={{ ds }}"      # If not defined will be derived from the Source info.
+  ],
+  accuracy=Accuracy.SNAPSHOT,          # This could be TEMPORAL for point in time correctness.
+  env={
+    "backfill": {                      # Execution environment variables for each of the modes for `run.py`
+      "EXECUTOR_MEMORY": "4G"
+     },
+  },
+  online=True,                         # True if this group by needs to be uploaded to a KV Store.
+  production=False                     # True if this group by is production level.
+)
+```
+
+##### Join
+
+A Join is a collection of feature values for the keys and (times if applicable) defined on the left (source). Example:
+
+```python
+# File <repo>/joins/example_team/example_join.py
+from ai.chronon.join import Join, JoinPart
+from sources import test_sources
+from group_bys.example_team import example_group_by
+
+v1 = Join(
+    left=test_sources.website__views,
+    right_parts=[
+        JoinPart(group_by=example_group_by.v0),
+    ],
+    online=True,       # True if this join will be fetched in production.
+    production=False,  # True if this join should not use non-production group bys.
+    env={"backfill": {"PARALLELISM": "10"}, "streaming": {"STREAMING_ENV_VAR": "VALUE"}},
+)
+```
+
+##### Pre-commit Setup
+
+1. Install pre-commit and other dev libraries:
+```
+pip install -r requirements/dev.txt
+```
+2. Run the following command under `api/python` to install the git hook scripts:
+```
+pre-commit install
+```
+
+To support more pre-commit hooks, add them to the `.pre-commit-config.yaml` file.

awx_zipline_ai-0.3.4/pyproject.toml

@@ -0,0 +1,23 @@
+[project]
+name="awx-zipline-ai"
+version="0.3.4"
+description="CLI tool for the Zipline AI platform"
+readme="README.md"
+dependencies=["boto3==1.40.26", "botocore==1.40.26", "cachetools==5.5.2", "certifi==2025.8.3", "charset-normalizer==3.4.3", "click==8.2.1", "crcmod==1.7", "gitdb==4.0.12", "gitpython==3.1.45", "google-api-core[grpc]==2.25.1", "google-auth==2.40.3", "google-cloud-bigquery-storage==2.33.0", "google-cloud-core==2.4.3", "google-cloud-iam==2.19.1", "google-cloud-storage==2.19.0", "google-crc32c==1.7.1", "google-resumable-media==2.7.2", "googleapis-common-protos[grpc]==1.70.0", "grpc-google-iam-v1==0.14.2", "grpcio==1.74.0", "grpcio-status==1.74.0", "idna==3.10", "importlib-resources==6.5.2", "jmespath==1.0.1", "markdown-it-py==4.0.0", "mdurl==0.1.2", "proto-plus==1.26.1", "protobuf==6.32.0", "py4j==0.10.9.7", "pyasn1==0.6.1", "pyasn1-modules==0.4.2", "pygments==2.19.2", "pyspark==3.5.4", "python-dateutil==2.9.0.post0", "requests==2.32.5", "rich==14.1.0", "rsa==4.9.1", "s3transfer==0.13.1", "six==1.17.0", "smmap==5.0.2", "thrift==0.21.0", "urllib3==2.5.0"]
+requires-python=">= 3.11"
+license={text="Apache License 2.0"}
+keywords=[]
+classifiers=[]
+authors=[{name="Zipline AI", email="hello@zipline.ai"}]
+
+[project.urls]
+"homepage"="https://zipline.ai"
+"documentation"="https://docs.zipline.ai"
+"github"="https://github.com/zipline-ai/chronon/"
+
+[build-system]
+requires=["setuptools"]
+build-backend="setuptools.build_meta"
+
+[project.scripts]
+zipline = "ai.chronon.repo.zipline:zipline"

awx_zipline_ai-0.3.4/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

awx_zipline_ai-0.3.4/src/agent/__init__.py

@@ -0,0 +1 @@
+__all__ = ['ttypes', 'constants']

awx_zipline_ai-0.3.4/src/agent/constants.py

@@ -0,0 +1,15 @@
+#
+# Autogenerated by Thrift Compiler (0.22.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+#  options string: py
+#
+
+from thrift.Thrift import TType, TMessageType, TFrozenDict, TException, TApplicationException
+from thrift.protocol.TProtocol import TProtocolException
+from thrift.TRecursive import fix_spec
+from uuid import UUID
+
+import sys
+from .ttypes import *