github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-01 02:50:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

openai[patch]: propagate service_tier to response metadata (#31089)

Browse Source
This commit is contained in:
ccurme
2025-05-01 13:50:48 -04:00
committed by GitHub
parent 6110c3ffc5
commit c51eadd54f
3 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
libs/partners/openai/langchain_openai/chat_models/base.py
View File

@@ -736,6 +736,8 @@ class BaseChatOpenAI(BaseChatModel):
generation_info["model_name"] = model_name
if system_fingerprint := chunk.get("system_fingerprint"):
generation_info["system_fingerprint"] = system_fingerprint
if service_tier := chunk.get("service_tier"):
generation_info["service_tier"] = service_tier
logprobs = choice.get("logprobs")
if logprobs:
@@ -1020,6 +1022,8 @@ class BaseChatOpenAI(BaseChatModel):
}
if "id" in response_dict:
llm_output["id"] = response_dict["id"]
if "service_tier" in response_dict:
llm_output["service_tier"] = response_dict["service_tier"]
if isinstance(response, openai.BaseModel) and getattr(
response, "choices", None
@@ -3243,6 +3247,7 @@ def _construct_lc_result_from_responses_api(
"status",
"user",
"model",
"service_tier",
)
}
if metadata: