hugging2ido commited on
Commit
ebc10b6
·
1 Parent(s): 7c73aca

updating openai key from env

Browse files
Files changed (1) hide show
  1. src/action.py +3 -1
src/action.py CHANGED
@@ -7,7 +7,7 @@ import argparse
7
  import yaml
8
  import os
9
  from dotenv import load_dotenv
10
-
11
  from relevancy import generate_relevance_score, process_subject_fields
12
  from download_new_papers import get_papers
13
 
@@ -280,8 +280,10 @@ if __name__ == "__main__":
280
  args = parser.parse_args()
281
  with open(args.config, "r") as f:
282
  config = yaml.safe_load(f)
 
283
  if "OPENAI_API_KEY" not in os.environ:
284
  raise RuntimeError("No openai api key found")
 
285
 
286
  topic = config["topic"]
287
  categories = config["categories"]
 
7
  import yaml
8
  import os
9
  from dotenv import load_dotenv
10
+ import openai
11
  from relevancy import generate_relevance_score, process_subject_fields
12
  from download_new_papers import get_papers
13
 
 
280
  args = parser.parse_args()
281
  with open(args.config, "r") as f:
282
  config = yaml.safe_load(f)
283
+
284
  if "OPENAI_API_KEY" not in os.environ:
285
  raise RuntimeError("No openai api key found")
286
+ openai.api_key = os.environ.get("OPENAI_API_KEY")
287
 
288
  topic = config["topic"]
289
  categories = config["categories"]