From 36ae7cfe52ad62fd0b82382f27bd8eafbee5588e Mon Sep 17 00:00:00 2001
From: bkellam <bshizzle1234@gmail.com>
Date: Wed, 23 Jul 2025 12:03:44 -0700
Subject: [PATCH] Add additional debug logs. Also fix build warning about
 environment variables

---
 packages/web/src/app/api/(server)/chat/route.ts |  4 +++-
 packages/web/src/features/chat/agent.ts         |  3 +++
 packages/web/src/instrumentation.ts             | 15 +++++++++------
 3 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/packages/web/src/app/api/(server)/chat/route.ts b/packages/web/src/app/api/(server)/chat/route.ts
index 59fdfd00..b51a8809 100644
--- a/packages/web/src/app/api/(server)/chat/route.ts
+++ b/packages/web/src/app/api/(server)/chat/route.ts
@@ -243,7 +243,7 @@ const chatHandler = ({ messages, id, selectedRepos, languageModelId }: ChatHandl
                     stream,
                 });
             } catch (error) {
-                logger.error("Error:", error)
+                logger.error(error)
                 logger.error("Error stack:", error instanceof Error ? error.stack : "No stack trace")
                 Sentry.captureException(error);
 
@@ -409,6 +409,8 @@ const getAISDKLanguageModelAndOptions = async (config: LanguageModel, orgId: num
 }
 
 const errorHandler = (error: unknown) => {
+    logger.error(error);
+
     if (error == null) {
         return 'unknown error';
     }
diff --git a/packages/web/src/features/chat/agent.ts b/packages/web/src/features/chat/agent.ts
index 3e808c37..670672b4 100644
--- a/packages/web/src/features/chat/agent.ts
+++ b/packages/web/src/features/chat/agent.ts
@@ -141,6 +141,9 @@ export const createAgentStream = async ({
                 langfuseTraceId: traceId,
             },
         },
+        onError: (error) => {
+            logger.error(error);
+        },
     });
 
     return stream;
diff --git a/packages/web/src/instrumentation.ts b/packages/web/src/instrumentation.ts
index 6e2ee936..7421c482 100644
--- a/packages/web/src/instrumentation.ts
+++ b/packages/web/src/instrumentation.ts
@@ -1,16 +1,19 @@
 import * as Sentry from '@sentry/nextjs';
 import { registerOTel } from '@vercel/otel';
 import { LangfuseExporter } from 'langfuse-vercel';
-import { env } from './env.mjs';
 
 export async function register() {
-    if (env.LANGFUSE_SECRET_KEY && env.NEXT_PUBLIC_LANGFUSE_PUBLIC_KEY) {
+    if (
+        process.env.LANGFUSE_SECRET_KEY &&
+        process.env.NEXT_PUBLIC_LANGFUSE_PUBLIC_KEY
+    ) {
+        console.log("Registering Langfuse");
         registerOTel({
             serviceName: 'sourcebot',
             traceExporter: new LangfuseExporter({
-                secretKey: env.LANGFUSE_SECRET_KEY,
-                publicKey: env.NEXT_PUBLIC_LANGFUSE_PUBLIC_KEY,
-                baseUrl: env.NEXT_PUBLIC_LANGFUSE_BASE_URL,
+                secretKey: process.env.LANGFUSE_SECRET_KEY,
+                publicKey: process.env.NEXT_PUBLIC_LANGFUSE_PUBLIC_KEY,
+                baseUrl: process.env.NEXT_PUBLIC_LANGFUSE_BASE_URL,
             }),
         });
     }
@@ -24,7 +27,7 @@ export async function register() {
     }
 
     if (process.env.NEXT_RUNTIME === 'nodejs') {
-        await import ('./initialize');
+        await import('./initialize');
     }
 }