frogml 1.2.38__py3-none-any.whl → 1.2.47__py3-none-any.whl

frogml/_proto/qwak/models/models_pb2.py

@@ -21,7 +21,7 @@ from frogml._proto.qwak.deployment import deployment_pb2 as qwak_dot_deployment_
 from frogml._proto.qwak.build.v1 import build_pb2 as qwak_dot_build_dot_v1_dot_build__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1eqwak/models/models_query.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"\x9b\x01\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\x12N\n\x12jfrog_project_spec\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"\xf0\x01\n\x1aListModelsPaginatedRequest\x12\x13\n\tpage_size\x18\x01 \x01(\x05H\x00\x12\x10\n\x06\x63ursor\x18\x02 \x01(\tH\x01\x12\x39\n\x0bsort_models\x18\x03 \x01(\x0b\x32\".qwak.models.management.SortModelsH\x02\x12\x30\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x1e.qwak.models.management.FilterH\x03\x42\x0f\n\rpage_size_optB\x0c\n\ncursor_optB\x11\n\x0fsort_models_optB\x0c\n\nfilter_opt\"\x8a\x01\n\x1bListModelsPaginatedResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\x12<\n\x08metadata\x18\x02 \x01(\x0b\x32*.qwak.models.management.PaginationMetadata\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01\"i\n CheckModelCreationAllowedRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x18\n\x10model_group_name\x18\x02 \x01(\t\x12\x19\n\x11jfrog_project_key\x18\x03 \x01(\t\"\x9b\x02\n!CheckModelCreationAllowedResponse\x12\x14\n\x0cis_creatable\x18\x01 \x01(\x08\x12;\n\x0cmodel_exists\x18\x02 \x01(\x0b\x32#.qwak.models.management.ModelExistsH\x00\x12\x42\n\x10model_not_exists\x18\x03 \x01(\x0b\x32&.qwak.models.management.ModelNotExistsH\x00\x12U\n\x1amodel_exists_with_mismatch\x18\x04 \x01(\x0b\x32/.qwak.models.management.ModelExistsWithMismatchH\x00\x42\x08\n\x06status\"\r\n\x0bModelExists\"\x10\n\x0eModelNotExists\"*\n\x17ModelExistsWithMismatch\x12\x0f\n\x07\x64\x65tails\x18\x01 \x03(\t*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xf3\x0b\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12~\n\x13ListModelsPaginated\x12\x32.qwak.models.management.ListModelsPaginatedRequest\x1a\x33.qwak.models.management.ListModelsPaginatedResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponse\x12\x90\x01\n\x19\x43heckModelCreationAllowed\x12\x38.qwak.models.management.CheckModelCreationAllowedRequest\x1a\x39.qwak.models.management.CheckModelCreationAllowedResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1eqwak/models/models_query.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"\x9b\x01\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\x12N\n\x12jfrog_project_spec\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"\xf0\x01\n\x1aListModelsPaginatedRequest\x12\x13\n\tpage_size\x18\x01 \x01(\x05H\x00\x12\x10\n\x06\x63ursor\x18\x02 \x01(\tH\x01\x12\x39\n\x0bsort_models\x18\x03 \x01(\x0b\x32\".qwak.models.management.SortModelsH\x02\x12\x30\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x1e.qwak.models.management.FilterH\x03\x42\x0f\n\rpage_size_optB\x0c\n\ncursor_optB\x11\n\x0fsort_models_optB\x0c\n\nfilter_opt\"\x8a\x01\n\x1bListModelsPaginatedResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\x12<\n\x08metadata\x18\x02 \x01(\x0b\x32*.qwak.models.management.PaginationMetadata\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\"E\n\x1bListModelsByModelIdsRequest\x12\x11\n\tmodel_ids\x18\x01 \x03(\t\x12\x13\n\x0bmodel_uuids\x18\x02 \x03(\t\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01\"i\n CheckModelCreationAllowedRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x18\n\x10model_group_name\x18\x02 \x01(\t\x12\x19\n\x11jfrog_project_key\x18\x03 \x01(\t\"\x9b\x02\n!CheckModelCreationAllowedResponse\x12\x14\n\x0cis_creatable\x18\x01 \x01(\x08\x12;\n\x0cmodel_exists\x18\x02 \x01(\x0b\x32#.qwak.models.management.ModelExistsH\x00\x12\x42\n\x10model_not_exists\x18\x03 \x01(\x0b\x32&.qwak.models.management.ModelNotExistsH\x00\x12U\n\x1amodel_exists_with_mismatch\x18\x04 \x01(\x0b\x32/.qwak.models.management.ModelExistsWithMismatchH\x00\x42\x08\n\x06status\"\r\n\x0bModelExists\"\x10\n\x0eModelNotExists\"*\n\x17ModelExistsWithMismatch\x12\x0f\n\x07\x64\x65tails\x18\x01 \x03(\t*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xec\x0c\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12~\n\x13ListModelsPaginated\x12\x32.qwak.models.management.ListModelsPaginatedRequest\x1a\x33.qwak.models.management.ListModelsPaginatedResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12w\n\x14ListModelsByModelIds\x12\x33.qwak.models.management.ListModelsByModelIdsRequest\x1a*.qwak.models.management.ListModelsResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponse\x12\x90\x01\n\x19\x43heckModelCreationAllowed\x12\x38.qwak.models.management.CheckModelCreationAllowedRequest\x1a\x39.qwak.models.management.CheckModelCreationAllowedResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -45,12 +45,12 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_MODELSMANAGEMENTSERVICE'].methods_by_name['DeleteBranch']._serialized_options = b'\210\002\001'
   _globals['_MODELSMANAGEMENTSERVICE'].methods_by_name['GetDeploymentDetails']._options = None
   _globals['_MODELSMANAGEMENTSERVICE'].methods_by_name['GetDeploymentDetails']._serialized_options = b'\210\002\001'
-  _globals['_DEPLOYMENTMODELSTATUS']._serialized_start=5370
-  _globals['_DEPLOYMENTMODELSTATUS']._serialized_end=5449
-  _globals['_MODELSTATUS']._serialized_start=5451
-  _globals['_MODELSTATUS']._serialized_end=5518
-  _globals['_MEMORYUNIT']._serialized_start=5520
-  _globals['_MEMORYUNIT']._serialized_end=5575
+  _globals['_DEPLOYMENTMODELSTATUS']._serialized_start=5441
+  _globals['_DEPLOYMENTMODELSTATUS']._serialized_end=5520
+  _globals['_MODELSTATUS']._serialized_start=5522
+  _globals['_MODELSTATUS']._serialized_end=5589
+  _globals['_MEMORYUNIT']._serialized_start=5591
+  _globals['_MEMORYUNIT']._serialized_end=5646
   _globals['_CREATEMODELREQUEST']._serialized_start=278
   _globals['_CREATEMODELREQUEST']._serialized_end=433
   _globals['_CREATEMODELRESPONSE']._serialized_start=435
@@ -67,76 +67,78 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_GETMODELREQUEST']._serialized_end=1043
   _globals['_GETMODELRESPONSE']._serialized_start=1045
   _globals['_GETMODELRESPONSE']._serialized_end=1109
-  _globals['_DELETEMODELREQUEST']._serialized_start=1111
-  _globals['_DELETEMODELREQUEST']._serialized_end=1169
-  _globals['_DELETEMODELRESPONSE']._serialized_start=1171
-  _globals['_DELETEMODELRESPONSE']._serialized_end=1192
-  _globals['_UPDATEMODELREQUEST']._serialized_start=1194
-  _globals['_UPDATEMODELREQUEST']._serialized_end=1307
-  _globals['_UPDATEMODELRESPONSE']._serialized_start=1309
-  _globals['_UPDATEMODELRESPONSE']._serialized_end=1385
-  _globals['_CREATEBRANCHREQUEST']._serialized_start=1387
-  _globals['_CREATEBRANCHREQUEST']._serialized_end=1447
-  _globals['_CREATEBRANCHRESPONSE']._serialized_start=1449
-  _globals['_CREATEBRANCHRESPONSE']._serialized_end=1490
-  _globals['_DELETEBRANCHREQUEST']._serialized_start=1492
-  _globals['_DELETEBRANCHREQUEST']._serialized_end=1532
-  _globals['_DELETEBRANCHRESPONSE']._serialized_start=1534
-  _globals['_DELETEBRANCHRESPONSE']._serialized_end=1556
-  _globals['_ISMODELEXISTSREQUEST']._serialized_start=1558
-  _globals['_ISMODELEXISTSREQUEST']._serialized_end=1636
-  _globals['_ISMODELEXISTSRESPONSE']._serialized_start=1638
-  _globals['_ISMODELEXISTSRESPONSE']._serialized_end=1677
-  _globals['_MODELSPEC']._serialized_start=1680
-  _globals['_MODELSPEC']._serialized_end=1898
-  _globals['_MODELBRIEF']._serialized_start=1901
-  _globals['_MODELBRIEF']._serialized_end=2228
-  _globals['_MODELSPECCONFIGURATION']._serialized_start=2230
-  _globals['_MODELSPECCONFIGURATION']._serialized_end=2281
-  _globals['_MODEL']._serialized_start=2284
-  _globals['_MODEL']._serialized_end=2912
-  _globals['_MODELBUILDTYPE']._serialized_start=2915
-  _globals['_MODELBUILDTYPE']._serialized_end=3065
-  _globals['_SOURCE']._serialized_start=3067
-  _globals['_SOURCE']._serialized_end=3075
-  _globals['_MODELREGISTRYIMAGE']._serialized_start=3077
-  _globals['_MODELREGISTRYIMAGE']._serialized_end=3128
-  _globals['_BRANCH']._serialized_start=3131
-  _globals['_BRANCH']._serialized_end=3366
-  _globals['_BRANCH_BRANCHSTATUS']._serialized_start=3298
-  _globals['_BRANCH_BRANCHSTATUS']._serialized_end=3366
-  _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_start=3368
-  _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_end=3434
-  _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_start=3436
-  _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_end=3537
-  _globals['_DEPLOYMENTDETAILS']._serialized_start=3540
-  _globals['_DEPLOYMENTDETAILS']._serialized_end=3815
-  _globals['_LISTMODELSMETADATAREQUEST']._serialized_start=3817
-  _globals['_LISTMODELSMETADATAREQUEST']._serialized_end=3864
-  _globals['_LISTMODELSMETADATARESPONSE']._serialized_start=3866
-  _globals['_LISTMODELSMETADATARESPONSE']._serialized_end=3957
-  _globals['_GETMODELMETADATAREQUEST']._serialized_start=3959
-  _globals['_GETMODELMETADATAREQUEST']._serialized_end=4002
-  _globals['_GETMODELMETADATARESPONSE']._serialized_start=4004
-  _globals['_GETMODELMETADATARESPONSE']._serialized_end=4093
-  _globals['_ENVIRONMENTROUTES']._serialized_start=4095
-  _globals['_ENVIRONMENTROUTES']._serialized_end=4199
-  _globals['_MODELMETADATA']._serialized_start=4202
-  _globals['_MODELMETADATA']._serialized_end=4898
-  _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_start=4675
-  _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_end=4780
-  _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_start=4782
-  _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_end=4898
-  _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_start=4900
-  _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_end=5005
-  _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_start=5008
-  _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_end=5291
-  _globals['_MODELEXISTS']._serialized_start=5293
-  _globals['_MODELEXISTS']._serialized_end=5306
-  _globals['_MODELNOTEXISTS']._serialized_start=5308
-  _globals['_MODELNOTEXISTS']._serialized_end=5324
-  _globals['_MODELEXISTSWITHMISMATCH']._serialized_start=5326
-  _globals['_MODELEXISTSWITHMISMATCH']._serialized_end=5368
-  _globals['_MODELSMANAGEMENTSERVICE']._serialized_start=5578
-  _globals['_MODELSMANAGEMENTSERVICE']._serialized_end=7101
+  _globals['_LISTMODELSBYMODELIDSREQUEST']._serialized_start=1111
+  _globals['_LISTMODELSBYMODELIDSREQUEST']._serialized_end=1180
+  _globals['_DELETEMODELREQUEST']._serialized_start=1182
+  _globals['_DELETEMODELREQUEST']._serialized_end=1240
+  _globals['_DELETEMODELRESPONSE']._serialized_start=1242
+  _globals['_DELETEMODELRESPONSE']._serialized_end=1263
+  _globals['_UPDATEMODELREQUEST']._serialized_start=1265
+  _globals['_UPDATEMODELREQUEST']._serialized_end=1378
+  _globals['_UPDATEMODELRESPONSE']._serialized_start=1380
+  _globals['_UPDATEMODELRESPONSE']._serialized_end=1456
+  _globals['_CREATEBRANCHREQUEST']._serialized_start=1458
+  _globals['_CREATEBRANCHREQUEST']._serialized_end=1518
+  _globals['_CREATEBRANCHRESPONSE']._serialized_start=1520
+  _globals['_CREATEBRANCHRESPONSE']._serialized_end=1561
+  _globals['_DELETEBRANCHREQUEST']._serialized_start=1563
+  _globals['_DELETEBRANCHREQUEST']._serialized_end=1603
+  _globals['_DELETEBRANCHRESPONSE']._serialized_start=1605
+  _globals['_DELETEBRANCHRESPONSE']._serialized_end=1627
+  _globals['_ISMODELEXISTSREQUEST']._serialized_start=1629
+  _globals['_ISMODELEXISTSREQUEST']._serialized_end=1707
+  _globals['_ISMODELEXISTSRESPONSE']._serialized_start=1709
+  _globals['_ISMODELEXISTSRESPONSE']._serialized_end=1748
+  _globals['_MODELSPEC']._serialized_start=1751
+  _globals['_MODELSPEC']._serialized_end=1969
+  _globals['_MODELBRIEF']._serialized_start=1972
+  _globals['_MODELBRIEF']._serialized_end=2299
+  _globals['_MODELSPECCONFIGURATION']._serialized_start=2301
+  _globals['_MODELSPECCONFIGURATION']._serialized_end=2352
+  _globals['_MODEL']._serialized_start=2355
+  _globals['_MODEL']._serialized_end=2983
+  _globals['_MODELBUILDTYPE']._serialized_start=2986
+  _globals['_MODELBUILDTYPE']._serialized_end=3136
+  _globals['_SOURCE']._serialized_start=3138
+  _globals['_SOURCE']._serialized_end=3146
+  _globals['_MODELREGISTRYIMAGE']._serialized_start=3148
+  _globals['_MODELREGISTRYIMAGE']._serialized_end=3199
+  _globals['_BRANCH']._serialized_start=3202
+  _globals['_BRANCH']._serialized_end=3437
+  _globals['_BRANCH_BRANCHSTATUS']._serialized_start=3369
+  _globals['_BRANCH_BRANCHSTATUS']._serialized_end=3437
+  _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_start=3439
+  _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_end=3505
+  _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_start=3507
+  _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_end=3608
+  _globals['_DEPLOYMENTDETAILS']._serialized_start=3611
+  _globals['_DEPLOYMENTDETAILS']._serialized_end=3886
+  _globals['_LISTMODELSMETADATAREQUEST']._serialized_start=3888
+  _globals['_LISTMODELSMETADATAREQUEST']._serialized_end=3935
+  _globals['_LISTMODELSMETADATARESPONSE']._serialized_start=3937
+  _globals['_LISTMODELSMETADATARESPONSE']._serialized_end=4028
+  _globals['_GETMODELMETADATAREQUEST']._serialized_start=4030
+  _globals['_GETMODELMETADATAREQUEST']._serialized_end=4073
+  _globals['_GETMODELMETADATARESPONSE']._serialized_start=4075
+  _globals['_GETMODELMETADATARESPONSE']._serialized_end=4164
+  _globals['_ENVIRONMENTROUTES']._serialized_start=4166
+  _globals['_ENVIRONMENTROUTES']._serialized_end=4270
+  _globals['_MODELMETADATA']._serialized_start=4273
+  _globals['_MODELMETADATA']._serialized_end=4969
+  _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_start=4746
+  _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_end=4851
+  _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_start=4853
+  _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_end=4969
+  _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_start=4971
+  _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_end=5076
+  _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_start=5079
+  _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_end=5362
+  _globals['_MODELEXISTS']._serialized_start=5364
+  _globals['_MODELEXISTS']._serialized_end=5377
+  _globals['_MODELNOTEXISTS']._serialized_start=5379
+  _globals['_MODELNOTEXISTS']._serialized_end=5395
+  _globals['_MODELEXISTSWITHMISMATCH']._serialized_start=5397
+  _globals['_MODELEXISTSWITHMISMATCH']._serialized_end=5439
+  _globals['_MODELSMANAGEMENTSERVICE']._serialized_start=5649
+  _globals['_MODELSMANAGEMENTSERVICE']._serialized_end=7293
 # @@protoc_insertion_point(module_scope)

frogml/_proto/qwak/models/models_pb2.pyi

@@ -260,6 +260,30 @@ class GetModelResponse(google.protobuf.message.Message):
 
 global___GetModelResponse = GetModelResponse
 
+@typing.final
+class ListModelsByModelIdsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_IDS_FIELD_NUMBER: builtins.int
+    MODEL_UUIDS_FIELD_NUMBER: builtins.int
+    @property
+    def model_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """The model ids"""
+
+    @property
+    def model_uuids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """The model uuids"""
+
+    def __init__(
+        self,
+        *,
+        model_ids: collections.abc.Iterable[builtins.str] | None = ...,
+        model_uuids: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["model_ids", b"model_ids", "model_uuids", b"model_uuids"]) -> None: ...
+
+global___ListModelsByModelIdsRequest = ListModelsByModelIdsRequest
+
 @typing.final
 class DeleteModelRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor

frogml/_proto/qwak/models/models_pb2_grpc.py

@@ -34,6 +34,11 @@ class ModelsManagementServiceStub(object):
                 request_serializer=qwak_dot_models_dot_models__pb2.GetModelRequest.SerializeToString,
                 response_deserializer=qwak_dot_models_dot_models__pb2.GetModelResponse.FromString,
                 )
+        self.ListModelsByModelIds = channel.unary_unary(
+                '/qwak.models.management.ModelsManagementService/ListModelsByModelIds',
+                request_serializer=qwak_dot_models_dot_models__pb2.ListModelsByModelIdsRequest.SerializeToString,
+                response_deserializer=qwak_dot_models_dot_models__pb2.ListModelsResponse.FromString,
+                )
         self.DeleteModel = channel.unary_unary(
                 '/qwak.models.management.ModelsManagementService/DeleteModel',
                 request_serializer=qwak_dot_models_dot_models__pb2.DeleteModelRequest.SerializeToString,
@@ -112,6 +117,13 @@ class ModelsManagementServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListModelsByModelIds(self, request, context):
+        """Get models by model IDs and model UUIDs
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def DeleteModel(self, request, context):
         """Delete model
         """
@@ -193,6 +205,11 @@ def add_ModelsManagementServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_models_dot_models__pb2.GetModelRequest.FromString,
                     response_serializer=qwak_dot_models_dot_models__pb2.GetModelResponse.SerializeToString,
             ),
+            'ListModelsByModelIds': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListModelsByModelIds,
+                    request_deserializer=qwak_dot_models_dot_models__pb2.ListModelsByModelIdsRequest.FromString,
+                    response_serializer=qwak_dot_models_dot_models__pb2.ListModelsResponse.SerializeToString,
+            ),
             'DeleteModel': grpc.unary_unary_rpc_method_handler(
                     servicer.DeleteModel,
                     request_deserializer=qwak_dot_models_dot_models__pb2.DeleteModelRequest.FromString,
@@ -316,6 +333,23 @@ class ModelsManagementService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def ListModelsByModelIds(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.models.management.ModelsManagementService/ListModelsByModelIds',
+            qwak_dot_models_dot_models__pb2.ListModelsByModelIdsRequest.SerializeToString,
+            qwak_dot_models_dot_models__pb2.ListModelsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def DeleteModel(request,
             target,

frogml/core/clients/administration/cluster_v2/__init__.py

@@ -0,0 +1 @@
+from .client import ClusterV2Client

frogml/core/clients/administration/cluster_v2/client.py

@@ -0,0 +1,122 @@
+"""Cluster V2 API client.
+
+Provides wrapper methods for the Cluster V2 gRPC service with proper error handling.
+"""
+
+from frogml._proto.qwak.administration.cluster.v2.cluster_pb2 import (
+    ClusterSpec,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2 import (
+    CreateClusterCreationRequestRequest,
+    CreateClusterCreationRequestResponse,
+    DeleteClusterRequest,
+    DeleteClusterResponse,
+    GetClusterRequest,
+    GetClusterResponse,
+    GetClusterStateRequest,
+    GetClusterStateResponse,
+    ListClustersRequest,
+    ListClustersResponse,
+    UpdateClusterConfigurationRequest,
+    UpdateClusterConfigurationResponse,
+)
+from frogml._proto.qwak.administration.cluster.v2.cluster_service_pb2_grpc import (
+    ClusterServiceStub,
+)
+from dependency_injector.wiring import Provide
+from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
+
+
+class ClusterV2Client:
+    """Client for interacting with the Cluster V2 API.
+
+    This client wraps the gRPC stub and provides methods with proper error handling
+    using the grpc_try_catch_wrapper decorator.
+    """
+
+    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
+        self.__cluster_service = ClusterServiceStub(grpc_channel)
+
+    @grpc_try_catch_wrapper("Failed to create cluster")
+    def create_cluster(
+        self, cluster_spec: ClusterSpec
+    ) -> CreateClusterCreationRequestResponse:
+        """Create a new cluster.
+
+        Args:
+            cluster_spec: The cluster specification.
+
+        Returns:
+            CreateClusterCreationRequestResponse with the request_id.
+        """
+        request = CreateClusterCreationRequestRequest(cluster_spec=cluster_spec)
+        return self.__cluster_service.CreateClusterCreationRequest(request)
+
+    @grpc_try_catch_wrapper("Failed to get cluster state for request {request_id}")
+    def get_cluster_state(self, request_id: str) -> GetClusterStateResponse:
+        """Get cluster state by request ID.
+
+        Args:
+            request_id: The request ID from create_cluster.
+
+        Returns:
+            GetClusterStateResponse with the cluster state.
+        """
+        request = GetClusterStateRequest(request_id=request_id)
+        return self.__cluster_service.GetClusterState(request)
+
+    @grpc_try_catch_wrapper("Failed to get cluster {cluster_id}")
+    def get_cluster(self, cluster_id: str) -> GetClusterResponse:
+        """Get cluster by ID.
+
+        Args:
+            cluster_id: The cluster ID.
+
+        Returns:
+            GetClusterResponse with the cluster state.
+        """
+        request = GetClusterRequest(cluster_id=cluster_id)
+        return self.__cluster_service.GetCluster(request)
+
+    @grpc_try_catch_wrapper("Failed to list clusters")
+    def list_clusters(self) -> ListClustersResponse:
+        """List all clusters.
+
+        Returns:
+            ListClustersResponse with list of clusters.
+        """
+        request = ListClustersRequest()
+        return self.__cluster_service.ListClusters(request)
+
+    @grpc_try_catch_wrapper("Failed to update cluster {cluster_id}")
+    def update_cluster(
+        self, cluster_id: str, cluster_spec: ClusterSpec
+    ) -> UpdateClusterConfigurationResponse:
+        """Update cluster configuration.
+
+        Args:
+            cluster_id: The cluster ID to update.
+            cluster_spec: The updated cluster specification.
+
+        Returns:
+            UpdateClusterConfigurationResponse with the request_id.
+        """
+        request = UpdateClusterConfigurationRequest(
+            cluster_id=cluster_id,
+            cluster_spec=cluster_spec,
+        )
+        return self.__cluster_service.UpdateClusterConfiguration(request)
+
+    @grpc_try_catch_wrapper("Failed to delete cluster {cluster_id}")
+    def delete_cluster(self, cluster_id: str) -> DeleteClusterResponse:
+        """Delete cluster by ID.
+
+        Args:
+            cluster_id: The cluster ID to delete.
+
+        Returns:
+            DeleteClusterResponse with the request_id.
+        """
+        request = DeleteClusterRequest(cluster_id=cluster_id)
+        return self.__cluster_service.DeleteCluster(request)

frogml/core/clients/administration/environment/client.py

@@ -4,17 +4,23 @@ from dependency_injector.wiring import Provide
 from frogml._proto.qwak.administration.v0.environments.environment_pb2 import (
     QwakEnvironmentStatus,
 )
+from frogml._proto.qwak.administration.v0.environments.configuration_pb2 import (
+    QwakEnvironmentConfiguration,
+)
 from frogml._proto.qwak.administration.v0.environments.environment_service_pb2 import (
     GetEnvironmentApplicationUserCredentialsRequest,
     GetEnvironmentApplicationUserCredentialsResponse,
     ListEnvironmentsRequest,
     ListEnvironmentsResponse,
+    UpdateEnvironmentConfigurationRequest,
+    UpdateEnvironmentConfigurationResponse,
 )
 from frogml._proto.qwak.administration.v0.environments.environment_service_pb2_grpc import (
     EnvironmentServiceStub,
 )
 from frogml.core.exceptions import FrogmlException
 from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
 
 
 class EnvironmentClient:
@@ -75,3 +81,25 @@ class EnvironmentClient:
             raise FrogmlException(
                 f"Failed to get application user, error is {e.details()}"
             )
+
+    @grpc_try_catch_wrapper(
+        "Failed to update environment configuration for {environment_id}"
+    )
+    def update_environment_configuration(
+        self, environment_id: str, configuration: QwakEnvironmentConfiguration
+    ) -> UpdateEnvironmentConfigurationResponse:
+        """
+        Update environment configuration.
+
+        Args:
+            environment_id: The environment ID to update.
+            configuration: The new environment configuration.
+
+        Returns:
+            UpdateEnvironmentConfigurationResponse
+        """
+        request = UpdateEnvironmentConfigurationRequest(
+            environment_id=environment_id,
+            configuration=configuration,
+        )
+        return self._environment_service.UpdateEnvironmentConfiguration(request)

frogml/core/clients/administration/environment_v1/__init__.py

@@ -0,0 +1 @@
+from .client import EnvironmentV1Client

frogml/core/clients/administration/environment_v1/client.py

@@ -0,0 +1,53 @@
+"""Environment V1 API client.
+
+Provides wrapper methods for the Environment V1 gRPC service with proper error handling.
+"""
+
+from frogml._proto.qwak.administration.v1.environments.environment_pb2 import (
+    EnvironmentRuntimeConfigSpec,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2 import (
+    CreateEnvironmentRequest,
+    CreateEnvironmentResponse,
+)
+from frogml._proto.qwak.administration.v1.environments.environment_service_pb2_grpc import (
+    EnvironmentServiceStub,
+)
+from dependency_injector.wiring import Provide
+from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
+
+
+class EnvironmentV1Client:
+    """Client for interacting with the Environment V1 API.
+
+    This client wraps the gRPC stub and provides methods with proper error handling
+    using the grpc_try_catch_wrapper decorator.
+    """
+
+    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
+        self.__environment_service = EnvironmentServiceStub(grpc_channel)
+
+    @grpc_try_catch_wrapper("Failed to create environment {environment_name}")
+    def create_environment(
+        self,
+        environment_name: str,
+        cluster_id: str,
+        spec: EnvironmentRuntimeConfigSpec,
+    ) -> CreateEnvironmentResponse:
+        """Create a new environment.
+
+        Args:
+            environment_name: The name of the environment.
+            cluster_id: The cluster ID the environment belongs to.
+            spec: The environment runtime configuration specification.
+
+        Returns:
+            CreateEnvironmentResponse with the created environment.
+        """
+        request = CreateEnvironmentRequest(
+            environment_name=environment_name,
+            cluster_id=cluster_id,
+            spec=spec,
+        )
+        return self.__environment_service.CreateEnvironment(request)

frogml/core/clients/model_deployment_manager/__init__.py

@@ -0,0 +1 @@
+from .client import ModelDeploymentManagerClient