qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of qwak-core might be problematic. Click here for more details.
- _qwak_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.py +29 -0
- _qwak_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.pyi +70 -0
- _qwak_proto/jfml/hosting_gateway/v1/build_upload_url_pb2_grpc.py +4 -0
- _qwak_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.py +32 -0
- _qwak_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.pyi +48 -0
- _qwak_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2_grpc.py +67 -0
- _qwak_proto/qwak/administration/account/v1/account_pb2.py +29 -60
- _qwak_proto/qwak/administration/account/v1/account_pb2.pyi +120 -2
- _qwak_proto/qwak/administration/account/v1/account_service_pb2.py +67 -0
- _qwak_proto/qwak/administration/account/v1/account_service_pb2.pyi +345 -0
- _qwak_proto/qwak/administration/account/v1/account_service_pb2_grpc.py +338 -0
- _qwak_proto/qwak/administration/account/v1/jfrog_tenant_details_pb2.py +49 -0
- _qwak_proto/qwak/administration/account/v1/jfrog_tenant_details_pb2.pyi +209 -0
- _qwak_proto/qwak/administration/account/v1/jfrog_tenant_details_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/account/v1/personalization_pb2.py +8 -24
- _qwak_proto/qwak/administration/account/v1/preferences_pb2.py +6 -14
- _qwak_proto/qwak/administration/account/v1/terms_pb2.py +35 -0
- _qwak_proto/qwak/administration/account/v1/terms_pb2.pyi +94 -0
- _qwak_proto/qwak/administration/account/v1/terms_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/authenticated_user/v1/authenticated_user_service_pb2.py +18 -67
- _qwak_proto/qwak/administration/authenticated_user/v1/credentials_pb2.py +8 -24
- _qwak_proto/qwak/administration/authenticated_user/v1/credentials_pb2.pyi +1 -1
- _qwak_proto/qwak/administration/authenticated_user/v1/details_pb2.py +14 -54
- _qwak_proto/qwak/administration/cluster/v2/cluster_pb2.py +62 -0
- _qwak_proto/qwak/administration/cluster/v2/cluster_pb2.pyi +354 -0
- _qwak_proto/qwak/administration/cluster/v2/cluster_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2.py +32 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2.pyi +65 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/container_registry_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/creds/auth_pb2.py +28 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/creds/auth_pb2.pyi +36 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/creds/auth_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/creds/secret_pb2.py +29 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/creds/secret_pb2.pyi +49 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/creds/secret_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/data_catalog_config_pb2.py +33 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/data_catalog_config_pb2.pyi +89 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/data_catalog_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.py +34 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2.pyi +97 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/databricks/auth_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/elasticsearch_config_pb2.py +28 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/elasticsearch_config_pb2.pyi +39 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/elasticsearch_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/kafka_config_pb2.py +34 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/kafka_config_pb2.pyi +79 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/kafka_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.py +29 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.pyi +46 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/redis_config_pb2.py +28 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/redis_config_pb2.pyi +39 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/redis_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.py +28 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.pyi +33 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/feature_store_config_pb2.py +32 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/feature_store_config_pb2.pyi +66 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/feature_store_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/aws/auth_pb2.py +42 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/aws/auth_pb2.pyi +159 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/aws/auth_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.py +39 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2.pyi +126 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting/azure/auth_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting_config_pb2.py +40 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting_config_pb2.pyi +146 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/hosting_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/logs_storage_config_pb2.py +28 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/logs_storage_config_pb2.pyi +33 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/logs_storage_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/model_analytics_storage_config_pb2.py +28 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/model_analytics_storage_config_pb2.pyi +33 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/model_analytics_storage_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/network_config_pb2.py +36 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/network_config_pb2.pyi +107 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/network_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/object_storage_config_pb2.py +33 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/object_storage_config_pb2.pyi +93 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/object_storage_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2.py +33 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2.pyi +79 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.py +36 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.pyi +72 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/sql_engine_config_pb2.py +33 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/sql_engine_config_pb2.pyi +83 -0
- _qwak_proto/qwak/administration/runtime_configuration/v0/sql_engine_config_pb2_grpc.py +4 -0
- _qwak_proto/qwak/administration/v0/authentication/authentication_service_pb2.py +14 -47
- _qwak_proto/qwak/administration/v0/environments/configuration_pb2.py +23 -55
- _qwak_proto/qwak/administration/v0/environments/configuration_pb2.pyi +103 -7
- _qwak_proto/qwak/administration/v0/environments/environment_pb2.py +15 -33
- _qwak_proto/qwak/administration/v0/environments/environment_pb2.pyi +50 -2
- _qwak_proto/qwak/administration/v0/environments/environment_service_pb2.py +36 -158
- _qwak_proto/qwak/administration/v0/environments/personalization_pb2.py +8 -24
- _qwak_proto/qwak/administration/v0/users/user_pb2.py +18 -71
- _qwak_proto/qwak/admiral/secret/v0/secret_pb2.py +53 -0
- _qwak_proto/qwak/admiral/secret/v0/secret_pb2.pyi +256 -0
- _qwak_proto/qwak/admiral/secret/v0/secret_pb2_grpc.py +4 -0
- _qwak_proto/qwak/admiral/secret/v0/secret_service_pb2.py +40 -0
- _qwak_proto/qwak/admiral/secret/v0/secret_service_pb2.pyi +93 -0
- _qwak_proto/qwak/admiral/secret/v0/secret_service_pb2_grpc.py +135 -0
- _qwak_proto/qwak/admiral/secret/v0/system_secret_service_pb2.py +40 -0
- _qwak_proto/qwak/admiral/secret/v0/system_secret_service_pb2.pyi +101 -0
- _qwak_proto/qwak/admiral/secret/v0/system_secret_service_pb2_grpc.py +135 -0
- _qwak_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2.py +39 -105
- _qwak_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2.pyi +131 -3
- _qwak_proto/qwak/admiral/user_application_instance/v0/user_application_instance_service_pb2.py +22 -87
- _qwak_proto/qwak/analytics/analytics_pb2.py +28 -123
- _qwak_proto/qwak/analytics/analytics_pb2.pyi +5 -1
- _qwak_proto/qwak/analytics/analytics_service_pb2.py +48 -149
- _qwak_proto/qwak/analytics/analytics_service_pb2.pyi +87 -0
- _qwak_proto/qwak/analytics/analytics_service_pb2_grpc.py +101 -0
- _qwak_proto/qwak/artifactory_settings/artifactory_settings_pb2.py +29 -0
- _qwak_proto/qwak/artifactory_settings/artifactory_settings_pb2.pyi +57 -0
- _qwak_proto/qwak/artifactory_settings/artifactory_settings_pb2_grpc.py +4 -0
- _qwak_proto/qwak/audience/v1/audience_api_pb2.py +30 -127
- _qwak_proto/qwak/audience/v1/audience_pb2.py +27 -114
- _qwak_proto/qwak/audience/v1/audience_pb2.pyi +5 -1
- _qwak_proto/qwak/auto_scaling/v1/auto_scaling_pb2.py +18 -79
- _qwak_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2.py +10 -27
- _qwak_proto/qwak/automation/v1/action_pb2.py +42 -164
- _qwak_proto/qwak/automation/v1/action_pb2.pyi +82 -7
- _qwak_proto/qwak/automation/v1/auto_scaling_pb2.py +19 -77
- _qwak_proto/qwak/automation/v1/auto_scaling_pb2.pyi +6 -0
- _qwak_proto/qwak/automation/v1/automation_execution_pb2.py +12 -43
- _qwak_proto/qwak/automation/v1/automation_management_service_pb2.py +56 -258
- _qwak_proto/qwak/automation/v1/automation_pb2.py +11 -35
- _qwak_proto/qwak/automation/v1/automation_pb2.pyi +8 -1
- _qwak_proto/qwak/automation/v1/common_pb2.py +11 -33
- _qwak_proto/qwak/automation/v1/common_pb2.pyi +6 -0
- _qwak_proto/qwak/automation/v1/notification_pb2.py +16 -68
- _qwak_proto/qwak/automation/v1/trigger_pb2.py +14 -54
- _qwak_proto/qwak/batch_job/v1/batch_job_events_pb2.py +27 -116
- _qwak_proto/qwak/batch_job/v1/batch_job_events_pb2.pyi +6 -1
- _qwak_proto/qwak/batch_job/v1/batch_job_resources_pb2.py +9 -26
- _qwak_proto/qwak/batch_job/v1/batch_job_resources_pb2.pyi +2 -0
- _qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.py +121 -502
- _qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi +111 -4
- _qwak_proto/qwak/batch_job/v1/batch_job_service_pb2_grpc.py +67 -0
- _qwak_proto/qwak/build/v1/build_api_pb2.py +98 -278
- _qwak_proto/qwak/build/v1/build_api_pb2.pyi +220 -0
- _qwak_proto/qwak/build/v1/build_api_pb2_grpc.py +198 -0
- _qwak_proto/qwak/build/v1/build_pb2.py +115 -417
- _qwak_proto/qwak/build/v1/build_pb2.pyi +294 -5
- _qwak_proto/qwak/build_settings/build_settings_api_pb2.py +40 -0
- _qwak_proto/qwak/build_settings/build_settings_api_pb2.pyi +106 -0
- _qwak_proto/qwak/build_settings/build_settings_api_pb2_grpc.py +132 -0
- _qwak_proto/qwak/build_settings/build_settings_pb2.py +39 -0
- _qwak_proto/qwak/build_settings/build_settings_pb2.pyi +148 -0
- _qwak_proto/qwak/build_settings/build_settings_pb2_grpc.py +4 -0
- _qwak_proto/qwak/builds/build_pb2.py +46 -200
- _qwak_proto/qwak/builds/build_pb2.pyi +111 -5
- _qwak_proto/qwak/builds/build_url_pb2.py +27 -54
- _qwak_proto/qwak/builds/build_url_pb2.pyi +131 -2
- _qwak_proto/qwak/builds/build_values_pb2.py +82 -0
- _qwak_proto/qwak/builds/build_values_pb2.pyi +559 -0
- _qwak_proto/qwak/builds/build_values_pb2_grpc.py +4 -0
- _qwak_proto/qwak/builds/builds_orchestrator_service_pb2.py +93 -168
- _qwak_proto/qwak/builds/builds_orchestrator_service_pb2.pyi +342 -3
- _qwak_proto/qwak/builds/builds_orchestrator_service_pb2_grpc.py +166 -0
- _qwak_proto/qwak/builds/builds_pb2.py +87 -528
- _qwak_proto/qwak/builds/builds_pb2.pyi +5 -192
- _qwak_proto/qwak/builds/builds_pb2_grpc.py +0 -233
- _qwak_proto/qwak/builds/internal_builds_orchestrator_service_pb2.py +12 -37
- _qwak_proto/qwak/data_versioning/data_versioning_pb2.py +9 -15
- _qwak_proto/qwak/data_versioning/data_versioning_pb2.pyi +21 -0
- _qwak_proto/qwak/data_versioning/data_versioning_service_pb2.py +15 -48
- _qwak_proto/qwak/data_versioning/data_versioning_service_pb2.pyi +12 -1
- _qwak_proto/qwak/deployment/alert_pb2.py +25 -108
- _qwak_proto/qwak/deployment/alert_pb2.pyi +2 -0
- _qwak_proto/qwak/deployment/alert_service_pb2.py +30 -127
- _qwak_proto/qwak/deployment/deployment_messages_pb2.py +6 -14
- _qwak_proto/qwak/deployment/deployment_pb2.py +127 -578
- _qwak_proto/qwak/deployment/deployment_pb2.pyi +51 -8
- _qwak_proto/qwak/deployment/deployment_service_pb2.py +91 -400
- _qwak_proto/qwak/deployment/deployment_service_pb2.pyi +30 -0
- _qwak_proto/qwak/deployment/deployment_service_pb2_grpc.py +34 -0
- _qwak_proto/qwak/ecosystem/jfrog/v0/token_pb2.py +72 -0
- _qwak_proto/qwak/ecosystem/jfrog/v0/token_pb2.pyi +489 -0
- _qwak_proto/qwak/ecosystem/jfrog/v0/token_pb2_grpc.py +4 -0
- _qwak_proto/qwak/ecosystem/jfrog/v0/token_service_pb2.py +44 -0
- _qwak_proto/qwak/ecosystem/jfrog/v0/token_service_pb2.pyi +144 -0
- _qwak_proto/qwak/ecosystem/jfrog/v0/token_service_pb2_grpc.py +169 -0
- _qwak_proto/qwak/ecosystem/v0/auth_pb2.py +28 -0
- _qwak_proto/qwak/ecosystem/v0/auth_pb2.pyi +37 -0
- _qwak_proto/qwak/ecosystem/v0/auth_pb2_grpc.py +4 -0
- _qwak_proto/qwak/ecosystem/v0/azure_credentials_pb2.py +37 -0
- _qwak_proto/qwak/ecosystem/v0/azure_credentials_pb2.pyi +117 -0
- _qwak_proto/qwak/ecosystem/v0/azure_credentials_pb2_grpc.py +4 -0
- _qwak_proto/qwak/ecosystem/v0/credentials_pb2.py +13 -26
- _qwak_proto/qwak/ecosystem/v0/credentials_pb2.pyi +33 -3
- _qwak_proto/qwak/ecosystem/v0/ecosystem_pb2.py +34 -140
- _qwak_proto/qwak/ecosystem/v0/ecosystem_pb2.pyi +45 -7
- _qwak_proto/qwak/ecosystem/v0/ecosystem_runtime_service_pb2.py +54 -229
- _qwak_proto/qwak/ecosystem/v0/ecosystem_runtime_service_pb2.pyi +22 -3
- _qwak_proto/qwak/execution/v1/backfill_pb2.py +18 -74
- _qwak_proto/qwak/execution/v1/batch_pb2.py +10 -34
- _qwak_proto/qwak/execution/v1/deletion_pb2.py +33 -0
- _qwak_proto/qwak/execution/v1/deletion_pb2.pyi +67 -0
- _qwak_proto/qwak/execution/v1/deletion_pb2_grpc.py +4 -0
- _qwak_proto/qwak/execution/v1/execution_pb2.py +20 -60
- _qwak_proto/qwak/execution/v1/execution_pb2.pyi +61 -4
- _qwak_proto/qwak/execution/v1/execution_service_pb2.py +70 -108
- _qwak_proto/qwak/execution/v1/execution_service_pb2.pyi +306 -0
- _qwak_proto/qwak/execution/v1/execution_service_pb2_grpc.py +304 -2
- _qwak_proto/qwak/execution/v1/internal/deployment/platform_details_pb2.py +27 -0
- _qwak_proto/qwak/execution/v1/internal/deployment/platform_details_pb2.pyi +29 -0
- _qwak_proto/qwak/execution/v1/internal/deployment/platform_details_pb2_grpc.py +4 -0
- _qwak_proto/qwak/execution/v1/jobs/job_pb2.py +19 -67
- _qwak_proto/qwak/execution/v1/jobs/job_pb2.pyi +33 -2
- _qwak_proto/qwak/execution/v1/jobs/job_service_pb2.py +56 -88
- _qwak_proto/qwak/execution/v1/jobs/job_service_pb2.pyi +311 -0
- _qwak_proto/qwak/execution/v1/jobs/job_service_pb2_grpc.py +231 -0
- _qwak_proto/qwak/execution/v1/jobs/reports/report_pb2.py +21 -75
- _qwak_proto/qwak/execution/v1/jobs/reports/report_pb2.pyi +29 -2
- _qwak_proto/qwak/execution/v1/state/execution_state_pb2.py +6 -14
- _qwak_proto/qwak/execution/v1/state/execution_state_service_pb2.py +31 -108
- _qwak_proto/qwak/execution/v1/state/execution_state_service_pb2.pyi +30 -0
- _qwak_proto/qwak/execution/v1/state/execution_state_service_pb2_grpc.py +34 -0
- _qwak_proto/qwak/execution/v1/state/featureset_state_pb2.py +6 -14
- _qwak_proto/qwak/execution/v1/state/spark_execution_state_pb2.py +14 -68
- _qwak_proto/qwak/execution/v1/streaming_aggregation_pb2.py +42 -0
- _qwak_proto/qwak/execution/v1/streaming_aggregation_pb2.pyi +165 -0
- _qwak_proto/qwak/execution/v1/streaming_aggregation_pb2_grpc.py +4 -0
- _qwak_proto/qwak/execution/v1/streaming_pb2.py +33 -0
- _qwak_proto/qwak/execution/v1/streaming_pb2.pyi +79 -0
- _qwak_proto/qwak/execution/v1/streaming_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/entities/entity_pb2.py +16 -57
- _qwak_proto/qwak/feature_store/entities/entity_service_pb2.py +30 -127
- _qwak_proto/qwak/feature_store/features/aggregation_pb2.py +38 -175
- _qwak_proto/qwak/feature_store/features/deployment_pb2.py +36 -167
- _qwak_proto/qwak/feature_store/features/deployment_service_pb2.py +10 -27
- _qwak_proto/qwak/feature_store/features/execution_pb2.py +14 -56
- _qwak_proto/qwak/feature_store/features/feature_set_attribute_pb2.py +28 -0
- _qwak_proto/qwak/feature_store/features/feature_set_attribute_pb2.pyi +33 -0
- _qwak_proto/qwak/feature_store/features/feature_set_attribute_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/features/feature_set_pb2.py +49 -106
- _qwak_proto/qwak/feature_store/features/feature_set_pb2.pyi +154 -8
- _qwak_proto/qwak/feature_store/features/feature_set_service_pb2.py +101 -349
- _qwak_proto/qwak/feature_store/features/feature_set_service_pb2.pyi +160 -0
- _qwak_proto/qwak/feature_store/features/feature_set_service_pb2_grpc.py +170 -0
- _qwak_proto/qwak/feature_store/features/feature_set_state_pb2.py +38 -181
- _qwak_proto/qwak/feature_store/features/feature_set_state_service_pb2.py +26 -107
- _qwak_proto/qwak/feature_store/features/feature_set_types_pb2.py +84 -298
- _qwak_proto/qwak/feature_store/features/feature_set_types_pb2.pyi +123 -7
- _qwak_proto/qwak/feature_store/features/monitoring_pb2.py +10 -34
- _qwak_proto/qwak/feature_store/features/real_time_feature_extractor_pb2.py +32 -143
- _qwak_proto/qwak/feature_store/jobs/job_pb2.py +12 -46
- _qwak_proto/qwak/feature_store/jobs/job_service_pb2.py +30 -127
- _qwak_proto/qwak/feature_store/jobs/v1/job_pb2.py +13 -36
- _qwak_proto/qwak/feature_store/jobs/v1/job_pb2.pyi +19 -0
- _qwak_proto/qwak/feature_store/jobs/v1/job_service_pb2.py +44 -148
- _qwak_proto/qwak/feature_store/jobs/v1/job_service_pb2.pyi +64 -0
- _qwak_proto/qwak/feature_store/jobs/v1/job_service_pb2_grpc.py +67 -0
- _qwak_proto/qwak/feature_store/platform/platform_details_pb2.py +30 -0
- _qwak_proto/qwak/feature_store/platform/platform_details_pb2.pyi +54 -0
- _qwak_proto/qwak/feature_store/platform/platform_details_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/reports/report_pb2.py +14 -53
- _qwak_proto/qwak/feature_store/repository/common/platform_pb2.py +31 -0
- _qwak_proto/qwak/feature_store/repository/common/platform_pb2.pyi +70 -0
- _qwak_proto/qwak/feature_store/repository/common/platform_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/serving/management_pb2.py +36 -0
- _qwak_proto/qwak/feature_store/serving/management_pb2.pyi +72 -0
- _qwak_proto/qwak/feature_store/serving/management_pb2_grpc.py +101 -0
- _qwak_proto/qwak/feature_store/serving/metadata_pb2.py +13 -35
- _qwak_proto/qwak/feature_store/serving/metadata_pb2.pyi +27 -0
- _qwak_proto/qwak/feature_store/serving/serving_pb2.py +50 -222
- _qwak_proto/qwak/feature_store/serving/v1/value_pb2.py +8 -24
- _qwak_proto/qwak/feature_store/sinks/sink_pb2.py +36 -0
- _qwak_proto/qwak/feature_store/sinks/sink_pb2.pyi +137 -0
- _qwak_proto/qwak/feature_store/sinks/sink_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/sources/batch_pb2.py +80 -175
- _qwak_proto/qwak/feature_store/sources/batch_pb2.pyi +461 -13
- _qwak_proto/qwak/feature_store/sources/data_source_attribute_pb2.py +28 -0
- _qwak_proto/qwak/feature_store/sources/data_source_attribute_pb2.pyi +33 -0
- _qwak_proto/qwak/feature_store/sources/data_source_attribute_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/sources/data_source_pb2.py +21 -55
- _qwak_proto/qwak/feature_store/sources/data_source_pb2.pyi +39 -4
- _qwak_proto/qwak/feature_store/sources/data_source_service_pb2.py +41 -148
- _qwak_proto/qwak/feature_store/sources/data_source_service_pb2.pyi +25 -0
- _qwak_proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py +34 -0
- _qwak_proto/qwak/feature_store/sources/streaming_pb2.py +34 -146
- _qwak_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2.py +27 -0
- _qwak_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2.pyi +30 -0
- _qwak_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.py +32 -0
- _qwak_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.pyi +77 -0
- _qwak_proto/qwak/feature_store/v1/common/source_code/source_code_pb2_grpc.py +4 -0
- _qwak_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.py +39 -0
- _qwak_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.pyi +99 -0
- _qwak_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2_grpc.py +99 -0
- _qwak_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2.py +31 -0
- _qwak_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2.pyi +46 -0
- _qwak_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2_grpc.py +66 -0
- _qwak_proto/qwak/features_operator/v1/features_operator_pb2.py +14 -54
- _qwak_proto/qwak/features_operator/v1/features_operator_service_pb2.py +14 -47
- _qwak_proto/qwak/features_operator/v2/features_operator_pb2.py +16 -64
- _qwak_proto/qwak/features_operator/v2/features_operator_service_pb2.py +12 -37
- _qwak_proto/qwak/features_operator/v3/features_operator_async_service_pb2.py +26 -107
- _qwak_proto/qwak/features_operator/v3/features_operator_pb2.py +16 -64
- _qwak_proto/qwak/file_versioning/file_versioning_pb2.py +11 -24
- _qwak_proto/qwak/file_versioning/file_versioning_pb2.pyi +21 -0
- _qwak_proto/qwak/file_versioning/file_versioning_service_pb2.py +15 -48
- _qwak_proto/qwak/file_versioning/file_versioning_service_pb2.pyi +12 -1
- _qwak_proto/qwak/fitness_service/constructs_pb2.py +23 -93
- _qwak_proto/qwak/fitness_service/constructs_pb2.pyi +59 -1
- _qwak_proto/qwak/fitness_service/fitness_pb2.py +7 -34
- _qwak_proto/qwak/fitness_service/fitness_pb2.pyi +0 -77
- _qwak_proto/qwak/fitness_service/status_pb2.py +17 -107
- _qwak_proto/qwak/fitness_service/status_pb2.pyi +1 -117
- _qwak_proto/qwak/inference/feedback/feedback_pb2.py +28 -96
- _qwak_proto/qwak/instance_template/instance_template_pb2.py +13 -41
- _qwak_proto/qwak/instance_template/instance_template_pb2.pyi +13 -1
- _qwak_proto/qwak/instance_template/instance_template_service_pb2.py +14 -47
- _qwak_proto/qwak/integration/hugging_face_integration_pb2.py +37 -0
- _qwak_proto/qwak/integration/hugging_face_integration_pb2.pyi +106 -0
- _qwak_proto/qwak/integration/hugging_face_integration_pb2_grpc.py +4 -0
- _qwak_proto/qwak/integration/integration_pb2.py +55 -0
- _qwak_proto/qwak/integration/integration_pb2.pyi +357 -0
- _qwak_proto/qwak/integration/integration_pb2_grpc.py +4 -0
- _qwak_proto/qwak/integration/integration_service_pb2.py +56 -0
- _qwak_proto/qwak/integration/integration_service_pb2.pyi +234 -0
- _qwak_proto/qwak/integration/integration_service_pb2_grpc.py +271 -0
- _qwak_proto/qwak/integration/open_a_i_integration_pb2.py +37 -0
- _qwak_proto/qwak/integration/open_a_i_integration_pb2.pyi +109 -0
- _qwak_proto/qwak/integration/open_a_i_integration_pb2_grpc.py +4 -0
- _qwak_proto/qwak/integration/opsgenie_integration_pb2.py +43 -0
- _qwak_proto/qwak/integration/opsgenie_integration_pb2.pyi +168 -0
- _qwak_proto/qwak/integration/opsgenie_integration_pb2_grpc.py +4 -0
- _qwak_proto/qwak/integration/pagerduty_integration_pb2.py +47 -0
- _qwak_proto/qwak/integration/pagerduty_integration_pb2.pyi +195 -0
- _qwak_proto/qwak/integration/pagerduty_integration_pb2_grpc.py +4 -0
- _qwak_proto/qwak/integration/slack_app_integration_pb2.py +47 -0
- _qwak_proto/qwak/integration/slack_app_integration_pb2.pyi +198 -0
- _qwak_proto/qwak/integration/slack_app_integration_pb2_grpc.py +4 -0
- _qwak_proto/qwak/kube_deployment_captain/alert_pb2.py +22 -97
- _qwak_proto/qwak/kube_deployment_captain/alerting_pb2.py +12 -44
- _qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.py +76 -249
- _qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi +283 -6
- _qwak_proto/qwak/kube_deployment_captain/deployment_pb2.py +45 -161
- _qwak_proto/qwak/kube_deployment_captain/deployment_pb2.pyi +52 -1
- _qwak_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2.py +33 -145
- _qwak_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2.pyi +6 -2
- _qwak_proto/qwak/kube_deployment_captain/kube_deployment_captain_service_pb2.py +111 -510
- _qwak_proto/qwak/kube_deployment_captain/kube_deployment_captain_service_pb2.pyi +24 -1
- _qwak_proto/qwak/kube_deployment_captain/kube_deployment_captain_service_pb2_grpc.py +34 -0
- _qwak_proto/qwak/kube_deployment_captain/traffic_mapping_pb2.py +8 -24
- _qwak_proto/qwak/logging/log_filter_pb2.py +12 -44
- _qwak_proto/qwak/logging/log_line_pb2.py +8 -25
- _qwak_proto/qwak/logging/log_reader_service_pb2.py +10 -27
- _qwak_proto/qwak/logging/log_source_pb2.py +30 -134
- _qwak_proto/qwak/model_descriptor/open_ai_descriptor_pb2.py +40 -0
- _qwak_proto/qwak/model_descriptor/open_ai_descriptor_pb2.pyi +215 -0
- _qwak_proto/qwak/model_descriptor/open_ai_descriptor_pb2_grpc.py +4 -0
- _qwak_proto/qwak/model_group/model_group_pb2.py +47 -0
- _qwak_proto/qwak/model_group/model_group_pb2.pyi +192 -0
- _qwak_proto/qwak/model_group/model_group_pb2_grpc.py +169 -0
- _qwak_proto/qwak/model_group/model_group_repository_details_pb2.py +43 -0
- _qwak_proto/qwak/model_group/model_group_repository_details_pb2.pyi +172 -0
- _qwak_proto/qwak/model_group/model_group_repository_details_pb2_grpc.py +4 -0
- _qwak_proto/qwak/models/models_pb2.py +101 -356
- _qwak_proto/qwak/models/models_pb2.pyi +221 -6
- _qwak_proto/qwak/models/models_pb2_grpc.py +68 -0
- _qwak_proto/qwak/models/models_query_pb2.py +37 -0
- _qwak_proto/qwak/models/models_query_pb2.pyi +157 -0
- _qwak_proto/qwak/models/models_query_pb2_grpc.py +4 -0
- _qwak_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.py +30 -127
- _qwak_proto/qwak/monitoring/v0/alerting_channel_pb2.py +34 -152
- _qwak_proto/qwak/monitoring/v0/alerting_channel_sync_service_pb2.py +14 -47
- _qwak_proto/qwak/offline/serving/v1/feature_values_pb2.py +27 -0
- _qwak_proto/qwak/offline/serving/v1/feature_values_pb2.pyi +37 -0
- _qwak_proto/qwak/offline/serving/v1/feature_values_pb2_grpc.py +4 -0
- _qwak_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2.py +55 -0
- _qwak_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2.pyi +279 -0
- _qwak_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2_grpc.py +165 -0
- _qwak_proto/qwak/offline/serving/v1/options_pb2.py +27 -0
- _qwak_proto/qwak/offline/serving/v1/options_pb2.pyi +30 -0
- _qwak_proto/qwak/offline/serving/v1/options_pb2_grpc.py +4 -0
- _qwak_proto/qwak/offline/serving/v1/population_pb2.py +35 -0
- _qwak_proto/qwak/offline/serving/v1/population_pb2.pyi +99 -0
- _qwak_proto/qwak/offline/serving/v1/population_pb2_grpc.py +4 -0
- _qwak_proto/qwak/projects/jfrog_project_spec_pb2.py +27 -0
- _qwak_proto/qwak/projects/jfrog_project_spec_pb2.pyi +32 -0
- _qwak_proto/qwak/projects/jfrog_project_spec_pb2_grpc.py +4 -0
- _qwak_proto/qwak/projects/projects_pb2.py +51 -142
- _qwak_proto/qwak/projects/projects_pb2.pyi +149 -3
- _qwak_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.py +77 -0
- _qwak_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.pyi +417 -0
- _qwak_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2_grpc.py +441 -0
- _qwak_proto/qwak/prompt/v1/prompt/prompt_pb2.py +69 -0
- _qwak_proto/qwak/prompt/v1/prompt/prompt_pb2.pyi +415 -0
- _qwak_proto/qwak/prompt/v1/prompt/prompt_pb2_grpc.py +4 -0
- _qwak_proto/qwak/secret_service/secret_service_pb2.py +18 -67
- _qwak_proto/qwak/secret_service/secret_service_pb2.pyi +1 -1
- _qwak_proto/qwak/self_service/account/v0/account_membership_pb2.py +30 -65
- _qwak_proto/qwak/self_service/account/v0/account_membership_pb2.pyi +99 -2
- _qwak_proto/qwak/self_service/account/v0/account_membership_service_pb2.py +51 -208
- _qwak_proto/qwak/self_service/account/v0/account_membership_service_pb2.pyi +28 -0
- _qwak_proto/qwak/self_service/account/v0/account_membership_service_pb2_grpc.py +34 -0
- _qwak_proto/qwak/self_service/account/v0/account_status_pb2.py +44 -0
- _qwak_proto/qwak/self_service/account/v0/account_status_pb2.pyi +172 -0
- _qwak_proto/qwak/self_service/account/v0/account_status_pb2_grpc.py +4 -0
- _qwak_proto/qwak/self_service/account/v0/managing_account_pb2.py +60 -0
- _qwak_proto/qwak/self_service/account/v0/managing_account_pb2.pyi +317 -0
- _qwak_proto/qwak/self_service/account/v0/managing_account_pb2_grpc.py +4 -0
- _qwak_proto/qwak/self_service/account/v0/managing_account_service_pb2.py +45 -0
- _qwak_proto/qwak/self_service/account/v0/managing_account_service_pb2.pyi +136 -0
- _qwak_proto/qwak/self_service/account/v0/managing_account_service_pb2_grpc.py +169 -0
- _qwak_proto/qwak/self_service/user/v1/api_key_pb2.py +6 -14
- _qwak_proto/qwak/self_service/user/v1/user_pb2.py +6 -14
- _qwak_proto/qwak/self_service/user/v1/user_service_pb2.py +30 -121
- _qwak_proto/qwak/service_discovery/service_discovery_location_pb2.py +33 -0
- _qwak_proto/qwak/service_discovery/service_discovery_location_pb2.pyi +73 -0
- _qwak_proto/qwak/service_discovery/service_discovery_location_pb2_grpc.py +4 -0
- _qwak_proto/qwak/service_discovery/service_discovery_location_service_pb2.py +32 -0
- _qwak_proto/qwak/service_discovery/service_discovery_location_service_pb2.pyi +41 -0
- _qwak_proto/qwak/service_discovery/service_discovery_location_service_pb2_grpc.py +231 -0
- _qwak_proto/qwak/traffic/v1/traffic_api_pb2.py +23 -88
- _qwak_proto/qwak/traffic/v1/traffic_api_pb2.pyi +13 -2
- _qwak_proto/qwak/traffic/v1/traffic_pb2.py +27 -114
- _qwak_proto/qwak/traffic/v1/traffic_pb2.pyi +5 -1
- _qwak_proto/qwak/user_application/common/v0/resources_pb2.py +31 -132
- _qwak_proto/qwak/user_application/common/v0/resources_pb2.pyi +54 -0
- _qwak_proto/qwak/user_application/v0/user_application_pb2.py +15 -39
- _qwak_proto/qwak/user_application/v0/user_application_pb2.pyi +16 -0
- _qwak_proto/qwak/vectors/v1/collection/collection_pb2.py +25 -104
- _qwak_proto/qwak/vectors/v1/collection/collection_pb2.pyi +17 -1
- _qwak_proto/qwak/vectors/v1/collection/collection_service_pb2.py +38 -167
- _qwak_proto/qwak/vectors/v1/collection/event/collection_event_pb2.py +7 -15
- _qwak_proto/qwak/vectors/v1/collection/event/collection_event_pb2.pyi +5 -1
- _qwak_proto/qwak/vectors/v1/filters_pb2.py +30 -134
- _qwak_proto/qwak/vectors/v1/vector_pb2.py +17 -45
- _qwak_proto/qwak/vectors/v1/vector_pb2.pyi +25 -2
- _qwak_proto/qwak/vectors/v1/vector_service_pb2.py +31 -108
- _qwak_proto/qwak/vectors/v1/vector_service_pb2.pyi +30 -7
- _qwak_proto/qwak/workspace/workspace_pb2.py +23 -108
- _qwak_proto/qwak/workspace/workspace_pb2.pyi +10 -2
- _qwak_proto/qwak/workspace/workspace_service_pb2.py +63 -248
- _qwak_proto/qwak/workspace/workspace_service_pb2.pyi +56 -0
- _qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py +66 -0
- qwak/__init__.py +2 -3
- qwak/automations/automation_executions.py +14 -10
- qwak/automations/automations.py +33 -23
- qwak/automations/build_and_deploy_action.py +147 -28
- qwak/automations/common.py +6 -0
- qwak/clients/administration/eco_system/client.py +2 -2
- qwak/clients/administration/eco_system/eco_system_utils.py +15 -0
- qwak/clients/administration/environment/client.py +16 -0
- qwak/clients/analytics/client.py +28 -3
- qwak/clients/batch_job_management/client.py +94 -3
- qwak/clients/batch_job_management/executions_config.py +18 -0
- qwak/clients/build_orchestrator/build_model_request_getter.py +3 -0
- qwak/clients/build_orchestrator/client.py +155 -12
- qwak/clients/data_versioning/client.py +19 -1
- qwak/clients/data_versioning/data_tag_filter.py +32 -0
- qwak/clients/deployment/client.py +14 -0
- qwak/clients/feature_store/execution_management_client.py +28 -0
- qwak/clients/feature_store/management_client.py +203 -146
- qwak/clients/feature_store/offline_serving_client.py +251 -0
- qwak/clients/feature_store/operator_client.py +21 -4
- qwak/clients/file_versioning/client.py +19 -1
- qwak/clients/file_versioning/file_tag_filter.py +32 -0
- qwak/clients/instance_template/client.py +11 -10
- qwak/clients/integration_management/integration_manager_client.py +34 -0
- qwak/clients/integration_management/integration_utils.py +27 -0
- qwak/clients/integration_management/openai/openai_system_secret.py +56 -0
- qwak/clients/kube_deployment_captain/client.py +3 -1
- qwak/clients/location_discovery/__init__.py +1 -0
- qwak/clients/location_discovery/client.py +73 -0
- qwak/clients/logging_client/client.py +13 -18
- qwak/clients/model_management/client.py +7 -2
- qwak/clients/project/client.py +7 -2
- qwak/clients/prompt_manager/__init__.py +0 -0
- qwak/clients/prompt_manager/model_descriptor_mapper.py +183 -0
- qwak/clients/prompt_manager/prompt_manager_client.py +167 -0
- qwak/clients/prompt_manager/prompt_proto_mapper.py +236 -0
- qwak/clients/secret_service/client.py +8 -3
- qwak/clients/system_secret/__init__.py +0 -0
- qwak/clients/system_secret/system_secret_client.py +30 -0
- qwak/clients/vector_store/management_client.py +2 -0
- qwak/clients/vector_store/serving_client.py +64 -18
- qwak/clients/workspace_manager/client.py +15 -0
- qwak/exceptions/__init__.py +7 -0
- qwak/exceptions/qwak_decode_exception.py +7 -0
- qwak/exceptions/qwak_exception.py +2 -1
- qwak/exceptions/qwak_external_exception.py +11 -0
- qwak/exceptions/qwak_general_build_exception.py +13 -0
- qwak/exceptions/qwak_grpc_address_exception.py +9 -0
- qwak/exceptions/qwak_load_configuration_exception.py +2 -0
- qwak/exceptions/qwak_remote_build_failed.py +5 -0
- qwak/exceptions/qwak_suggestion_exception.py +27 -0
- qwak/feature_store/_common/artifact_utils.py +59 -0
- qwak/feature_store/_common/feature_set_utils.py +142 -19
- qwak/feature_store/_common/functions.py +0 -43
- qwak/feature_store/_common/packaging.py +262 -0
- qwak/feature_store/_common/source_code_spec.py +108 -0
- qwak/feature_store/_common/source_code_spec_factory.py +104 -0
- qwak/feature_store/_common/value.py +10 -0
- qwak/feature_store/data_sources/__init__.py +53 -17
- qwak/feature_store/data_sources/attributes.py +23 -0
- qwak/feature_store/data_sources/base.py +125 -0
- qwak/feature_store/data_sources/batch/__init__.py +0 -0
- qwak/feature_store/data_sources/batch/_batch.py +9 -0
- qwak/feature_store/data_sources/{batch_sources → batch}/_jdbc.py +2 -1
- qwak/feature_store/data_sources/batch/athena.py +260 -0
- qwak/feature_store/data_sources/{batch_sources → batch}/big_query.py +5 -5
- qwak/feature_store/data_sources/batch/clickhouse.py +60 -0
- qwak/feature_store/data_sources/{batch_sources → batch}/csv.py +9 -5
- qwak/feature_store/data_sources/{batch_sources → batch}/elastic_search.py +5 -5
- qwak/feature_store/data_sources/batch/filesystem/__init__.py +0 -0
- qwak/feature_store/data_sources/{batch_sources/filesystem_config.py → batch/filesystem/aws.py} +30 -38
- qwak/feature_store/data_sources/batch/filesystem/base_config.py +13 -0
- qwak/feature_store/data_sources/batch/filesystem/gcp.py +50 -0
- qwak/feature_store/data_sources/batch/filesystem/utils.py +51 -0
- qwak/feature_store/data_sources/{batch_sources → batch}/mongodb.py +6 -5
- qwak/feature_store/data_sources/{batch_sources → batch}/mysql.py +5 -7
- qwak/feature_store/data_sources/{batch_sources → batch}/parquet.py +10 -7
- qwak/feature_store/data_sources/{batch_sources → batch}/postgres.py +5 -7
- qwak/feature_store/data_sources/{batch_sources → batch}/redshift.py +6 -7
- qwak/feature_store/data_sources/{batch_sources → batch}/snowflake.py +39 -13
- qwak/feature_store/data_sources/batch/unity_catalog.py +107 -0
- qwak/feature_store/data_sources/{batch_sources → batch}/vertica.py +6 -5
- qwak/feature_store/data_sources/source_authentication.py +33 -0
- qwak/feature_store/data_sources/streaming/__init__.py +0 -0
- qwak/feature_store/data_sources/streaming/_streaming.py +9 -0
- qwak/feature_store/data_sources/streaming/kafka/__init__.py +25 -0
- qwak/feature_store/data_sources/streaming/kafka/authentication.py +120 -0
- qwak/feature_store/data_sources/streaming/kafka/deserialization.py +109 -0
- qwak/feature_store/data_sources/streaming/kafka/kafka.py +120 -0
- qwak/feature_store/data_sources/time_partition_columns.py +166 -0
- qwak/feature_store/entities/entity.py +17 -4
- qwak/feature_store/execution/backfill.py +8 -2
- qwak/feature_store/execution/streaming_backfill.py +48 -0
- qwak/feature_store/feature_sets/_utils/_featureset_utils.py +45 -0
- qwak/feature_store/feature_sets/backfill.py +13 -0
- qwak/feature_store/feature_sets/base_feature_set.py +148 -0
- qwak/feature_store/feature_sets/batch.py +142 -151
- qwak/feature_store/feature_sets/execution_spec.py +6 -0
- qwak/feature_store/feature_sets/metadata.py +51 -1
- qwak/feature_store/feature_sets/streaming.py +655 -0
- qwak/feature_store/feature_sets/streaming_backfill.py +223 -0
- qwak/feature_store/feature_sets/transformations/__init__.py +33 -0
- qwak/feature_store/feature_sets/transformations/aggregations/__init__.py +0 -0
- qwak/feature_store/feature_sets/transformations/aggregations/aggregations.py +425 -0
- qwak/feature_store/feature_sets/transformations/aggregations/windows.py +70 -0
- qwak/feature_store/feature_sets/transformations/functions/__init__.py +10 -0
- qwak/feature_store/feature_sets/transformations/functions/qwak_pandas.py +64 -0
- qwak/feature_store/feature_sets/transformations/functions/schema.py +57 -0
- qwak/feature_store/feature_sets/transformations/transformations.py +451 -0
- qwak/feature_store/feature_sets/transformations/validations/validations_util.py +78 -0
- qwak/feature_store/offline/__init__.py +2 -2
- qwak/feature_store/offline/_offline_serving_validations.py +31 -0
- qwak/feature_store/offline/client_v2.py +327 -0
- qwak/feature_store/offline/feature_set_features.py +25 -0
- qwak/feature_store/online/client.py +152 -46
- qwak/feature_store/online/endpoint_utils.py +63 -0
- qwak/feature_store/sinks/__init__.py +0 -0
- qwak/feature_store/sinks/base.py +18 -0
- qwak/feature_store/sinks/kafka.py +59 -0
- qwak/feature_store/sinks/streaming/__init__.py +0 -0
- qwak/feature_store/sinks/streaming/attachment.py +30 -0
- qwak/feature_store/sinks/streaming/factory.py +60 -0
- qwak/feature_store/validations/__init__.py +0 -0
- qwak/feature_store/validations/validation_decorators.py +70 -0
- qwak/feature_store/validations/validation_options.py +77 -0
- qwak/feature_store/validations/validation_response.py +119 -0
- qwak/feature_store/validations/validator.py +114 -0
- qwak/inner/build_config/__init__.py +0 -0
- qwak/inner/build_config/build_config_v1.py +311 -0
- qwak/inner/build_logic/__init__.py +0 -0
- qwak/inner/build_logic/build_loggers/__init__.py +0 -0
- qwak/inner/build_logic/build_loggers/trigger_build_logger.py +51 -0
- qwak/inner/build_logic/constants/__init__.py +0 -0
- qwak/inner/build_logic/constants/dependencies.py +5 -0
- qwak/inner/build_logic/constants/host_resource.py +4 -0
- qwak/inner/build_logic/constants/messages.py +3 -0
- qwak/inner/build_logic/constants/temp_dir.py +1 -0
- qwak/inner/build_logic/constants/upload_tag.py +7 -0
- qwak/inner/build_logic/dependency_manager_type.py +8 -0
- qwak/inner/build_logic/execute_build_pipeline.py +62 -0
- qwak/inner/build_logic/interface/__init__.py +0 -0
- qwak/inner/build_logic/interface/build_logger_interface.py +28 -0
- qwak/inner/build_logic/interface/build_phase.py +24 -0
- qwak/inner/build_logic/interface/context_interface.py +66 -0
- qwak/inner/build_logic/interface/phase_run_handler.py +71 -0
- qwak/inner/build_logic/interface/step_inteface.py +26 -0
- qwak/inner/build_logic/interface/time_source.py +23 -0
- qwak/inner/build_logic/phases/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_model_step.py +43 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/common.py +32 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/fetch_strategy_manager.py +62 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/folder/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/folder/folder_strategy.py +131 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/git/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/git/git_strategy.py +186 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/strategy.py +48 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/zip/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/zip/zip_strategy.py +64 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/post_fetch_validation_step.py +140 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/pre_fetch_validation_step.py +266 -0
- qwak/inner/build_logic/phases/phase_010_fetch_model/set_version_step.py +33 -0
- qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/__init__.py +0 -0
- qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/cleanup_step.py +20 -0
- qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/start_remote_build_step.py +43 -0
- qwak/inner/build_logic/phases/phase_020_remote_register_qwak_build/upload_step.py +258 -0
- qwak/inner/build_logic/phases/phases_pipeline.py +44 -0
- qwak/inner/build_logic/run_handlers/__init__.py +0 -0
- qwak/inner/build_logic/run_handlers/programmatic_phase_run_handler.py +99 -0
- qwak/inner/build_logic/tools/__init__.py +0 -0
- qwak/inner/build_logic/tools/dependencies_tools.py +68 -0
- qwak/inner/build_logic/tools/files.py +237 -0
- qwak/inner/build_logic/tools/ignore_files.py +20 -0
- qwak/inner/build_logic/tools/text.py +9 -0
- qwak/inner/build_logic/trigger_build_context.py +10 -0
- qwak/inner/const.py +2 -4
- qwak/inner/di_configuration/__init__.py +1 -63
- qwak/inner/di_configuration/account.py +32 -15
- qwak/inner/di_configuration/config.yml +2 -0
- qwak/inner/di_configuration/containers.py +10 -2
- qwak/inner/di_configuration/dependency_wiring.py +98 -0
- qwak/inner/di_configuration/session.py +5 -0
- qwak/inner/instance_template/__init__.py +0 -0
- qwak/inner/instance_template/verify_template_id.py +48 -0
- qwak/inner/model_loggers_utils.py +10 -7
- qwak/inner/provider.py +6 -0
- qwak/inner/tool/auth.py +42 -7
- qwak/inner/tool/grpc/grpc_auth.py +27 -8
- qwak/inner/tool/grpc/grpc_tools.py +124 -4
- qwak/inner/tool/grpc/grpc_try_wrapping.py +139 -4
- qwak/inner/tool/protobuf_factory.py +45 -0
- qwak/inner/tool/run_config/__init__.py +2 -0
- qwak/inner/tool/run_config/base.py +32 -7
- qwak/inner/tool/run_config/utils.py +66 -1
- qwak/llmops/__init__.py +0 -0
- qwak/llmops/generation/__init__.py +0 -0
- qwak/llmops/generation/_steaming.py +77 -0
- qwak/llmops/generation/base.py +5 -0
- qwak/llmops/generation/chat/__init__.py +0 -0
- qwak/llmops/generation/chat/openai/LICENSE.txt +201 -0
- qwak/llmops/generation/chat/openai/types/__init__.py +0 -0
- qwak/llmops/generation/chat/openai/types/chat/__init__.py +0 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion.py +105 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_assistant_message_param.py +65 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py +188 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_content_part_text_param.py +28 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_function_call_option_param.py +25 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_function_message_param.py +33 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_message.py +59 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_message_param.py +34 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call.py +46 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call_param.py +44 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_named_tool_choice_param.py +32 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_role.py +20 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_system_message_param.py +35 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_token_logprob.py +71 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_tool_choice_option_param.py +28 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_tool_message_param.py +31 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_tool_param.py +29 -0
- qwak/llmops/generation/chat/openai/types/chat/chat_completion_user_message_param.py +35 -0
- qwak/llmops/generation/chat/openai/types/chat/completion_create_params.py +279 -0
- qwak/llmops/generation/chat/openai/types/completion_choice.py +47 -0
- qwak/llmops/generation/chat/openai/types/completion_create_params.py +209 -0
- qwak/llmops/generation/chat/openai/types/completion_usage.py +30 -0
- qwak/llmops/generation/chat/openai/types/model.py +35 -0
- qwak/llmops/generation/chat/openai/types/shared/__init__.py +3 -0
- qwak/llmops/generation/chat/openai/types/shared/error_object.py +27 -0
- qwak/llmops/generation/chat/openai/types/shared/function_definition.py +49 -0
- qwak/llmops/generation/chat/openai/types/shared/function_parameters.py +20 -0
- qwak/llmops/generation/chat/openai/types/shared_params/__init__.py +2 -0
- qwak/llmops/generation/chat/openai/types/shared_params/function_definition.py +49 -0
- qwak/llmops/generation/chat/openai/types/shared_params/function_parameters.py +22 -0
- qwak/llmops/generation/streaming.py +26 -0
- qwak/llmops/model/__init__.py +0 -0
- qwak/llmops/model/descriptor.py +39 -0
- qwak/llmops/prompt/__init__.py +0 -0
- qwak/llmops/prompt/base.py +133 -0
- qwak/llmops/prompt/chat/__init__.py +0 -0
- qwak/llmops/prompt/chat/message.py +24 -0
- qwak/llmops/prompt/chat/template.py +113 -0
- qwak/llmops/prompt/chat/value.py +10 -0
- qwak/llmops/prompt/manager.py +134 -0
- qwak/llmops/prompt/template.py +24 -0
- qwak/llmops/prompt/value.py +14 -0
- qwak/llmops/provider/__init__.py +0 -0
- qwak/llmops/provider/chat.py +42 -0
- qwak/llmops/provider/openai/__init__.py +0 -0
- qwak/llmops/provider/openai/client.py +123 -0
- qwak/llmops/provider/openai/provider.py +89 -0
- qwak/model/_entity_extraction.py +40 -46
- qwak/model/adapters/input_adapters/multi_input_adapter.py +17 -1
- qwak/model/base.py +1 -1
- qwak/model/decorators/api.py +3 -1
- qwak/model/decorators/impl/api_implementation.py +1 -0
- qwak/model/fs_info_mapping_retriever.py +58 -0
- qwak/model/schema.py +1 -1
- qwak/model/schema_entities.py +1 -1
- qwak/model/tools/__init__.py +1 -1
- qwak/model/tools/adapters/input_adapters/dataframe_input.py +1 -1
- qwak/model/tools/adapters/output.py +1 -1
- qwak/model/tools/run_model_locally.py +15 -3
- qwak/model/utils/feature_utils.py +69 -0
- qwak/model_loggers/artifact_logger.py +46 -9
- qwak/model_loggers/data_logger.py +16 -2
- qwak/qwak_client/batch_jobs/execution.py +49 -0
- qwak/qwak_client/batch_jobs/task.py +43 -0
- qwak/qwak_client/build_api_helpers/__init__.py +0 -0
- qwak/qwak_client/build_api_helpers/build_api_steps.py +53 -0
- qwak/qwak_client/build_api_helpers/messages.py +6 -0
- qwak/qwak_client/build_api_helpers/trigger_build_api.py +65 -0
- qwak/qwak_client/client.py +276 -24
- qwak/qwak_client/data_versioning/__init__.py +0 -0
- qwak/qwak_client/data_versioning/data_tag.py +24 -0
- qwak/qwak_client/deployments/deployment.py +11 -9
- qwak/qwak_client/file_versioning/__init__.py +0 -0
- qwak/qwak_client/file_versioning/file_tag.py +24 -0
- qwak/qwak_client/projects/project.py +4 -2
- qwak/tools/logger/logger.py +1 -1
- qwak/utils/__init__.py +0 -0
- qwak/utils/datetime_utils.py +18 -0
- qwak/utils/dict_utils.py +12 -0
- qwak/vector_store/client.py +8 -0
- qwak/vector_store/collection.py +173 -73
- qwak/vector_store/inference_client.py +30 -2
- qwak/vector_store/rest_helpers.py +13 -21
- qwak/vector_store/utils/filter_utils.py +4 -17
- qwak/vector_store/utils/upsert_utils.py +217 -0
- {qwak_core-0.1.37.dist-info → qwak_core-0.5.19.dist-info}/METADATA +20 -15
- qwak_core-0.5.19.dist-info/RECORD +1100 -0
- {qwak_core-0.1.37.dist-info → qwak_core-0.5.19.dist-info}/WHEEL +1 -1
- qwak_services_mock/mocks/authentication_service.py +5 -4
- qwak_services_mock/mocks/batch_job_manager_service.py +53 -6
- qwak_services_mock/mocks/build_orchestrator_build_api.py +18 -1
- qwak_services_mock/mocks/build_orchestrator_build_settings_api.py +35 -0
- qwak_services_mock/mocks/build_orchestrator_service_api.py +4 -0
- qwak_services_mock/mocks/data_versioning_service.py +26 -2
- qwak_services_mock/mocks/deployment_management_service.py +46 -18
- qwak_services_mock/mocks/ecosystem_service_api.py +0 -1
- qwak_services_mock/mocks/execution_management_service.py +9 -1
- qwak_services_mock/mocks/feature_store_data_sources_manager_api.py +98 -2
- qwak_services_mock/mocks/feature_store_entities_manager_api.py +46 -3
- qwak_services_mock/mocks/feature_store_feature_set_manager_api.py +197 -21
- qwak_services_mock/mocks/features_online_serving_api.py +9 -0
- qwak_services_mock/mocks/features_operator_v3_service.py +20 -0
- qwak_services_mock/mocks/features_set_state_service_api.py +0 -1
- qwak_services_mock/mocks/file_versioning_service.py +29 -2
- qwak_services_mock/mocks/fs_offline_serving_service.py +53 -0
- qwak_services_mock/mocks/instance_template_management_service.py +35 -1
- qwak_services_mock/mocks/integration_management_service.py +72 -0
- qwak_services_mock/mocks/location_discovery_service_api.py +104 -0
- qwak_services_mock/mocks/model_management_service.py +11 -5
- qwak_services_mock/mocks/prompt_manager_service.py +244 -0
- qwak_services_mock/mocks/qwak_mocks.py +20 -2
- qwak_services_mock/mocks/system_secret_service.py +54 -0
- qwak_services_mock/mocks/vector_serving_api.py +44 -8
- qwak_services_mock/mocks/workspace_manager_service_mock.py +24 -8
- qwak_services_mock/services_mock.py +78 -12
- _qwak_proto/qwak/fitness_service/fitness_service_pb2.py +0 -110
- _qwak_proto/qwak/fitness_service/fitness_service_pb2.pyi +0 -121
- _qwak_proto/qwak/fitness_service/fitness_service_pb2_grpc.py +0 -169
- qwak/clients/build_management/__init__.py +0 -1
- qwak/clients/build_management/client.py +0 -114
- qwak/feature_store/_common/featureset_asterisk_handler.py +0 -115
- qwak/feature_store/data_sources/batch_sources/__init__.py +0 -6
- qwak/feature_store/data_sources/batch_sources/_batch.py +0 -86
- qwak/feature_store/feature_sets/transformations.py +0 -48
- qwak/feature_store/offline/_query_engine.py +0 -32
- qwak/feature_store/offline/athena/athena_query_engine.py +0 -153
- qwak/feature_store/offline/client.py +0 -699
- qwak/feature_store/validation_options.py +0 -69
- qwak_core-0.1.37.dist-info/RECORD +0 -682
- qwak_services_mock/mocks/build_management.py +0 -100
- {_qwak_proto → qwak/clients/integration_management}/__init__.py +0 -0
- /qwak/{feature_store/offline/athena → clients/integration_management/openai}/__init__.py +0 -0
|
@@ -0,0 +1,266 @@
|
|
|
1
|
+
import uuid
|
|
2
|
+
from pathlib import Path
|
|
3
|
+
|
|
4
|
+
import grpc
|
|
5
|
+
|
|
6
|
+
from qwak.clients.secret_service import SecretServiceClient
|
|
7
|
+
from qwak.exceptions import (
|
|
8
|
+
QwakSuggestionException,
|
|
9
|
+
QwakException,
|
|
10
|
+
)
|
|
11
|
+
from qwak.inner.build_logic.constants.host_resource import HOST_TEMP_BUILD_DIR
|
|
12
|
+
from qwak.inner.build_logic.interface.step_inteface import Step
|
|
13
|
+
from qwak.inner.build_logic.tools.dependencies_tools import find_dependency_files
|
|
14
|
+
from qwak.inner.build_logic.tools.text import snake_case
|
|
15
|
+
from qwak.inner.instance_template.verify_template_id import verify_template_id
|
|
16
|
+
from qwak.inner.provider import Provider
|
|
17
|
+
from qwak.model.base import QwakModel
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
class PreFetchValidationStep(Step):
|
|
21
|
+
STEP_DESCRIPTION = "Pre model fetch validation"
|
|
22
|
+
DEFAULT_CPUS = 2
|
|
23
|
+
DEFAULT_MEMORY = "4Gi"
|
|
24
|
+
VALIDATION_FAILURE_CPU_AND_GPU_MESSAGE = (
|
|
25
|
+
"CPU and GPU configured together, Invalid configuration."
|
|
26
|
+
)
|
|
27
|
+
VALIDATION_FAILURE_CPU_AND_GPU_SUGGESTION = "Please configure only CPU or GPU"
|
|
28
|
+
VALIDATION_FAILURE_CPU_AND_INSTANCE_MESSAGE = (
|
|
29
|
+
"CPU and instance configured together, Invalid configuration."
|
|
30
|
+
)
|
|
31
|
+
VALIDATION_FAILURE_CPU_AND_INSTANCE_SUGGESTION = (
|
|
32
|
+
"Please configure only CPU or instance"
|
|
33
|
+
)
|
|
34
|
+
VALIDATION_FAILURE_GPU_AND_INSTANCE_MESSAGE = (
|
|
35
|
+
"GPU and instance configured together, Invalid configuration."
|
|
36
|
+
)
|
|
37
|
+
VALIDATION_FAILURE_GPU_AND_INSTANCE_SUGGESTION = (
|
|
38
|
+
"Please configure only GPU or instance"
|
|
39
|
+
)
|
|
40
|
+
EXISTING_MODEL_PASSED_MSG = "Build process was provided with pre built model"
|
|
41
|
+
INVALID_QWAK_MODEL_MSG_FORMAT = (
|
|
42
|
+
"The provided qwak model is not of {expected_class} but of {passed_class}"
|
|
43
|
+
)
|
|
44
|
+
SERVICE_ACCOUNT_NOT_FOUND = (
|
|
45
|
+
"GCP service account key secret '{secret_name}' wasn't found"
|
|
46
|
+
)
|
|
47
|
+
SERVICE_ACCOUNT_NOT_FOUND_SUGGESTION = "Make sure your GCP service account key secret is configured correctly in the environment"
|
|
48
|
+
|
|
49
|
+
def description(self) -> str:
|
|
50
|
+
return self.STEP_DESCRIPTION
|
|
51
|
+
|
|
52
|
+
def execute(self) -> None:
|
|
53
|
+
self.validate_build_properties()
|
|
54
|
+
self.create_build_id()
|
|
55
|
+
self.collect_build_name()
|
|
56
|
+
self.collect_model_id()
|
|
57
|
+
self.create_build_dir()
|
|
58
|
+
self.collect_git_credentials()
|
|
59
|
+
self.validate_dependencies()
|
|
60
|
+
self.validate_resources()
|
|
61
|
+
self.validate_deployment_resources()
|
|
62
|
+
self.validate_qwak_model()
|
|
63
|
+
self.validate_gcp_service_account_secret()
|
|
64
|
+
|
|
65
|
+
def validate_build_properties(self):
|
|
66
|
+
if not self.config.build_properties.model_uri.uri:
|
|
67
|
+
error_message = "Model uri wasn't set"
|
|
68
|
+
self.build_logger.error(f"{error_message}, failing...")
|
|
69
|
+
|
|
70
|
+
raise QwakSuggestionException(
|
|
71
|
+
message=error_message,
|
|
72
|
+
suggestion="Make sure your build properties object contains model uri argument",
|
|
73
|
+
)
|
|
74
|
+
if (
|
|
75
|
+
self.config.build_properties.model_uri.git_credentials_secret
|
|
76
|
+
and self.config.build_properties.model_uri.git_secret_ssh
|
|
77
|
+
):
|
|
78
|
+
error_message = (
|
|
79
|
+
"Only one of git credentials secret or git secret ssh can be configured"
|
|
80
|
+
)
|
|
81
|
+
raise QwakSuggestionException(
|
|
82
|
+
message=error_message,
|
|
83
|
+
suggestion="Make sure your git credentials or git secret ssh are configured correctly and only one of "
|
|
84
|
+
"them is configured",
|
|
85
|
+
)
|
|
86
|
+
|
|
87
|
+
def collect_model_id(self):
|
|
88
|
+
model_id = self.config.build_properties.model_id
|
|
89
|
+
model = self.context.client_models_management.get_model(
|
|
90
|
+
model_id=model_id,
|
|
91
|
+
exception_on_missing=False,
|
|
92
|
+
)
|
|
93
|
+
if not model:
|
|
94
|
+
suggestion = f"Create model {model_id} or check model ID spelling"
|
|
95
|
+
snake_case_model_id = snake_case(model_id)
|
|
96
|
+
if self.context.client_models_management.is_model_exists(
|
|
97
|
+
snake_case_model_id
|
|
98
|
+
):
|
|
99
|
+
suggestion = f"Try using model ID {snake_case_model_id} instead"
|
|
100
|
+
raise QwakSuggestionException(
|
|
101
|
+
message=f"Model ID {model_id} isn't found",
|
|
102
|
+
suggestion=suggestion,
|
|
103
|
+
)
|
|
104
|
+
self.context.project_uuid = model.project_id
|
|
105
|
+
self.context.model_uuid = model.uuid
|
|
106
|
+
self.context.model_id = model_id
|
|
107
|
+
|
|
108
|
+
def create_build_id(self):
|
|
109
|
+
if not self.context.build_id:
|
|
110
|
+
self.context.build_id = str(uuid.uuid4())
|
|
111
|
+
self.build_logger.info(f"Generated build ID - {self.context.build_id}")
|
|
112
|
+
else:
|
|
113
|
+
self.build_logger.info(f"Using given build ID - {self.context.build_id}")
|
|
114
|
+
|
|
115
|
+
def collect_build_name(self):
|
|
116
|
+
build_name: str = self.config.build_properties.build_name
|
|
117
|
+
|
|
118
|
+
if build_name:
|
|
119
|
+
self.build_logger.info(f"Using given build name - {build_name}")
|
|
120
|
+
self.context.build_name = build_name
|
|
121
|
+
|
|
122
|
+
def create_build_dir(self):
|
|
123
|
+
build_dir = HOST_TEMP_BUILD_DIR / self.context.model_id / self.context.build_id
|
|
124
|
+
build_dir.mkdir(exist_ok=True, parents=True)
|
|
125
|
+
self.context.host_temp_local_build_dir = build_dir
|
|
126
|
+
self.build_logger.debug(f"Build directory created - {build_dir}")
|
|
127
|
+
|
|
128
|
+
def collect_git_credentials(self):
|
|
129
|
+
if self.config.build_properties.model_uri.git_credentials:
|
|
130
|
+
self.context.git_credentials = (
|
|
131
|
+
self.config.build_properties.model_uri.git_credentials
|
|
132
|
+
)
|
|
133
|
+
elif self.config.build_properties.model_uri.git_credentials_secret:
|
|
134
|
+
self.context.git_credentials = SecretServiceClient().get_secret(
|
|
135
|
+
self.config.build_properties.model_uri.git_credentials_secret
|
|
136
|
+
)
|
|
137
|
+
elif self.config.build_properties.model_uri.git_secret_ssh:
|
|
138
|
+
self.context.git_ssh_key = SecretServiceClient().get_secret(
|
|
139
|
+
self.config.build_properties.model_uri.git_secret_ssh
|
|
140
|
+
)
|
|
141
|
+
else:
|
|
142
|
+
self.build_logger.debug("Git credentials isn't configured")
|
|
143
|
+
|
|
144
|
+
def validate_dependencies(self):
|
|
145
|
+
if (
|
|
146
|
+
Path(self.config.build_properties.model_uri.uri).is_dir()
|
|
147
|
+
and not self.config.build_env.python_env.dependency_file_path
|
|
148
|
+
):
|
|
149
|
+
model_uri, main_dir = (
|
|
150
|
+
Path(self.config.build_properties.model_uri.uri),
|
|
151
|
+
self.config.build_properties.model_uri.main_dir,
|
|
152
|
+
)
|
|
153
|
+
(
|
|
154
|
+
self.context.dependency_manager_type,
|
|
155
|
+
self.context.model_relative_dependency_file,
|
|
156
|
+
self.context.model_relative_dependency_lock_file,
|
|
157
|
+
) = find_dependency_files(model_uri, main_dir, self.build_logger)
|
|
158
|
+
|
|
159
|
+
if (
|
|
160
|
+
self.context.dependency_manager_type
|
|
161
|
+
and self.context.model_relative_dependency_file
|
|
162
|
+
):
|
|
163
|
+
return
|
|
164
|
+
|
|
165
|
+
self.build_logger.error("Dependency file wasn't found, failing...")
|
|
166
|
+
raise QwakSuggestionException(
|
|
167
|
+
message="Dependency file isn't found",
|
|
168
|
+
suggestion="Make sure your model include one of dependencies manager: pip/poetry/conda",
|
|
169
|
+
)
|
|
170
|
+
|
|
171
|
+
def validate_resources(self):
|
|
172
|
+
gpu_configured = (
|
|
173
|
+
self.config.build_env.remote.resources.gpu_type
|
|
174
|
+
or self.config.build_env.remote.resources.gpu_amount
|
|
175
|
+
)
|
|
176
|
+
cpu_configured = (
|
|
177
|
+
self.config.build_env.remote.resources.cpus
|
|
178
|
+
or self.config.build_env.remote.resources.memory
|
|
179
|
+
)
|
|
180
|
+
instance_configured = self.config.build_env.remote.resources.instance
|
|
181
|
+
|
|
182
|
+
if cpu_configured and gpu_configured:
|
|
183
|
+
raise QwakSuggestionException(
|
|
184
|
+
message=self.VALIDATION_FAILURE_CPU_AND_GPU_MESSAGE,
|
|
185
|
+
suggestion=self.VALIDATION_FAILURE_CPU_AND_GPU_SUGGESTION,
|
|
186
|
+
)
|
|
187
|
+
if cpu_configured and instance_configured:
|
|
188
|
+
raise QwakSuggestionException(
|
|
189
|
+
message=self.VALIDATION_FAILURE_CPU_AND_INSTANCE_MESSAGE,
|
|
190
|
+
suggestion=self.VALIDATION_FAILURE_CPU_AND_INSTANCE_SUGGESTION,
|
|
191
|
+
)
|
|
192
|
+
if gpu_configured and instance_configured:
|
|
193
|
+
raise QwakSuggestionException(
|
|
194
|
+
message=self.VALIDATION_FAILURE_GPU_AND_INSTANCE_MESSAGE,
|
|
195
|
+
suggestion=self.VALIDATION_FAILURE_GPU_AND_INSTANCE_SUGGESTION,
|
|
196
|
+
)
|
|
197
|
+
|
|
198
|
+
if instance_configured:
|
|
199
|
+
provider = self.__get_provider()
|
|
200
|
+
verify_template_id(
|
|
201
|
+
self.config.build_env.remote.resources.instance,
|
|
202
|
+
self.context.client_instance_template,
|
|
203
|
+
provider=provider,
|
|
204
|
+
)
|
|
205
|
+
|
|
206
|
+
if not (cpu_configured or gpu_configured or instance_configured):
|
|
207
|
+
self.config.build_env.remote.resources.cpus = self.DEFAULT_CPUS
|
|
208
|
+
self.config.build_env.remote.resources.memory = self.DEFAULT_MEMORY
|
|
209
|
+
|
|
210
|
+
def __get_provider(self):
|
|
211
|
+
user_context = self.context.client_ecosystem.get_authenticated_user_context()
|
|
212
|
+
provider = None
|
|
213
|
+
if (
|
|
214
|
+
user_context.user.environment_details.configuration.cloud_configuration.WhichOneof(
|
|
215
|
+
"configuration"
|
|
216
|
+
)
|
|
217
|
+
== "aws_cloud_configuration"
|
|
218
|
+
):
|
|
219
|
+
provider = Provider.AWS
|
|
220
|
+
elif (
|
|
221
|
+
user_context.user.environment_details.configuration.cloud_configuration.WhichOneof(
|
|
222
|
+
"configuration"
|
|
223
|
+
)
|
|
224
|
+
== "gcp_cloud_configuration"
|
|
225
|
+
):
|
|
226
|
+
provider = Provider.GCP
|
|
227
|
+
return provider
|
|
228
|
+
|
|
229
|
+
def validate_deployment_resources(self):
|
|
230
|
+
if self.config.deploy and self.config.deployment_instance:
|
|
231
|
+
provider = self.__get_provider()
|
|
232
|
+
verify_template_id(
|
|
233
|
+
self.config.deployment_instance,
|
|
234
|
+
self.context.client_instance_template,
|
|
235
|
+
provider=provider,
|
|
236
|
+
)
|
|
237
|
+
|
|
238
|
+
def validate_qwak_model(self):
|
|
239
|
+
if self.config.pre_built_model:
|
|
240
|
+
self.build_logger.debug(self.EXISTING_MODEL_PASSED_MSG)
|
|
241
|
+
if not isinstance(self.config.pre_built_model, QwakModel):
|
|
242
|
+
raise QwakException(
|
|
243
|
+
self.INVALID_QWAK_MODEL_MSG_FORMAT.format(
|
|
244
|
+
expected_class=QwakModel.__name__,
|
|
245
|
+
passed_class=self.config.pre_built_model.__class__,
|
|
246
|
+
)
|
|
247
|
+
)
|
|
248
|
+
|
|
249
|
+
def validate_gcp_service_account_secret(self):
|
|
250
|
+
if self.config.build_env.docker.service_account_key_secret_name:
|
|
251
|
+
self.build_logger.debug("Checking GCP service account key secret")
|
|
252
|
+
try:
|
|
253
|
+
SecretServiceClient().get_secret(
|
|
254
|
+
self.config.build_env.docker.service_account_key_secret_name
|
|
255
|
+
)
|
|
256
|
+
except QwakException as e:
|
|
257
|
+
if e.status_code == grpc.StatusCode.NOT_FOUND:
|
|
258
|
+
raise QwakSuggestionException(
|
|
259
|
+
message=self.SERVICE_ACCOUNT_NOT_FOUND.format(
|
|
260
|
+
secret_name=self.config.build_env.docker.service_account_key_secret_name
|
|
261
|
+
),
|
|
262
|
+
suggestion=self.SERVICE_ACCOUNT_NOT_FOUND_SUGGESTION,
|
|
263
|
+
)
|
|
264
|
+
raise e
|
|
265
|
+
|
|
266
|
+
self.build_logger.debug("service account key secret found")
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import importlib
|
|
2
|
+
|
|
3
|
+
from qwak.inner.build_logic.interface.step_inteface import Step
|
|
4
|
+
from qwak import __version__ as qwak_core_version
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
class SetVersionStep(Step):
|
|
8
|
+
STEP_DESCRIPTION = "Getting SDK Version"
|
|
9
|
+
SDK_VERSION_NOT_AVAILABLE_MSG_FORMAT = (
|
|
10
|
+
"Sdk version not available, using core version {qwak_core_version}"
|
|
11
|
+
)
|
|
12
|
+
SDK_VERSION_FOUND_MSG_FORMAT = "Found sdk version {qwak_sdk_version}"
|
|
13
|
+
|
|
14
|
+
def description(self) -> str:
|
|
15
|
+
return self.STEP_DESCRIPTION
|
|
16
|
+
|
|
17
|
+
def execute(self) -> None:
|
|
18
|
+
try:
|
|
19
|
+
self.build_logger.debug("Getting sdk version")
|
|
20
|
+
qwak_sdk_version = importlib.import_module("qwak_sdk").__version__
|
|
21
|
+
self.context.qwak_sdk_version = qwak_sdk_version
|
|
22
|
+
self.build_logger.debug(
|
|
23
|
+
self.SDK_VERSION_FOUND_MSG_FORMAT.format(
|
|
24
|
+
qwak_sdk_version=qwak_sdk_version
|
|
25
|
+
)
|
|
26
|
+
)
|
|
27
|
+
except ImportError:
|
|
28
|
+
self.build_logger.debug(
|
|
29
|
+
self.SDK_VERSION_NOT_AVAILABLE_MSG_FORMAT.format(
|
|
30
|
+
qwak_core_version=qwak_core_version
|
|
31
|
+
)
|
|
32
|
+
)
|
|
33
|
+
self.context.qwak_sdk_version = qwak_core_version
|
|
File without changes
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import os
|
|
2
|
+
|
|
3
|
+
from qwak.inner.build_logic.interface.step_inteface import Step
|
|
4
|
+
from qwak.inner.build_logic.tools.files import cleaning_up_after_build
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
class CleanupStep(Step):
|
|
8
|
+
STEP_DESCRIPTION = "Clean build artifacts"
|
|
9
|
+
|
|
10
|
+
def description(self) -> str:
|
|
11
|
+
return self.STEP_DESCRIPTION
|
|
12
|
+
|
|
13
|
+
def execute(self) -> None:
|
|
14
|
+
self.build_logger.debug(
|
|
15
|
+
"Cleanup environment - Deleting file and intermediate images"
|
|
16
|
+
)
|
|
17
|
+
if os.getenv("POD_NAME"):
|
|
18
|
+
self.build_logger.debug("Skipping cleanup - Remote executor is temporary")
|
|
19
|
+
else:
|
|
20
|
+
cleaning_up_after_build(self)
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
from copy import deepcopy
|
|
2
|
+
|
|
3
|
+
from _qwak_proto.qwak.builds.build_url_pb2 import BuildVersioningTagsType
|
|
4
|
+
from qwak.inner.build_logic.constants.upload_tag import (
|
|
5
|
+
BUILD_CONFIG_TAG,
|
|
6
|
+
QWAK_SDK_VERSION_TAG,
|
|
7
|
+
MODEL_CODE_TAG,
|
|
8
|
+
)
|
|
9
|
+
|
|
10
|
+
from qwak.inner.build_logic.interface.step_inteface import Step
|
|
11
|
+
|
|
12
|
+
|
|
13
|
+
class StartRemoteBuildStep(Step):
|
|
14
|
+
STEP_DESCRIPTION = "Start remote build"
|
|
15
|
+
|
|
16
|
+
def description(self) -> str:
|
|
17
|
+
return self.STEP_DESCRIPTION
|
|
18
|
+
|
|
19
|
+
def execute(self) -> None:
|
|
20
|
+
self.build_logger.info(f"Start remote build - {self.context.build_id}")
|
|
21
|
+
config_copy = deepcopy(self.config)
|
|
22
|
+
config_copy.build_properties.build_id = self.context.build_id
|
|
23
|
+
config_copy.build_properties.build_name = self.context.build_name
|
|
24
|
+
self.context.client_builds_orchestrator.build_model(
|
|
25
|
+
build_conf=config_copy,
|
|
26
|
+
build_v1_flag=False,
|
|
27
|
+
build_config_url=self.get_download_url(BUILD_CONFIG_TAG),
|
|
28
|
+
qwak_sdk_version_url=self.get_download_url(QWAK_SDK_VERSION_TAG),
|
|
29
|
+
resolved_model_url=self.get_download_url(MODEL_CODE_TAG),
|
|
30
|
+
git_commit_id=self.context.git_commit_id,
|
|
31
|
+
sdk_version=self.context.qwak_sdk_version,
|
|
32
|
+
)
|
|
33
|
+
self.build_logger.info("Remote build started successfully")
|
|
34
|
+
|
|
35
|
+
def get_download_url(self, tag: str) -> str:
|
|
36
|
+
return (
|
|
37
|
+
self.context.client_builds_orchestrator.get_build_versioning_download_url(
|
|
38
|
+
model_id=self.config.build_properties.model_id,
|
|
39
|
+
build_id=self.context.build_id,
|
|
40
|
+
tag=tag,
|
|
41
|
+
tag_type=BuildVersioningTagsType.FILE_TAG_TYPE,
|
|
42
|
+
).download_url
|
|
43
|
+
)
|
|
@@ -0,0 +1,258 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
import re
|
|
4
|
+
from pathlib import Path
|
|
5
|
+
from typing import Any, List, Tuple, Union, Optional
|
|
6
|
+
|
|
7
|
+
import joblib
|
|
8
|
+
import requests
|
|
9
|
+
from retrying import retry
|
|
10
|
+
|
|
11
|
+
from _qwak_proto.qwak.builds.build_url_pb2 import BuildVersioningTagsType
|
|
12
|
+
from _qwak_proto.qwak.builds.builds_orchestrator_service_pb2 import (
|
|
13
|
+
GetBuildVersioningUploadURLResponse,
|
|
14
|
+
)
|
|
15
|
+
from qwak.exceptions import QwakException
|
|
16
|
+
from qwak.exceptions import QwakGeneralBuildException
|
|
17
|
+
from qwak.inner.build_logic.constants.temp_dir import TEMP_LOCAL_MODEL_DIR
|
|
18
|
+
from qwak.inner.build_logic.constants.upload_tag import (
|
|
19
|
+
MODEL_CODE_TAG,
|
|
20
|
+
SKINNY_MODEL_CODE_TAG,
|
|
21
|
+
QWAK_SDK_VERSION_TAG,
|
|
22
|
+
BUILD_CONFIG_TAG,
|
|
23
|
+
QWAK_RUNTIME_WHEEL_TAG,
|
|
24
|
+
QWAK_CORE_WHEEL_TAG,
|
|
25
|
+
QWAK_BUILT_MODEL_TAG,
|
|
26
|
+
)
|
|
27
|
+
from qwak.inner.build_logic.interface.step_inteface import Step
|
|
28
|
+
from qwak.inner.build_logic.tools.files import (
|
|
29
|
+
QWAK_IGNORE_FILE_NAME,
|
|
30
|
+
IGNORED_PATTERNS_FOR_UPLOAD,
|
|
31
|
+
zip_model,
|
|
32
|
+
UploadInChunks,
|
|
33
|
+
)
|
|
34
|
+
from qwak.inner.build_logic.tools.ignore_files import load_patterns_from_ignore_file
|
|
35
|
+
|
|
36
|
+
_MAX_FILE_SIZE_BYTES = 10000000
|
|
37
|
+
|
|
38
|
+
|
|
39
|
+
def should_retry(qwak_exception: QwakException):
|
|
40
|
+
# when Got 403 from Jfrog it means that the reposity doesn't exist. It may happen when in the first build in the project
|
|
41
|
+
return str(qwak_exception.message).__contains__("403")
|
|
42
|
+
|
|
43
|
+
|
|
44
|
+
class UploadStep(Step):
|
|
45
|
+
STEP_DESCRIPTION = "Saving Qwak Model"
|
|
46
|
+
|
|
47
|
+
def description(self) -> str:
|
|
48
|
+
return self.STEP_DESCRIPTION
|
|
49
|
+
|
|
50
|
+
def execute(self) -> None:
|
|
51
|
+
files_tag_iterator = self.create_files_to_upload()
|
|
52
|
+
files_total_size = sum(
|
|
53
|
+
file.stat().st_size for (file, tag) in files_tag_iterator
|
|
54
|
+
)
|
|
55
|
+
upload_so_far = 0
|
|
56
|
+
for file, tag in files_tag_iterator:
|
|
57
|
+
if file.exists():
|
|
58
|
+
pre_signed_url = self.upload_file(
|
|
59
|
+
file=file,
|
|
60
|
+
tag=tag,
|
|
61
|
+
all_files_size_to_upload=files_total_size,
|
|
62
|
+
read_so_far=upload_so_far,
|
|
63
|
+
)
|
|
64
|
+
upload_so_far += file.stat().st_size
|
|
65
|
+
|
|
66
|
+
if tag == MODEL_CODE_TAG:
|
|
67
|
+
self.context.model_code_remote_url = str(pre_signed_url).split("?")[
|
|
68
|
+
0
|
|
69
|
+
]
|
|
70
|
+
|
|
71
|
+
def create_files_to_upload(self) -> Union[List[Tuple[Any, Any], Tuple[Path, Any]]]:
|
|
72
|
+
ignored_patterns = (
|
|
73
|
+
load_patterns_from_ignore_file(
|
|
74
|
+
build_logger=self.build_logger,
|
|
75
|
+
ignore_file_path=self.context.host_temp_local_build_dir
|
|
76
|
+
/ TEMP_LOCAL_MODEL_DIR
|
|
77
|
+
/ self.config.build_properties.model_uri.main_dir
|
|
78
|
+
/ QWAK_IGNORE_FILE_NAME,
|
|
79
|
+
)
|
|
80
|
+
+ IGNORED_PATTERNS_FOR_UPLOAD
|
|
81
|
+
)
|
|
82
|
+
|
|
83
|
+
# copy 'main' and 'tests' directories
|
|
84
|
+
dirs_to_include = [self.config.build_properties.model_uri.main_dir, "tests"]
|
|
85
|
+
deps_folders = []
|
|
86
|
+
for (
|
|
87
|
+
folder
|
|
88
|
+
) in self.config.build_properties.model_uri.dependency_required_folders:
|
|
89
|
+
destination_folder = folder
|
|
90
|
+
while destination_folder.startswith(".."):
|
|
91
|
+
destination_folder = re.sub(r"^\.\./", "", destination_folder)
|
|
92
|
+
deps_folders.append(destination_folder)
|
|
93
|
+
if deps_folders:
|
|
94
|
+
self.build_logger.debug(
|
|
95
|
+
f"Adding dependency folders to model code: {deps_folders}"
|
|
96
|
+
)
|
|
97
|
+
dirs_to_include += deps_folders
|
|
98
|
+
|
|
99
|
+
self.build_logger.debug("Zipping skinny model code")
|
|
100
|
+
skinny_size_zip_file = zip_model(
|
|
101
|
+
build_dir=self.context.host_temp_local_build_dir,
|
|
102
|
+
dependency_file=self.context.model_relative_dependency_file,
|
|
103
|
+
deps_lock_file=self.context.model_relative_dependency_lock_file,
|
|
104
|
+
dirs_to_include=dirs_to_include,
|
|
105
|
+
zip_name="skinny_size_model_code",
|
|
106
|
+
ignored_patterns=ignored_patterns,
|
|
107
|
+
max_bytes=_MAX_FILE_SIZE_BYTES,
|
|
108
|
+
)
|
|
109
|
+
|
|
110
|
+
# Full size model
|
|
111
|
+
self.build_logger.debug("Zipping full model code")
|
|
112
|
+
full_size_zip_file = zip_model(
|
|
113
|
+
build_dir=self.context.host_temp_local_build_dir,
|
|
114
|
+
dependency_file=self.context.model_relative_dependency_file,
|
|
115
|
+
deps_lock_file=self.context.model_relative_dependency_lock_file,
|
|
116
|
+
dirs_to_include=dirs_to_include,
|
|
117
|
+
zip_name="full_size_model_code",
|
|
118
|
+
ignored_patterns=ignored_patterns,
|
|
119
|
+
)
|
|
120
|
+
|
|
121
|
+
# Dump config file for upload
|
|
122
|
+
config_file_temp = self.context.host_temp_local_build_dir / "build.conf"
|
|
123
|
+
config_file_temp.write_text(self.config.to_yaml())
|
|
124
|
+
|
|
125
|
+
# Dump qwak-sdk version for upload
|
|
126
|
+
qwak_sdk_version_temp = self.context.host_temp_local_build_dir / "VERSION"
|
|
127
|
+
qwak_sdk_version_temp.write_text(self.context.qwak_sdk_version)
|
|
128
|
+
|
|
129
|
+
files_tag_iterator = [
|
|
130
|
+
(full_size_zip_file, MODEL_CODE_TAG),
|
|
131
|
+
(skinny_size_zip_file, SKINNY_MODEL_CODE_TAG),
|
|
132
|
+
(qwak_sdk_version_temp, QWAK_SDK_VERSION_TAG),
|
|
133
|
+
(config_file_temp, BUILD_CONFIG_TAG),
|
|
134
|
+
]
|
|
135
|
+
|
|
136
|
+
if self.context.custom_runtime_wheel:
|
|
137
|
+
files_tag_iterator.append(
|
|
138
|
+
(self.context.custom_runtime_wheel, QWAK_RUNTIME_WHEEL_TAG)
|
|
139
|
+
)
|
|
140
|
+
|
|
141
|
+
if self.context.custom_core_wheel:
|
|
142
|
+
files_tag_iterator.append(
|
|
143
|
+
(self.context.custom_core_wheel, QWAK_CORE_WHEEL_TAG)
|
|
144
|
+
)
|
|
145
|
+
|
|
146
|
+
if self.config.pre_built_model:
|
|
147
|
+
temp_model_file = (
|
|
148
|
+
self.context.host_temp_local_build_dir / QWAK_BUILT_MODEL_TAG
|
|
149
|
+
)
|
|
150
|
+
joblib.dump(self.config.pre_built_model, temp_model_file, compress=3)
|
|
151
|
+
files_tag_iterator.append((temp_model_file, QWAK_BUILT_MODEL_TAG))
|
|
152
|
+
|
|
153
|
+
return files_tag_iterator
|
|
154
|
+
|
|
155
|
+
def upload_file(
|
|
156
|
+
self, file: Path, tag: str, all_files_size_to_upload: int, read_so_far: int
|
|
157
|
+
):
|
|
158
|
+
self.build_logger.debug(f"Upload file {file}")
|
|
159
|
+
|
|
160
|
+
pre_signed_url_response: GetBuildVersioningUploadURLResponse = (
|
|
161
|
+
self.get_pre_signed_upload_url(
|
|
162
|
+
tag=tag, tag_type=BuildVersioningTagsType.FILE_TAG_TYPE
|
|
163
|
+
)
|
|
164
|
+
)
|
|
165
|
+
self.upload_file_to_remote_storge(
|
|
166
|
+
upload_url=pre_signed_url_response.upload_url,
|
|
167
|
+
file=file,
|
|
168
|
+
all_files_size_to_upload=all_files_size_to_upload,
|
|
169
|
+
read_so_far=read_so_far,
|
|
170
|
+
headers=pre_signed_url_response.headers,
|
|
171
|
+
)
|
|
172
|
+
|
|
173
|
+
self.build_logger.debug(f"Upload file {file} completed")
|
|
174
|
+
|
|
175
|
+
return pre_signed_url_response.upload_url
|
|
176
|
+
|
|
177
|
+
def get_pre_signed_upload_url(
|
|
178
|
+
self, tag: str, tag_type: Optional[BuildVersioningTagsType]
|
|
179
|
+
) -> GetBuildVersioningUploadURLResponse:
|
|
180
|
+
try:
|
|
181
|
+
self.build_logger.debug(f"Getting pre-signed url for upload - tag {tag}")
|
|
182
|
+
|
|
183
|
+
pre_signed_url_response: GetBuildVersioningUploadURLResponse = (
|
|
184
|
+
self.context.client_builds_orchestrator.get_build_versioning_upload_url(
|
|
185
|
+
build_id=self.context.build_id,
|
|
186
|
+
model_id=self.context.model_id,
|
|
187
|
+
tag=tag,
|
|
188
|
+
tag_type=tag_type,
|
|
189
|
+
)
|
|
190
|
+
)
|
|
191
|
+
|
|
192
|
+
self.build_logger.debug("Pre-signed url generated successfully")
|
|
193
|
+
|
|
194
|
+
return pre_signed_url_response
|
|
195
|
+
except QwakException as e:
|
|
196
|
+
raise QwakGeneralBuildException(
|
|
197
|
+
message="Unable to get pre-signed url for uploading model",
|
|
198
|
+
src_exception=e,
|
|
199
|
+
)
|
|
200
|
+
|
|
201
|
+
def upload_file_to_remote_storge(
|
|
202
|
+
self,
|
|
203
|
+
upload_url: str,
|
|
204
|
+
file: Path,
|
|
205
|
+
all_files_size_to_upload: int,
|
|
206
|
+
read_so_far: int,
|
|
207
|
+
headers: Optional[dict] = None,
|
|
208
|
+
):
|
|
209
|
+
if not headers:
|
|
210
|
+
headers = {}
|
|
211
|
+
|
|
212
|
+
try:
|
|
213
|
+
self.build_logger.debug(f"Upload file {file} to Qwak storage")
|
|
214
|
+
|
|
215
|
+
self.send_request(
|
|
216
|
+
upload_url, file, all_files_size_to_upload, read_so_far, headers
|
|
217
|
+
)
|
|
218
|
+
self.build_logger.debug(
|
|
219
|
+
f"File {file} uploaded to Qwak storage successfully"
|
|
220
|
+
)
|
|
221
|
+
except Exception as e:
|
|
222
|
+
raise QwakGeneralBuildException(
|
|
223
|
+
message="Fail uploading model to remote storage.",
|
|
224
|
+
src_exception=e,
|
|
225
|
+
)
|
|
226
|
+
|
|
227
|
+
@retry(retry_on_exception=should_retry, wait_fixed=15000, stop_max_delay=60000)
|
|
228
|
+
def send_request(
|
|
229
|
+
self,
|
|
230
|
+
upload_url: str,
|
|
231
|
+
file: Path,
|
|
232
|
+
all_files_size_to_upload: int,
|
|
233
|
+
read_so_far: int,
|
|
234
|
+
headers: Optional[dict],
|
|
235
|
+
):
|
|
236
|
+
if not headers:
|
|
237
|
+
headers = {}
|
|
238
|
+
|
|
239
|
+
# Adding to the current headers the content-type
|
|
240
|
+
headers["content-type"] = "text/plain"
|
|
241
|
+
|
|
242
|
+
http_response = requests.put( # nosec B113
|
|
243
|
+
url=upload_url,
|
|
244
|
+
data=UploadInChunks(
|
|
245
|
+
file=file,
|
|
246
|
+
build_logger=self.build_logger,
|
|
247
|
+
chunk_size_bytes=10,
|
|
248
|
+
all_files_size_to_upload=all_files_size_to_upload,
|
|
249
|
+
read_so_far=read_so_far,
|
|
250
|
+
),
|
|
251
|
+
headers=headers,
|
|
252
|
+
)
|
|
253
|
+
|
|
254
|
+
if http_response.status_code not in [200, 201]:
|
|
255
|
+
raise QwakException(
|
|
256
|
+
f"Status: [{http_response.status_code}], "
|
|
257
|
+
f"reason: [{http_response.reason}]"
|
|
258
|
+
)
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
from typing import List, Optional
|
|
2
|
+
|
|
3
|
+
from qwak.inner.build_config.build_config_v1 import BuildConfigV1
|
|
4
|
+
from ..interface.build_phase import BuildPhase
|
|
5
|
+
from ..interface.context_interface import Context
|
|
6
|
+
|
|
7
|
+
from qwak.inner.build_logic.interface.step_inteface import Step
|
|
8
|
+
|
|
9
|
+
|
|
10
|
+
class PhasesPipeline:
|
|
11
|
+
def __init__(
|
|
12
|
+
self,
|
|
13
|
+
config: Optional[BuildConfigV1],
|
|
14
|
+
context: Context,
|
|
15
|
+
build_phase: Optional[BuildPhase] = None,
|
|
16
|
+
) -> None:
|
|
17
|
+
self._phases: List[PhasesPipeline] = []
|
|
18
|
+
self._steps: List[Step] = []
|
|
19
|
+
self.context = context
|
|
20
|
+
self._config = config
|
|
21
|
+
self.build_phase = build_phase
|
|
22
|
+
|
|
23
|
+
@property
|
|
24
|
+
def phases(self) -> List["PhasesPipeline"]:
|
|
25
|
+
return self._phases
|
|
26
|
+
|
|
27
|
+
@property
|
|
28
|
+
def steps(self) -> List[Step]:
|
|
29
|
+
return self._steps
|
|
30
|
+
|
|
31
|
+
def add_phase(self, steps: List[Step], build_phase: BuildPhase):
|
|
32
|
+
phase = PhasesPipeline(
|
|
33
|
+
config=self._config,
|
|
34
|
+
context=self.context,
|
|
35
|
+
build_phase=build_phase,
|
|
36
|
+
)
|
|
37
|
+
|
|
38
|
+
for step in steps:
|
|
39
|
+
step.context = self.context
|
|
40
|
+
step.config = self._config
|
|
41
|
+
step.build_phase = build_phase
|
|
42
|
+
phase._steps.append(step)
|
|
43
|
+
|
|
44
|
+
self._phases.append(phase)
|
|
File without changes
|