From 673f6f618565f290b4746da2df16e94928898fa9 Mon Sep 17 00:00:00 2001
From: nai-degen <44111-khanon@users.noreply.gitgud.io>
Date: Fri, 14 Apr 2023 20:54:55 -0700
Subject: [PATCH] actually fixes prompt logger path detection

---
 src/proxy/middleware/response/log-prompt.ts | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/proxy/middleware/response/log-prompt.ts b/src/proxy/middleware/response/log-prompt.ts
index 9b73314..3d07c09 100644
--- a/src/proxy/middleware/response/log-prompt.ts
+++ b/src/proxy/middleware/response/log-prompt.ts
@@ -18,8 +18,13 @@ export const logPrompt: ProxyResHandlerWithBody = async (
     throw new Error("Expected body to be an object");
   }
 
+
   // Only log prompts if we're making a request to a completion endpoint
-  if (!req.originalUrl.endsWith(COMPLETE_ENDPOINT)) {
+  if (!req.path.startsWith(COMPLETE_ENDPOINT)) {
+    // Remove this once we're confident that we're not missing any prompts
+    req.log.info(
+      `Not logging prompt for ${req.path} because it's not a completion endpoint`
+    );
     return;
   }