From 501c15370d065e34f87b515b66e5ae26f5827bd6 Mon Sep 17 00:00:00 2001 From: case-k-git Date: Wed, 13 Mar 2024 19:46:46 +0900 Subject: [PATCH] Add Request Header Exlude Option For Debug logging --- src/openai/_base_client.py | 2 +- src/openai/_compat.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/src/openai/_base_client.py b/src/openai/_base_client.py index 73bd2411fd..ef6a752272 100644 --- a/src/openai/_base_client.py +++ b/src/openai/_base_client.py @@ -436,7 +436,7 @@ def _build_request( options: FinalRequestOptions, ) -> httpx.Request: if log.isEnabledFor(logging.DEBUG): - log.debug("Request options: %s", model_dump(options, exclude_unset=True)) + log.debug("Request options: %s", model_dump(options, exclude_unset=True, exclude_headers=True)) kwargs: dict[str, Any] = {} diff --git a/src/openai/_compat.py b/src/openai/_compat.py index 74c7639b4c..c240c50c12 100644 --- a/src/openai/_compat.py +++ b/src/openai/_compat.py @@ -135,15 +135,21 @@ def model_dump( *, exclude_unset: bool = False, exclude_defaults: bool = False, + exclude_headers: bool = False, ) -> dict[str, Any]: + exclude = {} + if exclude_headers: + exclude={"headers"} if PYDANTIC_V2: return model.model_dump( + exclude=exclude, exclude_unset=exclude_unset, exclude_defaults=exclude_defaults, ) return cast( "dict[str, Any]", model.dict( # pyright: ignore[reportDeprecated, reportUnnecessaryCast] + exclude=exclude, exclude_unset=exclude_unset, exclude_defaults=exclude_defaults, ),