File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Expand file tree Collapse file tree 1 file changed +9
-7
lines changed 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