chalkruntime 3.32.1__tar.gz

chalkruntime-3.32.1/PKG-INFO

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: chalkruntime
+Version: 3.32.1
+Summary: Runtime support library for Chalk AI
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: aiotools>=1.7.0
+Requires-Dist: boto3>=1.34.131
+Requires-Dist: datadog==0.47.0
+Requires-Dist: filelock>=3.12.4
+Requires-Dist: frozendict>=2.3.9
+Requires-Dist: fsspec>=2024.9.0
+Requires-Dist: gcsfs>=2024.9.0.post1
+Requires-Dist: adlfs>=2024.7.0
+Requires-Dist: google-cloud-storage>=2.18.2
+Requires-Dist: jinja2>=3.0.0
+Requires-Dist: numpy>=1.26.0
+Requires-Dist: pandas>=1.5.3
+Requires-Dist: pdb_attach>=3.0.1
+Requires-Dist: polars>=0.19.12
+Requires-Dist: pyarrow>=18.1.0
+Requires-Dist: pydantic<2.0
+Requires-Dist: s3fs>=2024.9.0
+Requires-Dist: sqlalchemy<2
+Requires-Dist: sqlglot<21.2.0,>=19.0.0
+Requires-Dist: typing_extensions>=4.12.2
+Requires-Dist: uvloop>=0.20.0
+Requires-Dist: opentelemetry-api>=1.38.0
+Requires-Dist: opentelemetry-sdk>=1.38.0
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.39.1
+Requires-Dist: opentelemetry-distro[otlp]>=0.60b1
+Requires-Dist: opentelemetry-instrumentation-fastapi>=0.59b0
+Requires-Dist: opentelemetry-instrumentation-requests>=0.59b0
+Requires-Dist: opentelemetry-instrumentation-httpx>=0.59b0
+Requires-Dist: opentelemetry-instrumentation-pymongo>=0.59b0
+Requires-Dist: opentelemetry-instrumentation-pymysql>=0.59b0
+Requires-Dist: opentelemetry-instrumentation-psycopg2>=0.59b0
+Requires-Dist: memray>=1.19.1
+Requires-Dist: viztracer>=1.0.0
+
+# chalkruntime
+
+Runtime support library used by Chalk AI's execution engine.
+
+This package is an internal component of the Chalk platform and is not intended
+to be consumed as a standalone library. See https://chalk.ai for more
+information.

chalkruntime-3.32.1/README.md

@@ -0,0 +1,7 @@
+# chalkruntime
+
+Runtime support library used by Chalk AI's execution engine.
+
+This package is an internal component of the Chalk platform and is not intended
+to be consumed as a standalone library. See https://chalk.ai for more
+information.

chalkruntime-3.32.1/chalkruntime/constants.py

@@ -0,0 +1,48 @@
+from datetime import datetime, timedelta, timezone
+
+import pyarrow as pa
+
+ID_FEATURE_COL_NAME = "__id__"
+TS_COL_NAME = "__ts__"
+TS_COL_TYPE = pa.timestamp("us", "UTC")
+
+PRESENCE_INDICATOR_COL_NAME = "__present__"
+
+INDEX_COL_NAME = "__index__"
+INDEX_COL_TYPE = pa.int64()
+"""The type of the `INDEX_COL_NAME` column."""
+
+CHILD_INDEX_COL_NAME = "__cidx__"
+OBSERVED_AT_FEATURE_COL_NAME = "__oat__"
+REPLACED_OBSERVED_AT_FEATURE_COL_NAME = "__rat__"
+SAMPLE_COL_NAME = "__sample__"
+IS_VALID_COL_NAME = "__is_valid__"
+RESOLVER_INPUT_PREFIX = "__CHALK_RESOLVER_INPUT__"
+
+HAS_MANY_SAMPLE_COL_NAME = "__hm_sample__"
+# Appears in the groups table of a has-many returning scalar resolver, equals __id__ of the input
+
+ONLINE_STORE_TTL_COL_NAME = "__exp__"
+
+CHALK_SPECIAL_COL_NAMES = (
+    ID_FEATURE_COL_NAME,
+    TS_COL_NAME,
+    INDEX_COL_NAME,
+    CHILD_INDEX_COL_NAME,
+    OBSERVED_AT_FEATURE_COL_NAME,
+    REPLACED_OBSERVED_AT_FEATURE_COL_NAME,
+    SAMPLE_COL_NAME,
+    IS_VALID_COL_NAME,
+    HAS_MANY_SAMPLE_COL_NAME,
+    ONLINE_STORE_TTL_COL_NAME,
+)
+
+INTERNAL_TTL_MAX = timedelta(days=365 * 10)
+CLOSE_TO_MIN_UTC = datetime.min.replace(tzinfo=timezone.utc) + timedelta(days=10)
+
+COUNT_COL_NAME = "__count__"
+MAX_OBSERVED_AT_COL_NAME = "__max_observed_at__"
+
+
+class SpecialResolverFQN:
+    CHALK_QUERY_INPUT = "__chalk__.query-input"

chalkruntime-3.32.1/chalkruntime/dataframe/__init__.py

@@ -0,0 +1,4 @@
+from .dataframe import ChalkDataFrame, chalk_dataframe_debug_context
+from .lazyframe import LazyFrame
+
+__all__ = ["ChalkDataFrame", "LazyFrame", "chalk_dataframe_debug_context"]

chalkruntime-3.32.1/chalkruntime/dataframe/dataframe.py

@@ -0,0 +1,18 @@
+"""Compatibility shims for the legacy ``chalkruntime.dataframe`` package.
+
+This module used to host the engine-facing ``ChalkDataFrame`` type before we
+introduced the standalone ``chalkdf`` package.  Some parts of the codebase (and
+older integrations) still import ``chalkruntime.dataframe.dataframe`` directly,
+so we keep a very small wrapper that simply forwards those imports to the new
+implementation.  Keeping this bridge avoids having to update every existing
+import site while the migration is in flight.
+"""
+
+from __future__ import annotations
+
+from chalkdf import DataFrame as _DataFrame
+from chalkdf.debug import chalk_dataframe_debug_context
+
+ChalkDataFrame = _DataFrame
+
+__all__ = ["ChalkDataFrame", "chalk_dataframe_debug_context"]

chalkruntime-3.32.1/chalkruntime/dataframe/lazyframe.py

@@ -0,0 +1,7 @@
+"""Compatibility shim for ``chalkruntime.dataframe.lazyframe`` imports."""
+
+from __future__ import annotations
+
+from chalkdf import LazyFrame
+
+__all__ = ["LazyFrame"]

chalkruntime-3.32.1/chalkruntime/exc/failed_argument.py

@@ -0,0 +1,7 @@
+import dataclasses
+
+
+@dataclasses.dataclass(slots=True, frozen=True)
+class FailedArgument:
+    expected_feature: str
+    value: object