fix: corrected the regex in the ReactAgent (run-llama#1022)

Signed-off-by: ranjanmangla1 <ranjanmangla1@gmail.com>
himself65 · Jul 9, 2024 · a0f424e · a0f424e
1 parent 3ae832c
commit a0f424e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.changeset/loud-bananas-shout.md b/.changeset/loud-bananas-shout.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+corrected the regex in the react.ts file in extractToolUse & extractJsonStr functions, as mentioned in https://github.com/run-llama/LlamaIndexTS/issues/1019
diff --git a/packages/llamaindex/src/agent/react.ts b/packages/llamaindex/src/agent/react.ts
@@ -66,7 +66,7 @@ function reasonFormatter(reason: Reason): string | Promise<string> {
 }
 
 function extractJsonStr(text: string): string {
-  const pattern = /\{.*}/s;
+  const pattern = /\{.*\}/s;
   const match = text.match(pattern);
 
   if (!match) {
@@ -98,7 +98,7 @@ function extractToolUse(
   inputText: string,
 ): [thought: string, action: string, input: string] {
   const pattern =
-    /\s*Thought: (.*?)\nAction: ([a-zA-Z0-9_]+).*?\.*Input: .*?(\{.*?})/s;
+    /\s*Thought: (.*?)\nAction: ([a-zA-Z0-9_]+).*?\.*Input: .*?(\{.*?\})/s;
 
   const match = inputText.match(pattern);