updating model_download function name, autogen files (#155)

Author: ian-scale

launch/api_client/apis/tags/default_api.py

@@ -187,8 +187,8 @@ class DefaultApi(
 CreateTriggerV1TriggersPost,
 DeleteFileV1FilesFileIdDelete,
 DeleteModelEndpointV1ModelEndpointsModelEndpointIdDelete,
-DownloadModelEndpointV1LlmModelEndpointsDownloadPost,
 DeleteTriggerV1TriggersTriggerIdDelete,
+DownloadModelEndpointV1LlmModelEndpointsDownloadPost,
 GetAsyncInferenceTaskV1AsyncTasksTaskIdGet,
 GetBatchJobV1BatchJobsBatchJobIdGet,
 GetDockerImageBatchJobModelBundleV1DockerImageBatchJobBundlesDockerImageBatchJobBundleIdGet,

launch/api_client/model/get_llm_model_endpoint_v1_response.py

@@ -36,6 +36,7 @@ class MetaOapg:
 "name",
 "id",
 "source",
+"status",
  }
 
 classproperties:
@@ -51,6 +52,10 @@ def inference_framework() -> typing.Type["LLMInferenceFramework"]:
 defsource() ->typing.Type["LLMSource"]:
 returnLLMSource
 
+@staticmethod
+defstatus() ->typing.Type["ModelEndpointStatus"]:
+returnModelEndpointStatus
+
 inference_framework_image_tag=schemas.StrSchema
 model_name=schemas.StrSchema
 num_shards=schemas.IntSchema
@@ -68,6 +73,7 @@ def spec() -> typing.Type["GetModelEndpointV1Response"]:
 "inference_framework": inference_framework,
 "name": name,
 "source": source,
+"status": status,
 "inference_framework_image_tag": inference_framework_image_tag,
 "model_name": model_name,
 "num_shards": num_shards,
@@ -79,6 +85,7 @@ def spec() -> typing.Type["GetModelEndpointV1Response"]:
 name: MetaOapg.properties.name
 id: MetaOapg.properties.id
 source: "LLMSource"
+status: "ModelEndpointStatus"
 
 @typing.overload
 def__getitem__(self, name: typing_extensions.Literal["id"]) ->MetaOapg.properties.id:
@@ -96,6 +103,10 @@ def __getitem__(self, name: typing_extensions.Literal["name"]) -> MetaOapg.prope
 def__getitem__(self, name: typing_extensions.Literal["source"]) ->"LLMSource":
  ...
 
+@typing.overload
+def__getitem__(self, name: typing_extensions.Literal["status"]) ->"ModelEndpointStatus":
+ ...
+
 @typing.overload
 def__getitem__(
 self, name: typing_extensions.Literal["inference_framework_image_tag"]
@@ -130,6 +141,7 @@ def __getitem__(
 "inference_framework",
 "name",
 "source",
+"status",
 "inference_framework_image_tag",
 "model_name",
 "num_shards",
@@ -158,6 +170,10 @@ def get_item_oapg(self, name: typing_extensions.Literal["name"]) -> MetaOapg.pro
 defget_item_oapg(self, name: typing_extensions.Literal["source"]) ->"LLMSource":
  ...
 
+@typing.overload
+defget_item_oapg(self, name: typing_extensions.Literal["status"]) ->"ModelEndpointStatus":
+ ...
+
 @typing.overload
 defget_item_oapg(
 self, name: typing_extensions.Literal["inference_framework_image_tag"]
@@ -198,6 +214,7 @@ def get_item_oapg(
 "inference_framework",
 "name",
 "source",
+"status",
 "inference_framework_image_tag",
 "model_name",
 "num_shards",
@@ -225,6 +242,7 @@ def __new__(
 str,
  ],
 source: "LLMSource",
+status: "ModelEndpointStatus",
 inference_framework_image_tag: typing.Union[
 MetaOapg.properties.inference_framework_image_tag, str, schemas.Unset
  ] =schemas.unset,
@@ -257,6 +275,7 @@ def __new__(
 name=name,
 id=id,
 source=source,
+status=status,
 inference_framework_image_tag=inference_framework_image_tag,
 model_name=model_name,
 num_shards=num_shards,
@@ -274,4 +293,5 @@ def __new__(
 LLMInferenceFramework,
 )
 fromlaunch.api_client.model.llm_sourceimportLLMSource
+fromlaunch.api_client.model.model_endpoint_statusimportModelEndpointStatus
 fromlaunch.api_client.model.quantizationimportQuantization

launch/api_client/model/get_llm_model_endpoint_v1_response.pyi

@@ -34,6 +34,7 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 "name",
 "id",
 "source",
+"status",
  }
 
 classproperties:
@@ -47,6 +48,9 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 @staticmethod
 defsource() ->typing.Type["LLMSource"]:
 returnLLMSource
+@staticmethod
+defstatus() ->typing.Type["ModelEndpointStatus"]:
+returnModelEndpointStatus
 inference_framework_image_tag=schemas.StrSchema
 model_name=schemas.StrSchema
 num_shards=schemas.IntSchema
@@ -62,6 +66,7 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 "inference_framework": inference_framework,
 "name": name,
 "source": source,
+"status": status,
 "inference_framework_image_tag": inference_framework_image_tag,
 "model_name": model_name,
 "num_shards": num_shards,
@@ -72,6 +77,7 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 name: MetaOapg.properties.name
 id: MetaOapg.properties.id
 source: "LLMSource"
+status: "ModelEndpointStatus"
 
 @typing.overload
 def__getitem__(self, name: typing_extensions.Literal["id"]) ->MetaOapg.properties.id: ...
@@ -82,6 +88,8 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 @typing.overload
 def__getitem__(self, name: typing_extensions.Literal["source"]) ->"LLMSource": ...
 @typing.overload
+def__getitem__(self, name: typing_extensions.Literal["status"]) ->"ModelEndpointStatus": ...
+@typing.overload
 def__getitem__(
 self, name: typing_extensions.Literal["inference_framework_image_tag"]
  ) ->MetaOapg.properties.inference_framework_image_tag: ...
@@ -103,6 +111,7 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 "inference_framework",
 "name",
 "source",
+"status",
 "inference_framework_image_tag",
 "model_name",
 "num_shards",
@@ -123,6 +132,8 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 @typing.overload
 defget_item_oapg(self, name: typing_extensions.Literal["source"]) ->"LLMSource": ...
 @typing.overload
+defget_item_oapg(self, name: typing_extensions.Literal["status"]) ->"ModelEndpointStatus": ...
+@typing.overload
 defget_item_oapg(
 self, name: typing_extensions.Literal["inference_framework_image_tag"]
  ) ->typing.Union[MetaOapg.properties.inference_framework_image_tag, schemas.Unset]: ...
@@ -152,6 +163,7 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 "inference_framework",
 "name",
 "source",
+"status",
 "inference_framework_image_tag",
 "model_name",
 "num_shards",
@@ -178,6 +190,7 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 str,
  ],
 source: "LLMSource",
+status: "ModelEndpointStatus",
 inference_framework_image_tag: typing.Union[
 MetaOapg.properties.inference_framework_image_tag, str, schemas.Unset
  ] =schemas.unset,
@@ -210,6 +223,7 @@ class GetLLMModelEndpointV1Response(schemas.DictSchema):
 name=name,
 id=id,
 source=source,
+status=status,
 inference_framework_image_tag=inference_framework_image_tag,
 model_name=model_name,
 num_shards=num_shards,
@@ -224,4 +238,5 @@ from launch_client.model.get_model_endpoint_v1_response import (
 )
 fromlaunch_client.model.llm_inference_frameworkimportLLMInferenceFramework
 fromlaunch_client.model.llm_sourceimportLLMSource
+fromlaunch_client.model.model_endpoint_statusimportModelEndpointStatus
 fromlaunch_client.model.quantizationimportQuantization

launch/client.py

@@ -3252,7 +3252,7 @@ def get_file_content(
 
 returnresp
 
-defdownload_model_weights(self, model_name: str, download_format: str="huggingface") ->ModelDownloadResponse:
+defmodel_download(self, model_name: str, download_format: str="hugging_face") ->ModelDownloadResponse:
 """
  download a finetuned model