Skip to content

Commit 88bc0c2

Browse files
committed
fix: streamline API key handling in http_client by consolidating header assignment
1 parent 07f1271 commit 88bc0c2

File tree

1 file changed

+6
-13
lines changed

1 file changed

+6
-13
lines changed

mineru_vl_utils/vlm_client/http_client.py

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -60,29 +60,22 @@ def __init__(
6060
)
6161
self.max_concurrency = max_concurrency
6262
self.debug = debug
63-
64-
# Work on a local copy of server_headers to avoid mutating the caller's dict.
65-
headers = dict(server_headers) if server_headers is not None else None
66-
67-
api_key = os.getenv("MINERU_VL_API_KEY")
68-
if api_key:
69-
if headers is None:
70-
headers = {}
71-
headers["Authorization"] = f"Bearer {api_key}"
72-
73-
self.headers = headers
7463
self.retry = Retry(total=max_retries, backoff_factor=retry_backoff_factor)
7564

7665
if not server_url:
7766
server_url = _get_env("MINERU_VL_SERVER")
78-
7967
if server_url.endswith("/"): # keep server_url if it ends with '/'
8068
server_url = server_url.rstrip("/")
8169
else: # use base_url if it does not end with '/' (backward compatibility)
8270
server_url = self._get_base_url(server_url)
83-
8471
self.server_url = server_url
72+
73+
api_key = os.getenv("MINERU_VL_API_KEY", "").strip()
74+
if api_key:
75+
server_headers = dict(server_headers) if server_headers else {}
76+
server_headers["Authorization"] = f"Bearer {api_key}"
8577
self.server_headers = server_headers
78+
8679
self.http_timeout = http_timeout
8780
self.max_retries = max_retries
8881
self.retry_backoff_factor = retry_backoff_factor

0 commit comments

Comments
 (0)