frogml-core 0.0.1__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.
- _frogml_proto/__init__.py +0 -0
- _frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.py +35 -0
- _frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.pyi +38 -0
- _frogml_proto/jfml/model_version/v1/model_repository_spec_pb2_grpc.py +4 -0
- _frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py +45 -0
- _frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi +41 -0
- _frogml_proto/jfml/model_version/v1/model_version_framework_pb2_grpc.py +4 -0
- _frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py +151 -0
- _frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi +230 -0
- _frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py +237 -0
- _frogml_proto/jfml/model_version/v1/model_version_pb2.py +88 -0
- _frogml_proto/jfml/model_version/v1/model_version_pb2.pyi +133 -0
- _frogml_proto/jfml/model_version/v1/model_version_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/account/v1/account_pb2.py +114 -0
- _frogml_proto/qwak/administration/account/v1/account_pb2.pyi +257 -0
- _frogml_proto/qwak/administration/account/v1/account_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/account/v1/account_service_pb2.py +212 -0
- _frogml_proto/qwak/administration/account/v1/account_service_pb2.pyi +344 -0
- _frogml_proto/qwak/administration/account/v1/account_service_pb2_grpc.py +338 -0
- _frogml_proto/qwak/administration/account/v1/jfrog_tenant_details_pb2.py +77 -0
- _frogml_proto/qwak/administration/account/v1/jfrog_tenant_details_pb2.pyi +109 -0
- _frogml_proto/qwak/administration/account/v1/jfrog_tenant_details_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/account/v1/personalization_pb2.py +45 -0
- _frogml_proto/qwak/administration/account/v1/personalization_pb2.pyi +49 -0
- _frogml_proto/qwak/administration/account/v1/personalization_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/account/v1/preferences_pb2.py +36 -0
- _frogml_proto/qwak/administration/account/v1/preferences_pb2.pyi +35 -0
- _frogml_proto/qwak/administration/account/v1/preferences_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/account/v1/terms_pb2.py +75 -0
- _frogml_proto/qwak/administration/account/v1/terms_pb2.pyi +94 -0
- _frogml_proto/qwak/administration/account/v1/terms_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/authenticated_user_service_pb2.py +91 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/authenticated_user_service_pb2.pyi +108 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/authenticated_user_service_pb2_grpc.py +135 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/credentials_pb2.py +46 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/credentials_pb2.pyi +62 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/credentials_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/details_pb2.py +75 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/details_pb2.pyi +140 -0
- _frogml_proto/qwak/administration/authenticated_user/v1/details_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/v0/authentication/authentication_service_pb2.py +69 -0
- _frogml_proto/qwak/administration/v0/authentication/authentication_service_pb2.pyi +92 -0
- _frogml_proto/qwak/administration/v0/authentication/authentication_service_pb2_grpc.py +67 -0
- _frogml_proto/qwak/administration/v0/environments/configuration_pb2.py +93 -0
- _frogml_proto/qwak/administration/v0/environments/configuration_pb2.pyi +233 -0
- _frogml_proto/qwak/administration/v0/environments/configuration_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/v0/environments/environment_pb2.py +66 -0
- _frogml_proto/qwak/administration/v0/environments/environment_pb2.pyi +156 -0
- _frogml_proto/qwak/administration/v0/environments/environment_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/v0/environments/environment_service_pb2.py +184 -0
- _frogml_proto/qwak/administration/v0/environments/environment_service_pb2.pyi +274 -0
- _frogml_proto/qwak/administration/v0/environments/environment_service_pb2_grpc.py +272 -0
- _frogml_proto/qwak/administration/v0/environments/personalization_pb2.py +45 -0
- _frogml_proto/qwak/administration/v0/environments/personalization_pb2.pyi +45 -0
- _frogml_proto/qwak/administration/v0/environments/personalization_pb2_grpc.py +4 -0
- _frogml_proto/qwak/administration/v0/users/user_pb2.py +97 -0
- _frogml_proto/qwak/administration/v0/users/user_pb2.pyi +249 -0
- _frogml_proto/qwak/administration/v0/users/user_pb2_grpc.py +4 -0
- _frogml_proto/qwak/admiral/secret/v0/secret_pb2.py +140 -0
- _frogml_proto/qwak/admiral/secret/v0/secret_pb2.pyi +237 -0
- _frogml_proto/qwak/admiral/secret/v0/secret_pb2_grpc.py +4 -0
- _frogml_proto/qwak/admiral/secret/v0/secret_service_pb2.py +89 -0
- _frogml_proto/qwak/admiral/secret/v0/secret_service_pb2.pyi +93 -0
- _frogml_proto/qwak/admiral/secret/v0/secret_service_pb2_grpc.py +135 -0
- _frogml_proto/qwak/admiral/secret/v0/system_secret_service_pb2.py +89 -0
- _frogml_proto/qwak/admiral/secret/v0/system_secret_service_pb2.pyi +101 -0
- _frogml_proto/qwak/admiral/secret/v0/system_secret_service_pb2_grpc.py +135 -0
- _frogml_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2.py +198 -0
- _frogml_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2.pyi +431 -0
- _frogml_proto/qwak/admiral/user_application_instance/v0/user_application_instance_pb2_grpc.py +4 -0
- _frogml_proto/qwak/admiral/user_application_instance/v0/user_application_instance_service_pb2.py +109 -0
- _frogml_proto/qwak/admiral/user_application_instance/v0/user_application_instance_service_pb2.pyi +128 -0
- _frogml_proto/qwak/admiral/user_application_instance/v0/user_application_instance_service_pb2_grpc.py +169 -0
- _frogml_proto/qwak/analytics/analytics_pb2.py +142 -0
- _frogml_proto/qwak/analytics/analytics_pb2.pyi +308 -0
- _frogml_proto/qwak/analytics/analytics_pb2_grpc.py +4 -0
- _frogml_proto/qwak/analytics/analytics_service_pb2.py +229 -0
- _frogml_proto/qwak/analytics/analytics_service_pb2.pyi +339 -0
- _frogml_proto/qwak/analytics/analytics_service_pb2_grpc.py +370 -0
- _frogml_proto/qwak/audience/v1/audience_api_pb2.py +149 -0
- _frogml_proto/qwak/audience/v1/audience_api_pb2.pyi +176 -0
- _frogml_proto/qwak/audience/v1/audience_api_pb2_grpc.py +240 -0
- _frogml_proto/qwak/audience/v1/audience_pb2.py +137 -0
- _frogml_proto/qwak/audience/v1/audience_pb2.pyi +313 -0
- _frogml_proto/qwak/audience/v1/audience_pb2_grpc.py +4 -0
- _frogml_proto/qwak/auto_scaling/v1/auto_scaling_pb2.py +100 -0
- _frogml_proto/qwak/auto_scaling/v1/auto_scaling_pb2.pyi +187 -0
- _frogml_proto/qwak/auto_scaling/v1/auto_scaling_pb2_grpc.py +4 -0
- _frogml_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2.py +49 -0
- _frogml_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2.pyi +55 -0
- _frogml_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2_grpc.py +67 -0
- _frogml_proto/qwak/automation/v1/action_pb2.py +220 -0
- _frogml_proto/qwak/automation/v1/action_pb2.pyi +513 -0
- _frogml_proto/qwak/automation/v1/action_pb2_grpc.py +4 -0
- _frogml_proto/qwak/automation/v1/auto_scaling_pb2.py +97 -0
- _frogml_proto/qwak/automation/v1/auto_scaling_pb2.pyi +177 -0
- _frogml_proto/qwak/automation/v1/auto_scaling_pb2_grpc.py +4 -0
- _frogml_proto/qwak/automation/v1/automation_execution_pb2.py +65 -0
- _frogml_proto/qwak/automation/v1/automation_execution_pb2.pyi +123 -0
- _frogml_proto/qwak/automation/v1/automation_execution_pb2_grpc.py +4 -0
- _frogml_proto/qwak/automation/v1/automation_management_service_pb2.py +283 -0
- _frogml_proto/qwak/automation/v1/automation_management_service_pb2.pyi +386 -0
- _frogml_proto/qwak/automation/v1/automation_management_service_pb2_grpc.py +429 -0
- _frogml_proto/qwak/automation/v1/automation_pb2.py +59 -0
- _frogml_proto/qwak/automation/v1/automation_pb2.pyi +121 -0
- _frogml_proto/qwak/automation/v1/automation_pb2_grpc.py +4 -0
- _frogml_proto/qwak/automation/v1/common_pb2.py +56 -0
- _frogml_proto/qwak/automation/v1/common_pb2.pyi +71 -0
- _frogml_proto/qwak/automation/v1/common_pb2_grpc.py +4 -0
- _frogml_proto/qwak/automation/v1/notification_pb2.py +93 -0
- _frogml_proto/qwak/automation/v1/notification_pb2.pyi +134 -0
- _frogml_proto/qwak/automation/v1/notification_pb2_grpc.py +4 -0
- _frogml_proto/qwak/automation/v1/trigger_pb2.py +76 -0
- _frogml_proto/qwak/automation/v1/trigger_pb2.pyi +112 -0
- _frogml_proto/qwak/automation/v1/trigger_pb2_grpc.py +4 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_events_pb2.py +148 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_events_pb2.pyi +307 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_events_pb2_grpc.py +4 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_resources_pb2.py +47 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_resources_pb2.pyi +61 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_resources_pb2_grpc.py +4 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_service_pb2.py +570 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi +1328 -0
- _frogml_proto/qwak/batch_job/v1/batch_job_service_pb2_grpc.py +574 -0
- _frogml_proto/qwak/build/v1/build_api_pb2.py +446 -0
- _frogml_proto/qwak/build/v1/build_api_pb2.pyi +682 -0
- _frogml_proto/qwak/build/v1/build_api_pb2_grpc.py +594 -0
- _frogml_proto/qwak/build/v1/build_pb2.py +575 -0
- _frogml_proto/qwak/build/v1/build_pb2.pyi +1328 -0
- _frogml_proto/qwak/build/v1/build_pb2_grpc.py +4 -0
- _frogml_proto/qwak/build_settings/build_settings_api_pb2.py +89 -0
- _frogml_proto/qwak/build_settings/build_settings_api_pb2.pyi +106 -0
- _frogml_proto/qwak/build_settings/build_settings_api_pb2_grpc.py +132 -0
- _frogml_proto/qwak/build_settings/build_settings_pb2.py +95 -0
- _frogml_proto/qwak/build_settings/build_settings_pb2.pyi +148 -0
- _frogml_proto/qwak/build_settings/build_settings_pb2_grpc.py +4 -0
- _frogml_proto/qwak/builds/build_pb2.py +233 -0
- _frogml_proto/qwak/builds/build_pb2.pyi +582 -0
- _frogml_proto/qwak/builds/build_pb2_grpc.py +4 -0
- _frogml_proto/qwak/builds/build_url_pb2.py +86 -0
- _frogml_proto/qwak/builds/build_url_pb2.pyi +179 -0
- _frogml_proto/qwak/builds/build_url_pb2_grpc.py +4 -0
- _frogml_proto/qwak/builds/builds_orchestrator_service_pb2.py +212 -0
- _frogml_proto/qwak/builds/builds_orchestrator_service_pb2.pyi +322 -0
- _frogml_proto/qwak/builds/builds_orchestrator_service_pb2_grpc.py +306 -0
- _frogml_proto/qwak/builds/builds_pb2.py +556 -0
- _frogml_proto/qwak/builds/builds_pb2.pyi +1271 -0
- _frogml_proto/qwak/builds/builds_pb2_grpc.py +237 -0
- _frogml_proto/qwak/builds/internal_builds_orchestrator_service_pb2.py +59 -0
- _frogml_proto/qwak/builds/internal_builds_orchestrator_service_pb2.pyi +64 -0
- _frogml_proto/qwak/builds/internal_builds_orchestrator_service_pb2_grpc.py +67 -0
- _frogml_proto/qwak/data_versioning/data_versioning_pb2.py +45 -0
- _frogml_proto/qwak/data_versioning/data_versioning_pb2.pyi +71 -0
- _frogml_proto/qwak/data_versioning/data_versioning_pb2_grpc.py +4 -0
- _frogml_proto/qwak/data_versioning/data_versioning_service_pb2.py +69 -0
- _frogml_proto/qwak/data_versioning/data_versioning_service_pb2.pyi +83 -0
- _frogml_proto/qwak/data_versioning/data_versioning_service_pb2_grpc.py +101 -0
- _frogml_proto/qwak/deployment/alert_pb2.py +131 -0
- _frogml_proto/qwak/deployment/alert_pb2.pyi +262 -0
- _frogml_proto/qwak/deployment/alert_pb2_grpc.py +4 -0
- _frogml_proto/qwak/deployment/alert_service_pb2.py +153 -0
- _frogml_proto/qwak/deployment/alert_service_pb2.pyi +218 -0
- _frogml_proto/qwak/deployment/alert_service_pb2_grpc.py +237 -0
- _frogml_proto/qwak/deployment/deployment_messages_pb2.py +36 -0
- _frogml_proto/qwak/deployment/deployment_messages_pb2.pyi +61 -0
- _frogml_proto/qwak/deployment/deployment_messages_pb2_grpc.py +4 -0
- _frogml_proto/qwak/deployment/deployment_pb2.py +644 -0
- _frogml_proto/qwak/deployment/deployment_pb2.pyi +1582 -0
- _frogml_proto/qwak/deployment/deployment_pb2_grpc.py +4 -0
- _frogml_proto/qwak/deployment/deployment_service_pb2.py +520 -0
- _frogml_proto/qwak/deployment/deployment_service_pb2.pyi +817 -0
- _frogml_proto/qwak/deployment/deployment_service_pb2_grpc.py +509 -0
- _frogml_proto/qwak/ecosystem/v0/credentials_pb2.py +56 -0
- _frogml_proto/qwak/ecosystem/v0/credentials_pb2.pyi +91 -0
- _frogml_proto/qwak/ecosystem/v0/credentials_pb2_grpc.py +4 -0
- _frogml_proto/qwak/ecosystem/v0/ecosystem_pb2.py +173 -0
- _frogml_proto/qwak/ecosystem/v0/ecosystem_pb2.pyi +332 -0
- _frogml_proto/qwak/ecosystem/v0/ecosystem_pb2_grpc.py +4 -0
- _frogml_proto/qwak/ecosystem/v0/ecosystem_runtime_service_pb2.py +264 -0
- _frogml_proto/qwak/ecosystem/v0/ecosystem_runtime_service_pb2.pyi +434 -0
- _frogml_proto/qwak/ecosystem/v0/ecosystem_runtime_service_pb2_grpc.py +238 -0
- _frogml_proto/qwak/execution/v1/backfill_pb2.py +97 -0
- _frogml_proto/qwak/execution/v1/backfill_pb2.pyi +167 -0
- _frogml_proto/qwak/execution/v1/backfill_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/batch_pb2.py +56 -0
- _frogml_proto/qwak/execution/v1/batch_pb2.pyi +76 -0
- _frogml_proto/qwak/execution/v1/batch_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/deletion_pb2.py +65 -0
- _frogml_proto/qwak/execution/v1/deletion_pb2.pyi +67 -0
- _frogml_proto/qwak/execution/v1/deletion_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/execution_pb2.py +96 -0
- _frogml_proto/qwak/execution/v1/execution_pb2.pyi +201 -0
- _frogml_proto/qwak/execution/v1/execution_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/execution_service_pb2.py +294 -0
- _frogml_proto/qwak/execution/v1/execution_service_pb2.pyi +431 -0
- _frogml_proto/qwak/execution/v1/execution_service_pb2_grpc.py +467 -0
- _frogml_proto/qwak/execution/v1/jobs/job_pb2.py +101 -0
- _frogml_proto/qwak/execution/v1/jobs/job_pb2.pyi +282 -0
- _frogml_proto/qwak/execution/v1/jobs/job_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/jobs/job_service_pb2.py +270 -0
- _frogml_proto/qwak/execution/v1/jobs/job_service_pb2.pyi +469 -0
- _frogml_proto/qwak/execution/v1/jobs/job_service_pb2_grpc.py +396 -0
- _frogml_proto/qwak/execution/v1/jobs/reports/report_pb2.py +109 -0
- _frogml_proto/qwak/execution/v1/jobs/reports/report_pb2.pyi +266 -0
- _frogml_proto/qwak/execution/v1/jobs/reports/report_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/state/execution_state_pb2.py +36 -0
- _frogml_proto/qwak/execution/v1/state/execution_state_pb2.pyi +45 -0
- _frogml_proto/qwak/execution/v1/state/execution_state_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/state/execution_state_service_pb2.py +150 -0
- _frogml_proto/qwak/execution/v1/state/execution_state_service_pb2.pyi +192 -0
- _frogml_proto/qwak/execution/v1/state/execution_state_service_pb2_grpc.py +237 -0
- _frogml_proto/qwak/execution/v1/state/featureset_state_pb2.py +35 -0
- _frogml_proto/qwak/execution/v1/state/featureset_state_pb2.pyi +37 -0
- _frogml_proto/qwak/execution/v1/state/featureset_state_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/state/spark_execution_state_pb2.py +91 -0
- _frogml_proto/qwak/execution/v1/state/spark_execution_state_pb2.pyi +156 -0
- _frogml_proto/qwak/execution/v1/state/spark_execution_state_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/streaming_aggregation_pb2.py +75 -0
- _frogml_proto/qwak/execution/v1/streaming_aggregation_pb2.pyi +95 -0
- _frogml_proto/qwak/execution/v1/streaming_aggregation_pb2_grpc.py +4 -0
- _frogml_proto/qwak/execution/v1/streaming_pb2.py +65 -0
- _frogml_proto/qwak/execution/v1/streaming_pb2.pyi +79 -0
- _frogml_proto/qwak/execution/v1/streaming_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/entities/entity_pb2.py +79 -0
- _frogml_proto/qwak/feature_store/entities/entity_pb2.pyi +166 -0
- _frogml_proto/qwak/feature_store/entities/entity_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/entities/entity_service_pb2.py +149 -0
- _frogml_proto/qwak/feature_store/entities/entity_service_pb2.pyi +203 -0
- _frogml_proto/qwak/feature_store/entities/entity_service_pb2_grpc.py +237 -0
- _frogml_proto/qwak/feature_store/features/aggregation_pb2.py +196 -0
- _frogml_proto/qwak/feature_store/features/aggregation_pb2.pyi +361 -0
- _frogml_proto/qwak/feature_store/features/aggregation_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/deployment_pb2.py +189 -0
- _frogml_proto/qwak/feature_store/features/deployment_pb2.pyi +379 -0
- _frogml_proto/qwak/feature_store/features/deployment_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/deployment_service_pb2.py +49 -0
- _frogml_proto/qwak/feature_store/features/deployment_service_pb2.pyi +45 -0
- _frogml_proto/qwak/feature_store/features/deployment_service_pb2_grpc.py +67 -0
- _frogml_proto/qwak/feature_store/features/execution_pb2.py +77 -0
- _frogml_proto/qwak/feature_store/features/execution_pb2.pyi +158 -0
- _frogml_proto/qwak/feature_store/features/execution_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/feature_set_attribute_pb2.py +36 -0
- _frogml_proto/qwak/feature_store/features/feature_set_attribute_pb2.pyi +33 -0
- _frogml_proto/qwak/feature_store/features/feature_set_attribute_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/feature_set_pb2.py +198 -0
- _frogml_proto/qwak/feature_store/features/feature_set_pb2.pyi +443 -0
- _frogml_proto/qwak/feature_store/features/feature_set_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/feature_set_service_pb2.py +467 -0
- _frogml_proto/qwak/feature_store/features/feature_set_service_pb2.pyi +663 -0
- _frogml_proto/qwak/feature_store/features/feature_set_service_pb2_grpc.py +746 -0
- _frogml_proto/qwak/feature_store/features/feature_set_state_pb2.py +203 -0
- _frogml_proto/qwak/feature_store/features/feature_set_state_pb2.pyi +329 -0
- _frogml_proto/qwak/feature_store/features/feature_set_state_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/feature_set_state_service_pb2.py +129 -0
- _frogml_proto/qwak/feature_store/features/feature_set_state_service_pb2.pyi +175 -0
- _frogml_proto/qwak/feature_store/features/feature_set_state_service_pb2_grpc.py +203 -0
- _frogml_proto/qwak/feature_store/features/feature_set_types_pb2.py +381 -0
- _frogml_proto/qwak/feature_store/features/feature_set_types_pb2.pyi +862 -0
- _frogml_proto/qwak/feature_store/features/feature_set_types_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/monitoring_pb2.py +55 -0
- _frogml_proto/qwak/feature_store/features/monitoring_pb2.pyi +67 -0
- _frogml_proto/qwak/feature_store/features/monitoring_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/features/real_time_feature_extractor_pb2.py +167 -0
- _frogml_proto/qwak/feature_store/features/real_time_feature_extractor_pb2.pyi +359 -0
- _frogml_proto/qwak/feature_store/features/real_time_feature_extractor_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/jobs/job_pb2.py +68 -0
- _frogml_proto/qwak/feature_store/jobs/job_pb2.pyi +169 -0
- _frogml_proto/qwak/feature_store/jobs/job_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/jobs/job_service_pb2.py +149 -0
- _frogml_proto/qwak/feature_store/jobs/job_service_pb2.pyi +237 -0
- _frogml_proto/qwak/feature_store/jobs/job_service_pb2_grpc.py +237 -0
- _frogml_proto/qwak/feature_store/jobs/v1/job_pb2.py +67 -0
- _frogml_proto/qwak/feature_store/jobs/v1/job_pb2.pyi +175 -0
- _frogml_proto/qwak/feature_store/jobs/v1/job_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/jobs/v1/job_service_pb2.py +210 -0
- _frogml_proto/qwak/feature_store/jobs/v1/job_service_pb2.pyi +349 -0
- _frogml_proto/qwak/feature_store/jobs/v1/job_service_pb2_grpc.py +338 -0
- _frogml_proto/qwak/feature_store/platform/platform_details_pb2.py +46 -0
- _frogml_proto/qwak/feature_store/platform/platform_details_pb2.pyi +54 -0
- _frogml_proto/qwak/feature_store/platform/platform_details_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/reports/report_pb2.py +75 -0
- _frogml_proto/qwak/feature_store/reports/report_pb2.pyi +159 -0
- _frogml_proto/qwak/feature_store/reports/report_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/repository/common/platform_pb2.py +55 -0
- _frogml_proto/qwak/feature_store/repository/common/platform_pb2.pyi +70 -0
- _frogml_proto/qwak/feature_store/repository/common/platform_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/serving/management_pb2.py +69 -0
- _frogml_proto/qwak/feature_store/serving/management_pb2.pyi +72 -0
- _frogml_proto/qwak/feature_store/serving/management_pb2_grpc.py +101 -0
- _frogml_proto/qwak/feature_store/serving/metadata_pb2.py +67 -0
- _frogml_proto/qwak/feature_store/serving/metadata_pb2.pyi +158 -0
- _frogml_proto/qwak/feature_store/serving/metadata_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/serving/serving_pb2.py +254 -0
- _frogml_proto/qwak/feature_store/serving/serving_pb2.pyi +476 -0
- _frogml_proto/qwak/feature_store/serving/serving_pb2_grpc.py +101 -0
- _frogml_proto/qwak/feature_store/serving/v1/value_pb2.py +46 -0
- _frogml_proto/qwak/feature_store/serving/v1/value_pb2.pyi +91 -0
- _frogml_proto/qwak/feature_store/serving/v1/value_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/sinks/sink_pb2.py +76 -0
- _frogml_proto/qwak/feature_store/sinks/sink_pb2.pyi +137 -0
- _frogml_proto/qwak/feature_store/sinks/sink_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/sources/batch_pb2.py +362 -0
- _frogml_proto/qwak/feature_store/sources/batch_pb2.pyi +911 -0
- _frogml_proto/qwak/feature_store/sources/batch_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/sources/data_source_attribute_pb2.py +36 -0
- _frogml_proto/qwak/feature_store/sources/data_source_attribute_pb2.pyi +33 -0
- _frogml_proto/qwak/feature_store/sources/data_source_attribute_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/sources/data_source_pb2.py +84 -0
- _frogml_proto/qwak/feature_store/sources/data_source_pb2.pyi +169 -0
- _frogml_proto/qwak/feature_store/sources/data_source_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/sources/data_source_service_pb2.py +169 -0
- _frogml_proto/qwak/feature_store/sources/data_source_service_pb2.pyi +237 -0
- _frogml_proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py +305 -0
- _frogml_proto/qwak/feature_store/sources/streaming_pb2.py +175 -0
- _frogml_proto/qwak/feature_store/sources/streaming_pb2.pyi +356 -0
- _frogml_proto/qwak/feature_store/sources/streaming_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.py +55 -0
- _frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.pyi +69 -0
- _frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2_grpc.py +4 -0
- _frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.py +68 -0
- _frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.pyi +71 -0
- _frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2_grpc.py +99 -0
- _frogml_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2.py +48 -0
- _frogml_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2.pyi +46 -0
- _frogml_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2_grpc.py +66 -0
- _frogml_proto/qwak/features_operator/v1/features_operator_pb2.py +76 -0
- _frogml_proto/qwak/features_operator/v1/features_operator_pb2.pyi +129 -0
- _frogml_proto/qwak/features_operator/v1/features_operator_pb2_grpc.py +4 -0
- _frogml_proto/qwak/features_operator/v1/features_operator_service_pb2.py +69 -0
- _frogml_proto/qwak/features_operator/v1/features_operator_service_pb2.pyi +98 -0
- _frogml_proto/qwak/features_operator/v1/features_operator_service_pb2_grpc.py +99 -0
- _frogml_proto/qwak/features_operator/v2/features_operator_pb2.py +86 -0
- _frogml_proto/qwak/features_operator/v2/features_operator_pb2.pyi +138 -0
- _frogml_proto/qwak/features_operator/v2/features_operator_pb2_grpc.py +4 -0
- _frogml_proto/qwak/features_operator/v2/features_operator_service_pb2.py +61 -0
- _frogml_proto/qwak/features_operator/v2/features_operator_service_pb2.pyi +89 -0
- _frogml_proto/qwak/features_operator/v2/features_operator_service_pb2_grpc.py +99 -0
- _frogml_proto/qwak/features_operator/v3/features_operator_async_service_pb2.py +132 -0
- _frogml_proto/qwak/features_operator/v3/features_operator_async_service_pb2.pyi +218 -0
- _frogml_proto/qwak/features_operator/v3/features_operator_async_service_pb2_grpc.py +165 -0
- _frogml_proto/qwak/features_operator/v3/features_operator_pb2.py +86 -0
- _frogml_proto/qwak/features_operator/v3/features_operator_pb2.pyi +143 -0
- _frogml_proto/qwak/features_operator/v3/features_operator_pb2_grpc.py +4 -0
- _frogml_proto/qwak/file_versioning/file_versioning_pb2.py +56 -0
- _frogml_proto/qwak/file_versioning/file_versioning_pb2.pyi +103 -0
- _frogml_proto/qwak/file_versioning/file_versioning_pb2_grpc.py +4 -0
- _frogml_proto/qwak/file_versioning/file_versioning_service_pb2.py +69 -0
- _frogml_proto/qwak/file_versioning/file_versioning_service_pb2.pyi +83 -0
- _frogml_proto/qwak/file_versioning/file_versioning_service_pb2_grpc.py +101 -0
- _frogml_proto/qwak/fitness_service/constructs_pb2.py +120 -0
- _frogml_proto/qwak/fitness_service/constructs_pb2.pyi +297 -0
- _frogml_proto/qwak/fitness_service/constructs_pb2_grpc.py +4 -0
- _frogml_proto/qwak/fitness_service/fitness_pb2.py +54 -0
- _frogml_proto/qwak/fitness_service/fitness_pb2.pyi +126 -0
- _frogml_proto/qwak/fitness_service/fitness_pb2_grpc.py +4 -0
- _frogml_proto/qwak/fitness_service/fitness_service_pb2.py +110 -0
- _frogml_proto/qwak/fitness_service/fitness_service_pb2.pyi +121 -0
- _frogml_proto/qwak/fitness_service/fitness_service_pb2_grpc.py +169 -0
- _frogml_proto/qwak/fitness_service/status_pb2.py +128 -0
- _frogml_proto/qwak/fitness_service/status_pb2.pyi +283 -0
- _frogml_proto/qwak/fitness_service/status_pb2_grpc.py +4 -0
- _frogml_proto/qwak/inference/feedback/feedback_pb2.py +118 -0
- _frogml_proto/qwak/inference/feedback/feedback_pb2.pyi +253 -0
- _frogml_proto/qwak/inference/feedback/feedback_pb2_grpc.py +100 -0
- _frogml_proto/qwak/instance_template/instance_template_pb2.py +62 -0
- _frogml_proto/qwak/instance_template/instance_template_pb2.pyi +108 -0
- _frogml_proto/qwak/instance_template/instance_template_pb2_grpc.py +4 -0
- _frogml_proto/qwak/instance_template/instance_template_service_pb2.py +69 -0
- _frogml_proto/qwak/instance_template/instance_template_service_pb2.pyi +79 -0
- _frogml_proto/qwak/instance_template/instance_template_service_pb2_grpc.py +101 -0
- _frogml_proto/qwak/integration/hugging_face_integration_pb2.py +85 -0
- _frogml_proto/qwak/integration/hugging_face_integration_pb2.pyi +106 -0
- _frogml_proto/qwak/integration/hugging_face_integration_pb2_grpc.py +4 -0
- _frogml_proto/qwak/integration/integration_pb2.py +153 -0
- _frogml_proto/qwak/integration/integration_pb2.pyi +357 -0
- _frogml_proto/qwak/integration/integration_pb2_grpc.py +4 -0
- _frogml_proto/qwak/integration/integration_service_pb2.py +169 -0
- _frogml_proto/qwak/integration/integration_service_pb2.pyi +234 -0
- _frogml_proto/qwak/integration/integration_service_pb2_grpc.py +271 -0
- _frogml_proto/qwak/integration/open_a_i_integration_pb2.py +85 -0
- _frogml_proto/qwak/integration/open_a_i_integration_pb2.pyi +109 -0
- _frogml_proto/qwak/integration/open_a_i_integration_pb2_grpc.py +4 -0
- _frogml_proto/qwak/integration/opsgenie_integration_pb2.py +105 -0
- _frogml_proto/qwak/integration/opsgenie_integration_pb2.pyi +168 -0
- _frogml_proto/qwak/integration/opsgenie_integration_pb2_grpc.py +4 -0
- _frogml_proto/qwak/integration/pagerduty_integration_pb2.py +127 -0
- _frogml_proto/qwak/integration/pagerduty_integration_pb2.pyi +195 -0
- _frogml_proto/qwak/integration/pagerduty_integration_pb2_grpc.py +4 -0
- _frogml_proto/qwak/integration/slack_app_integration_pb2.py +127 -0
- _frogml_proto/qwak/integration/slack_app_integration_pb2.pyi +198 -0
- _frogml_proto/qwak/integration/slack_app_integration_pb2_grpc.py +4 -0
- _frogml_proto/qwak/jfrog/gateway/v0/repository_pb2.py +95 -0
- _frogml_proto/qwak/jfrog/gateway/v0/repository_pb2.pyi +125 -0
- _frogml_proto/qwak/jfrog/gateway/v0/repository_pb2_grpc.py +4 -0
- _frogml_proto/qwak/jfrog/gateway/v0/repository_service_pb2.py +69 -0
- _frogml_proto/qwak/jfrog/gateway/v0/repository_service_pb2.pyi +71 -0
- _frogml_proto/qwak/jfrog/gateway/v0/repository_service_pb2_grpc.py +101 -0
- _frogml_proto/qwak/kube_deployment_captain/alert_pb2.py +120 -0
- _frogml_proto/qwak/kube_deployment_captain/alert_pb2.pyi +232 -0
- _frogml_proto/qwak/kube_deployment_captain/alert_pb2_grpc.py +4 -0
- _frogml_proto/qwak/kube_deployment_captain/alerting_pb2.py +65 -0
- _frogml_proto/qwak/kube_deployment_captain/alerting_pb2.pyi +101 -0
- _frogml_proto/qwak/kube_deployment_captain/alerting_pb2_grpc.py +4 -0
- _frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.py +357 -0
- _frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi +845 -0
- _frogml_proto/qwak/kube_deployment_captain/batch_job_pb2_grpc.py +4 -0
- _frogml_proto/qwak/kube_deployment_captain/deployment_pb2.py +182 -0
- _frogml_proto/qwak/kube_deployment_captain/deployment_pb2.pyi +482 -0
- _frogml_proto/qwak/kube_deployment_captain/deployment_pb2_grpc.py +4 -0
- _frogml_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2.py +172 -0
- _frogml_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2.pyi +370 -0
- _frogml_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2_grpc.py +4 -0
- _frogml_proto/qwak/kube_deployment_captain/kube_deployment_captain_service_pb2.py +562 -0
- _frogml_proto/qwak/kube_deployment_captain/kube_deployment_captain_service_pb2.pyi +924 -0
- _frogml_proto/qwak/kube_deployment_captain/kube_deployment_captain_service_pb2_grpc.py +1226 -0
- _frogml_proto/qwak/kube_deployment_captain/traffic_mapping_pb2.py +45 -0
- _frogml_proto/qwak/kube_deployment_captain/traffic_mapping_pb2.pyi +63 -0
- _frogml_proto/qwak/kube_deployment_captain/traffic_mapping_pb2_grpc.py +4 -0
- _frogml_proto/qwak/logging/log_filter_pb2.py +66 -0
- _frogml_proto/qwak/logging/log_filter_pb2.pyi +102 -0
- _frogml_proto/qwak/logging/log_filter_pb2_grpc.py +4 -0
- _frogml_proto/qwak/logging/log_line_pb2.py +48 -0
- _frogml_proto/qwak/logging/log_line_pb2.pyi +62 -0
- _frogml_proto/qwak/logging/log_line_pb2_grpc.py +4 -0
- _frogml_proto/qwak/logging/log_reader_service_pb2.py +51 -0
- _frogml_proto/qwak/logging/log_reader_service_pb2.pyi +83 -0
- _frogml_proto/qwak/logging/log_reader_service_pb2_grpc.py +67 -0
- _frogml_proto/qwak/logging/log_source_pb2.py +155 -0
- _frogml_proto/qwak/logging/log_source_pb2.pyi +321 -0
- _frogml_proto/qwak/logging/log_source_pb2_grpc.py +4 -0
- _frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2.py +96 -0
- _frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2.pyi +215 -0
- _frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2_grpc.py +4 -0
- _frogml_proto/qwak/models/models_pb2.py +427 -0
- _frogml_proto/qwak/models/models_pb2.pyi +851 -0
- _frogml_proto/qwak/models/models_pb2_grpc.py +402 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.py +149 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.pyi +183 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2_grpc.py +237 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_pb2.py +175 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_pb2.pyi +360 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_pb2_grpc.py +4 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_sync_service_pb2.py +70 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_sync_service_pb2.pyi +85 -0
- _frogml_proto/qwak/monitoring/v0/alerting_channel_sync_service_pb2_grpc.py +101 -0
- _frogml_proto/qwak/offline/serving/v1/feature_values_pb2.py +35 -0
- _frogml_proto/qwak/offline/serving/v1/feature_values_pb2.pyi +37 -0
- _frogml_proto/qwak/offline/serving/v1/feature_values_pb2_grpc.py +4 -0
- _frogml_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2.py +148 -0
- _frogml_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2.pyi +279 -0
- _frogml_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2_grpc.py +165 -0
- _frogml_proto/qwak/offline/serving/v1/options_pb2.py +35 -0
- _frogml_proto/qwak/offline/serving/v1/options_pb2.pyi +30 -0
- _frogml_proto/qwak/offline/serving/v1/options_pb2_grpc.py +4 -0
- _frogml_proto/qwak/offline/serving/v1/population_pb2.py +75 -0
- _frogml_proto/qwak/offline/serving/v1/population_pb2.pyi +99 -0
- _frogml_proto/qwak/offline/serving/v1/population_pb2_grpc.py +4 -0
- _frogml_proto/qwak/projects/projects_pb2.py +225 -0
- _frogml_proto/qwak/projects/projects_pb2.pyi +436 -0
- _frogml_proto/qwak/projects/projects_pb2_grpc.py +203 -0
- _frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.py +270 -0
- _frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.pyi +417 -0
- _frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2_grpc.py +441 -0
- _frogml_proto/qwak/prompt/v1/prompt/prompt_pb2.py +237 -0
- _frogml_proto/qwak/prompt/v1/prompt/prompt_pb2.pyi +415 -0
- _frogml_proto/qwak/prompt/v1/prompt/prompt_pb2_grpc.py +4 -0
- _frogml_proto/qwak/secret_service/secret_service_pb2.py +88 -0
- _frogml_proto/qwak/secret_service/secret_service_pb2.pyi +101 -0
- _frogml_proto/qwak/secret_service/secret_service_pb2_grpc.py +135 -0
- _frogml_proto/qwak/self_service/account/v0/account_membership_pb2.py +134 -0
- _frogml_proto/qwak/self_service/account/v0/account_membership_pb2.pyi +274 -0
- _frogml_proto/qwak/self_service/account/v0/account_membership_pb2_grpc.py +4 -0
- _frogml_proto/qwak/self_service/account/v0/account_membership_service_pb2.py +251 -0
- _frogml_proto/qwak/self_service/account/v0/account_membership_service_pb2.pyi +337 -0
- _frogml_proto/qwak/self_service/account/v0/account_membership_service_pb2_grpc.py +407 -0
- _frogml_proto/qwak/self_service/account/v0/account_status_pb2.py +93 -0
- _frogml_proto/qwak/self_service/account/v0/account_status_pb2.pyi +169 -0
- _frogml_proto/qwak/self_service/account/v0/account_status_pb2_grpc.py +4 -0
- _frogml_proto/qwak/self_service/account/v0/managing_account_pb2.py +155 -0
- _frogml_proto/qwak/self_service/account/v0/managing_account_pb2.pyi +279 -0
- _frogml_proto/qwak/self_service/account/v0/managing_account_pb2_grpc.py +4 -0
- _frogml_proto/qwak/self_service/account/v0/managing_account_service_pb2.py +110 -0
- _frogml_proto/qwak/self_service/account/v0/managing_account_service_pb2.pyi +136 -0
- _frogml_proto/qwak/self_service/account/v0/managing_account_service_pb2_grpc.py +169 -0
- _frogml_proto/qwak/self_service/user/v1/api_key_pb2.py +35 -0
- _frogml_proto/qwak/self_service/user/v1/api_key_pb2.pyi +30 -0
- _frogml_proto/qwak/self_service/user/v1/api_key_pb2_grpc.py +4 -0
- _frogml_proto/qwak/self_service/user/v1/user_pb2.py +37 -0
- _frogml_proto/qwak/self_service/user/v1/user_pb2.pyi +62 -0
- _frogml_proto/qwak/self_service/user/v1/user_pb2_grpc.py +4 -0
- _frogml_proto/qwak/self_service/user/v1/user_service_pb2.py +146 -0
- _frogml_proto/qwak/self_service/user/v1/user_service_pb2.pyi +209 -0
- _frogml_proto/qwak/self_service/user/v1/user_service_pb2_grpc.py +203 -0
- _frogml_proto/qwak/traffic/v1/traffic_api_pb2.py +109 -0
- _frogml_proto/qwak/traffic/v1/traffic_api_pb2.pyi +137 -0
- _frogml_proto/qwak/traffic/v1/traffic_api_pb2_grpc.py +169 -0
- _frogml_proto/qwak/traffic/v1/traffic_pb2.py +134 -0
- _frogml_proto/qwak/traffic/v1/traffic_pb2.pyi +285 -0
- _frogml_proto/qwak/traffic/v1/traffic_pb2_grpc.py +4 -0
- _frogml_proto/qwak/user_application/common/v0/resources_pb2.py +159 -0
- _frogml_proto/qwak/user_application/common/v0/resources_pb2.pyi +333 -0
- _frogml_proto/qwak/user_application/common/v0/resources_pb2_grpc.py +4 -0
- _frogml_proto/qwak/user_application/v0/user_application_pb2.py +69 -0
- _frogml_proto/qwak/user_application/v0/user_application_pb2.pyi +89 -0
- _frogml_proto/qwak/user_application/v0/user_application_pb2_grpc.py +4 -0
- _frogml_proto/qwak/vectors/v1/collection/collection_pb2.py +128 -0
- _frogml_proto/qwak/vectors/v1/collection/collection_pb2.pyi +287 -0
- _frogml_proto/qwak/vectors/v1/collection/collection_pb2_grpc.py +4 -0
- _frogml_proto/qwak/vectors/v1/collection/collection_service_pb2.py +189 -0
- _frogml_proto/qwak/vectors/v1/collection/collection_service_pb2.pyi +258 -0
- _frogml_proto/qwak/vectors/v1/collection/collection_service_pb2_grpc.py +304 -0
- _frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2.py +36 -0
- _frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2.pyi +41 -0
- _frogml_proto/qwak/vectors/v1/collection/event/collection_event_pb2_grpc.py +4 -0
- _frogml_proto/qwak/vectors/v1/filters_pb2.py +156 -0
- _frogml_proto/qwak/vectors/v1/filters_pb2.pyi +297 -0
- _frogml_proto/qwak/vectors/v1/filters_pb2_grpc.py +4 -0
- _frogml_proto/qwak/vectors/v1/vector_pb2.py +78 -0
- _frogml_proto/qwak/vectors/v1/vector_pb2.pyi +142 -0
- _frogml_proto/qwak/vectors/v1/vector_pb2_grpc.py +4 -0
- _frogml_proto/qwak/vectors/v1/vector_service_pb2.py +134 -0
- _frogml_proto/qwak/vectors/v1/vector_service_pb2.pyi +243 -0
- _frogml_proto/qwak/vectors/v1/vector_service_pb2_grpc.py +201 -0
- _frogml_proto/qwak/workspace/workspace_pb2.py +135 -0
- _frogml_proto/qwak/workspace/workspace_pb2.pyi +331 -0
- _frogml_proto/qwak/workspace/workspace_pb2_grpc.py +4 -0
- _frogml_proto/qwak/workspace/workspace_service_pb2.py +309 -0
- _frogml_proto/qwak/workspace/workspace_service_pb2.pyi +393 -0
- _frogml_proto/qwak/workspace/workspace_service_pb2_grpc.py +507 -0
- frogml/__init__.py +16 -0
- frogml/automations/__init__.py +70 -0
- frogml/automations/automation_executions.py +99 -0
- frogml/automations/automations.py +396 -0
- frogml/automations/batch_execution_action.py +336 -0
- frogml/automations/build_and_deploy_action.py +840 -0
- frogml/automations/common.py +106 -0
- frogml/clients/__init__.py +0 -0
- frogml/clients/_inner/__init__.py +0 -0
- frogml/clients/_inner/edge_communications.py +24 -0
- frogml/clients/administration/__init__.py +4 -0
- frogml/clients/administration/authenticated_user/__init__.py +0 -0
- frogml/clients/administration/authenticated_user/client.py +41 -0
- frogml/clients/administration/authentication/__init__.py +0 -0
- frogml/clients/administration/authentication/client.py +34 -0
- frogml/clients/administration/eco_system/__init__.py +0 -0
- frogml/clients/administration/eco_system/client.py +152 -0
- frogml/clients/administration/eco_system/eco_system_utils.py +15 -0
- frogml/clients/administration/environment/__init__.py +0 -0
- frogml/clients/administration/environment/client.py +77 -0
- frogml/clients/administration/self_service/__init__.py +0 -0
- frogml/clients/administration/self_service/client.py +73 -0
- frogml/clients/alert_management/__init__.py +1 -0
- frogml/clients/alert_management/client.py +64 -0
- frogml/clients/alerts_registry/__init__.py +1 -0
- frogml/clients/alerts_registry/channel.py +132 -0
- frogml/clients/alerts_registry/client.py +137 -0
- frogml/clients/analytics/__init__.py +1 -0
- frogml/clients/analytics/client.py +82 -0
- frogml/clients/audience/__init__.py +1 -0
- frogml/clients/audience/client.py +53 -0
- frogml/clients/automation_management/__init__.py +0 -0
- frogml/clients/automation_management/client.py +274 -0
- frogml/clients/autoscaling/__init__.py +1 -0
- frogml/clients/autoscaling/client.py +34 -0
- frogml/clients/batch_job_management/__init__.py +5 -0
- frogml/clients/batch_job_management/client.py +544 -0
- frogml/clients/batch_job_management/executions_config.py +201 -0
- frogml/clients/batch_job_management/results.py +73 -0
- frogml/clients/build_management/__init__.py +1 -0
- frogml/clients/build_management/client.py +115 -0
- frogml/clients/build_orchestrator/__init__.py +2 -0
- frogml/clients/build_orchestrator/build_model_request_getter.py +138 -0
- frogml/clients/build_orchestrator/client.py +456 -0
- frogml/clients/build_orchestrator/internal_client.py +114 -0
- frogml/clients/data_versioning/__init__.py +0 -0
- frogml/clients/data_versioning/client.py +76 -0
- frogml/clients/data_versioning/data_tag_filter.py +32 -0
- frogml/clients/deployment/__init__.py +0 -0
- frogml/clients/deployment/client.py +191 -0
- frogml/clients/feature_store/__init__.py +1 -0
- frogml/clients/feature_store/execution_management_client.py +102 -0
- frogml/clients/feature_store/job_registry_client.py +71 -0
- frogml/clients/feature_store/management_client.py +570 -0
- frogml/clients/feature_store/offline_serving_client.py +249 -0
- frogml/clients/feature_store/operator_client.py +164 -0
- frogml/clients/file_versioning/__init__.py +0 -0
- frogml/clients/file_versioning/client.py +77 -0
- frogml/clients/file_versioning/file_tag_filter.py +32 -0
- frogml/clients/instance_template/__init__.py +0 -0
- frogml/clients/instance_template/client.py +67 -0
- frogml/clients/integration_management/__init__.py +0 -0
- frogml/clients/integration_management/integration_manager_client.py +35 -0
- frogml/clients/integration_management/integration_utils.py +27 -0
- frogml/clients/integration_management/openai/__init__.py +0 -0
- frogml/clients/integration_management/openai/openai_system_secret.py +56 -0
- frogml/clients/jfrog_gateway/__init__.py +1 -0
- frogml/clients/jfrog_gateway/client.py +42 -0
- frogml/clients/kube_deployment_captain/__init__.py +1 -0
- frogml/clients/kube_deployment_captain/client.py +233 -0
- frogml/clients/logging_client/__init__.py +1 -0
- frogml/clients/logging_client/client.py +148 -0
- frogml/clients/model_management/__init__.py +1 -0
- frogml/clients/model_management/client.py +128 -0
- frogml/clients/model_version_manager/__init__.py +1 -0
- frogml/clients/model_version_manager/client.py +108 -0
- frogml/clients/project/__init__.py +0 -0
- frogml/clients/project/client.py +60 -0
- frogml/clients/prompt_manager/__init__.py +0 -0
- frogml/clients/prompt_manager/model_descriptor_mapper.py +196 -0
- frogml/clients/prompt_manager/prompt_manager_client.py +190 -0
- frogml/clients/prompt_manager/prompt_proto_mapper.py +262 -0
- frogml/clients/secret_service/__init__.py +1 -0
- frogml/clients/secret_service/client.py +96 -0
- frogml/clients/system_secret/__init__.py +0 -0
- frogml/clients/system_secret/system_secret_client.py +31 -0
- frogml/clients/user_application_instance/__init__.py +1 -0
- frogml/clients/user_application_instance/client.py +177 -0
- frogml/clients/vector_store/__init__.py +2 -0
- frogml/clients/vector_store/management_client.py +127 -0
- frogml/clients/vector_store/serving_client.py +157 -0
- frogml/clients/workspace_manager/__init__.py +1 -0
- frogml/clients/workspace_manager/client.py +224 -0
- frogml/exceptions/__init__.py +13 -0
- frogml/exceptions/frogml_decode_exception.py +7 -0
- frogml/exceptions/frogml_exception.py +7 -0
- frogml/exceptions/frogml_external_exception.py +11 -0
- frogml/exceptions/frogml_general_build_exception.py +13 -0
- frogml/exceptions/frogml_http_exception.py +19 -0
- frogml/exceptions/frogml_inference_exception.py +4 -0
- frogml/exceptions/frogml_load_configuration_exception.py +2 -0
- frogml/exceptions/frogml_load_model_failed_exception.py +10 -0
- frogml/exceptions/frogml_login_exception.py +8 -0
- frogml/exceptions/frogml_mock_http_exception.py +4 -0
- frogml/exceptions/frogml_model_initialization_exception.py +4 -0
- frogml/exceptions/frogml_not_found_exception.py +6 -0
- frogml/exceptions/frogml_remote_build_failed.py +5 -0
- frogml/exceptions/frogml_suggestion_exception.py +27 -0
- frogml/exceptions/quiet_error.py +22 -0
- frogml/feature_store/__init__.py +0 -0
- frogml/feature_store/_common/__init__.py +0 -0
- frogml/feature_store/_common/artifact_utils.py +57 -0
- frogml/feature_store/_common/feature_set_utils.py +261 -0
- frogml/feature_store/_common/featureset_asterisk_handler.py +115 -0
- frogml/feature_store/_common/functions.py +37 -0
- frogml/feature_store/_common/packaging.py +244 -0
- frogml/feature_store/_common/source_code_spec.py +62 -0
- frogml/feature_store/_common/source_code_spec_factory.py +51 -0
- frogml/feature_store/_common/value.py +10 -0
- frogml/feature_store/data_sources/__init__.py +65 -0
- frogml/feature_store/data_sources/attributes.py +25 -0
- frogml/feature_store/data_sources/base.py +121 -0
- frogml/feature_store/data_sources/batch/__init__.py +0 -0
- frogml/feature_store/data_sources/batch/_batch.py +9 -0
- frogml/feature_store/data_sources/batch/_jdbc.py +22 -0
- frogml/feature_store/data_sources/batch/athena.py +272 -0
- frogml/feature_store/data_sources/batch/big_query.py +80 -0
- frogml/feature_store/data_sources/batch/clickhouse.py +60 -0
- frogml/feature_store/data_sources/batch/csv.py +62 -0
- frogml/feature_store/data_sources/batch/elastic_search.py +61 -0
- frogml/feature_store/data_sources/batch/filesystem/__init__.py +0 -0
- frogml/feature_store/data_sources/batch/filesystem/aws.py +93 -0
- frogml/feature_store/data_sources/batch/filesystem/base_config.py +13 -0
- frogml/feature_store/data_sources/batch/filesystem/gcp.py +54 -0
- frogml/feature_store/data_sources/batch/filesystem/utils.py +59 -0
- frogml/feature_store/data_sources/batch/mongodb.py +58 -0
- frogml/feature_store/data_sources/batch/mysql.py +50 -0
- frogml/feature_store/data_sources/batch/parquet.py +57 -0
- frogml/feature_store/data_sources/batch/postgres.py +50 -0
- frogml/feature_store/data_sources/batch/redshift.py +85 -0
- frogml/feature_store/data_sources/batch/snowflake.py +77 -0
- frogml/feature_store/data_sources/batch/vertica.py +58 -0
- frogml/feature_store/data_sources/source_authentication.py +35 -0
- frogml/feature_store/data_sources/streaming/__init__.py +0 -0
- frogml/feature_store/data_sources/streaming/_streaming.py +9 -0
- frogml/feature_store/data_sources/streaming/kafka/__init__.py +25 -0
- frogml/feature_store/data_sources/streaming/kafka/authentication.py +124 -0
- frogml/feature_store/data_sources/streaming/kafka/deserialization.py +115 -0
- frogml/feature_store/data_sources/streaming/kafka/kafka.py +121 -0
- frogml/feature_store/data_sources/time_partition_columns.py +178 -0
- frogml/feature_store/entities/__init__.py +0 -0
- frogml/feature_store/entities/entity.py +73 -0
- frogml/feature_store/execution/__init__.py +0 -0
- frogml/feature_store/execution/backfill.py +197 -0
- frogml/feature_store/execution/execution.py +595 -0
- frogml/feature_store/execution/execution_query.py +91 -0
- frogml/feature_store/feature_sets/__init__.py +0 -0
- frogml/feature_store/feature_sets/_utils/_featureset_utils.py +45 -0
- frogml/feature_store/feature_sets/backfill.py +72 -0
- frogml/feature_store/feature_sets/base_feature_set.py +148 -0
- frogml/feature_store/feature_sets/batch.py +472 -0
- frogml/feature_store/feature_sets/context.py +13 -0
- frogml/feature_store/feature_sets/execution_spec.py +75 -0
- frogml/feature_store/feature_sets/metadata.py +75 -0
- frogml/feature_store/feature_sets/read_policies.py +245 -0
- frogml/feature_store/feature_sets/streaming.py +635 -0
- frogml/feature_store/feature_sets/streaming_backfill.py +266 -0
- frogml/feature_store/feature_sets/transformations/__init__.py +35 -0
- frogml/feature_store/feature_sets/transformations/aggregations/__init__.py +0 -0
- frogml/feature_store/feature_sets/transformations/aggregations/aggregations.py +443 -0
- frogml/feature_store/feature_sets/transformations/aggregations/windows.py +70 -0
- frogml/feature_store/feature_sets/transformations/functions/__init__.py +10 -0
- frogml/feature_store/feature_sets/transformations/functions/frogml_pandas.py +64 -0
- frogml/feature_store/feature_sets/transformations/functions/schema.py +57 -0
- frogml/feature_store/feature_sets/transformations/transformations.py +467 -0
- frogml/feature_store/feature_sets/transformations/validations/validations_util.py +78 -0
- frogml/feature_store/offline/__init__.py +4 -0
- frogml/feature_store/offline/_offline_serving_validations.py +32 -0
- frogml/feature_store/offline/_query_engine.py +32 -0
- frogml/feature_store/offline/athena/__init__.py +0 -0
- frogml/feature_store/offline/athena/athena_query_engine.py +154 -0
- frogml/feature_store/offline/client.py +719 -0
- frogml/feature_store/offline/client_v2.py +338 -0
- frogml/feature_store/offline/feature_set_features.py +25 -0
- frogml/feature_store/online/__init__.py +0 -0
- frogml/feature_store/online/client.py +352 -0
- frogml/feature_store/online/endpoint_utils.py +65 -0
- frogml/feature_store/sinks/__init__.py +0 -0
- frogml/feature_store/sinks/base.py +18 -0
- frogml/feature_store/sinks/kafka.py +59 -0
- frogml/feature_store/sinks/streaming/__init__.py +0 -0
- frogml/feature_store/sinks/streaming/attachment.py +34 -0
- frogml/feature_store/sinks/streaming/factory.py +60 -0
- frogml/feature_store/validations/__init__.py +0 -0
- frogml/feature_store/validations/validation_options.py +82 -0
- frogml/feature_store/validations/validation_response.py +121 -0
- frogml/feature_store/validations/validator.py +106 -0
- frogml/frogml_client/__init__.py +0 -0
- frogml/frogml_client/batch_jobs/execution.py +49 -0
- frogml/frogml_client/batch_jobs/task.py +43 -0
- frogml/frogml_client/build_api_helpers/__init__.py +0 -0
- frogml/frogml_client/build_api_helpers/build_api_steps.py +53 -0
- frogml/frogml_client/build_api_helpers/messages.py +6 -0
- frogml/frogml_client/build_api_helpers/trigger_build_api.py +65 -0
- frogml/frogml_client/builds/__init__.py +0 -0
- frogml/frogml_client/builds/build.py +97 -0
- frogml/frogml_client/builds/filters/__init__.py +0 -0
- frogml/frogml_client/builds/filters/metric_filter.py +38 -0
- frogml/frogml_client/builds/filters/parameter_filter.py +36 -0
- frogml/frogml_client/client.py +874 -0
- frogml/frogml_client/data_versioning/__init__.py +0 -0
- frogml/frogml_client/data_versioning/data_tag.py +24 -0
- frogml/frogml_client/deployments/__init__.py +0 -0
- frogml/frogml_client/deployments/deployment.py +409 -0
- frogml/frogml_client/file_versioning/__init__.py +0 -0
- frogml/frogml_client/file_versioning/file_tag.py +24 -0
- frogml/frogml_client/models/__init__.py +0 -0
- frogml/frogml_client/models/model.py +52 -0
- frogml/frogml_client/models/model_metadata.py +31 -0
- frogml/frogml_client/projects/__init__.py +0 -0
- frogml/frogml_client/projects/project.py +56 -0
- frogml/inner/__init__.py +11 -0
- frogml/inner/build_config/__init__.py +0 -0
- frogml/inner/build_config/build_config_v1.py +313 -0
- frogml/inner/build_logic/__init__.py +0 -0
- frogml/inner/build_logic/build_loggers/__init__.py +0 -0
- frogml/inner/build_logic/build_loggers/trigger_build_logger.py +50 -0
- frogml/inner/build_logic/constants/__init__.py +0 -0
- frogml/inner/build_logic/constants/dependencies.py +5 -0
- frogml/inner/build_logic/constants/host_resource.py +4 -0
- frogml/inner/build_logic/constants/messages.py +3 -0
- frogml/inner/build_logic/constants/temp_dir.py +1 -0
- frogml/inner/build_logic/constants/upload_tag.py +7 -0
- frogml/inner/build_logic/dependency_manager_type.py +8 -0
- frogml/inner/build_logic/execute_build_pipeline.py +64 -0
- frogml/inner/build_logic/interface/__init__.py +0 -0
- frogml/inner/build_logic/interface/build_logger_interface.py +28 -0
- frogml/inner/build_logic/interface/build_phase.py +24 -0
- frogml/inner/build_logic/interface/context_interface.py +56 -0
- frogml/inner/build_logic/interface/phase_run_handler.py +71 -0
- frogml/inner/build_logic/interface/step_inteface.py +26 -0
- frogml/inner/build_logic/interface/time_source.py +23 -0
- frogml/inner/build_logic/phases/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_model_step.py +41 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/common.py +32 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/fetch_strategy_manager.py +62 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/folder/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/folder/folder_strategy.py +131 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/git/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/git/git_strategy.py +184 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/strategy.py +48 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/zip/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/zip/zip_strategy.py +64 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/post_fetch_validation_step.py +139 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/pre_fetch_validation_step.py +263 -0
- frogml/inner/build_logic/phases/phase_010_fetch_model/set_version_step.py +33 -0
- frogml/inner/build_logic/phases/phase_020_remote_register_frogml_build/__init__.py +0 -0
- frogml/inner/build_logic/phases/phase_020_remote_register_frogml_build/cleanup_step.py +20 -0
- frogml/inner/build_logic/phases/phase_020_remote_register_frogml_build/start_remote_build_step.py +42 -0
- frogml/inner/build_logic/phases/phase_020_remote_register_frogml_build/upload_step.py +257 -0
- frogml/inner/build_logic/phases/phases_pipeline.py +44 -0
- frogml/inner/build_logic/run_handlers/__init__.py +0 -0
- frogml/inner/build_logic/run_handlers/programmatic_phase_run_handler.py +101 -0
- frogml/inner/build_logic/tools/__init__.py +0 -0
- frogml/inner/build_logic/tools/dependencies_tools.py +68 -0
- frogml/inner/build_logic/tools/files.py +237 -0
- frogml/inner/build_logic/tools/ignore_files.py +20 -0
- frogml/inner/build_logic/tools/text.py +9 -0
- frogml/inner/build_logic/trigger_build_context.py +10 -0
- frogml/inner/const.py +6 -0
- frogml/inner/di_configuration/__init__.py +73 -0
- frogml/inner/di_configuration/account.py +106 -0
- frogml/inner/di_configuration/config.yml +9 -0
- frogml/inner/di_configuration/containers.py +40 -0
- frogml/inner/instance_template/__init__.py +0 -0
- frogml/inner/instance_template/verify_template_id.py +48 -0
- frogml/inner/model_loggers_utils.py +110 -0
- frogml/inner/provider.py +6 -0
- frogml/inner/runtime_di/__init__.py +13 -0
- frogml/inner/runtime_di/containers.py +17 -0
- frogml/inner/singleton_meta.py +18 -0
- frogml/inner/tool/__init__.py +2 -0
- frogml/inner/tool/auth.py +70 -0
- frogml/inner/tool/grpc/__init__.py +0 -0
- frogml/inner/tool/grpc/grpc_auth.py +42 -0
- frogml/inner/tool/grpc/grpc_tools.py +209 -0
- frogml/inner/tool/grpc/grpc_try_wrapping.py +16 -0
- frogml/inner/tool/protobuf_factory.py +45 -0
- frogml/inner/tool/retry_utils.py +23 -0
- frogml/inner/tool/run_config/__init__.py +12 -0
- frogml/inner/tool/run_config/base.py +127 -0
- frogml/inner/tool/run_config/utils.py +316 -0
- frogml/llmops/__init__.py +0 -0
- frogml/llmops/generation/__init__.py +0 -0
- frogml/llmops/generation/_steaming.py +78 -0
- frogml/llmops/generation/base.py +5 -0
- frogml/llmops/generation/chat/__init__.py +0 -0
- frogml/llmops/generation/chat/openai/LICENSE.txt +201 -0
- frogml/llmops/generation/chat/openai/types/__init__.py +0 -0
- frogml/llmops/generation/chat/openai/types/chat/__init__.py +0 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion.py +88 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_assistant_message_param.py +65 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_chunk.py +153 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_content_part_text_param.py +28 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_function_call_option_param.py +25 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_function_message_param.py +33 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_message.py +56 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_message_param.py +34 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call.py +46 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_message_tool_call_param.py +44 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_named_tool_choice_param.py +32 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_role.py +20 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_system_message_param.py +35 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_token_logprob.py +71 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_tool_choice_option_param.py +28 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_tool_message_param.py +31 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_tool_param.py +29 -0
- frogml/llmops/generation/chat/openai/types/chat/chat_completion_user_message_param.py +35 -0
- frogml/llmops/generation/chat/openai/types/chat/completion_create_params.py +279 -0
- frogml/llmops/generation/chat/openai/types/completion_choice.py +47 -0
- frogml/llmops/generation/chat/openai/types/completion_create_params.py +209 -0
- frogml/llmops/generation/chat/openai/types/completion_usage.py +30 -0
- frogml/llmops/generation/chat/openai/types/model.py +35 -0
- frogml/llmops/generation/chat/openai/types/shared/__init__.py +3 -0
- frogml/llmops/generation/chat/openai/types/shared/error_object.py +27 -0
- frogml/llmops/generation/chat/openai/types/shared/function_definition.py +49 -0
- frogml/llmops/generation/chat/openai/types/shared/function_parameters.py +20 -0
- frogml/llmops/generation/chat/openai/types/shared_params/__init__.py +2 -0
- frogml/llmops/generation/chat/openai/types/shared_params/function_definition.py +49 -0
- frogml/llmops/generation/chat/openai/types/shared_params/function_parameters.py +22 -0
- frogml/llmops/generation/streaming.py +26 -0
- frogml/llmops/model/__init__.py +0 -0
- frogml/llmops/model/descriptor.py +40 -0
- frogml/llmops/prompt/__init__.py +0 -0
- frogml/llmops/prompt/base.py +136 -0
- frogml/llmops/prompt/chat/__init__.py +0 -0
- frogml/llmops/prompt/chat/message.py +24 -0
- frogml/llmops/prompt/chat/template.py +113 -0
- frogml/llmops/prompt/chat/value.py +10 -0
- frogml/llmops/prompt/manager.py +138 -0
- frogml/llmops/prompt/template.py +24 -0
- frogml/llmops/prompt/value.py +14 -0
- frogml/llmops/provider/__init__.py +0 -0
- frogml/llmops/provider/chat.py +44 -0
- frogml/llmops/provider/openai/__init__.py +0 -0
- frogml/llmops/provider/openai/client.py +126 -0
- frogml/llmops/provider/openai/provider.py +91 -0
- frogml/model/__init__.py +0 -0
- frogml/model/_entity_extraction.py +116 -0
- frogml/model/adapters/__init__.py +39 -0
- frogml/model/adapters/input_adapters/__init__.py +0 -0
- frogml/model/adapters/input_adapters/base_input_adapter.py +8 -0
- frogml/model/adapters/input_adapters/dataframe_input_adapter.py +11 -0
- frogml/model/adapters/input_adapters/file_input_adapter.py +8 -0
- frogml/model/adapters/input_adapters/image_input_adapter.py +8 -0
- frogml/model/adapters/input_adapters/json_input_adapter.py +8 -0
- frogml/model/adapters/input_adapters/multi_input_adapter.py +66 -0
- frogml/model/adapters/input_adapters/numpy_input_adapter.py +100 -0
- frogml/model/adapters/input_adapters/proto_input_adapter.py +32 -0
- frogml/model/adapters/input_adapters/string_input_adapter.py +8 -0
- frogml/model/adapters/input_adapters/tf_tensor_input_adapter.py +8 -0
- frogml/model/adapters/output_adapters/__init__.py +0 -0
- frogml/model/adapters/output_adapters/base_output_adapter.py +13 -0
- frogml/model/adapters/output_adapters/dataframe_output_adapter.py +12 -0
- frogml/model/adapters/output_adapters/default_output_adapter.py +9 -0
- frogml/model/adapters/output_adapters/json_output_adapter.py +9 -0
- frogml/model/adapters/output_adapters/numpy_output_adapter.py +32 -0
- frogml/model/adapters/output_adapters/proto_output_adapter.py +16 -0
- frogml/model/adapters/output_adapters/qwak_with_default_fallback.py +5 -0
- frogml/model/adapters/output_adapters/tf_tensor_output_adapter.py +9 -0
- frogml/model/analytics_logging.py +14 -0
- frogml/model/base.py +86 -0
- frogml/model/decorators/__init__.py +0 -0
- frogml/model/decorators/api.py +62 -0
- frogml/model/decorators/impl/__init__.py +0 -0
- frogml/model/decorators/impl/api_implementation.py +29 -0
- frogml/model/decorators/impl/timer_implementation.py +10 -0
- frogml/model/decorators/timer.py +20 -0
- frogml/model/fs_info_mapping_retriever.py +58 -0
- frogml/model/model_version_tracking.py +57 -0
- frogml/model/schema.py +52 -0
- frogml/model/schema_entities.py +107 -0
- frogml/model/tools/__init__.py +24 -0
- frogml/model/tools/adapters/__init__.py +0 -0
- frogml/model/tools/adapters/encoders.py +46 -0
- frogml/model/tools/adapters/input.py +66 -0
- frogml/model/tools/adapters/input_adapters/__init__.py +0 -0
- frogml/model/tools/adapters/input_adapters/base_input.py +21 -0
- frogml/model/tools/adapters/input_adapters/dataframe_input.py +48 -0
- frogml/model/tools/adapters/input_adapters/file_input.py +8 -0
- frogml/model/tools/adapters/input_adapters/image_input.py +45 -0
- frogml/model/tools/adapters/input_adapters/json_input.py +20 -0
- frogml/model/tools/adapters/input_adapters/string_input.py +6 -0
- frogml/model/tools/adapters/input_adapters/tf_tensor_input.py +44 -0
- frogml/model/tools/adapters/output.py +75 -0
- frogml/model/tools/adapters/output_adapters/__init__.py +0 -0
- frogml/model/tools/adapters/output_adapters/base_output.py +16 -0
- frogml/model/tools/adapters/output_adapters/dataframe_output.py +30 -0
- frogml/model/tools/adapters/output_adapters/default_output.py +58 -0
- frogml/model/tools/adapters/output_adapters/json_output.py +17 -0
- frogml/model/tools/adapters/output_adapters/tf_tensor_output.py +41 -0
- frogml/model/tools/run_model_locally.py +68 -0
- frogml/model/utils/__init__.py +0 -0
- frogml/model/utils/extract_wrapped_function.py +9 -0
- frogml/model/utils/feature_utils.py +69 -0
- frogml/model_loggers/__init__.py +0 -0
- frogml/model_loggers/artifact_logger.py +95 -0
- frogml/model_loggers/data_logger.py +172 -0
- frogml/model_loggers/model_logger.py +30 -0
- frogml/testing/__init__.py +0 -0
- frogml/testing/fixtures.py +7 -0
- frogml/tools/__init__.py +0 -0
- frogml/tools/logger/__init__.py +3 -0
- frogml/tools/logger/logger.py +287 -0
- frogml/tools/logger/logging.yml +79 -0
- frogml/utils/__init__.py +0 -0
- frogml/utils/datetime_utils.py +18 -0
- frogml/utils/dict_utils.py +12 -0
- frogml/vector_store/__init__.py +4 -0
- frogml/vector_store/client.py +151 -0
- frogml/vector_store/collection.py +429 -0
- frogml/vector_store/filters.py +359 -0
- frogml/vector_store/inference_client.py +105 -0
- frogml/vector_store/rest_helpers.py +81 -0
- frogml/vector_store/utils/__init__.py +0 -0
- frogml/vector_store/utils/filter_utils.py +23 -0
- frogml/vector_store/utils/upsert_utils.py +218 -0
- frogml_core-0.0.1.dist-info/METADATA +49 -0
- frogml_core-0.0.1.dist-info/RECORD +1013 -0
- frogml_core-0.0.1.dist-info/WHEEL +4 -0
- frogml_services_mock/__init__.py +1 -0
- frogml_services_mock/mocks/__init__.py +0 -0
- frogml_services_mock/mocks/alert_manager_service_api.py +56 -0
- frogml_services_mock/mocks/alert_registry_service_api.py +62 -0
- frogml_services_mock/mocks/analytics_api.py +54 -0
- frogml_services_mock/mocks/audience_service_api.py +79 -0
- frogml_services_mock/mocks/authentication_service.py +33 -0
- frogml_services_mock/mocks/automation_management_service.py +228 -0
- frogml_services_mock/mocks/autoscaling_service_api.py +31 -0
- frogml_services_mock/mocks/batch_job_manager_service.py +337 -0
- frogml_services_mock/mocks/build_management.py +101 -0
- frogml_services_mock/mocks/build_orchestrator_build_api.py +138 -0
- frogml_services_mock/mocks/build_orchestrator_build_settings_api.py +38 -0
- frogml_services_mock/mocks/build_orchestrator_service_api.py +132 -0
- frogml_services_mock/mocks/data_versioning_service.py +63 -0
- frogml_services_mock/mocks/deployment_management_service.py +461 -0
- frogml_services_mock/mocks/ecosystem_service_api.py +45 -0
- frogml_services_mock/mocks/execution_management_service.py +26 -0
- frogml_services_mock/mocks/feature_store_data_sources_manager_api.py +99 -0
- frogml_services_mock/mocks/feature_store_entities_manager_api.py +99 -0
- frogml_services_mock/mocks/feature_store_feature_set_manager_api.py +269 -0
- frogml_services_mock/mocks/features_online_serving_api.py +169 -0
- frogml_services_mock/mocks/features_operator_v3_service.py +29 -0
- frogml_services_mock/mocks/features_set_state_service_api.py +57 -0
- frogml_services_mock/mocks/feedback_service.py +35 -0
- frogml_services_mock/mocks/file_versioning_service.py +66 -0
- frogml_services_mock/mocks/frogml_mocks.py +133 -0
- frogml_services_mock/mocks/fs_offline_serving_service.py +53 -0
- frogml_services_mock/mocks/instance_template_management_service.py +150 -0
- frogml_services_mock/mocks/integration_management_service.py +73 -0
- frogml_services_mock/mocks/internal_build_orchestrator_service.py +27 -0
- frogml_services_mock/mocks/job_registry_service_api.py +74 -0
- frogml_services_mock/mocks/kube_captain_service_api.py +41 -0
- frogml_services_mock/mocks/logging_service.py +203 -0
- frogml_services_mock/mocks/model_management_service.py +115 -0
- frogml_services_mock/mocks/model_version_manager_service.py +100 -0
- frogml_services_mock/mocks/project_manager_service.py +96 -0
- frogml_services_mock/mocks/prompt_manager_service.py +281 -0
- frogml_services_mock/mocks/repository_service_mock.py +39 -0
- frogml_services_mock/mocks/secret_service.py +43 -0
- frogml_services_mock/mocks/self_service_user_service.py +31 -0
- frogml_services_mock/mocks/system_secret_service.py +55 -0
- frogml_services_mock/mocks/user_application_instance_service_api.py +107 -0
- frogml_services_mock/mocks/utils/__init__.py +0 -0
- frogml_services_mock/mocks/utils/exception_handlers.py +13 -0
- frogml_services_mock/mocks/vector_serving_api.py +159 -0
- frogml_services_mock/mocks/vectors_management_api.py +97 -0
- frogml_services_mock/mocks/workspace_manager_service_mock.py +202 -0
- frogml_services_mock/services_mock.py +542 -0
- frogml_services_mock/utils/__init__.py +0 -0
- frogml_services_mock/utils/service_utils.py +9 -0
@@ -0,0 +1,100 @@
|
|
1
|
+
import uuid
|
2
|
+
from datetime import datetime
|
3
|
+
from typing import Dict
|
4
|
+
|
5
|
+
from google.protobuf.timestamp_pb2 import Timestamp
|
6
|
+
|
7
|
+
from _frogml_proto.jfml.model_version.v1.model_repository_spec_pb2 import (
|
8
|
+
ModelRepositorySpec,
|
9
|
+
)
|
10
|
+
from _frogml_proto.jfml.model_version.v1.model_version_framework_pb2 import (
|
11
|
+
CatboostFramework,
|
12
|
+
)
|
13
|
+
from _frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import (
|
14
|
+
CreateModelVersionResponse,
|
15
|
+
CreateModelVersionRequest,
|
16
|
+
GetModelVersionByIdResponse,
|
17
|
+
GetModelVersionByIdRequest,
|
18
|
+
GetModelVersionByNameRequest,
|
19
|
+
GetModelVersionByNameResponse,
|
20
|
+
)
|
21
|
+
from _frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
|
22
|
+
ModelVersionManagerServiceServicer,
|
23
|
+
)
|
24
|
+
from _frogml_proto.jfml.model_version.v1.model_version_pb2 import (
|
25
|
+
ModelVersion,
|
26
|
+
ModelVersionCompletedStatus,
|
27
|
+
ModelVersionSpec,
|
28
|
+
)
|
29
|
+
|
30
|
+
timestamp = Timestamp()
|
31
|
+
timestamp.FromDatetime(datetime.now())
|
32
|
+
|
33
|
+
|
34
|
+
class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
|
35
|
+
def __init__(self):
|
36
|
+
super(ModelVersionManagerServiceMock, self).__init__()
|
37
|
+
self.model_versions: Dict[str, ModelVersion] = {}
|
38
|
+
|
39
|
+
def CreateModelVersion(self, request: CreateModelVersionRequest, context):
|
40
|
+
return CreateModelVersionResponse(
|
41
|
+
model_version=ModelVersion(
|
42
|
+
model_version_id=str(uuid.uuid4()),
|
43
|
+
spec=request.model_version,
|
44
|
+
status=ModelVersionCompletedStatus(),
|
45
|
+
created_by="mock",
|
46
|
+
created_at=timestamp,
|
47
|
+
updated_at=timestamp,
|
48
|
+
)
|
49
|
+
)
|
50
|
+
|
51
|
+
def GetModelVersionById(self, request: GetModelVersionByIdRequest, context):
|
52
|
+
return GetModelVersionByIdResponse(
|
53
|
+
model_version=ModelVersion(
|
54
|
+
model_version_id=request.model_version_id,
|
55
|
+
spec=ModelVersionSpec(
|
56
|
+
repository_spec=ModelRepositorySpec(
|
57
|
+
project_key="mock-project",
|
58
|
+
repository_key="mock-repository",
|
59
|
+
model_id=str(uuid.uuid4()),
|
60
|
+
),
|
61
|
+
name="mock-model-version-name",
|
62
|
+
framework=CatboostFramework(),
|
63
|
+
python_version="3.9",
|
64
|
+
),
|
65
|
+
status=ModelVersionCompletedStatus(),
|
66
|
+
created_by="mock",
|
67
|
+
created_at=timestamp,
|
68
|
+
updated_at=timestamp,
|
69
|
+
)
|
70
|
+
)
|
71
|
+
|
72
|
+
def GetModelVersionByName(self, request: GetModelVersionByNameRequest, context):
|
73
|
+
return GetModelVersionByNameResponse(
|
74
|
+
model_version=ModelVersion(
|
75
|
+
model_version_id=str(uuid.uuid4()),
|
76
|
+
spec=ModelVersionSpec(
|
77
|
+
repository_spec=ModelRepositorySpec(
|
78
|
+
project_key="mock-project",
|
79
|
+
repository_key="mock-repository",
|
80
|
+
model_id=request.model_id,
|
81
|
+
),
|
82
|
+
name=request.model_version_name,
|
83
|
+
framework=CatboostFramework(),
|
84
|
+
python_version="3.9",
|
85
|
+
),
|
86
|
+
status=ModelVersionCompletedStatus(),
|
87
|
+
created_by="mock",
|
88
|
+
created_at=timestamp,
|
89
|
+
updated_at=timestamp,
|
90
|
+
)
|
91
|
+
)
|
92
|
+
|
93
|
+
def InitPagination(self, request, context):
|
94
|
+
return super().InitPagination(request, context)
|
95
|
+
|
96
|
+
def ListModelVersions(self, request, context):
|
97
|
+
return super().ListModelVersions(request, context)
|
98
|
+
|
99
|
+
def DeleteModelVersionById(self, request, context):
|
100
|
+
return super().DeleteModelVersionById(request, context)
|
@@ -0,0 +1,96 @@
|
|
1
|
+
import uuid
|
2
|
+
from typing import Dict
|
3
|
+
|
4
|
+
import grpc
|
5
|
+
from google.protobuf.timestamp_pb2 import Timestamp
|
6
|
+
|
7
|
+
from _frogml_proto.qwak.projects.projects_pb2 import (
|
8
|
+
CreateProjectRequest,
|
9
|
+
CreateProjectResponse,
|
10
|
+
DeleteProjectRequest,
|
11
|
+
DeleteProjectResponse,
|
12
|
+
GetProjectRequest,
|
13
|
+
GetProjectResponse,
|
14
|
+
ListProjectsRequest,
|
15
|
+
ListProjectsResponse,
|
16
|
+
Project,
|
17
|
+
ProjectSpec,
|
18
|
+
)
|
19
|
+
from _frogml_proto.qwak.projects.projects_pb2_grpc import (
|
20
|
+
ProjectsManagementServiceServicer,
|
21
|
+
)
|
22
|
+
|
23
|
+
|
24
|
+
class ProjectManagerServiceMock(ProjectsManagementServiceServicer):
|
25
|
+
def __init__(self):
|
26
|
+
self.projects: Dict[str, ProjectSpec] = dict()
|
27
|
+
|
28
|
+
def ListProjects(
|
29
|
+
self, request: ListProjectsRequest, context
|
30
|
+
) -> ListProjectsResponse:
|
31
|
+
return ListProjectsResponse(
|
32
|
+
projects=[project for project in self.projects.values()]
|
33
|
+
)
|
34
|
+
|
35
|
+
def CreateProject(
|
36
|
+
self, request: CreateProjectRequest, context
|
37
|
+
) -> CreateProjectResponse:
|
38
|
+
if request.project_name in self.projects.keys():
|
39
|
+
description = f"Project with name {request.project_name} already exists"
|
40
|
+
context.set_code(grpc.StatusCode.ALREADY_EXISTS)
|
41
|
+
context.set_details(description)
|
42
|
+
raise Exception(description)
|
43
|
+
|
44
|
+
now = Timestamp()
|
45
|
+
now.GetCurrentTime()
|
46
|
+
|
47
|
+
project_spec = ProjectSpec(
|
48
|
+
project_id=str(uuid.uuid4()),
|
49
|
+
project_name=request.project_name,
|
50
|
+
project_description=request.project_description,
|
51
|
+
project_status=ProjectSpec.Status.ACTIVE,
|
52
|
+
created_at=now,
|
53
|
+
last_modified_at=now,
|
54
|
+
models_active=0,
|
55
|
+
models_count=0,
|
56
|
+
)
|
57
|
+
|
58
|
+
self.projects[request.project_name] = project_spec
|
59
|
+
|
60
|
+
return CreateProjectResponse(project=project_spec)
|
61
|
+
|
62
|
+
def GetProject(self, request: GetProjectRequest, context) -> GetProjectResponse:
|
63
|
+
matched_projects = list(
|
64
|
+
filter(
|
65
|
+
lambda project: request.project_id == project.project_id,
|
66
|
+
self.projects.values(),
|
67
|
+
)
|
68
|
+
)
|
69
|
+
|
70
|
+
if not matched_projects:
|
71
|
+
description = f"Project with id {request.project_name} was not found"
|
72
|
+
context.set_code(grpc.StatusCode.NOT_FOUND)
|
73
|
+
context.set_details(description)
|
74
|
+
raise Exception(description)
|
75
|
+
|
76
|
+
return GetProjectResponse(project=Project(spec=matched_projects[0]))
|
77
|
+
|
78
|
+
def DeleteProject(
|
79
|
+
self, request: DeleteProjectRequest, context
|
80
|
+
) -> DeleteProjectResponse:
|
81
|
+
matched_projects = list(
|
82
|
+
filter(
|
83
|
+
lambda project: request.project_id == project.project_id,
|
84
|
+
self.projects.values(),
|
85
|
+
)
|
86
|
+
)
|
87
|
+
|
88
|
+
if not matched_projects:
|
89
|
+
description = f"Project with id {request.project_name} was not found"
|
90
|
+
context.set_code(grpc.StatusCode.NOT_FOUND)
|
91
|
+
context.set_details(description)
|
92
|
+
raise Exception(description)
|
93
|
+
|
94
|
+
del self.projects[matched_projects[0].project_name]
|
95
|
+
|
96
|
+
return DeleteProjectResponse(info="Deleted")
|
@@ -0,0 +1,281 @@
|
|
1
|
+
from typing import Dict, List, Tuple
|
2
|
+
|
3
|
+
import grpc
|
4
|
+
from google.protobuf.timestamp_pb2 import Timestamp
|
5
|
+
|
6
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
7
|
+
CreatePromptRequest as ProtoCreatePromptRequest,
|
8
|
+
)
|
9
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
10
|
+
CreatePromptResponse as ProtoCreatePromptResponse,
|
11
|
+
)
|
12
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
13
|
+
CreatePromptVersionRequest as ProtoCreatePromptVersionRequest,
|
14
|
+
)
|
15
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
16
|
+
CreatePromptVersionResponse as ProtoCreatePromptVersionResponse,
|
17
|
+
)
|
18
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
19
|
+
DeletePromptRequest as ProtoDeletePromptRequest,
|
20
|
+
)
|
21
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
22
|
+
DeletePromptResponse as ProtoDeletePromptResponse,
|
23
|
+
)
|
24
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
25
|
+
DeletePromptVersionRequest as ProtoDeletePromptVersionRequest,
|
26
|
+
)
|
27
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
28
|
+
DeletePromptVersionResponse as ProtoDeletePromptVersionResponse,
|
29
|
+
)
|
30
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
31
|
+
GetPromptByNameRequest as ProtoGetPromptByNameRequest,
|
32
|
+
)
|
33
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
34
|
+
GetPromptByNameResponse as ProtoGetPromptByNameResponse,
|
35
|
+
)
|
36
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
37
|
+
GetPromptVersionByPromptNameRequest as ProtoGetPromptVersionByPromptNameRequest,
|
38
|
+
)
|
39
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
40
|
+
GetPromptVersionByPromptNameResponse as ProtoGetPromptVersionByPromptNameResponse,
|
41
|
+
)
|
42
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
43
|
+
SetDefaultPromptVersionRequest as ProtoSetDefaultPromptVersionRequest,
|
44
|
+
)
|
45
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
46
|
+
SetDefaultPromptVersionResponse as ProtoSetDefaultPromptVersionResponse,
|
47
|
+
)
|
48
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
49
|
+
UpdatePromptRequest as ProtoUpdatePromptRequest,
|
50
|
+
)
|
51
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
|
52
|
+
UpdatePromptResponse as ProtoUpdatePromptResponse,
|
53
|
+
)
|
54
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
|
55
|
+
PromptManagerServiceServicer,
|
56
|
+
)
|
57
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import Prompt as ProtoPrompt
|
58
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
|
59
|
+
PromptMetadata as ProtoPromptMetadata,
|
60
|
+
)
|
61
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import PromptSpec as ProtoPromptSpec
|
62
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
|
63
|
+
PromptVersion as ProtoPromptVersion,
|
64
|
+
)
|
65
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
|
66
|
+
PromptVersionDefinition as ProtoPromptVersionDefinition,
|
67
|
+
)
|
68
|
+
from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
|
69
|
+
PromptVersionMetadata as ProtoPromptVersionMetadata,
|
70
|
+
)
|
71
|
+
|
72
|
+
|
73
|
+
class PromptManagerServiceMock(PromptManagerServiceServicer):
|
74
|
+
prompts: Dict[str, ProtoPrompt] = {}
|
75
|
+
prompts_versions: Dict[Tuple[str, int], ProtoPromptVersionDefinition] = {}
|
76
|
+
prompts_next_version: Dict[str, int] = {}
|
77
|
+
|
78
|
+
def clear(self):
|
79
|
+
self.prompts = {}
|
80
|
+
self.prompts_versions = {}
|
81
|
+
self.prompts_next_version = {}
|
82
|
+
|
83
|
+
def _is_prompt_name_valid(self, test_str: str) -> bool:
|
84
|
+
import re
|
85
|
+
|
86
|
+
regex = r"^[a-z0-9](?:[-_]?[a-z0-9]+)+$"
|
87
|
+
return re.match(pattern=regex, string=test_str) is not None
|
88
|
+
|
89
|
+
def _raise_on_missing_prompt(self, prompt_name: str, context):
|
90
|
+
if prompt_name not in self.prompts:
|
91
|
+
context.set_code(grpc.StatusCode.NOT_FOUND)
|
92
|
+
context.set_details("Prompt name does not exist")
|
93
|
+
raise ValueError("Prompt name does not exist")
|
94
|
+
|
95
|
+
def _raise_on_missing_prompt_version(self, prompt_name: str, version: int, context):
|
96
|
+
if (
|
97
|
+
prompt_name not in self.prompts
|
98
|
+
or (prompt_name, version) not in self.prompts_versions
|
99
|
+
):
|
100
|
+
context.set_code(grpc.StatusCode.NOT_FOUND)
|
101
|
+
context.set_details("Prompt version does not exist")
|
102
|
+
raise ValueError("Prompt version does not exist")
|
103
|
+
|
104
|
+
def CreatePrompt(self, request: ProtoCreatePromptRequest, context):
|
105
|
+
"""Create a new prompt"""
|
106
|
+
|
107
|
+
if request.prompt_name in self.prompts:
|
108
|
+
context.set_code(grpc.StatusCode.ALREADY_EXISTS)
|
109
|
+
context.set_details("Prompt name already exists")
|
110
|
+
raise ValueError("Prompt name already exists")
|
111
|
+
|
112
|
+
if not self._is_prompt_name_valid(request.prompt_name):
|
113
|
+
context.set_code(grpc.StatusCode.INVALID_ARGUMENT)
|
114
|
+
context.set_details("Bad Prompt name")
|
115
|
+
raise ValueError("Bad Prompt name")
|
116
|
+
|
117
|
+
prompt_version_definition = ProtoPromptVersionDefinition(
|
118
|
+
version_number=1,
|
119
|
+
version_spec=request.prompt_version_spec,
|
120
|
+
version_metadata=ProtoPromptVersionMetadata(
|
121
|
+
created_by="nirh@frogml.com", created_at=Timestamp().GetCurrentTime()
|
122
|
+
),
|
123
|
+
)
|
124
|
+
|
125
|
+
prompt = ProtoPrompt(
|
126
|
+
name=request.prompt_name,
|
127
|
+
default_version_definition=prompt_version_definition,
|
128
|
+
metadata=ProtoPromptMetadata(
|
129
|
+
created_at=Timestamp().GetCurrentTime(), created_by="nirh@frogml.com"
|
130
|
+
),
|
131
|
+
prompt_spec=ProtoPromptSpec(description=request.prompt_spec.description),
|
132
|
+
)
|
133
|
+
|
134
|
+
self.prompts[prompt.name] = prompt
|
135
|
+
self.prompts_next_version[prompt.name] = 2
|
136
|
+
self.prompts_versions[
|
137
|
+
(prompt.name, prompt.default_version_definition.version_number)
|
138
|
+
] = prompt_version_definition
|
139
|
+
|
140
|
+
return ProtoCreatePromptResponse(prompt=prompt)
|
141
|
+
|
142
|
+
def CreatePromptVersion(self, request: ProtoCreatePromptVersionRequest, context):
|
143
|
+
"""
|
144
|
+
Create new prompt version.
|
145
|
+
Set new version in versions dict.
|
146
|
+
Bump prompt next version.
|
147
|
+
Set default version if needed
|
148
|
+
"""
|
149
|
+
self._raise_on_missing_prompt(prompt_name=request.prompt_name, context=context)
|
150
|
+
|
151
|
+
existing_prompt: ProtoPrompt = self.prompts[request.prompt_name]
|
152
|
+
new_prompt_version_number: int = self.prompts_next_version[request.prompt_name]
|
153
|
+
prompt_version_definition = ProtoPromptVersionDefinition(
|
154
|
+
version_number=new_prompt_version_number,
|
155
|
+
version_spec=request.prompt_version_spec,
|
156
|
+
version_metadata=ProtoPromptVersionMetadata(
|
157
|
+
created_by="nirh@frogml.com", created_at=Timestamp().GetCurrentTime()
|
158
|
+
),
|
159
|
+
)
|
160
|
+
|
161
|
+
self.prompts_next_version[request.prompt_name] = new_prompt_version_number + 1
|
162
|
+
|
163
|
+
if request.set_default:
|
164
|
+
existing_prompt.default_version_definition.CopyFrom(
|
165
|
+
prompt_version_definition
|
166
|
+
)
|
167
|
+
|
168
|
+
self.prompts_versions[
|
169
|
+
(request.prompt_name, new_prompt_version_number)
|
170
|
+
] = prompt_version_definition
|
171
|
+
return ProtoCreatePromptVersionResponse(
|
172
|
+
prompt_version=ProtoPromptVersion(
|
173
|
+
prompt_name=request.prompt_name,
|
174
|
+
prompt_version_definition=prompt_version_definition,
|
175
|
+
)
|
176
|
+
)
|
177
|
+
|
178
|
+
def UpdatePrompt(self, request: ProtoUpdatePromptRequest, context):
|
179
|
+
"""Update Prompt"""
|
180
|
+
self._raise_on_missing_prompt(prompt_name=request.prompt_name, context=context)
|
181
|
+
|
182
|
+
existing_prompt: ProtoPrompt = self.prompts[request.prompt_name]
|
183
|
+
existing_prompt.prompt_spec.CopyFrom(request.prompt_spec)
|
184
|
+
|
185
|
+
return ProtoUpdatePromptResponse()
|
186
|
+
|
187
|
+
def GetPromptByName(self, request: ProtoGetPromptByNameRequest, context):
|
188
|
+
"""Get prompt with its default version by name"""
|
189
|
+
self._raise_on_missing_prompt(prompt_name=request.prompt_name, context=context)
|
190
|
+
|
191
|
+
existing_prompt: ProtoPrompt = self.prompts[request.prompt_name]
|
192
|
+
return ProtoGetPromptByNameResponse(prompt=existing_prompt)
|
193
|
+
|
194
|
+
def GetPromptVersionByPromptName(
|
195
|
+
self, request: ProtoGetPromptVersionByPromptNameRequest, context
|
196
|
+
):
|
197
|
+
"""Get prompt version by name and optional version number"""
|
198
|
+
|
199
|
+
self._raise_on_missing_prompt(
|
200
|
+
prompt_name=request.prompt_name,
|
201
|
+
context=context,
|
202
|
+
)
|
203
|
+
|
204
|
+
self._raise_on_missing_prompt_version(
|
205
|
+
prompt_name=request.prompt_name,
|
206
|
+
version=request.version_number,
|
207
|
+
context=context,
|
208
|
+
)
|
209
|
+
|
210
|
+
return ProtoGetPromptVersionByPromptNameResponse(
|
211
|
+
prompt_version=ProtoPromptVersion(
|
212
|
+
prompt_name=request.prompt_name,
|
213
|
+
prompt_version_definition=self.prompts_versions[
|
214
|
+
(request.prompt_name, request.version_number)
|
215
|
+
],
|
216
|
+
)
|
217
|
+
)
|
218
|
+
|
219
|
+
def DeletePrompt(self, request: ProtoDeletePromptRequest, context):
|
220
|
+
"""Delete prompt"""
|
221
|
+
self._raise_on_missing_prompt(prompt_name=request.prompt_name, context=context)
|
222
|
+
|
223
|
+
del self.prompts[request.prompt_name]
|
224
|
+
del self.prompts_next_version[request.prompt_name]
|
225
|
+
|
226
|
+
versions_to_delete: List[int] = [
|
227
|
+
version
|
228
|
+
for (name, version) in self.prompts_versions.keys()
|
229
|
+
if name == request.prompt_name
|
230
|
+
]
|
231
|
+
|
232
|
+
for v in versions_to_delete:
|
233
|
+
del self.prompts_versions[(request.prompt_name, v)]
|
234
|
+
|
235
|
+
return ProtoDeletePromptResponse()
|
236
|
+
|
237
|
+
def DeletePromptVersion(self, request: ProtoDeletePromptVersionRequest, context):
|
238
|
+
"""Delete prompt version"""
|
239
|
+
self._raise_on_missing_prompt(
|
240
|
+
prompt_name=request.prompt_name,
|
241
|
+
context=context,
|
242
|
+
)
|
243
|
+
|
244
|
+
self._raise_on_missing_prompt_version(
|
245
|
+
prompt_name=request.prompt_name,
|
246
|
+
version=request.version_number,
|
247
|
+
context=context,
|
248
|
+
)
|
249
|
+
|
250
|
+
prompt_default_version: ProtoPrompt = self.prompts[request.prompt_name]
|
251
|
+
if (
|
252
|
+
prompt_default_version.default_version_definition.version_number
|
253
|
+
== request.version_number
|
254
|
+
):
|
255
|
+
context.set_code(grpc.StatusCode.FAILED_PRECONDITION)
|
256
|
+
context.set_details("Can't delete prompt default version")
|
257
|
+
raise ValueError("Can't delete prompt default version")
|
258
|
+
|
259
|
+
del self.prompts_versions[(request.prompt_name, request.version_number)]
|
260
|
+
return ProtoDeletePromptVersionResponse()
|
261
|
+
|
262
|
+
def SetDefaultPromptVersion(
|
263
|
+
self, request: ProtoSetDefaultPromptVersionRequest, context
|
264
|
+
):
|
265
|
+
"""Set default prompt version"""
|
266
|
+
self._raise_on_missing_prompt(
|
267
|
+
prompt_name=request.prompt_name,
|
268
|
+
context=context,
|
269
|
+
)
|
270
|
+
|
271
|
+
self._raise_on_missing_prompt_version(
|
272
|
+
prompt_name=request.prompt_name,
|
273
|
+
version=request.version_number,
|
274
|
+
context=context,
|
275
|
+
)
|
276
|
+
existing_prompt: ProtoPrompt = self.prompts[request.prompt_name]
|
277
|
+
prompt_version_definition: ProtoPromptVersionDefinition = self.prompts_versions[
|
278
|
+
(request.prompt_name, request.version_number)
|
279
|
+
]
|
280
|
+
existing_prompt.default_version_definition.CopyFrom(prompt_version_definition)
|
281
|
+
return ProtoSetDefaultPromptVersionResponse()
|
@@ -0,0 +1,39 @@
|
|
1
|
+
from datetime import datetime
|
2
|
+
from typing import Dict
|
3
|
+
|
4
|
+
from google.protobuf.timestamp_pb2 import Timestamp
|
5
|
+
|
6
|
+
from _frogml_proto.qwak.jfrog.gateway.v0.repository_pb2 import RepositorySpec
|
7
|
+
from _frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2 import (
|
8
|
+
GetRepositoryConfigurationResponse,
|
9
|
+
GetRepositoryConfigurationRequest,
|
10
|
+
)
|
11
|
+
from _frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2_grpc import (
|
12
|
+
RepositoryServiceServicer,
|
13
|
+
)
|
14
|
+
from frogml.exceptions import FrogmlNotFoundException
|
15
|
+
from frogml_services_mock.mocks.utils.exception_handlers import (
|
16
|
+
raise_not_found_grpc_error,
|
17
|
+
)
|
18
|
+
|
19
|
+
timestamp = Timestamp()
|
20
|
+
timestamp.FromDatetime(datetime.now())
|
21
|
+
|
22
|
+
|
23
|
+
class RepositoryServiceMock(RepositoryServiceServicer):
|
24
|
+
def __init__(self):
|
25
|
+
super(RepositoryServiceMock, self).__init__()
|
26
|
+
self.repositories: Dict[str, RepositorySpec] = {}
|
27
|
+
|
28
|
+
def GetRepositoryConfiguration(
|
29
|
+
self, request: GetRepositoryConfigurationRequest, context
|
30
|
+
) -> GetRepositoryConfigurationResponse:
|
31
|
+
repository_key: str = request.repository_key
|
32
|
+
|
33
|
+
if repository_key not in self.repositories:
|
34
|
+
raise_not_found_grpc_error(
|
35
|
+
context,
|
36
|
+
FrogmlNotFoundException(f"Repository {repository_key} not found"),
|
37
|
+
)
|
38
|
+
|
39
|
+
return self.repositories[repository_key]
|
@@ -0,0 +1,43 @@
|
|
1
|
+
import grpc
|
2
|
+
|
3
|
+
from _frogml_proto.qwak.secret_service.secret_service_pb2 import (
|
4
|
+
DeleteSecretRequest,
|
5
|
+
DeleteSecretResponse,
|
6
|
+
GetSecretRequest,
|
7
|
+
GetSecretResponse,
|
8
|
+
SetSecretRequest,
|
9
|
+
SetSecretResponse,
|
10
|
+
)
|
11
|
+
from _frogml_proto.qwak.secret_service.secret_service_pb2_grpc import (
|
12
|
+
SecretServiceServicer,
|
13
|
+
)
|
14
|
+
|
15
|
+
|
16
|
+
class SecretServiceMock(SecretServiceServicer):
|
17
|
+
secrets = {}
|
18
|
+
|
19
|
+
def SetSecret(self, request: SetSecretRequest, context) -> SetSecretResponse:
|
20
|
+
"""Set a value for a secret"""
|
21
|
+
self.secrets[request.name] = request.value
|
22
|
+
return SetSecretResponse()
|
23
|
+
|
24
|
+
def GetSecret(self, request: GetSecretRequest, context) -> GetSecretResponse:
|
25
|
+
"""Get a secret value"""
|
26
|
+
if request.name in self.secrets:
|
27
|
+
return GetSecretResponse(
|
28
|
+
name=request.name, value=self.secrets[request.name]
|
29
|
+
)
|
30
|
+
context.set_code(grpc.StatusCode.NOT_FOUND)
|
31
|
+
context.set_details("Secret not found")
|
32
|
+
return GetSecretResponse()
|
33
|
+
|
34
|
+
def DeleteSecret(
|
35
|
+
self, request: DeleteSecretRequest, context
|
36
|
+
) -> DeleteSecretResponse:
|
37
|
+
"""Remove a secret"""
|
38
|
+
if request.name in self.secrets:
|
39
|
+
del self.secrets[request.name]
|
40
|
+
return DeleteSecretResponse()
|
41
|
+
context.set_code(grpc.StatusCode.NOT_FOUND)
|
42
|
+
context.set_details("Secret not found")
|
43
|
+
return DeleteSecretResponse()
|
@@ -0,0 +1,31 @@
|
|
1
|
+
from typing import Dict
|
2
|
+
|
3
|
+
from _frogml_proto.qwak.self_service.user.v1.user_service_pb2 import (
|
4
|
+
GenerateApiKeyResponse,
|
5
|
+
RevokeApiKeyResponse,
|
6
|
+
)
|
7
|
+
from _frogml_proto.qwak.self_service.user.v1.user_service_pb2_grpc import (
|
8
|
+
UserServiceServicer,
|
9
|
+
)
|
10
|
+
|
11
|
+
API_KEY_FORMAT = "apikey-for-{}-{}"
|
12
|
+
|
13
|
+
|
14
|
+
class SelfServiceUserServiceMock(UserServiceServicer):
|
15
|
+
def __init__(self):
|
16
|
+
super(SelfServiceUserServiceMock, self).__init__()
|
17
|
+
self.apikeys: Dict[(str, str), str] = dict()
|
18
|
+
|
19
|
+
def RevokeApiKey(self, request, context) -> RevokeApiKeyResponse:
|
20
|
+
if self.apikeys.get((request.user_id, request.environment_id)):
|
21
|
+
self.apikeys.pop((request.user_id, request.environment_id))
|
22
|
+
return RevokeApiKeyResponse(status=RevokeApiKeyResponse.Status.REVOKED)
|
23
|
+
else:
|
24
|
+
return RevokeApiKeyResponse(
|
25
|
+
status=RevokeApiKeyResponse.Status.INVALID_USER_STATUS
|
26
|
+
)
|
27
|
+
|
28
|
+
def GenerateApiKey(self, request, context) -> GenerateApiKeyResponse:
|
29
|
+
api_key = API_KEY_FORMAT.format(request.user_id, request.environment_id)
|
30
|
+
self.apikeys[(request.user_id, request.environment_id)] = api_key
|
31
|
+
return GenerateApiKeyResponse(api_key=api_key)
|
@@ -0,0 +1,55 @@
|
|
1
|
+
from typing import Dict, Tuple
|
2
|
+
|
3
|
+
import grpc
|
4
|
+
|
5
|
+
from _frogml_proto.qwak.admiral.secret.v0.secret_pb2 import (
|
6
|
+
EnvironmentSecretIdentifier,
|
7
|
+
SetSystemEnvironmentSecretOptions,
|
8
|
+
SystemSecretDefinition,
|
9
|
+
)
|
10
|
+
from _frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2 import (
|
11
|
+
DeleteSystemSecretRequest,
|
12
|
+
DeleteSystemSecretResponse,
|
13
|
+
GetSystemSecretRequest,
|
14
|
+
GetSystemSecretResponse,
|
15
|
+
SetSystemSecretRequest,
|
16
|
+
SetSystemSecretResponse,
|
17
|
+
)
|
18
|
+
from _frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
|
19
|
+
SystemSecretServiceServicer,
|
20
|
+
)
|
21
|
+
|
22
|
+
|
23
|
+
class SystemSecretServiceMock(SystemSecretServiceServicer):
|
24
|
+
_secrets: Dict[Tuple[str, str], SystemSecretDefinition] = dict()
|
25
|
+
|
26
|
+
def clear(self):
|
27
|
+
self._secrets.clear()
|
28
|
+
|
29
|
+
def SetSystemSecret(self, request: SetSystemSecretRequest, context):
|
30
|
+
"""Set a value for a secret"""
|
31
|
+
|
32
|
+
options: SetSystemEnvironmentSecretOptions = request.options
|
33
|
+
identifier: EnvironmentSecretIdentifier = options.identifier
|
34
|
+
definition: SystemSecretDefinition = SystemSecretDefinition(
|
35
|
+
identifier=identifier, spec=options.spec
|
36
|
+
)
|
37
|
+
|
38
|
+
self._secrets[(identifier.name, identifier.environment_id)] = definition
|
39
|
+
return SetSystemSecretResponse()
|
40
|
+
|
41
|
+
def DeleteSystemSecret(self, request: DeleteSystemSecretRequest, context):
|
42
|
+
"""Delete secret"""
|
43
|
+
name: str = request.identifier.name
|
44
|
+
env_id: str = request.identifier.environment_id
|
45
|
+
|
46
|
+
if (name, env_id) in self._secrets:
|
47
|
+
del self._secrets[(name, env_id)]
|
48
|
+
|
49
|
+
return DeleteSystemSecretResponse()
|
50
|
+
|
51
|
+
def GetSystemSecret(self, request: GetSystemSecretRequest, context):
|
52
|
+
"""Get secret"""
|
53
|
+
name: str = request.identifier.name
|
54
|
+
env_id: str = request.identifier.environment_id
|
55
|
+
return GetSystemSecretResponse(secret_definition=self._secrets[(name, env_id)])
|