File tree Expand file tree Collapse file tree 1 file changed +6
-13
lines changed
mineru_vl_utils/vlm_client Expand file tree Collapse file tree 1 file changed +6
-13
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments