chalkpy 2.96.1__tar.gz → 2.96.2__tar.gz

{chalkpy-2.96.1 → chalkpy-2.96.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.96.1
+Version: 2.96.2
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

chalkpy-2.96.2/chalk/_version.py

@@ -0,0 +1 @@
+__version__ = "2.96.2"

{chalkpy-2.96.1 → chalkpy-2.96.2}/chalk/sql/_internal/integrations/snowflake.py

@@ -523,6 +523,9 @@ class SnowflakeSourceImpl(BaseSQLSource):
                             unload_job_identifier=job_id,
                             snowflake_unload_stage=query_execution_parameters.snowflake.snowflake_unload_stage,
                         )
+                    if env_var_bool("CHALK_SNOWFLAKE_TIMEZONE_UTC"):
+                        chalk_logger.info(f"Setting Snowflake timezone to UTC")
+                        cursor.execute(f"ALTER SESSION SET TIMEZONE = 'UTC';")
 
                     # add the query to the QueryRegistry so we can clean up on shutdown
                     chalk_logger.info(f"Compiled query: {repr(sql)}")
@@ -692,6 +695,10 @@ class SnowflakeSourceImpl(BaseSQLSource):
                         )
                     )
                 with con.cursor() as cursor:
+                    if env_var_bool("CHALK_SNOWFLAKE_TIMEZONE_UTC"):
+                        chalk_logger.info(f"Setting Snowflake timezone to UTC")
+                        cursor.execute(f"ALTER SESSION SET TIMEZONE = 'UTC';")
+
                     job_id = str(uuid.uuid4())
                     if query_execution_parameters.snowflake.snowflake_unload_stage is not None:
                         sql = _rewrite_query_for_unload(

chalkpy-2.96.1/chalk/_version.py

@@ -1 +0,0 @@
-__version__ = "2.96.1"