query-cache-common 1.6.0__tar.gz → 1.7.0__tar.gz

{query_cache_common-1.6.0 → query_cache_common-1.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: query-cache-common
-Version: 1.6.0
+Version: 1.7.0
 Summary: Common code for Query Cache shared across client and server
 License: Copyright (c) 2026 Fivetran, Inc.
         

{query_cache_common-1.6.0 → query_cache_common-1.7.0}/src/query_cache_common/models/services/execution_service_models.py

@@ -5,6 +5,7 @@ from dataclasses import field
 
 from query_cache_common.decorators import proto_dataclass
 from query_cache_common.models import shared_models
+from query_cache_common.models.services import sql_service_models
 from query_cache_common.models.base import BaseSerDeModel
 from query_cache_common.models.converters import dict_to_struct, struct_to_dict
 from query_cache_protobuf.query_cache.services import execution_service_pb2
@@ -47,6 +48,22 @@ class SQLExecution(BaseSerDeModel):
     semantic_extras: t.Dict[str, str] = field(default_factory=dict)
     labels: t.Dict[str, str] = field(default_factory=dict)
 
+    @classmethod
+    def from_submit_sql_request(
+        cls, req: sql_service_models.SubmitEnrichedSQLRequest
+    ) -> SQLExecution:
+        return cls(
+            target_table=req.target_table,
+            dialect=req.dialect,
+            default_catalog=req.default_catalog,
+            execution_type=req.execution_type,
+            sql=req.sql,
+            tables=req.tables,
+            query_dependencies=req.query_dependencies,
+            semantic_extras=req.semantic_extras,
+            labels=req.labels,
+        )
+
 
 @proto_dataclass(execution_service_pb2.ValuesExecution)
 class ValuesExecution(BaseSerDeModel):
@@ -57,6 +74,19 @@ class ValuesExecution(BaseSerDeModel):
     semantic_extras: t.Dict[str, str] = field(default_factory=dict)
     labels: t.Dict[str, str] = field(default_factory=dict)
 
+    @classmethod
+    def from_submit_values_request(
+        cls, req: sql_service_models.SubmitValuesRequest
+    ) -> ValuesExecution:
+        return cls(
+            target_table=req.target_table,
+            dialect=req.dialect,
+            default_catalog=req.default_catalog,
+            values_hash=req.values_hash,
+            semantic_extras=req.semantic_extras,
+            labels=req.labels,
+        )
+
 
 @proto_dataclass(execution_service_pb2.ExecutionOutcome)
 class ExecutionOutcome(BaseSerDeModel):