refac

backend/open_webui/retrieval/loaders/datalab_marker.py

@@ -64,7 +64,7 @@ class DatalabMarkerLoader:
         return mime_map.get(ext, "application/octet-stream")
 
     def check_marker_request_status(self, request_id: str) -> dict:
-        url = f"{self.api_base_url}/{request_id}"
+        url = f"{self.api_base_url}/marker/{request_id}"
         headers = {"X-Api-Key": self.api_key}
         try:
             response = requests.get(url, headers=headers)
@@ -85,7 +85,6 @@ class DatalabMarkerLoader:
             )
 
     def load(self) -> List[Document]:
-        url = self.api_base_url
         filename = os.path.basename(self.file_path)
         mime_type = self._get_mime_type(filename)
         headers = {"X-Api-Key": self.api_key}
@@ -112,7 +111,10 @@ class DatalabMarkerLoader:
             with open(self.file_path, "rb") as f:
                 files = {"file": (filename, f, mime_type)}
                 response = requests.post(
-                    url, data=form_data, files=files, headers=headers
+                    f"{self.api_base_url}/marker",
+                    data=form_data,
+                    files=files,
+                    headers=headers,
                 )
                 response.raise_for_status()
                 result = response.json()

backend/open_webui/retrieval/loaders/main.py

@@ -283,7 +283,7 @@ class Loader:
         ):
             api_base_url = self.kwargs.get("DATALAB_MARKER_API_BASE_URL", "")
             if not api_base_url or api_base_url.strip() == "":
-                api_base_url = "https://www.datalab.to/api/v1/marker"
+                api_base_url = "https://www.datalab.to/api/v1"
 
             loader = DatalabMarkerLoader(
                 file_path=file_path,

src/lib/components/admin/Settings/Documents.svelte

@@ -252,11 +252,6 @@
 			2
 		);
 
-		// Set default API Base URL if empty
-		if (!config.DATALAB_MARKER_API_BASE_URL) {
-			config.DATALAB_MARKER_API_BASE_URL = 'https://www.datalab.to/api/v1/marker';
-		}
-
 		RAGConfig = config;
 	});
 </script>