diff --git a/langfuse/openai.py b/langfuse/openai.py index fb321e848..ea0cd186a 100644 --- a/langfuse/openai.py +++ b/langfuse/openai.py @@ -170,6 +170,8 @@ def __init__( name=None, langfuse_prompt=None, # we cannot use prompt because it's an argument of the old OpenAI completions API langfuse_public_key=None, + trace_id=None, + parent_observation_id=None, **kwargs, ): self.args = {} @@ -187,6 +189,8 @@ def __init__( self.args["name"] = name self.args["langfuse_public_key"] = langfuse_public_key self.args["langfuse_prompt"] = langfuse_prompt + self.args["trace_id"] = trace_id + self.args["parent_observation_id"] = parent_observation_id self.kwargs = kwargs