small fix

Helicone · Dec 17, 2024 · cd65994 · cd65994
1 parent e5bc360
commit cd65994
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/valhalla/jawn/src/managers/experiment/ExperimentV2Manager.ts b/valhalla/jawn/src/managers/experiment/ExperimentV2Manager.ts
@@ -86,8 +86,14 @@ export class ExperimentV2Manager extends BaseManager {
     const res = await supabaseServer.client
       .from("prompt_input_record")
       .select("*")
-      .eq("request_id", requestId)
+      .eq("source_request", requestId)
       .single();
+
+    if (res.error || !res.data) {
+      return err("Failed to get prompt version from request");
+    }
+
+    return ok(res.data.prompt_version);
   }
 
   async hasAccessToExperiment(experimentId: string): Promise<boolean> {