Use ml cloud connector for getting the xml

Author: gabriel-piles

requirements.txt

@@ -2,13 +2,12 @@ fastapi==0.111.1
 python-multipart==0.0.9
 uvicorn==0.30.3
 gunicorn==22.0.0
-requests==2.32.3
 graypy==2.1.0
 PyYAML==6.0.1
 pymongo==4.8.0
 httpx==0.27.0
 sentry-sdk==2.8.0
 setuptools==75.4.0
 git+https://github.com/huridocs/queue-processor@fce76dc0c39a8d929128304eb77047985ccd344a
-git+https://github.com/huridocs/ml-cloud-connector@d62cb04509a33b54d02b2004255abb097d9a4614
+git+https://github.com/huridocs/ml-cloud-connector@6f8dd2aaa890b9cebbab3254dddec65b6bbce95f
 

src/tests/test_app.py

@@ -15,7 +15,7 @@ def test_info(self):
 
         self.assertEqual(200, response.status_code)
 
-    @unittest.skip("This test requires a running cloud service")
+    # @unittest.skip("This test requires a running cloud service")
     def test_cloud(self):
         test_file_path = Path(APP_PATH, "tests", "test_files", "test.pdf")
         with open(test_file_path, "rb") as stream:

src/use_cases/extract_segments_use_case.py

@@ -88,14 +88,14 @@ def extract_segments_cloud(pdf_file: PdfFile, task: Task, xml_file_name: str = "
         files=files,
         data={"fast": "False"},
     )
-    results, success, error = execute_on_cloud_use_case.execute(rest_call)
+    response, success, error = execute_on_cloud_use_case.execute(rest_call)
     if not success:
         return False, None
 
-    if not save_cloud_xml_file(cloud_provider, xml_file_name):
+    if not save_cloud_xml_file(execute_on_cloud_use_case, xml_file_name):
         return False, None
 
-    segments: list[SegmentBox] = [SegmentBox(**segment_box) for segment_box in results]
+    segments: list[SegmentBox] = [SegmentBox(**segment_box) for segment_box in response.json()]
 
     return True, ExtractionData(
         tenant=task.tenant,
@@ -106,9 +106,18 @@ def extract_segments_cloud(pdf_file: PdfFile, task: Task, xml_file_name: str = "
     )
 
 
-def save_cloud_xml_file(cloud_provider: CloudProviderRepository, xml_file_name: str) -> bool:
+def save_cloud_xml_file(execute_on_cloud_use_case: ExecuteOnCloudUseCase, xml_file_name: str) -> bool:
     try:
-        response = requests.get(f"http://{cloud_provider.get_ip()}:{DOCUMENT_LAYOUT_ANALYSIS_PORT}/get_xml/{xml_file_name}")
+        rest_call = RestCall(
+            port=DOCUMENT_LAYOUT_ANALYSIS_PORT,
+            endpoint=["get_xml", xml_file_name],
+            method="GET",
+        )
+        response, success, error = execute_on_cloud_use_case.execute(rest_call)
+
+        if not success:
+            return False
+
         xml_file_path = Path(DATA_PATH, xml_file_name)
         xml_file_path.write_bytes(response.content)
         return True