@@ -30,9 +30,17 @@ class UiPathChatBedrockConverse(UiPathBaseChatModel, ChatBedrockConverse):
3030 client : Any = WrappedBotoClient ()
3131 bedrock_client : Any = WrappedBotoClient ()
3232
33+ @model_validator (mode = "before" )
34+ @classmethod
35+ def setup_model_id (cls , values : Any ) -> Any :
36+ if isinstance (values , dict ) and "model_id" not in values :
37+ model = values .get ("model" ) or values .get ("model_name" )
38+ if model :
39+ values = {** values , "model_id" : model }
40+ return values
41+
3342 @model_validator (mode = "after" )
3443 def setup_uipath_client (self ) -> Self :
35- self .model_id = self .model_name
3644 self .client = WrappedBotoClient (self .uipath_sync_client )
3745 return self
3846
@@ -51,9 +59,17 @@ class UiPathChatBedrock(UiPathBaseChatModel, ChatBedrock):
5159 client : Any = WrappedBotoClient ()
5260 bedrock_client : Any = WrappedBotoClient ()
5361
62+ @model_validator (mode = "before" )
63+ @classmethod
64+ def setup_model_id (cls , values : Any ) -> Any :
65+ if isinstance (values , dict ) and "model_id" not in values :
66+ model = values .get ("model" ) or values .get ("model_name" )
67+ if model :
68+ values = {** values , "model_id" : model }
69+ return values
70+
5471 @model_validator (mode = "after" )
5572 def setup_uipath_client (self ) -> Self :
56- self .model_id = self .model_name
5773 self .client = WrappedBotoClient (self .uipath_sync_client )
5874 return self
5975
0 commit comments