query-cache-common 2.0.0__tar.gz → 2.2.0__tar.gz

{query_cache_common-2.0.0 → query_cache_common-2.2.0}/PKG-INFO

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

{query_cache_common-2.0.0 → query_cache_common-2.2.0}/src/query_cache_common/models/services/client_telemetry_service_models.py

@@ -14,6 +14,7 @@ from query_cache_common.models.converters import (
     struct_to_dict,
     timedelta_to_duration,
 )
+from query_cache_common.models.services.selector_service_models import SelectorCriteria
 
 
 @proto_dataclass(client_telemetry_service_pb2.SessionStartRequest)
@@ -81,6 +82,17 @@ class SessionEndRequest(BaseSerDeModel):
     )
 
 
+@proto_dataclass(client_telemetry_service_pb2.ClientSelectorEvent)
+class ClientSelectorEvent(BaseSerDeModel):
+    request_id: str
+    project_id: str
+    dbt_target: str
+    selector_criteria: SelectorCriteria
+    num_nodes: int
+    processing_time_ms: int
+    hash_calculation_time_ms: int
+
+
 @proto_dataclass(client_telemetry_service_pb2.ClientTelemetryEvent)
 class ClientTelemetryEvent(BaseSerDeModel):
     event_order: t.Optional[int] = None
@@ -88,6 +100,7 @@ class ClientTelemetryEvent(BaseSerDeModel):
     enriched_sql_prepared: t.Optional[ClientPrepareEnrichedSQLRequest] = None
     session_start: t.Optional[SessionStartRequest] = None
     session_end: t.Optional[SessionEndRequest] = None
+    client_selector_event: t.Optional[ClientSelectorEvent] = None
 
 
 @proto_dataclass(client_telemetry_service_pb2.SubmitTelemetryBatchRequest)

{query_cache_common-2.0.0 → query_cache_common-2.2.0}/src/query_cache_common/models/services/selector_service_models.py

@@ -41,6 +41,7 @@ class SelectorRequest(BaseSerDeModel):
     project_id: str
     nodes: t.List[DbtNodeData]
     selector_criteria: SelectorCriteria
+    request_id: str
 
 
 @proto_dataclass(selector_service_pb2.SelectorResponse)