refactor(agent): changed regex for tool use extraction to look for op…

…tional "Input:" prefix to comply with Cohere command-r models (#1654)
run-llama · Feb 14, 2025 · 6d37d44 · 6d37d44
1 parent e724f7e
commit 6d37d44
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.changeset/sweet-llamas-dress.md b/.changeset/sweet-llamas-dress.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+fix: Update react agent to support tool calls from Cohere command-r models
diff --git a/packages/llamaindex/src/agent/react.ts b/packages/llamaindex/src/agent/react.ts
@@ -100,7 +100,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);