Add langfuse for tracing

SuveenE · Jan 10, 2025 · 2ecdd63 · 2ecdd63
1 parent 4075983
commit 2ecdd63
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 3 deletions.
diff --git a/.env.example b/.env.example
@@ -1,2 +1,5 @@
 OPENAI_API_KEY=
 ENVIRONMENT=development
+LANGFUSE_SECRET_KEY="sk-lf-..."
+LANGFUSE_PUBLIC_KEY="pk-lf-..."
+LANGFUSE_BASEURL="https://us.cloud.langfuse.com"
diff --git a/app/api/gpt/route.ts b/app/api/gpt/route.ts
@@ -5,10 +5,13 @@ import { getSystemPrompt, generatePrompt } from "@/utils/prompts";
 import { ClueResponseSchema, GuessResponseSchema } from "@/types/requests";
 import { zodResponseFormat } from "openai/helpers/zod";
 import { delay } from "@/utils/gameUtils";
+import { observeOpenAI } from "langfuse";
 
-const openai = new OpenAI({
-  apiKey: process.env.OPENAI_API_KEY,
-});
+const openai = observeOpenAI(
+  new OpenAI({
+    apiKey: process.env.OPENAI_API_KEY,
+  }),
+);
 
 const MAX_RETRIES = 2;
 

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -14,6 +14,7 @@
     "class-variance-authority": "^0.7.1",
     "clsx": "^2.1.1",
     "fs": "^0.0.1-security",
+    "langfuse": "^3.32.0",
     "lucide-react": "^0.469.0",
     "next": "14.2.16",
     "openai": "^4.77.0",