We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 60c4841 commit 8c22e82Copy full SHA for 8c22e82
1 file changed
src/openlayer/lib/integrations/openai_tracer.py
@@ -150,6 +150,22 @@ def traced_responses_create_func(*args, **kwargs):
150
)
151
152
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
169
else:
170
logger.debug("Responses API not available in this OpenAI client version")
171
0 commit comments