fix: get lk secrets from extension first

jupyter-naas · May 3, 2024 · ed9afd4 · ed9afd4
1 parent a7188cd
commit ed9afd4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/models/growth-engine/__pipeline__.ipynb b/models/growth-engine/__pipeline__.ipynb
@@ -129,8 +129,8 @@
     "entity_name = pload(os.path.join(datalake_dir, \"entities\", entity_index), \"entity_name\") or \"\"\n",
     "emails = pload(os.path.join(datalake_dir, \"entities\", entity_index), \"emails\") or []\n",
     "linkedin_url = pload(os.path.join(datalake_dir, \"entities\", entity_index), \"linkedin_url\") or \"\"\n",
-    "li_at = naas.secret.get(\"LINKEDIN_LI_AT\")\n",
-    "JSESSIONID = naas.secret.get(\"LINKEDIN_JSESSIONID\")\n",
+    "li_at = naas.secret.get(\"li_at\") or naas.secret.get(\"LINKEDIN_LI_AT\")\n",
+    "JSESSIONID = naas.secret.get(\"JSESSIONID\") or naas.secret.get(\"LINKEDIN_JSESSIONID\")\n",
     "entity_dir = pload(os.path.join(datalake_dir, \"entities\", entity_index), \"entity_dir\") or \"\"\n",
     "\n",
     "# Google Sheets\n",

diff --git a/models/sales-engine/__pipeline__.ipynb b/models/sales-engine/__pipeline__.ipynb
@@ -133,8 +133,8 @@
     "entity_name = pload(os.path.join(datalake_dir, \"entities\", \"0\"), \"entity_name\") or \"\"\n",
     "emails = pload(os.path.join(datalake_dir, \"entities\", \"0\"), \"emails\") or []\n",
     "linkedin_url = pload(os.path.join(datalake_dir, \"entities\", \"0\"), \"linkedin_url\") or \"\"\n",
-    "li_at = naas.secret.get(\"LINKEDIN_LI_AT\")\n",
-    "JSESSIONID = naas.secret.get(\"LINKEDIN_JSESSIONID\")\n",
+    "li_at = naas.secret.get(\"li_at\") or naas.secret.get(\"LINKEDIN_LI_AT\")\n",
+    "JSESSIONID = naas.secret.get(\"JSESSIONID\") or naas.secret.get(\"LINKEDIN_JSESSIONID\")\n",
     "entity_dir = pload(os.path.join(datalake_dir, \"entities\", \"0\"), \"entity_dir\") or \"\"\n",
     "\n",
     "# Engine\n",