qwak-core 0.4.200__py3-none-any.whl → 0.4.201__py3-none-any.whl

qwak/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for qwak-core."""
 
 __author__ = "Qwak.ai"
-__version__ = "0.4.200"
+__version__ = "0.4.201"
 
 from qwak.inner.di_configuration import wire_dependencies
 from qwak.model.experiment_tracking import log_metric, log_param

qwak/clients/model_management/client.py

@@ -12,6 +12,7 @@ from _qwak_proto.qwak.models.models_pb2 import (
     ModelSpec,
 )
 from _qwak_proto.qwak.models.models_pb2_grpc import ModelsManagementServiceStub
+from _qwak_proto.qwak.projects.jfrog_project_spec_pb2 import ModelRepositoryJFrogSpec
 from dependency_injector.wiring import Provide
 from qwak.exceptions import QwakException
 from qwak.inner.di_configuration import QwakContainer
@@ -53,7 +54,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,
+        model_name,
+        model_description,
+        artifactory_project_key: Optional[str] = None,
+    ):
         try:
             return self._models_management_service.CreateModel(
                 CreateModelRequest(
@@ -62,7 +69,10 @@ class ModelsManagementClient:
                         display_name=model_name,
                         project_id=project_id,
                         model_description=model_description,
-                    )
+                    ),
+                    jfrog_project_spec=ModelRepositoryJFrogSpec(
+                        jfrog_project_key=artifactory_project_key,
+                    ),
                 )
             )
 

qwak/clients/project/client.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 import grpc
 from _qwak_proto.qwak.projects.projects_pb2 import (
     CreateProjectRequest,
@@ -6,6 +8,7 @@ from _qwak_proto.qwak.projects.projects_pb2 import (
     ListProjectsRequest,
 )
 from _qwak_proto.qwak.projects.projects_pb2_grpc import ProjectsManagementServiceStub
+from _qwak_proto.qwak.projects.jfrog_project_spec_pb2 import ModelRepositoryJFrogSpec
 from dependency_injector.wiring import Provide, inject
 from qwak.exceptions import QwakException
 from qwak.inner.di_configuration import QwakContainer
@@ -27,11 +30,20 @@ class ProjectsManagementClient:
         except grpc.RpcError as e:
             raise QwakException(f"Failed to list projects, error is {e.details()}")
 
-    def create_project(self, project_name, project_description):
+    def create_project(
+        self,
+        project_name,
+        project_description,
+        artifactory_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=artifactory_project_key
+                    ),
                 )
             )
 

qwak/qwak_client/client.py

@@ -357,20 +357,28 @@ class QwakClient:
         """
         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,
+        artifactory_project_key: Optional[str] = None,
+    ) -> str:
         """
         Create project
 
         Args:
             project_name (str): The requested name
             project_description (str): The requested description
+            artifactory_project_key (Optional[str]): The requested artifactory 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,
+            artifactory_project_key=artifactory_project_key,
         )
 
         return project.project.project_id
@@ -412,7 +420,11 @@ class QwakClient:
         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,
+        artifactory_project_key: Optional[str] = None,
     ) -> str:
         """
         Create model
@@ -421,6 +433,7 @@ class QwakClient:
             project_id (str): The project ID to associate the model
             model_name (str): The requested name
             model_description (str): The requested description
+            artifactory_project_key (Optional[str]): The artifactory project key
 
         Returns:
              str: The model ID of the newly created project
@@ -430,6 +443,7 @@ class QwakClient:
             project_id=project_id,
             model_name=model_name,
             model_description=model_description,
+            artifactory_project_key=artifactory_project_key,
         )
 
         return model.model_id

{qwak_core-0.4.200.dist-info → qwak_core-0.4.201.dist-info}/METADATA

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

{qwak_core-0.4.200.dist-info → qwak_core-0.4.201.dist-info}/RECORD

@@ -525,7 +525,7 @@ _qwak_proto/qwak/workspace/workspace_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 _qwak_proto/qwak/workspace/workspace_service_pb2.py,sha256=AB3C9S_AbOD7Nx1Ni4j1rW6PNtYTV1zjiqFQk-goQ74,21429
 _qwak_proto/qwak/workspace/workspace_service_pb2.pyi,sha256=nKKCHwnovZhsy8TSVmdz-Vtl0nviOOoX56HD-41Xo08,13726
 _qwak_proto/qwak/workspace/workspace_service_pb2_grpc.py,sha256=yKGuexxTBza99Ihe0DSTniV2ZSd_AG47inHenqfi890,27193
