query-cache-common 2.5.0__tar.gz → 2.7.0__tar.gz

{query_cache_common-2.5.0 → query_cache_common-2.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: query-cache-common
-Version: 2.5.0
+Version: 2.7.0
 Summary: Common code for Query Cache shared across client and server
 License: 
                                          Apache License

{query_cache_common-2.5.0 → query_cache_common-2.7.0}/src/query_cache_common/models/services/execution_service_models.py

@@ -33,7 +33,7 @@ class ConfirmExecutionRequest(BaseSerDeModel):
 
 @proto_dataclass(execution_service_pb2.ConfirmExecutionResponse)
 class ConfirmExecutionResponse(BaseSerDeModel):
-    request_id: str
+    execution_id: str
     success: bool
 
 
@@ -145,3 +145,14 @@ class ResolveDeferredRelationsRequest(BaseSerDeModel):
 @proto_dataclass(execution_service_pb2.ResolveDeferredRelationsResponse)
 class ResolveDeferredRelationsResponse(BaseSerDeModel):
     fqn_by_unique_id: t.Dict[str, str] = field(default_factory=dict)
+
+
+@proto_dataclass(execution_service_pb2.UpdateExecutionMetadataRequest)
+class UpdateExecutionMetadataRequest(BaseSerDeModel):
+    execution_id: str
+    last_modified_epoch: t.Optional[int] = None
+
+
+@proto_dataclass(execution_service_pb2.UpdateExecutionMetadataResponse)
+class UpdateExecutionMetadataResponse(BaseSerDeModel):
+    pass

{query_cache_common-2.5.0 → query_cache_common-2.7.0}/src/query_cache_common/models/services/sql_service_models.py

@@ -86,6 +86,7 @@ class UndecidedResponse(BaseSerDeModel):
 @proto_dataclass(sql_service_pb2.SkipExecutionResponse)
 class SkipExecutionResponse(BaseSerDeModel):
     explained_decision: shared_models.ExplainedDecision
+    execution_id: str
     transformed_nodes_by_query: TransformedNodesByQuery = transformed_nodes_by_query_field()
     execution_results: t.Dict[str, t.Any] = field(
         default_factory=dict,

{query_cache_common-2.5.0 → query_cache_common-2.7.0}/src/query_cache_common/models/shared_models.py

@@ -24,10 +24,19 @@ class ModelExecutionType(BaseSerDeEnum):
     DBT_CUSTOM = "DBT_CUSTOM"
 
 
+@proto_enum(shared_pb2.TableModifiedInfo.DependencyKind)
+class DependencyKind(BaseSerDeEnum):
+    UNKNOWN = "UNKNOWN"
+    SOURCE = "SOURCE"
+    SEED = "SEED"
+    MODEL = "MODEL"
+
+
 @proto_dataclass(shared_pb2.TableModifiedInfo)
 class TableModifiedInfo(BaseSerDeModel):
     name: str
     last_modified_epoch: int = 0
+    kind: DependencyKind = DependencyKind.UNKNOWN
 
 
 @proto_dataclass(shared_pb2.ColumnInfo)