From 4ab1f355153cc98c8dada2bcd802e37da6200a87 Mon Sep 17 00:00:00 2001 From: Yohann Paris Date: Fri, 18 Aug 2023 14:02:07 -0400 Subject: [PATCH 1/2] ignore PyCharm settings --- .gitignore | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.gitignore b/.gitignore index 1da5499..e1ce62a 100644 --- a/.gitignore +++ b/.gitignore @@ -1,6 +1,9 @@ # OSX .DS_Store +# IDE +.idea + # Local environment .env api.env From 3377ee620d09c35b5784408b0ddfe164a280a4e5 Mon Sep 17 00:00:00 2001 From: Yohann Paris Date: Fri, 18 Aug 2023 14:02:30 -0400 Subject: [PATCH 2/2] Remove the needs of OPEN_AI_KEY --- tests/conftest.py | 1 - workers/operations.py | 10 ++-------- 2 files changed, 2 insertions(+), 9 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index c39f495..700b9a3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -9,7 +9,6 @@ os.environ["TA1_UNIFIED_URL"] = "https://ta1:5" os.environ["MIT_TR_URL"] = "http://mit:10" os.environ["TDS_URL"] = "http://tds:15" -os.environ["OPENAI_API_KEY"] = "foo" os.environ["LOG_LEVEL"] = "INFO" diff --git a/workers/operations.py b/workers/operations.py index 43e7133..782f647 100644 --- a/workers/operations.py +++ b/workers/operations.py @@ -227,7 +227,6 @@ def pdf_extractions(*args, **kwargs): def data_card(*args, **kwargs): - openai_key = os.getenv("OPENAI_API_KEY") dataset_id = kwargs.get("dataset_id") artifact_id = kwargs.get("artifact_id") @@ -251,8 +250,6 @@ def data_card(*args, **kwargs): ) dataset_json = dataset_response.json() - params = {"gpt_key": openai_key} - files = { "csv_file": ("csv_file", dataset_csv_string.encode()), "doc_file": ("doc_file", doc_file), @@ -260,7 +257,7 @@ def data_card(*args, **kwargs): url = f"{MIT_API}/cards/get_data_card" logger.info(f"Sending dataset {dataset_id} to MIT service at {url}") - resp = requests.post(url, params=params, files=files) + resp = requests.post(url, files=files) if resp.status_code != 200: raise Exception(f"Failed response from MIT: {resp.status_code}") @@ -308,7 +305,6 @@ def data_card(*args, **kwargs): def model_card(*args, **kwargs): - openai_key = os.getenv("OPENAI_API_KEY") model_id = kwargs.get("model_id") paper_artifact_id = kwargs.get("paper_artifact_id") @@ -339,8 +335,6 @@ def model_card(*args, **kwargs): amr = get_model_from_tds(model_id).json() - params = {"gpt_key": openai_key} - files = { "text_file": ("text_file", text_file), "code_file": ("doc_file", code_file), @@ -349,7 +343,7 @@ def model_card(*args, **kwargs): url = f"{MIT_API}/cards/get_model_card" logger.info(f"Sending model {model_id} to MIT service at {url}") - resp = requests.post(url, params=params, files=files) + resp = requests.post(url, files=files) logger.info(f"Response received from MIT with status: {resp.status_code}") logger.debug(f"TA 1 response object: {resp.json()}")