From 8466cde82ebebd7fe219bdb5bde1824f90f93c47 Mon Sep 17 00:00:00 2001
From: Laurens Profittlich <profikid@gmail.com>
Date: Wed, 16 Oct 2024 16:36:03 +0700
Subject: [PATCH] add langfuseprompt tracing

---
 integrations/open-ai.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/integrations/open-ai.ts b/integrations/open-ai.ts
index d0ea88a..6faf8f7 100644
--- a/integrations/open-ai.ts
+++ b/integrations/open-ai.ts
@@ -1,6 +1,7 @@
 import { task, logger } from '@trigger.dev/sdk/v3';
 import OpenAI from 'openai';
 import { observeOpenAI } from 'langfuse';
+import type { LangfusePromptClient } from 'langfuse-core';
 import type { ChatCompletionCreateParams } from 'openai/resources/index.mjs';
 
 export const DallE = task({
@@ -22,10 +23,10 @@ export const DallE = task({
 
 export const openAi = task({
 	id: 'open-ai',
-	run: async ({ payload }: { payload: ChatCompletionCreateParams }) => {
-		const openai = observeOpenAI(new OpenAI());
+	run: async ({ payload, langfusePrompt }: { payload: ChatCompletionCreateParams, langfusePrompt: LangfusePromptClient | undefined }) => {
+		const openai = observeOpenAI(new OpenAI(), {langfusePrompt});
 		logger.log(payload.model);
-		payload.messages.forEach((message) => logger.log(message.content, { message }));
+		payload.messages.forEach((message) => logger.log(message?.content?.toString() || '', { message }));
 
 		const result = await openai.chat.completions.create(payload);
 		logger.log('Result from open ai', { result });