Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions langfuse/openai.py
Original file line number Diff line number Diff line change
Expand Up @@ -391,7 +391,7 @@ def _get_langfuse_data_from_kwargs(resource: OpenAiDefinition, kwargs: Any) -> A

if resource.type == "completion":
prompt = kwargs.get("prompt", None)
elif resource.object == "Responses":
elif resource.object == "Responses" or resource.object == "AsyncResponses":
Comment thread
hassiebp marked this conversation as resolved.
prompt = kwargs.get("input", None)
elif resource.type == "chat":
prompt = _extract_chat_prompt(kwargs)
Expand Down Expand Up @@ -672,7 +672,7 @@ def _get_langfuse_data_from_default_response(

completion = choice.text if _is_openai_v1() else choice.get("text", None)

elif resource.object == "Responses":
elif resource.object == "Responses" or resource.object == "AsyncResponses":
output = response.get("output", {})

if not isinstance(output, list):
Expand Down Expand Up @@ -922,6 +922,7 @@ def _finalize(self) -> None:
model, completion, usage, metadata = (
_extract_streamed_response_api_response(self.items)
if self.resource.object == "Responses"
or self.resource.object == "AsyncResponses"
else _extract_streamed_openai_response(self.resource, self.items)
)

Expand Down Expand Up @@ -992,6 +993,7 @@ async def _finalize(self) -> None:
model, completion, usage, metadata = (
_extract_streamed_response_api_response(self.items)
if self.resource.object == "Responses"
or self.resource.object == "AsyncResponses"
else _extract_streamed_openai_response(self.resource, self.items)
)

Expand Down