Skip to content

Commit ecad23e

Browse files
authored
Fix bedrock clients model-id (#37)
1 parent 803e5e5 commit ecad23e

3 files changed

Lines changed: 24 additions & 3 deletions

File tree

packages/uipath_langchain_client/CHANGELOG.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,11 @@
22

33
All notable changes to `uipath_langchain_client` will be documented in this file.
44

5+
## [1.2.7] - 2026-02-26
6+
7+
### Fix
8+
- Fix Bedrock clients model_id
9+
510
## [1.2.6] - 2026-02-26
611

712
### Refactor
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
__title__ = "UiPath LangChain Client"
22
__description__ = "A Python client for interacting with UiPath's LLM services via LangChain."
3-
__version__ = "1.2.6"
3+
__version__ = "1.2.7"

packages/uipath_langchain_client/src/uipath_langchain_client/clients/bedrock/chat_models.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)