Skip to content

Commit 2a32193

Browse files
committed
fix: update imports to use langchain_aws for Bedrock embeddings and chat models
1 parent 8a9de59 commit 2a32193

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

apps/models_provider/impl/aws_bedrock_model_provider/model/embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from langchain_community.embeddings import BedrockEmbeddings
1+
from langchain_aws import BedrockEmbeddings
22

33
from models_provider.base_model_provider import MaxKBBaseModel
44
from typing import Dict, List

apps/models_provider/impl/aws_bedrock_model_provider/model/llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from typing import Dict, List
44

55
from botocore.config import Config
6-
from langchain_community.chat_models import BedrockChat
6+
from langchain_aws import ChatBedrock
77
from langchain_core.messages import BaseMessage, get_buffer_string
88

99
from common.config.tokenizer_manage_config import TokenizerManage
@@ -33,7 +33,7 @@ def get_max_tokens_keyword(model_name):
3333
return 'max_tokens'
3434

3535

36-
class BedrockModel(MaxKBBaseModel, BedrockChat):
36+
class BedrockModel(MaxKBBaseModel, ChatBedrock):
3737

3838
@staticmethod
3939
def is_cache_model():

0 commit comments

Comments
 (0)