feat: OAuth client implementation for Python chatbot

Author: clareliguori

examples/chatbots/python/main.py

@@ -14,6 +14,10 @@
     LambdaFunctionUrlClient,
     LambdaFunctionUrlConfig,
 )
+from server_clients.interactive_oauth import (
+    InteractiveOAuthClient,
+    InteractiveOAuthConfig,
+)
 
 # Configure logging
 logging.basicConfig(
@@ -77,14 +81,20 @@ async def main() -> None:
     """Initialize and run the chat session."""
     config = Configuration()
     server_config = config.load_config("servers_config.json")
+
+    # Initialize stdio servers
     servers = [
         StdioServer(name, srv_config)
-        for name, srv_config in server_config["stdioServers"].items()
+        for name, srv_config in server_config.get("stdioServers", {}).items()
     ]
+
+    # Initialize lambda function servers
     servers.extend(
         [
             LambdaFunctionClient(name, srv_config)
-            for name, srv_config in server_config["lambdaFunctionServers"].items()
+            for name, srv_config in server_config.get(
+                "lambdaFunctionServers", {}
+            ).items()
         ]
     )
 
@@ -98,6 +108,14 @@ async def main() -> None:
         ]
     )
 
+    # Initialize interactive OAuth servers
+    servers.extend(
+        [
+            InteractiveOAuthClient(name, InteractiveOAuthConfig(**srv_config))
+            for name, srv_config in server_config.get("oAuthServers", {}).items()
+        ]
+    )
+
     llm_client = LLMClient(config.bedrock_client, config.model_id)
     chat_session = ChatSession(servers, llm_client)
     await chat_session.start()