chalkpy 2.99.1__py3-none-any.whl → 2.99.3__py3-none-any.whl

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.99.1"
+__version__ = "2.99.3"

chalk/client/__init__.py

@@ -19,6 +19,7 @@ from chalk.client.models import (
     FeatureResult,
     GetIncrementalProgressResponse,
     OfflineQueryContext,
+    OfflineQueryDeadlineOptions,
     OnlineQuery,
     OnlineQueryContext,
     OnlineQueryResponse,
@@ -54,6 +55,7 @@ __all__ = [
     "FeatureResult",
     "GetIncrementalProgressResponse",
     "OfflineQueryContext",
+    "OfflineQueryDeadlineOptions",
     "OnlineQuery",
     "OnlineQueryContext",
     "OnlineQueryResponse",

chalk/client/client.py

@@ -1174,6 +1174,7 @@ class ChalkClient:
         self,
         name: str,
         query_version: str | None = None,
+        branch: str | None = None,
     ) -> List[NamedQueryMetadata]:
         """
         Get the metadata associated with named queries.
@@ -1184,6 +1185,8 @@ class ChalkClient:
             The name of the named query.
         query_version
             The query version of the named query. Returns all versions of the named query by default.
+        branch
+            The branch name to get the named query from. By default will use the mainline deployment.
 
         Returns
         -------

chalk/client/client_grpc.py

@@ -681,6 +681,12 @@ class ChalkGRPCClient:
         if token_config is None:
             raise ChalkAuthException()
 
+        # using for instantiation of ChalkClient(). Can remove if we exclusively start using GRPC client
+        self._client_id = token_config.clientId
+        self._client_secret = token_config.clientSecret
+        self._environment = token_config.activeEnvironment
+        self._api_server = token_config.apiServer
+
         self._stub_refresher = StubRefresher(
             token_config=token_config,
             query_server=query_server,
@@ -1394,6 +1400,7 @@ class ChalkGRPCClient:
         self,
         name: str,
         query_version: str | None = None,
+        branch: str | None = None,
     ) -> List[NamedQueryMetadata]:
         """
         Get the metadata associated with named queries.
@@ -1404,6 +1411,8 @@ class ChalkGRPCClient:
             The name of the named query.
         query_version
             The query version of the named query. Returns all versions of the named query by default.
+        branch
+            The branch name to get the named query from. By default will use the mainline deployment.
 
         Returns
         -------
@@ -1412,12 +1421,29 @@ class ChalkGRPCClient:
 
         Examples
         --------
-        >>> from chalk.client import ChalkClient
-        >>> ChalkClient().get_named_query_metadata(
+        >>> from chalk.client.client_grpc import ChalkGRPCClient
+        >>> ChalkGRPCClient().get_named_query_metadata(
         ...     name="my_named_query",
         ...     query_version="1.1.0",
         ... )
         """
+        # TODO update underlying logic to be done server side and/or not split between GRPC and python client.
+        if branch:
+            from chalk.client import ChalkClient
+
+            client = ChalkClient(
+                client_id=self._client_id,
+                client_secret=self._client_secret,
+                environment=self._environment,
+                api_server=self._api_server,
+            )
+
+            return client.get_named_query_metadata(
+                name=name,
+                query_version=query_version,
+                branch=branch,
+            )
+
         proto_resp = self._stub_refresher.call_get_named_query_metadata(
             lambda x: x.GetNamedQueryByName(
                 request=GetNamedQueryByNameRequest(

chalk/client/client_impl.py

@@ -2571,6 +2571,7 @@ https://docs.chalk.ai/cli/apply
         self,
         name: str,
         query_version: str | None = None,
+        branch: str | None = None,
     ) -> List[NamedQueryMetadata]:
         """
         Get the metadata associated with named queries.
@@ -2581,6 +2582,8 @@ https://docs.chalk.ai/cli/apply
             The name of the named query.
         query_version
             The query version of the named query. Returns all versions of the named query by default.
+        branch
+            The branch name to get the named query from. By default will use the mainline deployment.
 
         Returns
         -------
@@ -2604,13 +2607,37 @@ https://docs.chalk.ai/cli/apply
             api_server=self._api_server,
         )
 
-        resp = client_grpc.get_named_query_metadata(
+        if not branch:
+            branch = self.get_branch()
+
+        # TODO update underlying logic to be done server side and/or not split between GRPC and python client.
+
+        # named queries aren't easily available until full chalk apply, so grabbing through layered logic
+        if branch:
+            _branch_metadata_response = self._get_branch_info()
+            latest_deployment_id = next(
+                (
+                    branch_info.latest_deployment
+                    for branch_info in _branch_metadata_response.branches
+                    if branch_info.name == branch
+                ),
+                None,
+            )
+            if not latest_deployment_id:
+                raise FileNotFoundError(f"Could not find a deployment associated with branch: {branch}")
+
+            desired_named_query_list = [
+                x
+                for x in client_grpc.get_graph(latest_deployment_id).named_queries
+                if x.name == name and (not query_version or x.query_version == query_version)
+            ]
+            return [NamedQueryMetadata.from_proto(nq) for nq in desired_named_query_list]
+
+        return client_grpc.get_named_query_metadata(
             name=name,
             query_version=query_version,
         )
 
-        return resp
-
     def prompt_evaluation(
         self,
         prompts: list[Prompt | str],

{chalkpy-2.99.1.dist-info → chalkpy-2.99.3.dist-info}/METADATA

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

{chalkpy-2.99.1.dist-info → chalkpy-2.99.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=veq7UDoQd0z2ACc6n5r66tpMpzpZd-wj3jbKxjfT1KQ,23
+chalk/_version.py,sha256=5e5zIu_nDTY-AvXFwpeYAQ8-9xsDJzUSurX4_yPRN38,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -611,12 +611,12 @@ chalk/_validation/feature_validation.py,sha256=AfEEHbDNzLpvPXb-KBE3MnpLCvtXzZVWb
 chalk/_validation/validation.py,sha256=9cCMfZa9-1wxkXLme_ylmD5vIA1qExJD6aqbYvbmKn4,2169
 chalk/byte_transmit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/byte_transmit/model.py,sha256=LFX8pj9X_CWXeap7fDnMl9YmXsYTgq7jBAbEWkxoYoE,13048
-chalk/client/__init__.py,sha256=wu3WQVzE5gRj6noQQDOdYJUgDaz_9QtbjXH4KuoIlXQ,1782
-chalk/client/client.py,sha256=VfaLfbWyk-VaImPLNrNeAP4FvuJwQ5yEtJrkLST_-H4,104808
+chalk/client/__init__.py,sha256=2sxazRlPQZDXetDJGvZsqxRZQ9KelIflzSwjujsfdAE,1850
+chalk/client/client.py,sha256=3muGd7X1ZdcZuwbV5-4wEUgNes1fCWKWAdfnibJIz68,104960
 chalk/client/client_async.py,sha256=YmuExumlDkenlGc8ROtmPjUE2m6nQUZisUDvJmwTRdk,51546
 chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
-chalk/client/client_grpc.py,sha256=z1WeFWjmKKCU8dK7fdvWEOsCUlBi-v_2-HijC2_oIVs,108854
-chalk/client/client_impl.py,sha256=Izb7a1sOLLI3vK8_Mi51iPZc5kmaqPxAhbqjHvzRF90,216305
+chalk/client/client_grpc.py,sha256=UNhVLLY7I65tymPMjJ7A9CCHOkJOcnHgFKcPb4bmeXE,109930
+chalk/client/client_impl.py,sha256=acWBpgU1xsMyFUHbrA0sdyL0NmKwwRRL4vdZzSqBrwU,217576
 chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
 chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
 chalk/client/models.py,sha256=ApXv28A44dNXUULKFkG5L6e48vDscZyFGenaXItxL94,67598
@@ -827,8 +827,8 @@ chalk/utils/tracing.py,sha256=NiiM-9dbuJhSCv6R1npR1uYNKWlkqTR6Ygm0Voi2NrY,13078
 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.99.1.dist-info/METADATA,sha256=0h_o7NzQ0REbB61J16f_NpXpes2lu_0S-trfMzb5ZdQ,27754
-chalkpy-2.99.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-chalkpy-2.99.1.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.99.1.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.99.1.dist-info/RECORD,,
+chalkpy-2.99.3.dist-info/METADATA,sha256=2uN7S8xkv-AQrL23F_48VyqoLm5t_IxzWfrtpxtTL8U,27754
+chalkpy-2.99.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+chalkpy-2.99.3.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.99.3.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.99.3.dist-info/RECORD,,