From 65c3d492a7ec14208a2ab87336612a810c075c50 Mon Sep 17 00:00:00 2001 From: Yohann Paris Date: Tue, 5 Mar 2024 15:37:36 -0500 Subject: [PATCH 1/2] new URL --- lib/settings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/settings.py b/lib/settings.py index 99a9bb4..53aadac 100644 --- a/lib/settings.py +++ b/lib/settings.py @@ -31,7 +31,7 @@ class Settings(BaseSettings): TDS_URL: str = "http://data-service.staging.terarium.ai:8000" TDS_USER: str = "user" TDS_PASSWORD: str = "password" - COSMOS_URL: str = "http://xdd.wisc.edu/cosmos_service" + COSMOS_URL: str = "http://cosmos0004.chtc.wisc.edu:8088" OPENAI_API_KEY: str = "foo" LOG_LEVEL: str = "INFO" PDF_EXTRACTOR: ExtractionServices = "cosmos" From 390c21fe4b17c87127d1ff6c2f6fa14c74e856c3 Mon Sep 17 00:00:00 2001 From: Yohann Paris Date: Tue, 5 Mar 2024 15:37:43 -0500 Subject: [PATCH 2/2] new endpoint --- worker/operations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/worker/operations.py b/worker/operations.py index 1f8a0fd..ab0d8e2 100644 --- a/worker/operations.py +++ b/worker/operations.py @@ -166,7 +166,7 @@ def cosmos_extraction(document_id, filename, downloaded_document, force_run=Fals logger.info("COSMOS response object: %s", extraction_json) status_endpoint = extraction_json["status_endpoint"] result_endpoint = f"{extraction_json['result_endpoint']}" - result_endpoint_text = f"{result_endpoint}/text" + result_endpoint_text = f"{result_endpoint}/text-layer" equations_endpoint = f"{result_endpoint}/extractions/equations" figures_endpoint = f"{result_endpoint}/extractions/figures" tables_endpoint = f"{result_endpoint}/extractions/tables"