File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
apps/models_provider/impl/aws_bedrock_model_provider/model Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 1- from langchain_community . embeddings import BedrockEmbeddings
1+ from langchain_aws import BedrockEmbeddings
22
33from models_provider .base_model_provider import MaxKBBaseModel
44from typing import Dict , List
Original file line number Diff line number Diff line change 33from typing import Dict , List
44
55from botocore .config import Config
6- from langchain_community . chat_models import BedrockChat
6+ from langchain_aws import ChatBedrock
77from langchain_core .messages import BaseMessage , get_buffer_string
88
99from 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 ():
You can’t perform that action at this time.
0 commit comments