From bc3917d9aeaf6a1e020a0b5cb3da64601c7995b4 Mon Sep 17 00:00:00 2001
From: Hassieb Pakzad <68423100+hassiebp@users.noreply.github.com>
Date: Mon, 7 Jul 2025 18:51:26 +0200
Subject: [PATCH] fix(openai): allow setting trace_id and parent_observation_id

---
 langfuse/openai.py | 4 ++++
 1 file changed, 4 insertions(+)

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