gpt4free

公开

关注 0 Fork 0 Star 0

XFEstudio/gpt4free

Refactor update_headers to use del for authorization header removal and enhance provider retrieval logic in Api class

f4dc6515

hlohaus <hlohaus@users.noreply.github.com>

提交于 2026-02-23 12:02

父提交 eb9a552

1 个文件 +4 -2

Modified g4f/api/__init__.py +4 -2

@@ -181,7 +181,7 @@ def update_headers(request: Request, new_api_key: str = None, user: str = None)

181 181 if new_api_key:

182 182 new_headers["authorization"] = f"Bearer {new_api_key}"

183 183 else:

184 new_headers.pop("authorization", None)

184 del new_headers["authorization"]

185 185 if user:

186 186 new_headers["x-user"] = user

187 187 request.scope["headers"] = new_headers.raw

@@ -284,6 +284,8 @@ class Api:

284 284 return ErrorResponse.from_message(e.detail, e.status_code, e.headers)

285 285 if user_g4f_api_key and update_authorization:

286 286 new_api_key = user_g4f_api_key.pop()

287 if secrets.compare_digest(AppConfig.g4f_api_key, new_api_key):

288 new_api_key = None

287 289 else:

288 290 new_api_key = None

289 291 request = update_headers(request, new_api_key, user)

@@ -455,7 +457,7 @@ class Api:

455 457 if provider is None:

456 458 provider = AppConfig.provider

457 459 try:

458 provider = ProviderUtils.get_by_label(provider)

460 provider = ProviderUtils.get_by_label(provider).__name__

459 461 except ValueError as e:

460 462 if provider in model_map:

461 463 config.model = provider