highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-09 20:55:06 +02:00
Code Issues Packages Projects Releases Wiki Activity

app should be initialized once per connector

Browse Source
This commit is contained in:
Richard Kuo (Danswer)
2025-01-28 15:55:52 -08:00
parent 98abd7d3fa
commit 1308b6cbe8
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
backend/onyx/connectors/sharepoint/connector.py
View File

@@ -73,6 +73,7 @@ class SharepointConnector(LoadConnector, PollConnector):
self.batch_size = batch_size self.batch_size = batch_size
self.graph_client: GraphClient | None = None self.graph_client: GraphClient | None = None
self.site_data: list[SiteData] = self._extract_site_and_folder(sites) self.site_data: list[SiteData] = self._extract_site_and_folder(sites)
self.msal_app: msal.ConfidentialClientApplication | None = None
@staticmethod @staticmethod
def _extract_site_and_folder(site_urls: list[str]) -> list[SiteData]: def _extract_site_and_folder(site_urls: list[str]) -> list[SiteData]:
@@ -182,17 +183,21 @@ class SharepointConnector(LoadConnector, PollConnector):
sp_client_secret = credentials["sp_client_secret"] sp_client_secret = credentials["sp_client_secret"]
sp_directory_id = credentials["sp_directory_id"] sp_directory_id = credentials["sp_directory_id"]
authority_url = f"https://login.microsoftonline.com/{sp_directory_id}"
self.msal_app = msal.ConfidentialClientApplication(
authority=authority_url,
client_id=sp_client_id,
client_credential=sp_client_secret,
)
def _acquire_token_func() -> dict[str, Any]: def _acquire_token_func() -> dict[str, Any]:
""" """
Acquire token via MSAL Acquire token via MSAL
""" """
authority_url = f"https://login.microsoftonline.com/{sp_directory_id}" if self.msal_app is None:
app = msal.ConfidentialClientApplication( raise RuntimeError("MSAL app is not initialized")
authority=authority_url,
client_id=sp_client_id, token = self.msal_app.acquire_token_for_client(
client_credential=sp_client_secret,
)
token = app.acquire_token_for_client(
scopes=["https://graph.microsoft.com/.default"] scopes=["https://graph.microsoft.com/.default"]
) )
return token return token