3131
3232def _get_model_info (
3333 model_name : str ,
34+ * ,
3435 client_settings : UiPathBaseSettings ,
3536 byo_connection_id : str | None = None ,
3637) -> dict [str , Any ]:
@@ -67,6 +68,7 @@ def _get_model_info(
6768
6869def get_chat_model (
6970 model_name : str ,
71+ * ,
7072 byo_connection_id : str | None = None ,
7173 client_settings : UiPathBaseSettings | None = None ,
7274 client_type : Literal ["passthrough" , "normalized" ] = "passthrough" ,
@@ -88,7 +90,9 @@ def get_chat_model(
8890 ValueError: If the model is not found in available models or vendor is not supported
8991 """
9092 client_settings = client_settings or get_default_client_settings ()
91- model_info = _get_model_info (model_name , client_settings , byo_connection_id )
93+ model_info = _get_model_info (
94+ model_name , client_settings = client_settings , byo_connection_id = byo_connection_id
95+ )
9296 is_uipath_owned = model_info .get ("modelSubscriptionType" ) == "UiPathOwned"
9397 if not is_uipath_owned :
9498 client_settings .validate_byo_model (model_info )
@@ -215,6 +219,7 @@ def get_chat_model(
215219
216220def get_embedding_model (
217221 model_name : str ,
222+ * ,
218223 byo_connection_id : str | None = None ,
219224 client_settings : UiPathBaseSettings | None = None ,
220225 client_type : Literal ["passthrough" , "normalized" ] = "passthrough" ,
@@ -243,7 +248,9 @@ def get_embedding_model(
243248 >>> vectors = embeddings.embed_documents(["Hello world"])
244249 """
245250 client_settings = client_settings or get_default_client_settings ()
246- model_info = _get_model_info (model_name , client_settings , byo_connection_id )
251+ model_info = _get_model_info (
252+ model_name , client_settings = client_settings , byo_connection_id = byo_connection_id
253+ )
247254 is_uipath_owned = model_info .get ("modelSubscriptionType" ) == "UiPathOwned"
248255 if not is_uipath_owned :
249256 client_settings .validate_byo_model (model_info )
0 commit comments