chalkpy 2.95.4__py3-none-any.whl → 2.95.6__py3-none-any.whl

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.95.4"
+__version__ = "2.95.6"

chalk/client/client_grpc.py

@@ -30,6 +30,7 @@ from chalk._gen.chalk.graph.v1.graph_pb2 import Graph
 from chalk._gen.chalk.models.v1 import model_artifact_pb2 as _model_artifact_pb2
 from chalk._gen.chalk.protosql.v1.sql_service_pb2 import (
     ExecuteSqlQueryRequest,
+    ExecuteSqlResultPersistenceSettings,
     GetDbCatalogsRequest,
     GetDbSchemasRequest,
     GetTablesRequest,
@@ -1453,8 +1454,15 @@ class ChalkGRPCClient:
         """
         return self._stub_refresher.call_team_stub(lambda x: x.ListServiceTokens(ListServiceTokensRequest()))
 
-    def run_sql(self, sql: str):
-        return self._stub_refresher.call_sql_stub(lambda x: x.ExecuteSqlQuery(ExecuteSqlQueryRequest(query=sql)))
+    def run_sql(self, sql: str, persistence_settings: Optional[Dict[str, Any]] = None):
+        request = ExecuteSqlQueryRequest(query=sql)
+
+        if persistence_settings is not None:
+            # Convert dict to proto message
+            enabled = persistence_settings.get("enabled", False)
+            request.persistence_settings.CopyFrom(ExecuteSqlResultPersistenceSettings(enabled=enabled))
+
+        return self._stub_refresher.call_sql_stub(lambda x: x.ExecuteSqlQuery(request))
 
     def explain_sql(self, sql: str):
         return self._stub_refresher.call_sql_stub(lambda x: x.PlanSqlQuery(PlanSqlQueryRequest(query=sql)))

{chalkpy-2.95.4.dist-info → chalkpy-2.95.6.dist-info}/METADATA

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

{chalkpy-2.95.4.dist-info → chalkpy-2.95.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=806ZJ4Nmzvk8iZwepGS8Gxx4y0klotQk8lkbn8S8CIo,23
+chalk/_version.py,sha256=-CDItrmCodozEPr2Fvom-vIUtFY7wbiK7w67cIoXGck,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -606,7 +606,7 @@ chalk/client/__init__.py,sha256=wu3WQVzE5gRj6noQQDOdYJUgDaz_9QtbjXH4KuoIlXQ,1782
 chalk/client/client.py,sha256=fqw75x4yiAk3CXryGbuF_qg09WDIlluhYeFdQ-BfTG8,103337
 chalk/client/client_async.py,sha256=nFFTWJbdBlb7zksyjOMBY566tZTAyNXQhCnq06LHWl0,50803
 chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
-chalk/client/client_grpc.py,sha256=VB3JDdSzpk5jMIRUi15_wGs2U9pumeAav7x46VFCFew,106317
+chalk/client/client_grpc.py,sha256=rCjKVe6uMNfKjat_Fng3SRUoMaut8Vz5QRw02v_d9xc,106699
 chalk/client/client_impl.py,sha256=R2ZUiEQf662q3O1eo_NjzYqLGB71B5mWzcx8opC3Os0,211639
 chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
 chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
@@ -818,8 +818,8 @@ chalk/utils/tracing.py,sha256=qAyMRdFBkL4Q2_Bn-C31atDfJirUtBvcSjkYC2R51sM,13005
 chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
 chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
-chalkpy-2.95.4.dist-info/METADATA,sha256=KWZ99My6RPafBho_fntwsoMUHCFT5oaLH2qaD6zDVm0,27494
-chalkpy-2.95.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chalkpy-2.95.4.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.95.4.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.95.4.dist-info/RECORD,,
+chalkpy-2.95.6.dist-info/METADATA,sha256=lM-d5lkB7Ck9FZenhyd-Pmkmpg63VDeY8gu0tiAtK2M,27494
+chalkpy-2.95.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chalkpy-2.95.6.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.95.6.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.95.6.dist-info/RECORD,,