PyPI - frogml - Versions diffs - 1.2.37__py3-none-any.whl → 1.2.39__py3-none-any.whl - Mend

frogml 1.2.37py3-none-any.whl → 1.2.39py3-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.

Files changed (9) hide show

frogml/__init__.py +1 -1
frogml/_proto/qwak/models/models_pb2.py +81 -79
frogml/_proto/qwak/models/models_pb2.pyi +24 -0
frogml/_proto/qwak/models/models_pb2_grpc.py +34 -0
frogml/core/clients/batch_job_management/client.py +14 -6
frogml/core/clients/logging_client/client.py +118 -33
{frogml-1.2.37.dist-info → frogml-1.2.39.dist-info}/METADATA +1 -1
{frogml-1.2.37.dist-info → frogml-1.2.39.dist-info}/RECORD +9 -9
{frogml-1.2.37.dist-info → frogml-1.2.39.dist-info}/WHEEL +0 -0

frogml/__init__.py CHANGED Viewed

@@ -1,5 +1,5 @@
 __author__ = "jfrog"
-__version__ = "1.2.37"
+__version__ = "1.2.39"
 from frogml.sdk.model.decorators.api import api_decorator as api
 from frogml.sdk.model_loggers.model_logger import load_model, log_model

frogml/_proto/qwak/models/models_pb2.py CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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/batch_job_management/client.py CHANGED Viewed

@@ -71,7 +71,7 @@ from .results import (
     GetExecutionReportResult,
     StartExecutionResult,
 )
-from typing import List
+from typing import List, Optional
 CLIENT_TIMEOUT = 180  # Seconds
@@ -359,15 +359,21 @@ class BatchJobManagerClient:
         except grpc.RpcError as e:
             raise FrogmlException(f"Failed to cancel execution, error is {e}")
-    def get_execution_report(self, execution_id: str) -> GetExecutionReportResult:
+    def get_execution_report(
+        self,
+        execution_id: str,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
+    ) -> GetExecutionReportResult:
         """
         Args:
-            execution_id: the batch execution id to get the current status of
+            execution_id: The batch execution id to get the report for.
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
         Returns:
-            A full report of all the events that occurred as part of the execution job
+            A full report of all the events that occurred as part of the execution job.
         """
         try:
             self.logging_client = (
@@ -383,7 +389,9 @@ class BatchJobManagerClient:
             try:
                 execution_log_response: ReadLogsResponse = (
                     self.logging_client.read_execution_models_logs(
-                        execution_id=execution_id
+                        execution_id=execution_id,
+                        model_id=model_id,
+                        model_group_name=model_group_name,
                     )
                 )
                 execution_logs = execution_log_response.log_line

frogml/core/clients/logging_client/client.py CHANGED Viewed

@@ -49,15 +49,41 @@ class LoggingClient:
     def read_build_logs(
         self,
-        build_id=None,
-        before_offset=None,
-        after_offset=None,
-        max_number_of_results=None,
-        log_text_filter=None,
-    ):
+        build_id: Optional[str] = None,
+        before_offset: Optional[str] = None,
+        after_offset: Optional[str] = None,
+        max_number_of_results: Optional[int] = None,
+        log_text_filter: Optional[str] = None,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
+    ) -> ReadLogsResponse:
+        """
+        Read logs for a remote build.
+        Args:
+            build_id: The build ID to fetch logs for
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
+        Returns:
+            ReadLogsResponse containing the log lines
+        Raises:
+            FrogmlException: If reading logs fails
+        """
         try:
+            remote_build_source = RemoteBuildSource(
+                build_id=build_id,
+                model_id=model_id,
+                model_group_name=model_group_name,
+            )
             response = self.read_logs(
-                source=LogSource(remote_build=RemoteBuildSource(build_id=build_id)),
+                source=LogSource(remote_build=remote_build_source),
                 before_offset=before_offset,
                 after_offset=after_offset,
                 log_text_filter=log_text_filter,
@@ -70,20 +96,44 @@ class LoggingClient:
     def read_model_runtime_logs(
         self,
-        build_id=None,
-        deployment_id=None,
-        before_offset=None,
-        after_offset=None,
-        max_number_of_results=None,
-        log_text_filter=None,
+        build_id: Optional[str] = None,
+        deployment_id: Optional[str] = None,
+        before_offset: Optional[str] = None,
+        after_offset: Optional[str] = None,
+        max_number_of_results: Optional[int] = None,
+        log_text_filter: Optional[str] = None,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
     ) -> ReadLogsResponse:
+        """
+        Read logs for a model runtime (deployment).
+        Args:
+            build_id: The build ID to fetch logs for
+            deployment_id: The deployment ID to fetch logs for
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
+        Returns:
+            ReadLogsResponse containing the log lines
+        Raises:
+            FrogmlException: If reading logs fails
+        """
         try:
+            model_runtime_source = ModelRuntimeSource(
+                build_id=build_id,
+                deployment_id=deployment_id,
+                model_id=model_id,
+                model_group_name=model_group_name,
+            )
             response = self.read_logs(
-                source=LogSource(
-                    model_runtime=ModelRuntimeSource(
-                        build_id=build_id, deployment_id=deployment_id
-                    )
-                ),
+                source=LogSource(model_runtime=model_runtime_source),
                 before_offset=before_offset,
                 after_offset=after_offset,
                 log_text_filter=log_text_filter,
@@ -96,19 +146,41 @@ class LoggingClient:
     def read_execution_models_logs(
         self,
-        execution_id,
-        before_offset=None,
-        after_offset=None,
-        max_number_of_results=None,
-        log_text_filter=None,
+        execution_id: str,
+        before_offset: Optional[str] = None,
+        after_offset: Optional[str] = None,
+        max_number_of_results: Optional[int] = None,
+        log_text_filter: Optional[str] = None,
+        model_id: Optional[str] = None,
+        model_group_name: Optional[str] = None,
     ) -> ReadLogsResponse:
+        """
+        Read logs for a model inference execution (batch job).
+        Args:
+            execution_id: The execution/inference job ID to fetch logs for
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+            model_id: The model ID. This is the display name(identifier), not the UUID.
+            model_group_name: The model group name.
+        Returns:
+            ReadLogsResponse containing the log lines
+        Raises:
+            FrogmlException: If reading logs fails
+        """
         try:
+            inference_execution_source = InferenceExecutionSource(
+                inference_job_id=execution_id,
+                model_id=model_id,
+                model_group_name=model_group_name,
+            )
             response = self.read_logs(
-                source=LogSource(
-                    inference_execution=InferenceExecutionSource(
-                        inference_job_id=execution_id
-                    )
-                ),
+                source=LogSource(inference_execution=inference_execution_source),
                 before_offset=before_offset,
                 after_offset=after_offset,
                 log_text_filter=log_text_filter,
@@ -122,12 +194,25 @@ class LoggingClient:
     @grpc_try_catch_wrapper("Failed to read logs request")
     def read_logs(
         self,
-        source,
-        before_offset,
-        after_offset,
-        max_number_of_results,
-        log_text_filter,
+        source: LogSource,
+        before_offset: Optional[str],
+        after_offset: Optional[str],
+        max_number_of_results: Optional[int],
+        log_text_filter: Optional[str],
     ) -> ReadLogsResponse:
+        """
+        Low-level method to read logs from any source.
+        Args:
+            source: The log source to read from
+            before_offset: Optional offset to read logs before this point
+            after_offset: Optional offset to read logs after this point
+            max_number_of_results: Optional maximum number of log lines to return
+            log_text_filter: Optional text filter to search for in logs
+        Returns:
+            ReadLogsResponse containing the log lines
+        """
         response: ReadLogsResponse = self._logging_service.ReadLogs(
             ReadLogsRequest(
                 source=source,

{frogml-1.2.37.dist-info → frogml-1.2.39.dist-info}/METADATA RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml
-Version: 1.2.37
+Version: 1.2.39
 Summary: frogml contains the necessary objects and communication tools for using the JFrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml-1.2.37.dist-info → frogml-1.2.39.dist-info}/RECORD RENAMED Viewed

@@ -1,4 +1,4 @@
-frogml/__init__.py,sha256=ho5VcYluDFFuPfQz7VnMgeySBaT4a_YS4li3VUGFxqs,741
+frogml/__init__.py,sha256=R9rCscnmMzh7Gs7DDZeWQZkE-FFQRsoJIjhY6yVZQ6Q,741
 frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.py,sha256=pY7-QarLJWbL9uNmZ69RfE7IPVzwpVIhtd3A0wztSNY,1942
 frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.pyi,sha256=505I4uUrCmv2hsUw2H5eT-nlkNOCQOWodpvEkH2fHIo,3073
 frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -542,9 +542,9 @@ frogml/_proto/qwak/model_group/model_group_pb2_grpc.py,sha256=YgDt3DtagEdZ7UtxcP
 frogml/_proto/qwak/model_group/model_group_repository_details_pb2.py,sha256=BxaVjDayd9h5Va2ZId3pNPeJFAqVXb8TJHIWmYsURN4,3806
 frogml/_proto/qwak/model_group/model_group_repository_details_pb2.pyi,sha256=3oCsYmmfvBjGkA1nRcx2cg11WNr40uuR3gAWXC7Wz2I,7245
 frogml/_proto/qwak/model_group/model_group_repository_details_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml/_proto/qwak/models/models_pb2.py,sha256=XF3-e8xW62ymbg7xGRCwc4fNsgXV2UZJ1lPgv47Ylkk,19046
-frogml/_proto/qwak/models/models_pb2.pyi,sha256=m2NGA7K3Jz3YFp8nOW4l_AB5kgjJUEDmr3xMUPYu3Dc,42638
-frogml/_proto/qwak/models/models_pb2_grpc.py,sha256=iOfvU_txYazEBCR2rlFlD3BXLseuSjKzXqCcwfI60h4,23627
+frogml/_proto/qwak/models/models_pb2.py,sha256=jtw5KSbMwX5q6NVhfA5eblpJIANVavURyYvmXYvFVr0,19427
+frogml/_proto/qwak/models/models_pb2.pyi,sha256=c8e6l8fxCXXJmlyV9GSZ7PTThelw79sCYiEpMKaMRZE,43564
+frogml/_proto/qwak/models/models_pb2_grpc.py,sha256=7HZ5RwIk9I0MOHMOvZhvPFq_UIKwVdjZ79N3h-dlk8o,25406
 frogml/_proto/qwak/models/models_query_pb2.py,sha256=ZDEx_w4JgAL1q2wNOq5qkXiO5DctBgVRWbjrpOmOzbo,2880
 frogml/_proto/qwak/models/models_query_pb2.pyi,sha256=JEWkO0TXbGD2uKElgZB6VNPfYoq99GQ3Nj3jHObBQ2s,7490
 frogml/_proto/qwak/models/models_query_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -656,7 +656,7 @@ frogml/core/clients/automation_management/client.py,sha256=Dk6SsZGMLaVDLLzvFBysL
 frogml/core/clients/autoscaling/__init__.py,sha256=A-zuZOWaZf16NjnpX38204m730c5Lde491KaXneGirE,38
 frogml/core/clients/autoscaling/client.py,sha256=d1BoUH4hBiqoSR6ZM2_Ds6H7gV9HZPM2IIYNDbjPGgY,1299
 frogml/core/clients/batch_job_management/__init__.py,sha256=zywxYf2JupkIWnGVB2C6ugZs5CrOdxTPrksD5P6uQu4,211
-frogml/core/clients/batch_job_management/client.py,sha256=yym_GRJlB2qWnFf58EK2i7iD4llSj3FwmjdmIEdBSRc,22621
+frogml/core/clients/batch_job_management/client.py,sha256=TBN_9MkHXW342PLRepUckuQmA_gLPPrq5VavkkDtnCQ,22979
 frogml/core/clients/batch_job_management/executions_config.py,sha256=tyQbNiIAo3U4t_vyQp8nY17U0oQIyuE5GLW5WqhDxjA,7045
 frogml/core/clients/batch_job_management/results.py,sha256=nrFGZ3y538FBhej94CYV4cwqQssK1aUQOLGubYdQVrc,1846
 frogml/core/clients/build_orchestrator/__init__.py,sha256=ReW7Lbjws06MbVlCTRdsymDZiS2CKuczXIZ1xnPKSdg,105
@@ -691,7 +691,7 @@ frogml/core/clients/kube_deployment_captain/client.py,sha256=la4V7WsVL1z4zuLQtPZ
 frogml/core/clients/location_discovery/__init__.py,sha256=sqGQ75YHFE6nvOcir38fykUUmAa6cFEIze8PJYgYWRc,44
 frogml/core/clients/location_discovery/client.py,sha256=ZcBbbLqzZ9E7oLyamBTLTpp611LldhfFOSkzWroyIQg,2895
 frogml/core/clients/logging_client/__init__.py,sha256=1OCHnigQBYThBwGbxCreYA0BgP0HcuLFzNEWd3Yxh-c,34
-frogml/core/clients/logging_client/client.py,sha256=vq2Fv7UAEnrG910Npp1COxxWQ2_NWofbAStPkLhiz1w,4836
+frogml/core/clients/logging_client/client.py,sha256=ZTRRPrcM2cqrZX9T3DvIr0YICrhDK91P4CDP-ei_ZjQ,8599
 frogml/core/clients/model_group_management/__init__.py,sha256=f-nExP4vGWoWMUrRstY8QdtPqXnbsl2Ux0KfFkvGBkg,47
 frogml/core/clients/model_group_management/client.py,sha256=F8YCbfyphcljb8WxpeY7H7wnPpksyaKQ2ZOKi03RU6M,2016
 frogml/core/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
@@ -1147,6 +1147,6 @@ frogml_services_mock/mocks/utils/exception_handlers.py,sha256=k_8mez3cwjNjKE9yGQ
 frogml_services_mock/services_mock.py,sha256=y1_h189Ldu5MdVUW4nj-WMBtvUfkM_aKse7UfAb6Rxk,19419
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml-1.2.37.dist-info/METADATA,sha256=twB7XLyJSr6AgiZSI-mgf4EpcBJ54r9pw1kG1HU7LUw,5599
-frogml-1.2.37.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml-1.2.37.dist-info/RECORD,,
+frogml-1.2.39.dist-info/METADATA,sha256=H5aSy_IkGZt6boZicvagD2u8ca7pMhADbJ8SbXBnaOc,5599
+frogml-1.2.39.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml-1.2.39.dist-info/RECORD,,

{frogml-1.2.37.dist-info → frogml-1.2.39.dist-info}/WHEEL RENAMED Viewed

File without changes

frogml 1.2.37__py3-none-any.whl → 1.2.39__py3-none-any.whl

frogml 1.2.37py3-none-any.whl → 1.2.39py3-none-any.whl