frogml-core 0.0.17__py3-none-any.whl → 0.0.19__py3-none-any.whl

frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.17"
+__version__ = "0.0.19"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

frogml_core/clients/model_management/client.py

@@ -17,6 +17,7 @@ from frogml_proto.qwak.models.models_pb2 import (
 from frogml_proto.qwak.models.models_pb2_grpc import ModelsManagementServiceStub
 from frogml_core.exceptions import FrogmlException
 from frogml_core.inner.di_configuration import FrogmlContainer
+from frogml_proto.qwak.projects.jfrog_project_spec_pb2 import ModelRepositoryJFrogSpec
 
 
 class ModelsManagementClient:
@@ -55,7 +56,13 @@ class ModelsManagementClient:
         model = self.get_model(model_id)
         return model.uuid
 
-    def create_model(self, project_id, model_name, model_description):
+    def create_model(
+        self,
+        project_id: str,
+        model_name: str,
+        model_description: str,
+        jfrog_project_key: Optional[str] = None,
+    ):
         try:
             return self._models_management_service.CreateModel(
                 CreateModelRequest(
@@ -64,7 +71,10 @@ class ModelsManagementClient:
                         display_name=model_name,
                         project_id=project_id,
                         model_description=model_description,
-                    )
+                    ),
+                    jfrog_project_spec=ModelRepositoryJFrogSpec(
+                        jfrog_project_key=jfrog_project_key,
+                    ),
                 )
             )
 

frogml_core/clients/project/client.py

@@ -1,6 +1,9 @@
+from typing import Optional
+
 import grpc
 from dependency_injector.wiring import Provide, inject
 
+from frogml_proto.qwak.projects.jfrog_project_spec_pb2 import ModelRepositoryJFrogSpec
 from frogml_proto.qwak.projects.projects_pb2 import (
     CreateProjectRequest,
     DeleteProjectRequest,
@@ -28,11 +31,20 @@ class ProjectsManagementClient:
         except grpc.RpcError as e:
             raise FrogmlException(f"Failed to list projects, error is {e.details()}")
 
-    def create_project(self, project_name, project_description):
+    def create_project(
+        self,
+        project_name: str,
+        project_description: str,
+        jfrog_project_key: Optional[str] = None,
+    ):
         try:
             return self._projects_management_service.CreateProject(
                 CreateProjectRequest(
-                    project_name=project_name, project_description=project_description
+                    project_name=project_name,
+                    project_description=project_description,
+                    jfrog_spec=ModelRepositoryJFrogSpec(
+                        jfrog_project_key=jfrog_project_key,
+                    ),
                 )
             )
 

frogml_core/frogml_client/client.py

@@ -374,20 +374,28 @@ class FrogMLClient:
         """
         self._get_build_orchestrator().register_tags(build_id=build_id, tags=tags)
 
-    def create_project(self, project_name: str, project_description: str) -> str:
+    def create_project(
+        self,
+        project_name: str,
+        project_description: str,
+        jfrog_project_key: Optional[str] = None,
+    ) -> str:
         """
         Create project
 
         Args:
             project_name (str): The requested name
             project_description (str): The requested description
+            jfrog_project_key (Optional[str]): The JFrog project key
 
         Returns:
              str: The project ID of the newly created project
 
         """
         project = self._get_project_management().create_project(
-            project_name=project_name, project_description=project_description
+            project_name=project_name,
+            project_description=project_description,
+            jfrog_project_key=jfrog_project_key,
         )
 
         return project.project.project_id
@@ -429,7 +437,11 @@ class FrogMLClient:
         return self._get_project_management().delete_project(project_id=project_id)
 
     def create_model(
-        self, project_id: str, model_name: str, model_description: str
+        self,
+        project_id: str,
+        model_name: str,
+        model_description: str,
+        jfrog_project_key: Optional[str] = None,
     ) -> str:
         """
         Create model
@@ -438,6 +450,7 @@ class FrogMLClient:
             project_id (str): The project ID to associate the model
             model_name (str): The requested name
             model_description (str): The requested description
+            jfrog_project_key (Optional[str]): The JFrog project key
 
         Returns:
              str: The model ID of the newly created project
@@ -447,6 +460,7 @@ class FrogMLClient:
             project_id=project_id,
             model_name=model_name,
             model_description=model_description,
+            jfrog_project_key=jfrog_project_key,
         )
 
         return model.model_id

