Skip to content

Commit 13d6f2a

Browse files
committed
feat: Automated OAuth client with client creds for integ tests
1 parent 72898f8 commit 13d6f2a

File tree

2 files changed

+588
-0
lines changed

2 files changed

+588
-0
lines changed

e2e_tests/python/main.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,10 @@
1414
LambdaFunctionUrlClient,
1515
LambdaFunctionUrlConfig,
1616
)
17+
from server_clients.automated_oauth import (
18+
AutomatedOAuthClient,
19+
AutomatedOAuthConfig,
20+
)
1721

1822
# Configure logging
1923
logging.basicConfig(
@@ -98,6 +102,15 @@ async def main() -> None:
98102
].items()
99103
]
100104
)
105+
106+
# Add automated OAuth servers if they exist in config
107+
if "oAuthServers" in server_config:
108+
servers.extend(
109+
[
110+
AutomatedOAuthClient(name, AutomatedOAuthConfig(**srv_config))
111+
for name, srv_config in server_config["oAuthServers"].items()
112+
]
113+
)
101114
llm_client = LLMClient(config.bedrock_client, config.model_id)
102115
user_utterances = [
103116
"Hello!",

0 commit comments

Comments
 (0)