diff --git a/packages/components/package.json b/packages/components/package.json
index f3371d26..a2565430 100644
--- a/packages/components/package.json
+++ b/packages/components/package.json
@@ -53,9 +53,9 @@
         "husky": "^8.0.3",
         "ioredis": "^5.3.2",
         "langchain": "^0.0.214",
-        "langfuse": "^1.2.0",
-        "langfuse-langchain": "^1.0.31",
-        "langsmith": "^0.0.53",
+        "langfuse": "2.0.2",
+        "langfuse-langchain": "2.0.2",
+        "langsmith": "0.0.53",
         "linkifyjs": "^4.1.1",
         "llmonitor": "^0.5.5",
         "mammoth": "^1.5.1",
diff --git a/packages/components/src/handler.ts b/packages/components/src/handler.ts
index 29aff3e2..1eb05a51 100644
--- a/packages/components/src/handler.ts
+++ b/packages/components/src/handler.ts
@@ -538,7 +538,7 @@ export class AnalyticHandler {
             if (trace) {
                 const generation = trace.generation({
                     name,
-                    prompt: input
+                    input: input
                 })
                 this.handlers['langFuse'].generation = { [generation.id]: generation }
                 returnIds['langFuse'].generation = generation.id
@@ -583,7 +583,7 @@ export class AnalyticHandler {
             const generation: LangfuseGenerationClient | undefined = this.handlers['langFuse'].generation[returnIds['langFuse'].generation]
             if (generation) {
                 generation.end({
-                    completion: output
+                    output: output
                 })
             }
         }
@@ -618,7 +618,7 @@ export class AnalyticHandler {
             const generation: LangfuseGenerationClient | undefined = this.handlers['langFuse'].generation[returnIds['langFuse'].generation]
             if (generation) {
                 generation.end({
-                    completion: error
+                    output: error
                 })
             }
         }