Skip to content

Commit b6bba2e

Browse files
authored
fix(openai): allow setting trace_id and parent_observation_id (#1245)
1 parent 358fbd2 commit b6bba2e

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

langfuse/openai.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,8 @@ def __init__(
170170
name=None,
171171
langfuse_prompt=None, # we cannot use prompt because it's an argument of the old OpenAI completions API
172172
langfuse_public_key=None,
173+
trace_id=None,
174+
parent_observation_id=None,
173175
**kwargs,
174176
):
175177
self.args = {}
@@ -187,6 +189,8 @@ def __init__(
187189
self.args["name"] = name
188190
self.args["langfuse_public_key"] = langfuse_public_key
189191
self.args["langfuse_prompt"] = langfuse_prompt
192+
self.args["trace_id"] = trace_id
193+
self.args["parent_observation_id"] = parent_observation_id
190194

191195
self.kwargs = kwargs
192196

0 commit comments

Comments
 (0)