frogml-core 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

frogml_services_mock/mocks/model_version_manager_service.py

@@ -4,13 +4,13 @@ from typing import Dict
 
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from _frogml_proto.jfml.model_version.v1.model_repository_spec_pb2 import (
+from frogml_proto.jfml.model_version.v1.model_repository_spec_pb2 import (
     ModelRepositorySpec,
 )
-from _frogml_proto.jfml.model_version.v1.model_version_framework_pb2 import (
+from frogml_proto.jfml.model_version.v1.model_version_framework_pb2 import (
     CatboostFramework,
 )
-from _frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import (
+from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import (
     CreateModelVersionResponse,
     CreateModelVersionRequest,
     GetModelVersionByIdResponse,
@@ -18,13 +18,13 @@ from _frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 impor
     GetModelVersionByNameRequest,
     GetModelVersionByNameResponse,
 )
-from _frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
+from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     ModelVersionManagerServiceServicer,
 )
-from _frogml_proto.jfml.model_version.v1.model_version_pb2 import (
+from frogml_proto.jfml.model_version.v1.model_version_pb2 import (
     ModelVersion,
     ModelVersionCompletedStatus,
-    ModelVersionSpec,
+    ModelVersionSpec, ModelVersionStatus,
 )
 
 timestamp = Timestamp()
@@ -41,7 +41,7 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
             model_version=ModelVersion(
                 model_version_id=str(uuid.uuid4()),
                 spec=request.model_version,
-                status=ModelVersionCompletedStatus(),
+                status=ModelVersionStatus(completed=ModelVersionCompletedStatus()),
                 created_by="mock",
                 created_at=timestamp,
                 updated_at=timestamp,
@@ -62,7 +62,7 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
                     framework=CatboostFramework(),
                     python_version="3.9",
                 ),
-                status=ModelVersionCompletedStatus(),
+                status=ModelVersionStatus(completed=ModelVersionCompletedStatus()),
                 created_by="mock",
                 created_at=timestamp,
                 updated_at=timestamp,

frogml_services_mock/mocks/project_manager_service.py

@@ -4,7 +4,7 @@ from typing import Dict
 import grpc
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from _frogml_proto.qwak.projects.projects_pb2 import (
+from frogml_proto.qwak.projects.projects_pb2 import (
     CreateProjectRequest,
     CreateProjectResponse,
     DeleteProjectRequest,
@@ -16,7 +16,7 @@ from _frogml_proto.qwak.projects.projects_pb2 import (
     Project,
     ProjectSpec,
 )
-from _frogml_proto.qwak.projects.projects_pb2_grpc import (
+from frogml_proto.qwak.projects.projects_pb2_grpc import (
     ProjectsManagementServiceServicer,
 )
 

frogml_services_mock/mocks/prompt_manager_service.py

@@ -3,69 +3,69 @@ from typing import Dict, List, Tuple
 import grpc
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     CreatePromptRequest as ProtoCreatePromptRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     CreatePromptResponse as ProtoCreatePromptResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     CreatePromptVersionRequest as ProtoCreatePromptVersionRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     CreatePromptVersionResponse as ProtoCreatePromptVersionResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     DeletePromptRequest as ProtoDeletePromptRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     DeletePromptResponse as ProtoDeletePromptResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     DeletePromptVersionRequest as ProtoDeletePromptVersionRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     DeletePromptVersionResponse as ProtoDeletePromptVersionResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     GetPromptByNameRequest as ProtoGetPromptByNameRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     GetPromptByNameResponse as ProtoGetPromptByNameResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     GetPromptVersionByPromptNameRequest as ProtoGetPromptVersionByPromptNameRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     GetPromptVersionByPromptNameResponse as ProtoGetPromptVersionByPromptNameResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     SetDefaultPromptVersionRequest as ProtoSetDefaultPromptVersionRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     SetDefaultPromptVersionResponse as ProtoSetDefaultPromptVersionResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     UpdatePromptRequest as ProtoUpdatePromptRequest,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
     UpdatePromptResponse as ProtoUpdatePromptResponse,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
     PromptManagerServiceServicer,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import Prompt as ProtoPrompt
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import Prompt as ProtoPrompt
+from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
     PromptMetadata as ProtoPromptMetadata,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import PromptSpec as ProtoPromptSpec
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import PromptSpec as ProtoPromptSpec
+from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
     PromptVersion as ProtoPromptVersion,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
     PromptVersionDefinition as ProtoPromptVersionDefinition,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
     PromptVersionMetadata as ProtoPromptVersionMetadata,
 )
 
@@ -165,9 +165,9 @@ class PromptManagerServiceMock(PromptManagerServiceServicer):
                 prompt_version_definition
             )
 
-        self.prompts_versions[
-            (request.prompt_name, new_prompt_version_number)
-        ] = prompt_version_definition
+        self.prompts_versions[(request.prompt_name, new_prompt_version_number)] = (
+            prompt_version_definition
+        )
         return ProtoCreatePromptVersionResponse(
             prompt_version=ProtoPromptVersion(
                 prompt_name=request.prompt_name,

frogml_services_mock/mocks/repository_service_mock.py

@@ -3,15 +3,15 @@ from typing import Dict
 
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from _frogml_proto.qwak.jfrog.gateway.v0.repository_pb2 import RepositorySpec
-from _frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2 import (
+from frogml_proto.qwak.jfrog.gateway.v0.repository_pb2 import RepositorySpec
+from frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2 import (
     GetRepositoryConfigurationResponse,
     GetRepositoryConfigurationRequest,
 )
-from _frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2_grpc import (
+from frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2_grpc import (
     RepositoryServiceServicer,
 )
-from frogml.exceptions import FrogmlNotFoundException
+from frogml_core.exceptions import FrogmlNotFoundException
 from frogml_services_mock.mocks.utils.exception_handlers import (
     raise_not_found_grpc_error,
 )
@@ -36,4 +36,6 @@ class RepositoryServiceMock(RepositoryServiceServicer):
                 FrogmlNotFoundException(f"Repository {repository_key} not found"),
             )
 
-        return self.repositories[repository_key]
+        return GetRepositoryConfigurationResponse(
+            repository_spec=self.repositories[repository_key]
+        )

frogml_services_mock/mocks/secret_service.py

@@ -1,6 +1,6 @@
 import grpc
 
-from _frogml_proto.qwak.secret_service.secret_service_pb2 import (
+from frogml_proto.qwak.secret_service.secret_service_pb2 import (
     DeleteSecretRequest,
     DeleteSecretResponse,
     GetSecretRequest,
@@ -8,7 +8,7 @@ from _frogml_proto.qwak.secret_service.secret_service_pb2 import (
     SetSecretRequest,
     SetSecretResponse,
 )
-from _frogml_proto.qwak.secret_service.secret_service_pb2_grpc import (
+from frogml_proto.qwak.secret_service.secret_service_pb2_grpc import (
     SecretServiceServicer,
 )
 

frogml_services_mock/mocks/self_service_user_service.py

@@ -1,10 +1,10 @@
 from typing import Dict
 
-from _frogml_proto.qwak.self_service.user.v1.user_service_pb2 import (
+from frogml_proto.qwak.self_service.user.v1.user_service_pb2 import (
     GenerateApiKeyResponse,
     RevokeApiKeyResponse,
 )
-from _frogml_proto.qwak.self_service.user.v1.user_service_pb2_grpc import (
+from frogml_proto.qwak.self_service.user.v1.user_service_pb2_grpc import (
     UserServiceServicer,
 )
 

frogml_services_mock/mocks/system_secret_service.py

@@ -2,12 +2,12 @@ from typing import Dict, Tuple
 
 import grpc
 
-from _frogml_proto.qwak.admiral.secret.v0.secret_pb2 import (
+from frogml_proto.qwak.admiral.secret.v0.secret_pb2 import (
     EnvironmentSecretIdentifier,
     SetSystemEnvironmentSecretOptions,
     SystemSecretDefinition,
 )
-from _frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2 import (
+from frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2 import (
     DeleteSystemSecretRequest,
     DeleteSystemSecretResponse,
     GetSystemSecretRequest,
@@ -15,7 +15,7 @@ from _frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2 import (
     SetSystemSecretRequest,
     SetSystemSecretResponse,
 )
-from _frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
+from frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
     SystemSecretServiceServicer,
 )
 

frogml_services_mock/mocks/user_application_instance_service_api.py

@@ -1,10 +1,10 @@
-from _frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_pb2 import (
+from frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_pb2 import (
     Description,
     Identifier,
     Metadata,
     SpecDescription,
 )
-from _frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_service_pb2 import (
+from frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_service_pb2 import (
     CreateUserApplicationInstanceRequest,
     CreateUserApplicationInstanceResponse,
     DeleteUserApplicationInstanceRequest,
@@ -14,7 +14,7 @@ from _frogml_proto.qwak.admiral.user_application_instance.v0.user_application_in
     UpdateUserApplicationInstanceSpecRequest,
     UpdateUserApplicationInstanceSpecResponse,
 )
-from _frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_service_pb2_grpc import (
+from frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_service_pb2_grpc import (
     UserApplicationInstanceServiceServicer,
 )
 from frogml_services_mock.mocks.utils.exception_handlers import (

frogml_services_mock/mocks/vector_serving_api.py

@@ -5,14 +5,14 @@ import grpc
 from numpy import dot
 from numpy.linalg import norm
 
-from _frogml_proto.qwak.vectors.v1.vector_pb2 import (
+from frogml_proto.qwak.vectors.v1.vector_pb2 import (
     DoubleVector,
     Property,
     SearchResult,
     StoredVector,
     VectorIdentifier,
 )
-from _frogml_proto.qwak.vectors.v1.vector_service_pb2 import (
+from frogml_proto.qwak.vectors.v1.vector_service_pb2 import (
     DeleteVectorsResponse,
     FetchVectorRequest,
     FetchVectorResponse,
@@ -21,7 +21,7 @@ from _frogml_proto.qwak.vectors.v1.vector_service_pb2 import (
     UpsertVectorsRequest,
     UpsertVectorsResponse,
 )
-from _frogml_proto.qwak.vectors.v1.vector_service_pb2_grpc import VectorServiceServicer
+from frogml_proto.qwak.vectors.v1.vector_service_pb2_grpc import VectorServiceServicer
 
 
 @dataclass

frogml_services_mock/mocks/vectors_management_api.py

@@ -4,13 +4,13 @@ from datetime import datetime
 import grpc
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from _frogml_proto.qwak.vectors.v1.collection.collection_pb2 import (
+from frogml_proto.qwak.vectors.v1.collection.collection_pb2 import (
     COLLECTION_STATUS_CREATE_REQUESTED,
     QwakMetadata,
     VectorCollection,
     VectorCollectionDefinition,
 )
-from _frogml_proto.qwak.vectors.v1.collection.collection_service_pb2 import (
+from frogml_proto.qwak.vectors.v1.collection.collection_service_pb2 import (
     CreateCollectionResponse,
     DeleteCollectionByIdResponse,
     DeleteCollectionByNameResponse,
@@ -18,7 +18,7 @@ from _frogml_proto.qwak.vectors.v1.collection.collection_service_pb2 import (
     GetCollectionByNameResponse,
     ListCollectionsResponse,
 )
-from _frogml_proto.qwak.vectors.v1.collection.collection_service_pb2_grpc import (
+from frogml_proto.qwak.vectors.v1.collection.collection_service_pb2_grpc import (
     VectorCollectionServiceServicer,
 )
 

frogml_services_mock/mocks/workspace_manager_service_mock.py

@@ -4,16 +4,16 @@ from dataclasses import dataclass
 
 import grpc
 
-from _frogml_proto.qwak.user_application.common.v0.resources_pb2 import (
+from frogml_proto.qwak.user_application.common.v0.resources_pb2 import (
     ClientPodComputeResources,
     PodComputeResourceTemplateSpec,
 )
-from _frogml_proto.qwak.workspace.workspace_pb2 import (
+from frogml_proto.qwak.workspace.workspace_pb2 import (
     DefaultWorkspaceDetails,
     Workspace,
     WorkspaceImage,
 )
-from _frogml_proto.qwak.workspace.workspace_service_pb2 import (
+from frogml_proto.qwak.workspace.workspace_service_pb2 import (
     CountWorkspacesRequest,
     CountWorkspacesResponse,
     CreateWorkspaceResponse,
@@ -25,7 +25,7 @@ from _frogml_proto.qwak.workspace.workspace_service_pb2 import (
     ListWorkspacesResponse,
     UpdateWorkspaceResponse,
 )
-from _frogml_proto.qwak.workspace.workspace_service_pb2_grpc import (
+from frogml_proto.qwak.workspace.workspace_service_pb2_grpc import (
     WorkspaceManagementServiceServicer,
 )
 

frogml_services_mock/services_mock.py

@@ -4,123 +4,125 @@ from typing import Any, Generator, List, Tuple
 import grpc
 import pytest
 
-from _frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
+from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     add_ModelVersionManagerServiceServicer_to_server,
 )
-from _frogml_proto.qwak.administration.v0.authentication.authentication_service_pb2_grpc import (
+from frogml_proto.qwak.administration.v0.authentication.authentication_service_pb2_grpc import (
     add_AuthenticationServiceServicer_to_server,
 )
-from _frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
+from frogml_proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
     add_SystemSecretServiceServicer_to_server,
 )
-from _frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_service_pb2_grpc import (
+from frogml_proto.qwak.admiral.user_application_instance.v0.user_application_instance_service_pb2_grpc import (
     add_UserApplicationInstanceServiceServicer_to_server,
 )
-from _frogml_proto.qwak.analytics.analytics_service_pb2_grpc import (
+from frogml_proto.qwak.analytics.analytics_service_pb2_grpc import (
     add_AnalyticsQueryServiceServicer_to_server,
 )
-from _frogml_proto.qwak.audience.v1.audience_api_pb2_grpc import (
+from frogml_proto.qwak.audience.v1.audience_api_pb2_grpc import (
     add_AudienceAPIServicer_to_server,
 )
-from _frogml_proto.qwak.auto_scaling.v1.auto_scaling_service_pb2_grpc import (
+from frogml_proto.qwak.auto_scaling.v1.auto_scaling_service_pb2_grpc import (
     add_AutoScalingServiceServicer_to_server,
 )
-from _frogml_proto.qwak.automation.v1.automation_management_service_pb2_grpc import (
+from frogml_proto.qwak.automation.v1.automation_management_service_pb2_grpc import (
     add_AutomationManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.batch_job.v1.batch_job_service_pb2_grpc import (
+from frogml_proto.qwak.batch_job.v1.batch_job_service_pb2_grpc import (
     add_BatchJobManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.build.v1.build_api_pb2_grpc import add_BuildAPIServicer_to_server
-from _frogml_proto.qwak.build_settings.build_settings_api_pb2_grpc import (
+from frogml_proto.qwak.build.v1.build_api_pb2_grpc import add_BuildAPIServicer_to_server
+from frogml_proto.qwak.build_settings.build_settings_api_pb2_grpc import (
     add_BuildSettingsApiServicer_to_server,
 )
-from _frogml_proto.qwak.builds.builds_orchestrator_service_pb2_grpc import (
+from frogml_proto.qwak.builds.builds_orchestrator_service_pb2_grpc import (
     add_BuildsOrchestratorServiceServicer_to_server,
 )
-from _frogml_proto.qwak.builds.builds_pb2_grpc import (
+from frogml_proto.qwak.builds.builds_pb2_grpc import (
     add_BuildsManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.builds.internal_builds_orchestrator_service_pb2_grpc import (
+from frogml_proto.qwak.builds.internal_builds_orchestrator_service_pb2_grpc import (
     add_InternalBuildsOrchestratorServiceServicer_to_server,
 )
-from _frogml_proto.qwak.data_versioning.data_versioning_service_pb2_grpc import (
+from frogml_proto.qwak.data_versioning.data_versioning_service_pb2_grpc import (
     add_DataVersioningManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.deployment.alert_service_pb2_grpc import (
+from frogml_proto.qwak.deployment.alert_service_pb2_grpc import (
     add_AlertManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.deployment.deployment_service_pb2_grpc import (
+from frogml_proto.qwak.deployment.deployment_service_pb2_grpc import (
     add_DeploymentManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.ecosystem.v0.ecosystem_runtime_service_pb2_grpc import (
+from frogml_proto.qwak.ecosystem.v0.ecosystem_runtime_service_pb2_grpc import (
     add_QwakEcosystemRuntimeServicer_to_server,
 )
-from _frogml_proto.qwak.execution.v1.execution_service_pb2_grpc import (
+from frogml_proto.qwak.execution.v1.execution_service_pb2_grpc import (
     add_FeatureStoreExecutionServiceServicer_to_server,
 )
-from _frogml_proto.qwak.feature_store.entities.entity_service_pb2_grpc import (
+from frogml_proto.qwak.feature_store.entities.entity_service_pb2_grpc import (
     add_EntityServiceServicer_to_server,
 )
-from _frogml_proto.qwak.feature_store.features.feature_set_service_pb2_grpc import (
+from frogml_proto.qwak.feature_store.features.feature_set_service_pb2_grpc import (
     add_FeatureSetServiceServicer_to_server,
 )
-from _frogml_proto.qwak.feature_store.jobs.v1.job_service_pb2_grpc import (
+from frogml_proto.qwak.feature_store.jobs.v1.job_service_pb2_grpc import (
     add_JobServiceServicer_to_server,
 )
-from _frogml_proto.qwak.feature_store.sources.data_source_service_pb2_grpc import (
+from frogml_proto.qwak.feature_store.sources.data_source_service_pb2_grpc import (
     add_DataSourceServiceServicer_to_server,
 )
-from _frogml_proto.qwak.features_operator.v3.features_operator_async_service_pb2_grpc import (
+from frogml_proto.qwak.features_operator.v3.features_operator_async_service_pb2_grpc import (
     add_FeaturesOperatorAsyncServiceServicer_to_server,
 )
-from _frogml_proto.qwak.file_versioning.file_versioning_service_pb2_grpc import (
+from frogml_proto.qwak.file_versioning.file_versioning_service_pb2_grpc import (
     add_FileVersioningManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.instance_template.instance_template_service_pb2_grpc import (
+from frogml_proto.qwak.instance_template.instance_template_service_pb2_grpc import (
     add_InstanceTemplateManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.integration.integration_service_pb2_grpc import (
+from frogml_proto.qwak.integration.integration_service_pb2_grpc import (
     add_IntegrationManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2_grpc import add_RepositoryServiceServicer_to_server
-from _frogml_proto.qwak.kube_deployment_captain.kube_deployment_captain_service_pb2_grpc import (
+from frogml_proto.qwak.jfrog.gateway.v0.repository_service_pb2_grpc import (
+    add_RepositoryServiceServicer_to_server,
+)
+from frogml_proto.qwak.kube_deployment_captain.kube_deployment_captain_service_pb2_grpc import (
     add_KubeDeploymentCaptainServicer_to_server,
 )
-from _frogml_proto.qwak.logging.log_reader_service_pb2_grpc import (
+from frogml_proto.qwak.logging.log_reader_service_pb2_grpc import (
     add_LogReaderServiceServicer_to_server,
 )
-from _frogml_proto.qwak.models.models_pb2_grpc import (
+from frogml_proto.qwak.models.models_pb2_grpc import (
     add_ModelsManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.monitoring.v0.alerting_channel_management_service_pb2_grpc import (
+from frogml_proto.qwak.monitoring.v0.alerting_channel_management_service_pb2_grpc import (
     add_AlertingChannelManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.offline.serving.v1.offline_serving_async_service_pb2_grpc import (
+from frogml_proto.qwak.offline.serving.v1.offline_serving_async_service_pb2_grpc import (
     add_FeatureStoreOfflineServingAsyncServiceServicer_to_server,
 )
-from _frogml_proto.qwak.projects.projects_pb2_grpc import (
+from frogml_proto.qwak.projects.projects_pb2_grpc import (
     add_ProjectsManagementServiceServicer_to_server,
 )
-from _frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
+from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
     add_PromptManagerServiceServicer_to_server,
 )
-from _frogml_proto.qwak.secret_service.secret_service_pb2_grpc import (
+from frogml_proto.qwak.secret_service.secret_service_pb2_grpc import (
     add_SecretServiceServicer_to_server,
 )
-from _frogml_proto.qwak.self_service.user.v1.user_service_pb2_grpc import (
+from frogml_proto.qwak.self_service.user.v1.user_service_pb2_grpc import (
     add_UserServiceServicer_to_server,
 )
-from _frogml_proto.qwak.vectors.v1.collection.collection_service_pb2_grpc import (
+from frogml_proto.qwak.vectors.v1.collection.collection_service_pb2_grpc import (
     add_VectorCollectionServiceServicer_to_server,
 )
-from _frogml_proto.qwak.vectors.v1.vector_service_pb2_grpc import (
+from frogml_proto.qwak.vectors.v1.vector_service_pb2_grpc import (
     add_VectorServiceServicer_to_server,
 )
-from _frogml_proto.qwak.workspace.workspace_service_pb2_grpc import (
+from frogml_proto.qwak.workspace.workspace_service_pb2_grpc import (
     add_WorkspaceManagementServiceServicer_to_server,
 )
-from frogml.inner.di_configuration import FrogmlContainer
+from frogml_core.inner.di_configuration import FrogmlContainer
 from frogml_services_mock.mocks.alert_manager_service_api import (
     AlertManagerServiceApiMock,
 )
@@ -232,7 +234,7 @@ def frogml_container():
             },
         },
     )
-    from frogml.clients import (
+    from frogml_core.clients import (
         alert_management,
         alerts_registry,
         analytics,
@@ -256,13 +258,17 @@ def frogml_container():
         vector_store,
         workspace_manager,
         model_version_manager,
-        jfrog_gateway
+        jfrog_gateway,
+    )
+    from frogml_core.clients.administration import (
+        authentication,
+        eco_system,
+        self_service,
     )
-    from frogml.clients.administration import authentication, eco_system, self_service
-    from frogml.clients.integration_management import integration_manager_client
-    from frogml.clients.prompt_manager import prompt_manager_client
-    from frogml.clients.system_secret import system_secret_client
-    from frogml.vector_store.utils import upsert_utils
+    from frogml_core.clients.integration_management import integration_manager_client
+    from frogml_core.clients.prompt_manager import prompt_manager_client
+    from frogml_core.clients.system_secret import system_secret_client
+    from frogml_core.vector_store.utils import upsert_utils
 
     container.wire(
         packages=[
@@ -296,7 +302,7 @@ def frogml_container():
             system_secret_client,
             prompt_manager_client,
             model_version_manager,
-            jfrog_gateway
+            jfrog_gateway,
         ]
     )
 

_frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.py

@@ -1,35 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: _frogml_proto.jfml.model_version/v1/model_repository_spec.proto
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
-from google.protobuf import symbol_database as _symbol_database
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1jfml/model_version/v1/model_repository_spec.proto\x12\x15jfml.model_version.v1\"l\n\x13ModelRepositorySpec\x12\x15\n\x0bproject_key\x18\x01 \x01(\tH\x00\x12\x16\n\x0erepository_key\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\tB\x14\n\x12project_identifierB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
-
-
-
-_MODELREPOSITORYSPEC = DESCRIPTOR.message_types_by_name['ModelRepositorySpec']
-ModelRepositorySpec = _reflection.GeneratedProtocolMessageType('ModelRepositorySpec', (_message.Message,), {
-  'DESCRIPTOR' : _MODELREPOSITORYSPEC,
-  '__module__' : 'jfml.model_version.v1.model_repository_spec_pb2'
-  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.ModelRepositorySpec)
-  })
-_sym_db.RegisterMessage(ModelRepositorySpec)
-
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
-  _MODELREPOSITORYSPEC._serialized_start=76
-  _MODELREPOSITORYSPEC._serialized_end=184
-# @@protoc_insertion_point(module_scope)

_frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py

@@ -1,45 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: _frogml_proto.jfml.model_version/v1/model_version_framework.proto
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
-from google.protobuf import symbol_database as _symbol_database
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3jfml/model_version/v1/model_version_framework.proto\x12\x15jfml.model_version.v1\"]\n\x15ModelVersionFramework\x12<\n\x08\x63\x61tboost\x18\x01 \x01(\x0b\x32(.jfml.model_version.v1.CatboostFrameworkH\x00\x42\x06\n\x04name\"\x13\n\x11\x43\x61tboostFrameworkB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
-
-
-
-_MODELVERSIONFRAMEWORK = DESCRIPTOR.message_types_by_name['ModelVersionFramework']
-_CATBOOSTFRAMEWORK = DESCRIPTOR.message_types_by_name['CatboostFramework']
-ModelVersionFramework = _reflection.GeneratedProtocolMessageType('ModelVersionFramework', (_message.Message,), {
-  'DESCRIPTOR' : _MODELVERSIONFRAMEWORK,
-  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
-  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.ModelVersionFramework)
-  })
-_sym_db.RegisterMessage(ModelVersionFramework)
-
-CatboostFramework = _reflection.GeneratedProtocolMessageType('CatboostFramework', (_message.Message,), {
-  'DESCRIPTOR' : _CATBOOSTFRAMEWORK,
-  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
-  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.CatboostFramework)
-  })
-_sym_db.RegisterMessage(CatboostFramework)
-
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
-  _MODELVERSIONFRAMEWORK._serialized_start=78
-  _MODELVERSIONFRAMEWORK._serialized_end=171
-  _CATBOOSTFRAMEWORK._serialized_start=173
-  _CATBOOSTFRAMEWORK._serialized_end=192
-# @@protoc_insertion_point(module_scope)