Bugfix/add regex to csv and airtable agent (FlowiseAI#2983)

add regex to csv and airtable agent
patrickreinan · Sep 3, 2024 · 0694131 · 0694131
1 parent 1107c17
commit 0694131
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/packages/components/nodes/agents/AirtableAgent/AirtableAgent.ts b/packages/components/nodes/agents/AirtableAgent/AirtableAgent.ts
@@ -166,13 +166,16 @@ json.dumps(my_dict)`
             }
             const res = await chain.call(inputs, [loggerHandler, ...callbacks])
             pythonCode = res?.text
+            // Regex to get rid of markdown code blocks syntax
+            pythonCode = pythonCode.replace(/^```[a-z]+\n|\n```$/gm, '')
         }
 
         // Then run the code using Pyodide
         let finalResult = ''
         if (pythonCode) {
             try {
                 const code = `import pandas as pd\n${pythonCode}`
+                // TODO: get print console output
                 finalResult = await pyodide.runPythonAsync(code)
             } catch (error) {
                 throw new Error(`Sorry, I'm unable to find answer for question: "${input}" using follwoing code: "${pythonCode}"`)

diff --git a/packages/components/nodes/agents/CSVAgent/CSVAgent.ts b/packages/components/nodes/agents/CSVAgent/CSVAgent.ts
@@ -171,13 +171,16 @@ json.dumps(my_dict)`
             }
             const res = await chain.call(inputs, [loggerHandler, ...callbacks])
             pythonCode = res?.text
+            // Regex to get rid of markdown code blocks syntax
+            pythonCode = pythonCode.replace(/^```[a-z]+\n|\n```$/gm, '')
         }
 
         // Then run the code using Pyodide
         let finalResult = ''
         if (pythonCode) {
             try {
                 const code = `import pandas as pd\n${pythonCode}`
+                // TODO: get print console output
                 finalResult = await pyodide.runPythonAsync(code)
             } catch (error) {
                 throw new Error(`Sorry, I'm unable to find answer for question: "${input}" using following code: "${pythonCode}"`)