diff --git a/backend/open_webui/retrieval/loaders/datalab_marker.py b/backend/open_webui/retrieval/loaders/datalab_marker.py index edcf502e7a..cc6c7ce79d 100644 --- a/backend/open_webui/retrieval/loaders/datalab_marker.py +++ b/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() diff --git a/backend/open_webui/retrieval/loaders/main.py b/backend/open_webui/retrieval/loaders/main.py index b502827adf..241cd7dbe8 100644 --- a/backend/open_webui/retrieval/loaders/main.py +++ b/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, diff --git a/src/lib/components/admin/Settings/Documents.svelte b/src/lib/components/admin/Settings/Documents.svelte index ac332d3e1f..cdf032de24 100644 --- a/src/lib/components/admin/Settings/Documents.svelte +++ b/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; });