fix(openai): support passing in traceName (#489)

Co-authored-by: Selby Kendrick <[email protected]>
langfuse · Dec 16, 2024 · 21ed800 · 21ed800
1 parent 21f8676
commit 21ed800
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/langfuse/src/openai/observeOpenAI.ts b/langfuse/src/openai/observeOpenAI.ts
@@ -38,7 +38,8 @@ export const observeOpenAI = <SDKType extends object>(
 
       const defaultGenerationName = `${sdk.constructor?.name}.${propKey.toString()}`;
       const generationName = langfuseConfig?.generationName ?? defaultGenerationName;
-      const config = { ...langfuseConfig, generationName };
+      const traceName = langfuseConfig && "traceName" in langfuseConfig ? langfuseConfig.traceName : generationName;
+      const config = { ...langfuseConfig, generationName, traceName };
 
       // Add a flushAsync method to the OpenAI SDK that flushes the Langfuse client
       if (propKey === "flushAsync") {

diff --git a/langfuse/src/openai/traceMethod.ts b/langfuse/src/openai/traceMethod.ts
@@ -60,6 +60,7 @@ const wrapMethod = async <T extends GenericMethod>(
       ...config,
       ...observationData,
       id: config?.traceId,
+      name: config?.traceName,
       timestamp: observationData.startTime,
     });
   }

diff --git a/langfuse/src/openai/types.ts b/langfuse/src/openai/types.ts
@@ -24,7 +24,11 @@ type LangfuseGenerationConfig = Pick<
   "metadata" | "version" | "promptName" | "promptVersion"
 >;
 
-export type LangfuseNewTraceConfig = LangfuseTraceConfig & { traceId?: string; clientInitParams?: LangfuseInitParams };
+export type LangfuseNewTraceConfig = LangfuseTraceConfig & {
+  traceId?: string;
+  traceName?: string;
+  clientInitParams?: LangfuseInitParams;
+};
 export type LangfuseParent = LangfuseTraceClient | LangfuseSpanClient | LangfuseGenerationClient;
 export type LangfuseWithParentConfig = LangfuseGenerationConfig & { parent: LangfuseParent };