Skip to content

Commit 8c22e82

Browse files
fix(closes OPEN-10593): openAI parse responses not getting patched
1 parent 60c4841 commit 8c22e82

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

src/openlayer/lib/integrations/openai_tracer.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,22 @@ def traced_responses_create_func(*args, **kwargs):
150150
)
151151

152152
client.responses.create = traced_responses_create_func
153+
154+
if hasattr(client.responses, "parse"):
155+
responses_parse_func = client.responses.parse
156+
157+
@wraps(responses_parse_func)
158+
def traced_responses_parse_func(*args, **kwargs):
159+
inference_id = kwargs.pop("inference_id", None)
160+
return handle_responses_non_streaming_create(
161+
*args,
162+
**kwargs,
163+
create_func=responses_parse_func,
164+
inference_id=inference_id,
165+
is_azure_openai=is_azure_openai,
166+
)
167+
168+
client.responses.parse = traced_responses_parse_func
153169
else:
154170
logger.debug("Responses API not available in this OpenAI client version")
155171

0 commit comments

Comments
 (0)