diff --git a/evaluation/data_collection.ipynb b/evaluation/data_collection.ipynb index 2acefbe9..f34c8e5b 100644 --- a/evaluation/data_collection.ipynb +++ b/evaluation/data_collection.ipynb @@ -40,7 +40,9 @@ "import os\n", "from groq import Groq\n", "from bs4 import BeautifulSoup\n", - "from html.parser import HTMLParser\n" + "from html.parser import HTMLParser\n", + "from dotenv import load_dotenv\n", + "load_dotenv()\n" ] }, { @@ -63,7 +65,7 @@ " Args:\n", " prompt (str): the prompt for groq completion\n", " \"\"\"\n", - " groq_API_key = \"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\"\n", + " groq_API_key = os.environ(\"GROQ_API_KEY\")\n", "\n", " client = Groq(\n", " api_key=groq_API_key,\n", @@ -181,7 +183,7 @@ "####################### biorxiv_data QA generation #######################\n", "\n", "df = pd.read_csv(\"./biorxiv_data_abstract.csv\")\n", - "client = OpenAI(api_key=\"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\")\n", + "client = OpenAI(api_key=os.getenv(\"OPENAPI_KEY\"))\n", "\n", "\n", "def generate_questions(df, client):\n", @@ -673,7 +675,7 @@ "\n", "################ QUESTION ANSWERS GENERATION ################\n", "\n", - "client = OpenAI(api_key=\"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\")\n", + "client = OpenAI(api_key=os.getenv(\"OPENAPI_KEY\"))\n", "\n", "\n", "def generate_questions(df, client):\n",