qwak-core 0.4.266__py3-none-any.whl → 0.4.268__py3-none-any.whl

_qwak_proto/qwak/builds/builds_orchestrator_service_pb2.py

@@ -16,7 +16,7 @@ from _qwak_proto.qwak.builds import build_pb2 as qwak_dot_builds_dot_build__pb2
 from _qwak_proto.qwak.builds import build_url_pb2 as qwak_dot_builds_dot_build__url__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/builds/builds_orchestrator_service.proto\x12\x18qwak.builds.orchestrator\x1a\x17qwak/builds/build.proto\x1a\x1bqwak/builds/build_url.proto\"R\n\x11\x42uildModelRequest\x12=\n\nbuild_spec\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.RemoteBuildSpec\"\x14\n\x12\x42uildModelResponse\"+\n\x17\x43\x61ncelBuildModelRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"\x1a\n\x18\x43\x61ncelBuildModelResponse\"*\n\x16\x43reateUploadURLRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"C\n\x17\x43reateUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x14\n\x0c\x64ownload_url\x18\x02 \x01(\t\"h\n\"GetBuildVersioningUploadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xa7\x02\n#GetBuildVersioningUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12Q\n\x15\x61uthentication_detail\x18\x03 \x01(\x0b\x32..qwak.builds.orchestrator.AuthenticationDetailB\x02\x18\x01\x12[\n\x07headers\x18\x04 \x03(\x0b\x32J.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x86\x01\n\x14\x41uthenticationDetail\x12Z\n\x1bjfrog_authentication_detail\x18\x01 \x01(\x0b\x32\x33.qwak.builds.orchestrator.JfrogAuthenticationDetailH\x00\x42\x12\n\x10integration_type\"<\n\x19JfrogAuthenticationDetail\x12\x10\n\x08username\x18\x01 \x01(\t\x12\r\n\x05token\x18\x02 \x01(\t\"j\n$GetBuildVersioningDownloadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xdf\x01\n%GetBuildVersioningDownloadURLResponse\x12\x14\n\x0c\x64ownload_url\x18\x01 \x01(\t\x12\x11\n\tfile_size\x18\x02 \x01(\x03\x12]\n\x07headers\x18\x03 \x03(\x0b\x32L.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"n\n\x1eListBuildVersioningTagsRequest\x12L\n\x15\x62uild_versioning_tags\x18\x01 \x01(\x0b\x32-.qwak.builds.orchestrator.BuildVersioningTags\"\x9c\x01\n\x1fListBuildVersioningTagsResponse\x12\x61\n build_versioning_tags_properties\x18\x01 \x03(\x0b\x32\x37.qwak.builds.orchestrator.BuildVersioningTagsProperties\x12\x16\n\x0eui_direct_link\x18\x02 \x01(\t\"\x95\x01\n\x16\x43reateDataTableRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12<\n\x05table\x18\x04 \x01(\x0b\x32-.qwak.builds.orchestrator.DataTableDefinition\x12\x0c\n\x04path\x18\x05 \x01(\t\"\x19\n\x17\x43reateDataTableResponse\"n\n\x1dGetBaseDockerImageNameRequest\x12M\n\x16\x62\x61se_docker_image_type\x18\x01 \x01(\x0e\x32-.qwak.builds.orchestrator.BaseDockerImageType\"@\n\x1eGetBaseDockerImageNameResponse\x12\x1e\n\x16\x62\x61se_docker_image_name\x18\x01 \x01(\t\"f\n\x17SaveArtifactPathRequest\x12K\n\x12\x61rtifact_path_spec\x18\x01 \x01(\x0b\x32/.qwak.builds.orchestrator.BuildArtifactPathSpec\"\x1a\n\x18SaveArtifactPathResponse\"-\n\x19ListBuildArtifactsRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"X\n\x1aListBuildArtifactsResponse\x12:\n\tartifacts\x18\x01 \x03(\x0b\x32\'.qwak.builds.orchestrator.BuildArtifact2\xcf\n\n\x19\x42uildsOrchestratorService\x12g\n\nBuildModel\x12+.qwak.builds.orchestrator.BuildModelRequest\x1a,.qwak.builds.orchestrator.BuildModelResponse\x12y\n\x10\x43\x61ncelBuildModel\x12\x31.qwak.builds.orchestrator.CancelBuildModelRequest\x1a\x32.qwak.builds.orchestrator.CancelBuildModelResponse\x12{\n\x0f\x43reateUploadURL\x12\x30.qwak.builds.orchestrator.CreateUploadURLRequest\x1a\x31.qwak.builds.orchestrator.CreateUploadURLResponse\"\x03\x88\x02\x01\x12\x9a\x01\n\x1bGetBuildVersioningUploadURL\x12<.qwak.builds.orchestrator.GetBuildVersioningUploadURLRequest\x1a=.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse\x12\xa0\x01\n\x1dGetBuildVersioningDownloadURL\x12>.qwak.builds.orchestrator.GetBuildVersioningDownloadURLRequest\x1a?.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse\x12\x8e\x01\n\x17ListBuildVersioningTags\x12\x38.qwak.builds.orchestrator.ListBuildVersioningTagsRequest\x1a\x39.qwak.builds.orchestrator.ListBuildVersioningTagsResponse\x12v\n\x0f\x43reateDataTable\x12\x30.qwak.builds.orchestrator.CreateDataTableRequest\x1a\x31.qwak.builds.orchestrator.CreateDataTableResponse\x12\x8b\x01\n\x16GetBaseDockerImageName\x12\x37.qwak.builds.orchestrator.GetBaseDockerImageNameRequest\x1a\x38.qwak.builds.orchestrator.GetBaseDockerImageNameResponse\x12y\n\x10SaveArtifactPath\x12\x31.qwak.builds.orchestrator.SaveArtifactPathRequest\x1a\x32.qwak.builds.orchestrator.SaveArtifactPathResponse\x12\x7f\n\x12ListBuildArtifacts\x12\x33.qwak.builds.orchestrator.ListBuildArtifactsRequest\x1a\x34.qwak.builds.orchestrator.ListBuildArtifactsResponseB\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/builds/builds_orchestrator_service.proto\x12\x18qwak.builds.orchestrator\x1a\x17qwak/builds/build.proto\x1a\x1bqwak/builds/build_url.proto\"R\n\x11\x42uildModelRequest\x12=\n\nbuild_spec\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.RemoteBuildSpec\"\x14\n\x12\x42uildModelResponse\"+\n\x17\x43\x61ncelBuildModelRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"\x1a\n\x18\x43\x61ncelBuildModelResponse\"*\n\x16\x43reateUploadURLRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"C\n\x17\x43reateUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x14\n\x0c\x64ownload_url\x18\x02 \x01(\t\"h\n\"GetBuildVersioningUploadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xa7\x02\n#GetBuildVersioningUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12Q\n\x15\x61uthentication_detail\x18\x03 \x01(\x0b\x32..qwak.builds.orchestrator.AuthenticationDetailB\x02\x18\x01\x12[\n\x07headers\x18\x04 \x03(\x0b\x32J.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x86\x01\n\x14\x41uthenticationDetail\x12Z\n\x1bjfrog_authentication_detail\x18\x01 \x01(\x0b\x32\x33.qwak.builds.orchestrator.JfrogAuthenticationDetailH\x00\x42\x12\n\x10integration_type\"<\n\x19JfrogAuthenticationDetail\x12\x10\n\x08username\x18\x01 \x01(\t\x12\r\n\x05token\x18\x02 \x01(\t\"j\n$GetBuildVersioningDownloadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xdf\x01\n%GetBuildVersioningDownloadURLResponse\x12\x14\n\x0c\x64ownload_url\x18\x01 \x01(\t\x12\x11\n\tfile_size\x18\x02 \x01(\x03\x12]\n\x07headers\x18\x03 \x03(\x0b\x32L.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"n\n\x1eListBuildVersioningTagsRequest\x12L\n\x15\x62uild_versioning_tags\x18\x01 \x01(\x0b\x32-.qwak.builds.orchestrator.BuildVersioningTags\"\x9c\x01\n\x1fListBuildVersioningTagsResponse\x12\x61\n build_versioning_tags_properties\x18\x01 \x03(\x0b\x32\x37.qwak.builds.orchestrator.BuildVersioningTagsProperties\x12\x16\n\x0eui_direct_link\x18\x02 \x01(\t\"\x95\x01\n\x16\x43reateDataTableRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12<\n\x05table\x18\x04 \x01(\x0b\x32-.qwak.builds.orchestrator.DataTableDefinition\x12\x0c\n\x04path\x18\x05 \x01(\t\"\x19\n\x17\x43reateDataTableResponse\"n\n\x1dGetBaseDockerImageNameRequest\x12M\n\x16\x62\x61se_docker_image_type\x18\x01 \x01(\x0e\x32-.qwak.builds.orchestrator.BaseDockerImageType\"@\n\x1eGetBaseDockerImageNameResponse\x12\x1e\n\x16\x62\x61se_docker_image_name\x18\x01 \x01(\t\"f\n\x17SaveArtifactPathRequest\x12K\n\x12\x61rtifact_path_spec\x18\x01 \x01(\x0b\x32/.qwak.builds.orchestrator.BuildArtifactPathSpec\"\x1a\n\x18SaveArtifactPathResponse\"-\n\x19ListBuildArtifactsRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"X\n\x1aListBuildArtifactsResponse\x12:\n\tartifacts\x18\x01 \x03(\x0b\x32\'.qwak.builds.orchestrator.BuildArtifact\"A\n$DetachBuildsFromModelVersionsRequest\x12\x19\n\x11model_version_ids\x18\x01 \x03(\t\"\'\n%DetachBuildsFromModelVersionsResponse2\xf2\x0b\n\x19\x42uildsOrchestratorService\x12g\n\nBuildModel\x12+.qwak.builds.orchestrator.BuildModelRequest\x1a,.qwak.builds.orchestrator.BuildModelResponse\x12y\n\x10\x43\x61ncelBuildModel\x12\x31.qwak.builds.orchestrator.CancelBuildModelRequest\x1a\x32.qwak.builds.orchestrator.CancelBuildModelResponse\x12{\n\x0f\x43reateUploadURL\x12\x30.qwak.builds.orchestrator.CreateUploadURLRequest\x1a\x31.qwak.builds.orchestrator.CreateUploadURLResponse\"\x03\x88\x02\x01\x12\x9a\x01\n\x1bGetBuildVersioningUploadURL\x12<.qwak.builds.orchestrator.GetBuildVersioningUploadURLRequest\x1a=.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse\x12\xa0\x01\n\x1dGetBuildVersioningDownloadURL\x12>.qwak.builds.orchestrator.GetBuildVersioningDownloadURLRequest\x1a?.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse\x12\x8e\x01\n\x17ListBuildVersioningTags\x12\x38.qwak.builds.orchestrator.ListBuildVersioningTagsRequest\x1a\x39.qwak.builds.orchestrator.ListBuildVersioningTagsResponse\x12v\n\x0f\x43reateDataTable\x12\x30.qwak.builds.orchestrator.CreateDataTableRequest\x1a\x31.qwak.builds.orchestrator.CreateDataTableResponse\x12\x8b\x01\n\x16GetBaseDockerImageName\x12\x37.qwak.builds.orchestrator.GetBaseDockerImageNameRequest\x1a\x38.qwak.builds.orchestrator.GetBaseDockerImageNameResponse\x12y\n\x10SaveArtifactPath\x12\x31.qwak.builds.orchestrator.SaveArtifactPathRequest\x1a\x32.qwak.builds.orchestrator.SaveArtifactPathResponse\x12\x7f\n\x12ListBuildArtifacts\x12\x33.qwak.builds.orchestrator.ListBuildArtifactsRequest\x1a\x34.qwak.builds.orchestrator.ListBuildArtifactsResponse\x12\xa0\x01\n\x1d\x44\x65tachBuildsFromModelVersions\x12>.qwak.builds.orchestrator.DetachBuildsFromModelVersionsRequest\x1a?.qwak.builds.orchestrator.DetachBuildsFromModelVersionsResponseB\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
 
 
 
@@ -44,6 +44,8 @@ _SAVEARTIFACTPATHREQUEST = DESCRIPTOR.message_types_by_name['SaveArtifactPathReq
 _SAVEARTIFACTPATHRESPONSE = DESCRIPTOR.message_types_by_name['SaveArtifactPathResponse']
 _LISTBUILDARTIFACTSREQUEST = DESCRIPTOR.message_types_by_name['ListBuildArtifactsRequest']
 _LISTBUILDARTIFACTSRESPONSE = DESCRIPTOR.message_types_by_name['ListBuildArtifactsResponse']
+_DETACHBUILDSFROMMODELVERSIONSREQUEST = DESCRIPTOR.message_types_by_name['DetachBuildsFromModelVersionsRequest']
+_DETACHBUILDSFROMMODELVERSIONSRESPONSE = DESCRIPTOR.message_types_by_name['DetachBuildsFromModelVersionsResponse']
 BuildModelRequest = _reflection.GeneratedProtocolMessageType('BuildModelRequest', (_message.Message,), {
   'DESCRIPTOR' : _BUILDMODELREQUEST,
   '__module__' : 'qwak.builds.builds_orchestrator_service_pb2'
@@ -214,6 +216,20 @@ ListBuildArtifactsResponse = _reflection.GeneratedProtocolMessageType('ListBuild
   })
 _sym_db.RegisterMessage(ListBuildArtifactsResponse)
 
+DetachBuildsFromModelVersionsRequest = _reflection.GeneratedProtocolMessageType('DetachBuildsFromModelVersionsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _DETACHBUILDSFROMMODELVERSIONSREQUEST,
+  '__module__' : 'qwak.builds.builds_orchestrator_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.builds.orchestrator.DetachBuildsFromModelVersionsRequest)
+  })
+_sym_db.RegisterMessage(DetachBuildsFromModelVersionsRequest)
+
+DetachBuildsFromModelVersionsResponse = _reflection.GeneratedProtocolMessageType('DetachBuildsFromModelVersionsResponse', (_message.Message,), {
+  'DESCRIPTOR' : _DETACHBUILDSFROMMODELVERSIONSRESPONSE,
+  '__module__' : 'qwak.builds.builds_orchestrator_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.builds.orchestrator.DetachBuildsFromModelVersionsResponse)
+  })
+_sym_db.RegisterMessage(DetachBuildsFromModelVersionsResponse)
+
 _BUILDSORCHESTRATORSERVICE = DESCRIPTOR.services_by_name['BuildsOrchestratorService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -275,6 +291,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _LISTBUILDARTIFACTSREQUEST._serialized_end=2163
   _LISTBUILDARTIFACTSRESPONSE._serialized_start=2165
   _LISTBUILDARTIFACTSRESPONSE._serialized_end=2253
-  _BUILDSORCHESTRATORSERVICE._serialized_start=2256
-  _BUILDSORCHESTRATORSERVICE._serialized_end=3615
+  _DETACHBUILDSFROMMODELVERSIONSREQUEST._serialized_start=2255
+  _DETACHBUILDSFROMMODELVERSIONSREQUEST._serialized_end=2320
+  _DETACHBUILDSFROMMODELVERSIONSRESPONSE._serialized_start=2322
+  _DETACHBUILDSFROMMODELVERSIONSRESPONSE._serialized_end=2361
+  _BUILDSORCHESTRATORSERVICE._serialized_start=2364
+  _BUILDSORCHESTRATORSERVICE._serialized_end=3886
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/builds/builds_orchestrator_service_pb2.pyi

@@ -417,3 +417,28 @@ class ListBuildArtifactsResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["artifacts", b"artifacts"]) -> None: ...
 
 global___ListBuildArtifactsResponse = ListBuildArtifactsResponse
+
+class DetachBuildsFromModelVersionsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_VERSION_IDS_FIELD_NUMBER: builtins.int
+    @property
+    def model_version_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """The model versions to detach"""
+    def __init__(
+        self,
+        *,
+        model_version_ids: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_version_ids", b"model_version_ids"]) -> None: ...
+
+global___DetachBuildsFromModelVersionsRequest = DetachBuildsFromModelVersionsRequest
+
+class DetachBuildsFromModelVersionsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___DetachBuildsFromModelVersionsResponse = DetachBuildsFromModelVersionsResponse

_qwak_proto/qwak/builds/builds_orchestrator_service_pb2_grpc.py

@@ -64,6 +64,11 @@ class BuildsOrchestratorServiceStub(object):
                 request_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsRequest.SerializeToString,
                 response_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsResponse.FromString,
                 )
+        self.DetachBuildsFromModelVersions = channel.unary_unary(
+                '/qwak.builds.orchestrator.BuildsOrchestratorService/DetachBuildsFromModelVersions',
+                request_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsResponse.FromString,
+                )
 
 
 class BuildsOrchestratorServiceServicer(object):
@@ -139,6 +144,12 @@ class BuildsOrchestratorServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def DetachBuildsFromModelVersions(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_BuildsOrchestratorServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -192,6 +203,11 @@ def add_BuildsOrchestratorServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsRequest.FromString,
                     response_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsResponse.SerializeToString,
             ),
+            'DetachBuildsFromModelVersions': grpc.unary_unary_rpc_method_handler(
+                    servicer.DetachBuildsFromModelVersions,
+                    request_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.builds.orchestrator.BuildsOrchestratorService', rpc_method_handlers)
@@ -371,3 +387,20 @@ class BuildsOrchestratorService(object):
             qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DetachBuildsFromModelVersions(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.builds.orchestrator.BuildsOrchestratorService/DetachBuildsFromModelVersions',
+            qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

qwak/__init__.py

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

qwak/clients/instance_template/client.py

@@ -49,12 +49,10 @@ class InstanceTemplateManagementClient:
 
     def list_instance_templates(self) -> List[InstanceTemplateSpec]:
         try:
-            result: ListInstanceTemplatesResponse = (
-                self._instance_template_service.ListInstanceTemplates(
-                    ListInstanceTemplatesRequest(
-                        optional_instance_filter=InstanceFilter(
-                            instance_type_filter=InstanceTypeFilter.INSTANCE_TYPE_FILTER_ALL
-                        )
+            result: ListInstanceTemplatesResponse = self._instance_template_service.ListInstanceTemplates(
+                ListInstanceTemplatesRequest(
+                    optional_instance_filter=InstanceFilter(
+                        instance_type_filter=InstanceTypeFilter.INSTANCE_TYPE_FILTER_ALL
                     )
                 )
             )

qwak/clients/prompt_manager/model_descriptor_mapper.py

@@ -44,10 +44,10 @@ class ModelDescriptorMapper:
         model_id: str, openai_chat_params: ProtoOpenAIChatModelParams
     ) -> OpenAIChat:
         p = openai_chat_params
-        _tool_choice: Union[str, ChatCompletionNamedToolChoiceParam] = (
-            ModelDescriptorMapper._from_tool_choice(
-                openai_chat_params=openai_chat_params
-            )
+        _tool_choice: Union[
+            str, ChatCompletionNamedToolChoiceParam
+        ] = ModelDescriptorMapper._from_tool_choice(
+            openai_chat_params=openai_chat_params
         )
         _tools: List[ChatCompletionToolParam] = []
 
@@ -57,23 +57,21 @@ class ModelDescriptorMapper:
 
         return OpenAIChat(
             model_id=model_id,
-            frequency_penalty=(
-                p.frequency_penalty if p.HasField("frequency_penalty") else None
-            ),
-            logit_bias=(
-                {k: int(v) for k, v in p.logit_bias.items()}
-                if p.HasField("logit_bias")
-                else None
-            ),
+            frequency_penalty=p.frequency_penalty
+            if p.HasField("frequency_penalty")
+            else None,
+            logit_bias={k: int(v) for k, v in p.logit_bias.items()}
+            if p.HasField("logit_bias")
+            else None,
             logprobs=p.logprobs if p.HasField("logprobs") else None,
             max_tokens=p.max_tokens if p.HasField("max_tokens") else None,
             n=p.n if p.HasField("n") else None,
-            presence_penalty=(
-                p.presence_penalty if p.HasField("presence_penalty") else None
-            ),
-            response_format=(
-                p.response_format if p.HasField("response_format") else None
-            ),  # noqa
+            presence_penalty=p.presence_penalty
+            if p.HasField("presence_penalty")
+            else None,
+            response_format=p.response_format
+            if p.HasField("response_format")
+            else None,  # noqa
             seed=p.seed if p.HasField("seed") else None,
             stop=[_ for _ in p.stop] if p.HasField("stop") else None,
             temperature=p.temperature if p.HasField("temperature") else None,
@@ -152,9 +150,9 @@ class ModelDescriptorMapper:
                     max_tokens=d.max_tokens,
                     n=d.n,
                     presence_penalty=d.presence_penalty,
-                    response_format=(
-                        d.response_format if d.response_format else None
-                    ),  # noqa
+                    response_format=d.response_format
+                    if d.response_format
+                    else None,  # noqa
                     seed=d.seed,
                     stop=stop_list_value if d.stop else None,
                     temperature=d.temperature,

qwak/feature_store/_common/artifact_utils.py

@@ -45,9 +45,9 @@ class ArtifactsUploader:
         featureset_name: str,
         __instance_module_path__: str,
     ) -> Optional[ArtifactSpec]:
-        transformation_functions: Optional[List[Callable[..., Any]]] = (
-            transformation.get_functions()
-        )
+        transformation_functions: Optional[
+            List[Callable[..., Any]]
+        ] = transformation.get_functions()
         if transformation_functions is not None and transformation_functions:
             return ArtifactSpec(
                 artifact_name=featureset_name,

qwak/feature_store/data_sources/base.py

@@ -63,10 +63,10 @@ class BaseSource(ABC):
             uploaded_artifact_url = self._upload_artifact()
 
         if source_definition_path:
-            presign_url: (
-                str
-            ) = FeatureRegistryClient().get_datasource_source_code_presign_url(
-                ds_name=self.name
+            presign_url: str = (
+                FeatureRegistryClient().get_datasource_source_code_presign_url(
+                    ds_name=self.name
+                )
             )
             source_code_spec: SourceCodeSpec = (
                 SourceCodeSpecFactory.get_zip_source_code_spec(

qwak/feature_store/data_sources/batch/athena.py

@@ -118,9 +118,9 @@ class AthenaSource(JdbcSource):
             )
         )
 
-        time_partition_columns: Optional[TimePartitionColumns] = (
-            AthenaSource._extract_partition_column(proto_athena_source)
-        )
+        time_partition_columns: Optional[
+            TimePartitionColumns
+        ] = AthenaSource._extract_partition_column(proto_athena_source)
         workgroup: Optional[str] = (
             proto_athena_source.workgroup
             if proto_athena_source.HasField("workgroup")

qwak/feature_store/feature_sets/streaming.py

@@ -120,9 +120,9 @@ def feature_set(
             offline_scheduling_policy=offline_scheduling_policy,
         )
 
-        streaming_backfill: Optional[StreamingBackfill] = (
-            StreamingBackfill.get_streaming_backfill_from_function(function=function)
-        )
+        streaming_backfill: Optional[
+            StreamingBackfill
+        ] = StreamingBackfill.get_streaming_backfill_from_function(function=function)
 
         fs_name = name or function.__name__
         streaming_feature_set = StreamingFeatureSet(

qwak/feature_store/feature_sets/streaming_backfill.py

@@ -218,7 +218,7 @@ class StreamingBackfill:
 
     @staticmethod
     def _get_normalized_backfill_sources_spec(
-        data_sources: Union[List[str], List[DataSourceBackfillSpec]],
+        data_sources: Union[List[str], List[DataSourceBackfillSpec]]
     ) -> List[DataSourceBackfillSpec]:
         # reformat all data source specs to 'DataSourceBackfillSpec'
         return [

qwak/feature_store/online/client.py

@@ -182,12 +182,12 @@ class OnlineClient:
         )
         ordered_entities = [entity[0] for entity in ordered_entities_tuple]
 
-        request_chunks: List[Tuple[RequestedEntitiesMatrix, pd.DataFrame]] = (
-            OnlineClient._split_entities(
-                entity_names=ordered_entities,
-                population_df=df,
-                max_entities_per_split=max_keys_per_request,
-            )
+        request_chunks: List[
+            Tuple[RequestedEntitiesMatrix, pd.DataFrame]
+        ] = OnlineClient._split_entities(
+            entity_names=ordered_entities,
+            population_df=df,
+            max_entities_per_split=max_keys_per_request,
         )
 
         results: List[pd.DataFrame] = []

qwak/feature_store/sinks/streaming/factory.py

@@ -25,7 +25,7 @@ class StreamingSinkFactory:
     def get_streaming_sink(proto_streaming_sink: ProtoStreamingSink) -> BaseSink:
         sink_type = proto_streaming_sink.WhichOneof("sink_type")
 
-        auth_conf: BaseAuthentication  # noqa: F842
+        auth_conf: BaseAuthentication
         if sink_type == "kafka_sink":
             proto_kafka_sink: ProtoKafkaSink = proto_streaming_sink.kafka_sink
             auth_configuration: BaseAuthentication = cast(

qwak/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/git/git_strategy.py

@@ -163,9 +163,9 @@ def make_ssh_key_file(git_ssh_key: str) -> str:
 
 
 def add_ssh_file_to_env(ssh_key_file_path: str) -> None:
-    os.environ["GIT_SSH_COMMAND"] = (
-        f"ssh -i {ssh_key_file_path} -o StrictHostKeyChecking=no"
-    )
+    os.environ[
+        "GIT_SSH_COMMAND"
+    ] = f"ssh -i {ssh_key_file_path} -o StrictHostKeyChecking=no"
     os.environ["GIT_SSH"] = f"ssh -i {ssh_key_file_path} -o StrictHostKeyChecking=no"
 
 

qwak/llmops/provider/openai/provider.py

@@ -26,9 +26,9 @@ class OpenAIProvider:
         self.client = OpenAIClient()
 
     def _get_random_openai_api_key(self) -> Optional[str]:
-        openai_api_keys: List[OpenAIApiKeySystemSecret] = (
-            IntegrationUtils().get_openai_api_keys()
-        )
+        openai_api_keys: List[
+            OpenAIApiKeySystemSecret
+        ] = IntegrationUtils().get_openai_api_keys()
         if len(openai_api_keys) == 0:
             return None
 

qwak/model/tools/adapters/output.py

@@ -46,7 +46,7 @@ def get_output_adapter(
 
             first_result = (
                 return_result[0]
-                if isinstance(return_result, list) and len(return_result) > 0
+                if type(return_result) == list and len(return_result) > 0
                 else return_result
             )
             if issubclass(type(first_result), Message):

qwak/model/utils/feature_utils.py

@@ -30,19 +30,15 @@ def validate_and_sanitize_features_name(
     ecosystem_utils = EcosystemUtils()
     current_env_name = ecosystem_utils.get_current_environment_name()
     return [
-        (
-            cast(
-                FeatureStoreInput,
-                dataclasses.replace(
-                    feature,
-                    name=validate_and_sanitize_feature_name(
-                        feature.name, current_env_name
-                    ),
-                ),
-            )
-            if isinstance(feature, FeatureStoreInput)
-            else feature
+        cast(
+            FeatureStoreInput,
+            dataclasses.replace(
+                feature,
+                name=validate_and_sanitize_feature_name(feature.name, current_env_name),
+            ),
         )
+        if isinstance(feature, FeatureStoreInput)
+        else feature
         for feature in features
     ]
 

qwak/model_loggers/artifact_logger.py

@@ -92,13 +92,13 @@ def load_file(
         )
 
     model_id = validate_model(model_id)
-    download_url_response: (
-        GetBuildVersioningDownloadURLResponse
-    ) = BuildOrchestratorClient().get_build_versioning_download_url(
-        build_id=build_id,
-        model_id=model_id,
-        tag=tag,
-        tag_type=BuildVersioningTagsType.FILE_TAG_TYPE,
+    download_url_response: GetBuildVersioningDownloadURLResponse = (
+        BuildOrchestratorClient().get_build_versioning_download_url(
+            build_id=build_id,
+            model_id=model_id,
+            tag=tag,
+            tag_type=BuildVersioningTagsType.FILE_TAG_TYPE,
+        )
     )
 
     try:

qwak/tools/logger/logger.py

@@ -275,7 +275,7 @@ def set_file_handler_log_file(
     logger: logging.Logger, handler_name: str, log_file: Path
 ):
     existing_handler = get_handler_from_logger(logger, handler_name)
-    if not isinstance(existing_handler, RotatingFileHandler):
+    if type(existing_handler) != RotatingFileHandler:
         raise QwakException(
             f"Error in setting log file. Error message: handler of name {handler_name} is not a file logger handler"
         )

qwak_core-0.4.268.dist-info/METADATA

@@ -0,0 +1,53 @@
+Metadata-Version: 2.3
+Name: qwak-core
+Version: 0.4.268
+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
+Author: Qwak
+Author-email: info@qwak.com
+Requires-Python: >=3.9,<3.12
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Provides-Extra: feature-store
+Provides-Extra: local-core-dependencies
+Requires-Dist: PyYAML
+Requires-Dist: cachetools
+Requires-Dist: chevron (==0.14.0)
+Requires-Dist: cloudpickle (==2.2.1) ; extra == "feature-store"
+Requires-Dist: dacite (==1.8.1)
+Requires-Dist: dependency-injector (>=4.0)
+Requires-Dist: filelock
+Requires-Dist: frogml-storage (>=0.11.2)
+Requires-Dist: grpcio (>=1.57.0)
+Requires-Dist: grpcio-tools (>=1.47.0) ; (python_full_version >= "3.7.1" and python_version < "3.10") and (extra == "local-core-dependencies")
+Requires-Dist: grpcio-tools (>=1.56.2) ; (python_version >= "3.10") and (extra == "local-core-dependencies")
+Requires-Dist: joblib (>=1.3.2,<2.0.0)
+Requires-Dist: marshmallow-dataclass (>=8.5.8,<9.0.0)
+Requires-Dist: mypy-protobuf (>=3.0.0,<4.0.0) ; extra == "local-core-dependencies"
+Requires-Dist: protobuf (>=3.10,<4) ; python_full_version >= "3.7.1" and python_version < "3.10"
+Requires-Dist: protobuf (>=4.21.6) ; python_version >= "3.10"
+Requires-Dist: pyarrow (>=6.0.0) ; extra == "feature-store"
+Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
+Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
+Requires-Dist: python-jose[cryptography] (>=3.4.0)
+Requires-Dist: python-json-logger (>=2.0.2)
+Requires-Dist: requests
+Requires-Dist: retrying (==1.3.4)
+Requires-Dist: typeguard (>=2,<3)
+Project-URL: Home page, https://www.qwak.com/
+Description-Content-Type: text/markdown
+
+# Qwak Core
+
+Qwak is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+Qwak Core contains all the objects and tools necessary to use the Qwak Platform
+