[Freestyler] Parse backticks out of the code

Bug: 340805507 Change-Id: I24f92dce2b860a31eb2520cf18352ff4474e8dbf Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/5633337 Commit-Queue: Alex Rudenko <[email protected]> Reviewed-by: Ergün Erdoğmuş <[email protected]>
paulirish · Jun 14, 2024 · 4e2ef92 · 4e2ef92
1 parent f377833
commit 4e2ef92
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/front_end/panels/freestyler/FreestylerAgent.test.ts b/front_end/panels/freestyler/FreestylerAgent.test.ts
@@ -119,6 +119,17 @@ c`;
       });
     });
 
+    it('parses an action with backticks in the code', async () => {
+      const payload = `const data = {
+  someKey: "value",
+}`;
+      assert.deepStrictEqual(FreestylerAgent.parseResponse(`ACTION\n\`\`\`\n${payload}\n\`\`\`\nSTOP`), {
+        action: payload,
+        thought: undefined,
+        answer: undefined,
+      });
+    });
+
     it('parses a thought and an action', async () => {
       const actionPayload = `const data = {
   someKey: "value",

diff --git a/front_end/panels/freestyler/FreestylerAgent.ts b/front_end/panels/freestyler/FreestylerAgent.ts
@@ -134,11 +134,12 @@ export class FreestylerAgent {
         const actionLines = [];
         let j = i + 1;
         while (j < lines.length && lines[j].trim() !== 'STOP') {
-          // TODO: handle markdown backticks.
           actionLines.push(lines[j]);
           j++;
         }
-        action = actionLines.join('\n').trim();
+        // TODO: perhaps trying to parse with a Markdown parser would
+        // yield more reliable results.
+        action = actionLines.join('\n').replaceAll('```', '').trim();
         i = j + 1;
       } else if (trimmed.startsWith('ANSWER:') && !answer) {
         const answerLines = [