From 258d18373d09f4a2c61b75497c7bdff326bb47dc Mon Sep 17 00:00:00 2001
From: Hassieb Pakzad <68423100+hassiebp@users.noreply.github.com>
Date: Mon, 9 Dec 2024 10:27:32 +0100
Subject: [PATCH] fix(openai): pass metadata when model distillation feat is
 used

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

diff --git a/langfuse/openai.py b/langfuse/openai.py
index da449dbb2..06bd20371 100644
--- a/langfuse/openai.py
+++ b/langfuse/openai.py
@@ -170,6 +170,11 @@ def get_langfuse_args(self):
         return {**self.args, **self.kwargs}
 
     def get_openai_args(self):
+        # If OpenAI model distillation is enabled, we need to add the metadata to the kwargs
+        # https://platform.openai.com/docs/guides/distillation
+        if self.kwargs.get("store", False):
+            self.kwargs["metadata"] = self.args.get("metadata", {})
+
         return self.kwargs