{frogml_core-0.0.17.dist-info → frogml_core-0.0.19.dist-info}/METADATA

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

{frogml_core-0.0.17.dist-info → frogml_core-0.0.19.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=BKbXlzCDAHp1YE0_mvZtBo3GU34A8lkBwj9p0IIVHKI,777
+frogml_core/__init__.py,sha256=SuxasxuXh-oOyiHcALJ5CRq404CcZQCY54VJ06ftIF0,777
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
 frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -71,11 +71,11 @@ frogml_core/clients/kube_deployment_captain/client.py,sha256=oz7VF37TSO0S07MqXOY
 frogml_core/clients/logging_client/__init__.py,sha256=1OCHnigQBYThBwGbxCreYA0BgP0HcuLFzNEWd3Yxh-c,34
 frogml_core/clients/logging_client/client.py,sha256=A7qQJWW54Ve7O0DuJX9gmSJlUQyRIKC0VmfGS6bUwjo,4950
 frogml_core/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
-frogml_core/clients/model_management/client.py,sha256=rXnzcP3-pfXCVG0KlcCwbdeIJ1iFzV84JYtyNC8n-x8,4575
+frogml_core/clients/model_management/client.py,sha256=hOWbpO6KkcScT6sn3eNHSp7n2HfQIPug7lm-e-bkzEs,4915
 frogml_core/clients/model_version_manager/__init__.py,sha256=4Pnfzj4Egps48__dProdbSKQl5pNip0hGJd75w67BfY,46
 frogml_core/clients/model_version_manager/client.py,sha256=WeU9XXLu5K0ViS-2YeENq121qV4ESXmI6pzc_K8QkvM,9457
 frogml_core/clients/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-frogml_core/clients/project/client.py,sha256=VvnUk7mo-UkE9Gkj9jVkCPGTVx8qmuekMz7oH9Uzv2Y,2159
+frogml_core/clients/project/client.py,sha256=2L4LJS3xIGTAeWuL2HOgW0OyUNiQUHZ4mXfukRLjX2U,2527
 frogml_core/clients/prompt_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/clients/prompt_manager/model_descriptor_mapper.py,sha256=-j97vEEx0RUNBH245vIIMYw_xrFE-4AgV1hmg7g14U8,7856
 frogml_core/clients/prompt_manager/prompt_manager_client.py,sha256=pl5SZ0j81PBMBeJc7jOkDJtJ61TGMqeq3Splg6Tsr90,7958
@@ -207,7 +207,7 @@ frogml_core/frogml_client/builds/build.py,sha256=5LysOp1nAhjQaL1Jdg-rtb81jOaxZL-
 frogml_core/frogml_client/builds/filters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/frogml_client/builds/filters/metric_filter.py,sha256=hDhyKGavGmZXRbmhGjtZejY1veQR277F1FUgBg-ZjjM,1186
 frogml_core/frogml_client/builds/filters/parameter_filter.py,sha256=3qEL2KJdZNytx646XnBjDO7phxCDO98tAUqMurp5Kq0,1098
-frogml_core/frogml_client/client.py,sha256=cE3seZBsIb5CRSTC1khrEkDsOCCRp6_cMoMBM4r5HLY,29356
+frogml_core/frogml_client/client.py,sha256=xQb2A9y-pLRekBVEdsdELfZaL_sF_QiZIxwEa3uSHFs,29759
 frogml_core/frogml_client/data_versioning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/frogml_client/data_versioning/data_tag.py,sha256=WadGhSMgKIajVV9jWIIwXhi-xBR5bsQnu9dN-XPX_XU,807
 frogml_core/frogml_client/deployments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1019,6 +1019,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=_34z6rFCHFwcSni-9eyJlrH264xsL_QDNmG-EBv0zMc,20281
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_core-0.0.17.dist-info/METADATA,sha256=1GPZDOPCEU2tNmyV0MP88shHnWvenLdbQJM9vEqWvlI,2099
-frogml_core-0.0.17.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.17.dist-info/RECORD,,
+frogml_core-0.0.19.dist-info/METADATA,sha256=KWodBht9OUKQIs7JZoyf-s72TtV5RnpA-23CIanMwUU,2099
+frogml_core-0.0.19.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.19.dist-info/RECORD,,