Load secrets as dotenv

edenia · Apr 27, 2024 · 28db2dc · 28db2dc
1 parent ce103c8
commit 28db2dc
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/analysis/ethereum_smart_contracts/GPT_finetuning_data_collection.ipynb b/analysis/ethereum_smart_contracts/GPT_finetuning_data_collection.ipynb
@@ -31,6 +31,11 @@
     "import pandas as pd\n",
     "from tqdm import tqdm\n",
     "import requests\n",
+    "from dotenv import load_dotenv\n",
+    "\n",
+    "# Load secrets\n",
+    "dotenv_path = '../.env'\n",
+    "load_dotenv(dotenv_path)\n",
     "\n",
     "tqdm.pandas()\n",
     "# disable warning logs from evmdasm tool\n",
@@ -39,10 +44,10 @@
     "blockchains = [\"ethereum\", \"polygon\", \"bsc\"]\n",
     "current_blockchain = \"ethereum\"\n",
     "\n",
-    "ZETTABLOCK_API_KEY = \"519a0243-0add-4593-9adf-85c6a6a5ef22\"\n",
-    "ZETTABLOCK_URL = \"https://api.zettablock.com/api/v1/databases/realtimeDB/queries\"\n",
-    "ZETTABLOCK_DATA_LAKE_ENDPOINT = \"https://api.zettablock.com/api/v1/databases/AwsDataCatalog/queries\"\n",
-    "EXECUTE_URL = \"https://api.zettablock.com/api/v1/queries/\"\n",
+    "ZETTABLOCK_API_KEY = os.environ.get(\"ZETTABLOCK_API_KEY\")\n",
+    "ZETTABLOCK_URL = os.environ.get(\"ZETTABLOCK_URL\")\n",
+    "ZETTABLOCK_DATA_LAKE_ENDPOINT = os.environ.get(\"ZETTABLOCK_DATA_LAKE_ENDPOINT\")\n",
+    "EXECUTE_URL = os.environ.get(\"EXECUTE_URL\")\n",
     "\n",
     "headers = {\n",
     "    \"accept\": \"application/json\",\n",

diff --git a/analysis/ethereum_smart_contracts/GTP_pretraining_data_collection.ipynb b/analysis/ethereum_smart_contracts/GTP_pretraining_data_collection.ipynb
@@ -30,6 +30,11 @@
     "import pandas as pd\n",
     "from tqdm import tqdm\n",
     "from web3 import Web3\n",
+    "from dotenv import load_dotenv\n",
+    "\n",
+    "# Load secrets\n",
+    "dotenv_path = '../.env'\n",
+    "load_dotenv(dotenv_path)\n",
     "\n",
     "tqdm.pandas()\n",
     "# disable warning logs from evmdasm tool\n",
@@ -39,7 +44,7 @@
     "processed_data_file = \"/data/forta/ethereum/text/pretraining/big_pretrain_data\"\n",
     "\n",
     "# Your Zettablock API key\n",
-    "API_KEY = \"XXXXX-XXXX-XXXX-XXX-XXXXXXXX\"\n",
+    "API_KEY = os.environ.get(\"ZETTABLOCK_API_KEY\")\n",
     "headers = {\n",
     "    \"accept\": \"application/json\",\n",
     "    \"content-type\": \"application/json\",\n",
@@ -48,7 +53,7 @@
     "}\n",
     "\n",
     "# Zettablock endpoint\n",
-    "ZETTABLOCK_DATA_LAKE_ENDPOINT = \"https://api.zettablock.com/api/v1/databases/AwsDataCatalog/queries\"\n",
+    "ZETTABLOCK_DATA_LAKE_ENDPOINT = os.environ.get(\"ZETTABLOCK_DATA_LAKE_ENDPOINT\")\n",
     "\n",
     "# Configure the blockchains we are interested in\n",
     "blockchains = [\"ethereum_mainnet\", \"polygon_mainnet\", \"bsc_mainnet\"]\n",