qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

qwak/clients/analytics/client.py

@@ -1,6 +1,8 @@
 from datetime import timedelta
 from time import sleep, time
+from typing import Optional
 
+import grpc
 from _qwak_proto.qwak.analytics.analytics_pb2 import (
     QueryResultDownloadURLParams,
     QueryStatus,
@@ -14,8 +16,9 @@ from _qwak_proto.qwak.analytics.analytics_service_pb2 import (
 from _qwak_proto.qwak.analytics.analytics_service_pb2_grpc import (
     AnalyticsQueryServiceStub,
 )
-from dependency_injector.wiring import Provide
+from qwak.clients.location_discovery import LocationDiscoveryClient
 from qwak.inner.di_configuration import QwakContainer
+from qwak.inner.tool.grpc.grpc_tools import create_grpc_channel
 
 
 class AnalyticsEngineError(RuntimeError):
@@ -26,8 +29,8 @@ class AnalyticsEngineError(RuntimeError):
 
 
 class AnalyticsEngineClient:
-    def __init__(self, grpc_channel=Provide[QwakContainer.core_grpc_channel]):
-        self.grpc_client = AnalyticsQueryServiceStub(grpc_channel)
+    def __init__(self, grpc_channel: Optional[QwakContainer.core_grpc_channel] = None):
+        self.grpc_client = self._create_grpc_client(grpc_channel)
 
     def get_analytics_data(self, query: str, timeout: timedelta = None) -> str:
         """
@@ -79,3 +82,25 @@ class AnalyticsEngineClient:
         request = GetQueryResultDownloadURLRequest(params=request_params)
         response = self.grpc_client.GetQueryResultDownloadURL(request)
         return response.download_url
+
+    def _create_grpc_client(
+        self, grpc_channel: Optional[grpc.Channel] = None
+    ) -> AnalyticsQueryServiceStub:
+        if grpc_channel:
+            return AnalyticsQueryServiceStub(grpc_channel)
+
+        channel = create_grpc_channel(
+            url=self._get_analytics_engine_url(),
+            status_for_retry=(
+                grpc.StatusCode.UNAVAILABLE,
+                grpc.StatusCode.CANCELLED,
+            ),
+        )
+        return AnalyticsQueryServiceStub(channel)
+
+    @staticmethod
+    def _get_analytics_engine_url() -> str:
+        """
+        Fetches the analytics engine service URL from the LocationDiscoveryService.
+        """
+        return LocationDiscoveryClient().get_analytics_engine().service_url

qwak/clients/batch_job_management/client.py

@@ -14,6 +14,8 @@ from _qwak_proto.qwak.batch_job.v1.batch_job_service_pb2 import (
     CancelBatchJobResponse,
     CancelWarmupJobRequest,
     CancelWarmupJobResponse,
+    GetBatchJobDetailsRequest,
+    GetBatchJobDetailsResponse,
     GetBatchJobDownloadDetailsRequest,
     GetBatchJobDownloadDetailsResponse,
     GetBatchJobPreSignedDownloadUrlRequest,
@@ -26,10 +28,16 @@ from _qwak_proto.qwak.batch_job.v1.batch_job_service_pb2 import (
     GetBatchJobStatusResponse,
     GetBatchJobUploadDetailsRequest,
     GetBatchJobUploadDetailsResponse,
+    ListBatchJobsRequest,
+    ListBatchJobsResponse,
     StartBatchJobRequest,
     StartBatchJobResponse,
     StartWarmupJobRequest,
     StartWarmupJobResponse,
+    UpdateTasksDetailsResponse,
+    UpdateTasksDetailsRequest,
+    BatchTaskDetails,
+    InputFileDetails,
 )
 from _qwak_proto.qwak.batch_job.v1.batch_job_service_pb2_grpc import (
     BatchJobManagementServiceStub,
@@ -52,6 +60,7 @@ from .executions_config import (
     INPUT_FORMATTERS_MAP,
     OUTPUT_FORMATTERS_MAP,
     ExecutionConfig,
+    PURCHASE_OPTION_SET,
 )
 from .results import (
     CancelExecutionResult,
@@ -62,6 +71,8 @@ from .results import (
     StartExecutionResult,
 )
 
+from typing import List
+
 CLIENT_TIMEOUT = 180  # Seconds
 
 
@@ -100,9 +111,18 @@ class BatchJobManagerClient:
 
             if user_raw_output_format not in OUTPUT_FORMATTERS_MAP:
                 raise ValueError(
-                    f"Invalid output format - please choose one of {list(OUTPUT_FORMATTERS_MAP.keys())}"
+                    f"Invalid input format - please choose one of {list(OUTPUT_FORMATTERS_MAP.keys())}"
                 )
 
+            user_purchase_option = execution_config.advanced_options.purchase_option
+            if user_purchase_option is not None:
+                if user_purchase_option not in PURCHASE_OPTION_SET:
+                    raise ValueError(
+                        f"Invalid purchase option - please choose one of {list(PURCHASE_OPTION_SET)}"
+                    )
+                else:
+                    user_purchase_option = user_purchase_option.replace("-", "")
+
             start_job_result: StartBatchJobResponse = self._start_batch_job(
                 batch_job_request=BatchJobRequest(
                     model_details=BatchJobModelDetails(
@@ -126,13 +146,16 @@ class BatchJobManagerClient:
                         ),
                         token_secret=execution_config.execution.access_token_name,
                         secret_secret=execution_config.execution.access_secret_name,
+                        service_account_json_key_secret=execution_config.execution.service_account_key_secret_name,
                     ),
                     execution_details=BatchJobExecutionDetails(
                         job_timeout=execution_config.execution.job_timeout,
                         task_timeout=execution_config.execution.file_timeout,
                         batch_job_deployment_size=job_size,
                         advanced_deployment_options=AdvancedDeploymentOptions(
-                            custom_iam_role_arn=execution_config.advanced_options.custom_iam_role_arn
+                            custom_iam_role_arn=execution_config.advanced_options.custom_iam_role_arn,
+                            purchase_option=user_purchase_option,
+                            service_account_key_secret_name=execution_config.advanced_options.service_account_key_secret_name,
                         ),
                         parameters=BatchJobManagerClient._batch_job_parameters_as_list(
                             execution_config.execution.parameters
@@ -488,4 +511,72 @@ class BatchJobManagerClient:
             )
 
         except grpc.RpcError as e:
-            raise QwakException(f"Failed to get download details error is: {e}")
+            raise QwakException(f"Failed to get download details, error is: {e}")
+
+    def list_batch_jobs(self, model_id: str, build_id: str) -> ListBatchJobsResponse:
+        """
+        List batch jobs by its model ID
+        Args:
+        model_id (str): The model ID
+        build_id (str): The build ID - optional.
+
+        Returns: ListBatchJobsResponse which contains list of batch jobs details
+        """
+        try:
+            return self.batch_job_management.ListBatchJobs(
+                ListBatchJobsRequest(model_id=model_id, build_id=build_id),
+                timeout=CLIENT_TIMEOUT,
+            )
+
+        except grpc.RpcError as e:
+            raise QwakException(f"Failed to list batch jobs, error is: {e}")
+
+    def get_batch_job_details(self, job_id: str) -> GetBatchJobDetailsResponse:
+        """
+        Get batch jos by its job ID
+        Args:
+        job_id: The job ID
+
+        Returns: GetBatchJobDetailsResponse which contains list of task execution details
+        """
+        try:
+            return self.batch_job_management.GetBatchJobDetails(
+                GetBatchJobDetailsRequest(job_id=job_id),
+                timeout=CLIENT_TIMEOUT,
+            )
+
+        except grpc.RpcError as e:
+            raise QwakException(f"Failed to get batch job details, error is: {e}")
+
+    def update_task_details(self, task_id: str, input_file_paths: List[str]):
+        """
+        Update task details for a specific task using the provided list of input file paths.
+
+        :param task_id: The ID of the task to update.
+        :param input_file_paths: A list of paths for the input files associated with the task.
+        """
+        # Create InputFileDetails messages for each input file path
+        input_files_details = [
+            InputFileDetails(path=file_path) for file_path in input_file_paths
+        ]
+
+        # Create BatchTaskDetails message for the single task
+        batch_task_details = BatchTaskDetails(
+            task_id=task_id, input_files_details=input_files_details
+        )
+
+        update_request = UpdateTasksDetailsRequest(
+            tasks_details=[batch_task_details]  # Wrap in a list
+        )
+
+        try:
+            response: UpdateTasksDetailsResponse = (
+                self.batch_job_management.UpdateTasksDetails(
+                    update_request, timeout=CLIENT_TIMEOUT
+                )
+            )
+            return response
+        except grpc.RpcError as e:
+            raise QwakException(
+                f"Failed to update task details for task '{task_id}', error is: {e}"
+            )

qwak/clients/batch_job_management/executions_config.py

@@ -50,6 +50,12 @@ GPU_TYPE_MAP = {
     "NVIDIA_A10G": NVIDIA_A10G,
 }
 
+PURCHASE_OPTION_SET = (
+    "on-demand",
+    "spot",
+)
+
+
 CONFIG_MAPPING: List[ConfigCliMap] = [
     ConfigCliMap("model_id", "execution.model_id", validate_string, True),
     ConfigCliMap("build_id", "execution.build_id", validate_string, False),
@@ -84,12 +90,21 @@ CONFIG_MAPPING: List[ConfigCliMap] = [
     ConfigCliMap(
         "access_secret_name", "execution.access_secret_name", validate_string, False
     ),
+    ConfigCliMap(
+        "service_account_key_secret_name",
+        "execution.service_account_key_secret_name",
+        validate_string,
+        False,
+    ),
     ConfigCliMap("pods", "resources.pods", validate_int, False),
     ConfigCliMap("cpus", "resources.cpus", validate_float, False),
     ConfigCliMap("memory", "resources.memory", validate_int, False),
     ConfigCliMap("gpu_type", "resources.gpu_type", validate_string, False),
     ConfigCliMap("gpu_amount", "resources.gpu_amount", validate_int, False),
     ConfigCliMap("instance", "resources.instance_size", validate_string, False),
+    ConfigCliMap(
+        "purchase_option", "advanced_options.purchase_option", validate_string, False
+    ),
 ]
 
 
@@ -139,6 +154,7 @@ class ExecutionConfig(YamlConfigMixin, QwakConfigBase):
         job_timeout: int = field(default=0)
         file_timeout: int = field(default=0)
         parameters: dict = field(default_factory=lambda: defaultdict(str))
+        service_account_key_secret_name: str = field(default="")
 
     @dataclass
     class Warmup:
@@ -158,6 +174,8 @@ class ExecutionConfig(YamlConfigMixin, QwakConfigBase):
     @dataclass
     class AdvancedOptions:
         custom_iam_role_arn: str = field(default=None)
+        purchase_option: str = field(default=None)
+        service_account_key_secret_name: str = field(default=None)
 
     warmup: Warmup = field(default_factory=Warmup)
     execution: Execution = field(default_factory=Execution)

qwak/clients/build_orchestrator/build_model_request_getter.py

@@ -39,6 +39,7 @@ def _get_build_model_spec(
     build_spec = RemoteBuildSpec(
         build_properties=BuildProperties(
             build_id=build_conf.build_properties.build_id,
+            build_name=build_conf.build_properties.build_name,
             model_id=build_conf.build_properties.model_id,
             branch=build_conf.build_properties.branch,
             tags=build_conf.build_properties.tags,
@@ -56,6 +57,7 @@ def _get_build_model_spec(
             docker_env=DockerEnv(
                 base_image=build_conf.build_env.docker.base_image,
                 assumed_iam_role_arn=build_conf.build_env.docker.assumed_iam_role_arn,
+                service_account_key_secret_name=build_conf.build_env.docker.service_account_key_secret_name,
                 no_cache=not build_conf.build_env.docker.cache,
                 env_vars=build_conf.build_env.docker.env_vars,
             ),
@@ -78,6 +80,7 @@ def _get_build_model_spec(
         build_v1_flag=build_v1_flag,
         build_steps=build_steps,
         purchase_option=_purchase_option_to_enum(build_conf.purchase_option),
+        provision_instance_timeout=build_conf.provision_instance_timeout,
     )
 
     if build_conf.build_env.remote.resources.instance:

qwak/clients/build_orchestrator/client.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass
 from typing import Dict, List, Optional
 
 import grpc
@@ -16,11 +17,30 @@ from _qwak_proto.qwak.build.v1.build_api_pb2 import (
     RegisterTagsResponse,
     UpdateBuildStatusRequest,
     UpdateBuildStatusResponse,
+    SaveFrameworkModelsRequest,
+    SaveFrameworkModelsResponse,
 )
 from _qwak_proto.qwak.build.v1.build_api_pb2_grpc import BuildAPIStub
-from _qwak_proto.qwak.build.v1.build_pb2 import BuildFilter, BuildStatus, ModelSchema
+from _qwak_proto.qwak.build.v1.build_pb2 import (
+    BuildFilter,
+    BuildStatus,
+    ModelSchema,
+    FrameworkModelsSpec,
+    HuggingFaceModelSpec,
+    FrameworkModel,
+)
+from _qwak_proto.qwak.build_settings.build_settings_api_pb2 import (
+    GetBuildSettingsResponse,
+    GetBuildSettingsRequest,
+)
+from _qwak_proto.qwak.build_settings.build_settings_api_pb2_grpc import (
+    BuildSettingsApiStub,
+)
 from _qwak_proto.qwak.builds.build_pb2 import BaseDockerImageType, DataTableDefinition
-from _qwak_proto.qwak.builds.build_url_pb2 import BuildVersioningUrlParams
+from _qwak_proto.qwak.builds.build_url_pb2 import (
+    BuildVersioningUrlParams,
+    BuildVersioningTagsType,
+)
 from _qwak_proto.qwak.builds.builds_orchestrator_service_pb2 import (
     BuildModelRequest,
     CancelBuildModelRequest,
@@ -45,10 +65,30 @@ from qwak.inner.di_configuration import QwakContainer
 from qwak.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
 
 
+@dataclass
+class FrameworkModelDataClass:
+    pass
+
+
+@dataclass
+class HuggingModelDataClass(FrameworkModelDataClass):
+    models_name_to_versions: Dict[str, List[str]]
+    repository: Optional[str] = None
+
+
+@dataclass
+class HuggingFaceModel(FrameworkModelDataClass):
+    model_name: str
+    version: str
+    repository: str
+    sha1: str
+
+
 class BuildOrchestratorClient:
     def __init__(self, grpc_channel=Provide[QwakContainer.core_grpc_channel]):
         self._builds_orchestrator_stub_build_api = BuildAPIStub(grpc_channel)
         self._builds_orchestrator_stub = BuildsOrchestratorServiceStub(grpc_channel)
+        self._build_settings_stub = BuildSettingsApiStub(grpc_channel)
 
     @grpc_try_catch_wrapper("Failed to update build status")
     def update_build_status(
@@ -86,6 +126,61 @@ class BuildOrchestratorClient:
             request
         )
 
+    @grpc_try_catch_wrapper("Failed to save framework models")
+    def save_framework_models(
+        self, build_id: str, frameworks_model_data_class: List[FrameworkModelDataClass]
+    ) -> SaveFrameworkModelsResponse:
+        framework_models = []
+        for framework_model_data_class in frameworks_model_data_class:
+            if isinstance(framework_model_data_class, HuggingModelDataClass):
+                for (
+                    huggingface_model_name,
+                    versions,
+                ) in framework_model_data_class.models_name_to_versions.items():
+                    for version in versions:
+                        framework_models.append(
+                            FrameworkModel(
+                                huggingface_model_spec=HuggingFaceModelSpec(
+                                    version=version,
+                                    model_name=huggingface_model_name,
+                                    repository=framework_model_data_class.repository,
+                                )
+                            )
+                        )
+
+        request = SaveFrameworkModelsRequest(
+            spec=FrameworkModelsSpec(
+                build_id=build_id, framework_models=framework_models
+            )
+        )
+
+        return self._builds_orchestrator_stub_build_api.SaveFrameworkModels(request)
+
+    @grpc_try_catch_wrapper("Failed to save framework models")
+    def save_used_framework_models(
+        self, build_id: str, models: List[HuggingFaceModel]
+    ) -> SaveFrameworkModelsResponse:
+        framework_models = []
+        for model in models:
+            if isinstance(model, HuggingFaceModel):
+                framework_models.append(
+                    FrameworkModel(
+                        huggingface_model_spec=HuggingFaceModelSpec(
+                            version=model.version,
+                            model_name=model.model_name,
+                            sha1=model.sha1,
+                            repository=model.repository,
+                        )
+                    )
+                )
+        request = SaveFrameworkModelsRequest(
+            spec=FrameworkModelsSpec(
+                build_id=build_id, framework_models=framework_models
+            )
+        )
+
+        return self._builds_orchestrator_stub_build_api.SaveFrameworkModels(request)
+
     @grpc_try_catch_wrapper("Failed to register tags")
     def register_tags(
         self,
@@ -126,7 +221,11 @@ class BuildOrchestratorClient:
 
     @grpc_try_catch_wrapper("Failed to get build versioning upload url")
     def get_build_versioning_upload_url(
-        self, build_id: str, model_id: str, tag: str
+        self,
+        build_id: str,
+        model_id: str,
+        tag: str,
+        tag_type: BuildVersioningTagsType = BuildVersioningTagsType.INVALID_TAG_TYPE,
     ) -> GetBuildVersioningUploadURLResponse:
         """Get Upload pre signed url
 
@@ -134,37 +233,81 @@ class BuildOrchestratorClient:
             model_id: Model ID
             build_id: Build ID
             tag: the tag to save the artifact by
+            tag_type: the type of the file to upload
 
         Returns:
-            the pre signed upload url
+            the upload url
         """
         return self._builds_orchestrator_stub.GetBuildVersioningUploadURL(
             GetBuildVersioningUploadURLRequest(
                 params=BuildVersioningUrlParams(
-                    build_id=build_id, model_id=model_id, tag=tag
+                    build_id=build_id,
+                    model_id=model_id,
+                    tag=tag,
+                    tag_type=tag_type,
                 )
             )
         )
 
-    @grpc_try_catch_wrapper("Failed to create download url")
     def get_build_versioning_download_url(
-        self, build_id: str, model_id: str, tag: str
+        self,
+        build_id: str,
+        model_id: str,
+        tag: str,
+        tag_type: BuildVersioningTagsType = BuildVersioningTagsType.INVALID_TAG_TYPE,
     ) -> GetBuildVersioningDownloadURLResponse:
-        """Get Download pre signed url
+        """Get Download url
 
         Args:
             model_id: Model ID
             build_id: Build ID
             tag: the tag to save the artifact by
+            tag_type: the type of the file to download
 
         Returns:
             the pre signed download url
         """
-        return self._builds_orchestrator_stub.GetBuildVersioningDownloadURL(
-            GetBuildVersioningDownloadURLRequest(
-                params=BuildVersioningUrlParams(
-                    build_id=build_id, model_id=model_id, tag=tag
+        try:
+            return self._builds_orchestrator_stub.GetBuildVersioningDownloadURL(
+                GetBuildVersioningDownloadURLRequest(
+                    params=BuildVersioningUrlParams(
+                        build_id=build_id, model_id=model_id, tag=tag, tag_type=tag_type
+                    )
+                )
+            )
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                print(
+                    "The specified file cannot be found. Please verify the file name before trying again"
+                )
+                raise QwakException(
+                    "The specified file cannot be found. Please verify the file name before trying again"
+                )
+            else:
+                print(f"Failed to get build versioning download url. Error is {e}")
+                raise QwakException(
+                    f"The specified file cannot be loaded. Error is {e}"
                 )
+
+        except Exception as e:
+            print(f"Failed to get build versioning download url. Error is {e}")
+            raise QwakException(
+                f"Failed to get build versioning download url. Error is {e}"
+            )
+
+    @grpc_try_catch_wrapper("Failed to get build settings")
+    def get_build_settings(self, environment_id: str) -> GetBuildSettingsResponse:
+        """Get BuildSettings
+
+        Args:
+            environment_id: Environment ID
+
+        Returns:
+            the build settings by the environment ID
+        """
+        return self._build_settings_stub.GetBuildSettings(
+            GetBuildSettingsRequest(
+                environment_id=environment_id,
             )
         )
 

qwak/clients/data_versioning/client.py

@@ -1,8 +1,12 @@
 from typing import Optional
 
 import grpc
-from _qwak_proto.qwak.data_versioning.data_versioning_pb2 import DataTagSpec
+from _qwak_proto.qwak.data_versioning.data_versioning_pb2 import (
+    DataTagFilter,
+    DataTagSpec,
+)
 from _qwak_proto.qwak.data_versioning.data_versioning_service_pb2 import (
+    GetModelDataTagsRequest,
     RegisterDataTagRequest,
 )
 from _qwak_proto.qwak.data_versioning.data_versioning_service_pb2_grpc import (
@@ -55,3 +59,17 @@ class DataVersioningManagementClient:
                 raise QwakException(
                     f"Failed to register data tag, error is {e.details()}"
                 )
+
+    def get_model_data_tags(
+        self, model_id: str, build_id: str, data_tag_filter: DataTagFilter = None
+    ):
+        try:
+            return self._data_management_service.GetModelDataTags(
+                GetModelDataTagsRequest(
+                    model_id=model_id, build_id=build_id, filter=data_tag_filter
+                )
+            )
+        except grpc.RpcError as e:
+            raise QwakException(
+                f"Failed to list model data tags, error is {e.details()}"
+            )

qwak/clients/data_versioning/data_tag_filter.py

@@ -0,0 +1,32 @@
+from dataclasses import dataclass, field
+from enum import Enum
+
+from _qwak_proto.qwak.data_versioning.data_versioning_pb2 import (
+    DataTagFilter as DataTagFilterProto,
+)
+
+
+class DataTagFilerType(Enum):
+    DATA_TAG_FILTER_TYPE_INVALID = 0
+    DATA_TAG_FILTER_TYPE_CONTAINS = 1
+    DATA_TAG_FILTER_TYPE_PREFIX = 2
+
+
+@dataclass
+class DataTagFilter:
+    value: str = field(default=None)
+    type: DataTagFilerType = field(
+        default=DataTagFilerType.DATA_TAG_FILTER_TYPE_INVALID
+    )
+
+    def to_proto(self):
+        if self.type is DataTagFilerType.DATA_TAG_FILTER_TYPE_CONTAINS:
+            return DataTagFilterProto(
+                tag_contains=self.value,
+            )
+        elif self.type is DataTagFilerType.DATA_TAG_FILTER_TYPE_PREFIX:
+            return DataTagFilterProto(
+                tag_prefix=self.value,
+            )
+        else:
+            return None

qwak/clients/deployment/client.py

@@ -22,6 +22,8 @@ from _qwak_proto.qwak.deployment.deployment_service_pb2 import (
     UndeployModelRequest,
     UpdateDeploymentRuntimeSettingsRequest,
     UpdateDeploymentRuntimeSettingsResponse,
+    GetDeploymentHistoryRequest,
+    GetDeploymentHistoryResponse,
 )
 from _qwak_proto.qwak.deployment.deployment_service_pb2_grpc import (
     DeploymentManagementServiceStub,
@@ -172,3 +174,15 @@ class DeploymentManagementClient:
             raise QwakException(
                 f"Failed to apply model traffic, error is {e.details()}"
             )
+
+    def get_deployment_history(
+        self, model_uuid: str = None, build_id: str = None
+    ) -> GetDeploymentHistoryResponse:
+        try:
+            return self.deployment_management.GetDeploymentHistory(
+                GetDeploymentHistoryRequest(model_uuid=model_uuid, build_id=build_id)
+            )
+        except grpc.RpcError as e:
+            raise QwakException(
+                f"Failed to get deployment history, error is {e.details()}"
+            )

qwak/clients/feature_store/execution_management_client.py

@@ -1,4 +1,7 @@
 from _qwak_proto.qwak.execution.v1.backfill_pb2 import BackfillSpec
+from _qwak_proto.qwak.execution.v1.streaming_aggregation_pb2 import (
+    StreamingAggregationBackfillIngestion,
+)
 from _qwak_proto.qwak.execution.v1.batch_pb2 import BatchIngestion
 from _qwak_proto.qwak.execution.v1.execution_service_pb2 import (
     GetExecutionEntryRequest,
@@ -9,6 +12,8 @@ from _qwak_proto.qwak.execution.v1.execution_service_pb2 import (
     TriggerBackfillResponse,
     TriggerBatchFeaturesetRequest,
     TriggerBatchFeaturesetResponse,
+    TriggerStreamingAggregationBackfillRequest,
+    TriggerStreamingAggregationBackfillResponse,
 )
 from _qwak_proto.qwak.execution.v1.execution_service_pb2_grpc import (
     FeatureStoreExecutionServiceStub,
@@ -29,6 +34,29 @@ class ExecutionManagementClient:
             grpc_channel
         )
 
+    def trigger_streaming_aggregation_backfill(
+        self, backfill_ingestion: StreamingAggregationBackfillIngestion
+    ) -> TriggerStreamingAggregationBackfillResponse:
+        """
+        Receives a configured streaming aggregation backfill proto and triggers a streaming aggregation
+        backfill against the execution manager
+
+        Args:
+            backfill_ingestion (StreamingAggregationBackfillIngestion): A protobuf message
+                containing the backfill specification details
+
+        Returns:
+            TriggerStreamingAggregationBackfillResponse: response object from the execution manager
+        """
+        try:
+            return self._feature_store_execution_service.TriggerStreamingAggregationBackfill(
+                TriggerStreamingAggregationBackfillRequest(backfill=backfill_ingestion)
+            )
+        except RpcError as e:
+            raise QwakException(
+                f"Failed to trigger streaming aggregation backfill job, error encountered {e}"
+            )
+
     def trigger_batch_backfill(
         self, batch_backfill_spec: BackfillSpec
     ) -> TriggerBackfillResponse: