We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 72898f8 commit 13d6f2aCopy full SHA for 13d6f2a
e2e_tests/python/main.py
@@ -14,6 +14,10 @@
14
LambdaFunctionUrlClient,
15
LambdaFunctionUrlConfig,
16
)
17
+from server_clients.automated_oauth import (
18
+ AutomatedOAuthClient,
19
+ AutomatedOAuthConfig,
20
+)
21
22
# Configure logging
23
logging.basicConfig(
@@ -98,6 +102,15 @@ async def main() -> None:
98
102
].items()
99
103
]
100
104
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
+ )
101
114
llm_client = LLMClient(config.bedrock_client, config.model_id)
115
user_utterances = [
116
"Hello!",
0 commit comments