From 26ba266d51b40bbd151d20cd8db317998623d880 Mon Sep 17 00:00:00 2001
From: celeste <parafactual@gmail.com>
Date: Mon, 27 Feb 2023 01:20:42 -0800
Subject: [PATCH] escape lt

---
 main.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/main.ts b/main.ts
index 53b5e99..543dd04 100644
--- a/main.ts
+++ b/main.ts
@@ -451,10 +451,14 @@ export default class LoomPlugin extends Plugin {
     const trailingSpace = prompt.match(/\s+$/);
     prompt = prompt.replace(/\s+$/, "");
 
+    prompt = prompt.replace(/\\</g, "<");
+
     const bpe = tokenizer.encode(prompt).bpe;
     const tokens = bpe.slice(Math.max(0, bpe.length - (8000 - this.settings.maxTokens)), bpe.length);
     prompt = tokenizer.decode(tokens);
 
+    console.log("prompt", prompt);
+
     let completions;
     try {
       completions = (
@@ -479,7 +483,7 @@ export default class LoomPlugin extends Plugin {
 
     let ids = [];
     for (const completion of completions) {
-      let completion_ = completion;
+      let completion_ = completion?.replace(/</g, "\\<");
       if (!completion_) continue; // i've never seen this happen
 
       if (trailingSpace && completion_[0] === " ") completion_ = completion_.slice(1);