From 2bb6063dcb496d3e9f6a0f43e7531b2f2ad8d716 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Thu, 28 Aug 2025 03:03:31 +0400 Subject: [PATCH] refac/fix: marker --- backend/open_webui/retrieval/loaders/datalab_marker.py | 4 ++-- backend/open_webui/retrieval/loaders/main.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/open_webui/retrieval/loaders/datalab_marker.py b/backend/open_webui/retrieval/loaders/datalab_marker.py index cc6c7ce79d..8d14be0a40 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}/marker/{request_id}" + url = f"{self.api_base_url}/{request_id}" headers = {"X-Api-Key": self.api_key} try: response = requests.get(url, headers=headers) @@ -111,7 +111,7 @@ class DatalabMarkerLoader: with open(self.file_path, "rb") as f: files = {"file": (filename, f, mime_type)} response = requests.post( - f"{self.api_base_url}/marker", + f"{self.api_base_url}", data=form_data, files=files, headers=headers, diff --git a/backend/open_webui/retrieval/loaders/main.py b/backend/open_webui/retrieval/loaders/main.py index c301d0b7c8..9b90dca041 100644 --- a/backend/open_webui/retrieval/loaders/main.py +++ b/backend/open_webui/retrieval/loaders/main.py @@ -284,7 +284,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" + api_base_url = "https://www.datalab.to/api/v1/marker" # https://github.com/open-webui/open-webui/pull/16867#issuecomment-3218424349 loader = DatalabMarkerLoader( file_path=file_path,