From 1502bcea129ab98fb7d3c5e2f9fc8b8bd8577572 Mon Sep 17 00:00:00 2001 From: "Richard Kuo (Danswer)" Date: Tue, 28 Jan 2025 19:03:54 -0800 Subject: [PATCH] do teams too --- backend/onyx/connectors/teams/connector.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/backend/onyx/connectors/teams/connector.py b/backend/onyx/connectors/teams/connector.py index 166c43e42..a69ee57ae 100644 --- a/backend/onyx/connectors/teams/connector.py +++ b/backend/onyx/connectors/teams/connector.py @@ -186,17 +186,21 @@ class TeamsConnector(LoadConnector, PollConnector): teams_client_secret = credentials["teams_client_secret"] teams_directory_id = credentials["teams_directory_id"] + authority_url = f"https://login.microsoftonline.com/{teams_directory_id}" + self.msal_app = msal.ConfidentialClientApplication( + authority=authority_url, + client_id=teams_client_id, + client_credential=teams_client_secret, + ) + def _acquire_token_func() -> dict[str, Any]: """ Acquire token via MSAL """ - authority_url = f"https://login.microsoftonline.com/{teams_directory_id}" - app = msal.ConfidentialClientApplication( - authority=authority_url, - client_id=teams_client_id, - client_credential=teams_client_secret, - ) - token = app.acquire_token_for_client( + if self.msal_app is None: + raise RuntimeError("MSAL app is not initialized") + + token = self.msal_app.acquire_token_for_client( scopes=["https://graph.microsoft.com/.default"] ) return token