From 7a4aa5f0aa0e001a625cffa973abf7e02c9267b2 Mon Sep 17 00:00:00 2001
From: David Corbitt <davidlcorbitt@gmail.com>
Date: Tue, 8 Aug 2023 13:45:46 -0700
Subject: [PATCH] Use openpipe optionally in app

---
 app/src/env.mjs                | 2 ++
 app/src/server/utils/openai.ts | 8 ++++++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/app/src/env.mjs b/app/src/env.mjs
index bec0e5e..f0f5f3f 100644
--- a/app/src/env.mjs
+++ b/app/src/env.mjs
@@ -20,6 +20,7 @@ export const env = createEnv({
     REPLICATE_API_TOKEN: z.string().default("placeholder"),
     ANTHROPIC_API_KEY: z.string().default("placeholder"),
     SENTRY_AUTH_TOKEN: z.string().optional(),
+    USE_OPENPIPE: z.string().optional(),
   },
 
   /**
@@ -54,6 +55,7 @@ export const env = createEnv({
     ANTHROPIC_API_KEY: process.env.ANTHROPIC_API_KEY,
     NEXT_PUBLIC_SENTRY_DSN: process.env.NEXT_PUBLIC_SENTRY_DSN,
     SENTRY_AUTH_TOKEN: process.env.SENTRY_AUTH_TOKEN,
+    USE_OPENPIPE: process.env.USE_OPENPIPE,
   },
   /**
    * Run `build` or `dev` with `SKIP_ENV_VALIDATION` to skip env validation.
diff --git a/app/src/server/utils/openai.ts b/app/src/server/utils/openai.ts
index 5397e2c..a18df5e 100644
--- a/app/src/server/utils/openai.ts
+++ b/app/src/server/utils/openai.ts
@@ -1,7 +1,11 @@
 import { env } from "~/env.mjs";
 
-// import OpenAI from "openai";
+import { default as OriginalOpenAI } from "openai";
 import { OpenAI } from "openpipe";
 
+const openAIConfig = { apiKey: env.OPENAI_API_KEY ?? "dummy-key" };
+
 // Set a dummy key so it doesn't fail at build time
-export const openai = new OpenAI.OpenAI({ apiKey: env.OPENAI_API_KEY ?? "dummy-key" });
+export const openai = env.USE_OPENPIPE
+  ? new OpenAI.OpenAI(openAIConfig)
+  : new OriginalOpenAI(openAIConfig);