Spaces:
Sleeping
Sleeping
Commit
·
ebc10b6
1
Parent(s):
7c73aca
updating openai key from env
Browse files- 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"]
|