Update mentions_dashboard.py

mentions_dashboard.py

@@ -11,6 +11,8 @@ from requests_oauthlib import OAuth2Session
 from textblob import TextBlob
 import matplotlib.pyplot as plt
 
+from sessions import create_session
+
 
 def extract_text_from_commentary(commentary):
     return re.sub(r"{.*?}", "", commentary).strip()
@@ -21,13 +23,12 @@ def analyze_sentiment(text):
 
 
 def generate_mentions_dashboard(comm_client_id, comm_token_dict):
-    linkedin = OAuth2Session(comm_client_id, token=comm_token_dict)
 
     org_urn = "urn:li:organization:19010008"
     encoded_urn = quote(org_urn, safe='')
 
-    linkedin.headers.update({
-        "LinkedIn-Version": "202502",
+    session = create_session(comm_client_id, token=comm_token_dict)
+    session.headers.update({
         "X-Restli-Protocol-Version": "2.0.0"
     })
 
@@ -44,7 +45,7 @@ def generate_mentions_dashboard(comm_client_id, comm_token_dict):
 
     while True:
         url = f"{base_url}&start={start}"
-        resp = linkedin.get(url)
+        resp = session.get(url)
         if resp.status_code != 200:
             break
 
@@ -67,7 +68,7 @@ def generate_mentions_dashboard(comm_client_id, comm_token_dict):
             continue
         encoded_share_urn = quote(share_urn, safe='')
         share_url = f"https://api.linkedin.com/rest/posts/{encoded_share_urn}"
-        response = linkedin.get(share_url)
+        response = session.get(share_url)
 
         if response.status_code != 200:
             continue