Change regex to allow multi-line inputs

raeeceip · Jan 31, 2024 · b0e1283 · b0e1283
1 parent a8f1d95
commit b0e1283
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/agents/markl_test.go b/agents/markl_test.go
@@ -26,6 +26,16 @@ func TestMRKLOutputParser(t *testing.T) {
 			expectedFinish: nil,
 			expectedErr:    nil,
 		},
+		{
+			input: "Action: foo\nAction Input:\nbar\nbaz",
+			expectedActions: []schema.AgentAction{{
+				Tool:      "foo",
+				ToolInput: "bar\nbaz",
+				Log:       "Action: foo\nAction Input:\nbar\nbaz",
+			}},
+			expectedFinish: nil,
+			expectedErr:    nil,
+		},
 	}
 
 	a := OneShotZeroAgent{}

diff --git a/agents/mrkl.go b/agents/mrkl.go
@@ -127,7 +127,7 @@ func (a *OneShotZeroAgent) parseOutput(output string) ([]schema.AgentAction, *sc
 		}, nil
 	}
 
-	r := regexp.MustCompile(`Action:\s*(.+)\s*Action Input:\s*(.+)`)
+	r := regexp.MustCompile(`Action:\s*(.+)\s*Action Input:\s(?s)*(.+)`)
 	matches := r.FindStringSubmatch(output)
 	if len(matches) == 0 {
 		return nil, nil, fmt.Errorf("%w: %s", ErrUnableToParseOutput, output)