Coverage for mindsdb / interfaces / agents / providers.py: 13%

1import importlib.util 

2 

3 

4def get_bedrock_chat_model(): 

5 try: 

6 from langchain_aws.chat_models import ChatBedrock 

7 except ModuleNotFoundError: 

8 raise RuntimeError("bedrock connector is not installed. Please install it with `pip install langchain-aws`") 

9 

10 if not importlib.util.find_spec("transformers"): 

11 raise RuntimeError( 

12 "`transformers` module is required for bedrock to count tokens. Please install it with `pip install transformers`" 

13 ) 

14 

15 class ChatBedrockPatched(ChatBedrock): 

16 def _prepare_input_and_invoke(self, *args, **kwargs): 

17 kwargs.pop("stop_sequences", None) 

18 return super()._prepare_input_and_invoke(*args, **kwargs) 

19 

20 return ChatBedrockPatched