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 4
4
5
5
from adalflow .core .types import ModelType , GeneratorOutput
6
6
from adalflow .components .model_client import BedrockAPIClient
7
- from adalflow import setup_env
8
7
9
8
10
9
def getenv_side_effect (key ):
@@ -21,12 +20,15 @@ def getenv_side_effect(key):
21
20
class TestBedrockClient (unittest .TestCase ):
22
21
def setUp (self ):
23
22
# 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
+ ):
30
32
# Now patch os.getenv to return mocked environment variable values
31
33
with patch ("os.getenv" , side_effect = getenv_side_effect ):
32
34
self .client = BedrockAPIClient ()
You can’t perform that action at this time.
0 commit comments