File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 44
55from adalflow .core .types import ModelType , GeneratorOutput
66from adalflow .components .model_client import BedrockAPIClient
7- from adalflow import setup_env
87
98
109def getenv_side_effect (key ):
@@ -21,12 +20,15 @@ def getenv_side_effect(key):
2120class TestBedrockClient (unittest .TestCase ):
2221 def setUp (self ):
2322 # Patch os.environ to ensure all environment variables are set
24- with patch .dict (os .environ , {
25- "AWS_ACCESS_KEY_ID" : "fake_api_key" ,
26- "AWS_SECRET_ACCESS_KEY" : "fake_api_key" ,
27- "AWS_REGION_NAME" : "fake_api_key" ,
28- "AWS_PROFILE_NAME" : "fake_profile" # Adding additional profile if needed
29- }):
23+ with patch .dict (
24+ os .environ ,
25+ {
26+ "AWS_ACCESS_KEY_ID" : "fake_api_key" ,
27+ "AWS_SECRET_ACCESS_KEY" : "fake_api_key" ,
28+ "AWS_REGION_NAME" : "fake_api_key" ,
29+ "AWS_PROFILE_NAME" : "fake_profile" , # Adding additional profile if needed
30+ },
31+ ):
3032 # Now patch os.getenv to return mocked environment variable values
3133 with patch ("os.getenv" , side_effect = getenv_side_effect ):
3234 self .client = BedrockAPIClient ()
You can’t perform that action at this time.
0 commit comments