-qwak/__init__.py,sha256=fY6x2ezLuFob-1iTd0NSs5b-o9a4kOD12IZ5bg0gQ3U,587
+qwak/__init__.py,sha256=SG4Z1wJtu_O4GMIpTHNBsVfgfEBRUOJtfbF68u633fQ,587
 qwak/automations/__init__.py,sha256=qFZRvCxUUn8gcxkJR0v19ulHW2oJ0x6-Rif7HiheDP4,1522
 qwak/automations/automation_executions.py,sha256=5MeH_epYYWb8NKXgAozwT_jPyyUDednBHG7izloi7RY,3228
 qwak/automations/automations.py,sha256=3yx8e2v0uSKDnXbqyknasyEoQ5vxGni6K40Hbi1_zkk,12599
@@ -596,9 +596,9 @@ qwak/clients/kube_deployment_captain/client.py,sha256=g5nLf4l7A4SFpdfF8Q5KwEincS
 qwak/clients/logging_client/__init__.py,sha256=1OCHnigQBYThBwGbxCreYA0BgP0HcuLFzNEWd3Yxh-c,34
 qwak/clients/logging_client/client.py,sha256=IdPa93JjlVm0B_Gdc4lH9LYHoVn5DjT_gIPaZ-W3Hdw,4906
 qwak/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
-qwak/clients/model_management/client.py,sha256=mlD0rIN3j763BPzTM7gt5eZ3ugLpRc96Ikik-lpb-AM,4431
+qwak/clients/model_management/client.py,sha256=MK_3T6DLGG2bs4nt-0lGmMgrp69LRp667E-7PKeZ51w,4767
 qwak/clients/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak/clients/project/client.py,sha256=Inb_NjiBuDdJUFWoREyNOPdtGMKlMXCsdeFAWDaURuY,2128
+qwak/clients/project/client.py,sha256=CSEA7jYQsWMXT3sdGGv71aPWaaCalhItzDN4sTFagBE,2496
 qwak/clients/prompt_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/clients/prompt_manager/model_descriptor_mapper.py,sha256=rRQXjqLvq7SVxYSM5_IFo_6z70Kzf5WPOtxFzm-69m0,7313
 qwak/clients/prompt_manager/prompt_manager_client.py,sha256=Mg1uqJjwj57jFfNKx6tomzfOe6C8keTKpxvXFSPo0yA,6952
@@ -922,7 +922,7 @@ qwak/qwak_client/builds/build.py,sha256=1dBRZkvZV04uw8RI-7iOs_yvfgUzglh9ZyzgHJyU
 qwak/qwak_client/builds/filters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/qwak_client/builds/filters/metric_filter.py,sha256=i0ZsjqN1iw-cX_uxvbJzRT-EYAOEqejKkukLiyYmYqE,1185
 qwak/qwak_client/builds/filters/parameter_filter.py,sha256=IY-J6bAIEM_KyVzESMDx0xv_fAScRp_AsltdMJk0IDY,1088
-qwak/qwak_client/client.py,sha256=1FxTRoljv_ZnFY32TRcCkk3dNZ4H1_tlNfSxmp-sZzQ,28361
+qwak/qwak_client/client.py,sha256=RLQMWDuA6WW5DDIAvUEHHeqPmIndMf4V-ZG6ZnZsSro,28834
 qwak/qwak_client/data_versioning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak/qwak_client/data_versioning/data_tag.py,sha256=B7_TsBrp0LZK-yLqGpodVhoh67dpWBsiyEgWq3RCKcA,806
 qwak/qwak_client/deployments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1001,6 +1001,6 @@ qwak_services_mock/mocks/workspace_manager_service_mock.py,sha256=O9ZSwln4T4kHVk
 qwak_services_mock/services_mock.py,sha256=CABZ2teXoSJ-X7CD3iRo5ht-w3jhW-Cvpb47APUHXEA,19089
 qwak_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-qwak_core-0.4.200.dist-info/METADATA,sha256=sFbULF5YQqm8XjcR58GPSNajI3jGEPVZ5za2o1_pXx4,2158
-qwak_core-0.4.200.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-qwak_core-0.4.200.dist-info/RECORD,,
+qwak_core-0.4.201.dist-info/METADATA,sha256=-LwjASRv_Vxex9WRYvb6m6lpE1u2Eeiw3AZ1SJGfRQc,2158
+qwak_core-0.4.201.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+qwak_core-0.4.201.dist-info/RECORD,,