File tree Expand file tree Collapse file tree 3 files changed +8
-7
lines changed Expand file tree Collapse file tree 3 files changed +8
-7
lines changed Original file line number Diff line number Diff line change @@ -1088,7 +1088,7 @@ def list_hf_models(query: str) -> List[str]:
1088
1088
raise format_hf_custom_error_message (err ) from err
1089
1089
1090
1090
1091
- def generate_tei_cmd_vars (os_path : str ) -> List [str ]:
1091
+ def generate_tei_cmd_var (os_path : str ) -> List [str ]:
1092
1092
"""This utility functions generates CMD params for Text Embedding Inference container. Only the
1093
1093
essential parameters for OCI model deployment are added, defaults are used for the rest.
1094
1094
Parameters
@@ -1098,15 +1098,16 @@ def generate_tei_cmd_vars(os_path: str) -> List[str]:
1098
1098
1099
1099
Returns
1100
1100
-------
1101
- List of command line arguments
1101
+ cmd_var:
1102
+ List of command line arguments
1102
1103
"""
1103
1104
1104
1105
cmd_prefix = "--"
1105
- cmd_vars = [
1106
+ cmd_var = [
1106
1107
cmd_prefix + TextEmbeddingInferenceContainerParams .MODEL_ID ,
1107
1108
f"{ AQUA_MODEL_DEPLOYMENT_FOLDER } { ObjectStorageDetails .from_path (os_path .rstrip ('/' )).filepath } /" ,
1108
1109
cmd_prefix + TextEmbeddingInferenceContainerParams .PORT ,
1109
1110
TEI_CONTAINER_DEFAULT_HOST ,
1110
1111
]
1111
1112
1112
- return cmd_vars
1113
+ return cmd_var
Original file line number Diff line number Diff line change 21
21
_build_resource_identifier ,
22
22
copy_model_config ,
23
23
create_word_icon ,
24
- generate_tei_cmd_vars ,
24
+ generate_tei_cmd_var ,
25
25
get_artifact_path ,
26
26
get_hf_model_info ,
27
27
list_os_files_with_extension ,
@@ -713,7 +713,7 @@ def _create_model_catalog_entry(
713
713
category = "Other" ,
714
714
)
715
715
716
- cmd_vars = generate_tei_cmd_vars (os_path )
716
+ cmd_vars = generate_tei_cmd_var (os_path )
717
717
metadata .add (
718
718
key = AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME ,
719
719
value = "," .join (cmd_vars ),
Original file line number Diff line number Diff line change @@ -263,7 +263,7 @@ def create(
263
263
cmd_var_string = aqua_model .custom_metadata_list .get (
264
264
AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME
265
265
).value
266
- cmd_var .append (cmd_var_string .split ("," ))
266
+ cmd_var .extend (cmd_var_string .split ("," ))
267
267
except ValueError as err :
268
268
raise AquaValueError (
269
269
f"{ AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME } key is not available in the custom metadata "
You can’t perform that action at this time.
0 commit comments