hotfix create experiment from prompt just created in ui (#3349)

Helicone · Mar 1, 2025 · c9fdab6 · c9fdab6
1 parent 1a74b55
commit c9fdab6
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/valhalla/jawn/src/managers/prompt/PromptManager.ts b/valhalla/jawn/src/managers/prompt/PromptManager.ts
@@ -770,6 +770,17 @@ export class PromptManager extends BaseManager {
       );
     }
 
+    const createPromptInputKeysResult = await this.createPromptInputKeys(
+      insertVersionResult.data[0].id,
+      JSON.stringify(params.prompt)
+    );
+
+    if (createPromptInputKeysResult.error) {
+      return err(
+        `Failed to create prompt input keys: ${createPromptInputKeysResult.error}`
+      );
+    }
+
     return ok({
       id: promptId,
       prompt_version_id: insertVersionResult.data[0].id,
@@ -848,11 +859,16 @@ export class PromptManager extends BaseManager {
 
       // Helper function to find keys in a string
       const findKeys = (str: string): string[] => {
-        const regex = /<helicone-prompt-input key="([^"]+)"\s*\/>/g;
+        const regex = /<helicone-prompt-input key=\\?"([^"]+)\\?"\s*\/>/g;
         const matches = str.match(regex);
         return matches
           ? matches.map((match) =>
-              match.replace(/<helicone-prompt-input key="|"\s*\/>/g, "")
+              match
+                .replace(
+                  /<helicone-prompt-input key=\\?"(.*?)\\?"\s*\/>/g,
+                  "$1"
+                )
+                .replace(/\\/g, "")
             )
           : [];
       };