Reduce wait time for tools

OpenAgentsInc · May 23, 2024 · 7693e95 · 7693e95
1 parent 5bd3e93
commit 7693e95
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/rag/index.js b/src/rag/index.js
@@ -16,6 +16,7 @@ async function run() {
         let toolsResultTemplate ="{{TOOL_RESULT}}";
         let expectedResults = 1;
         let maxWaitTime = 1000 * 60 * 2;
+        let toolWaitTime = 1000 * 10;
 
 
         const documents = []; // plain text documents
@@ -160,7 +161,7 @@ async function run() {
                 });
 
                 Job.log("Merging tools result...");
-                let toolResult = await Job.waitForContent(toolReq, expectedResults, maxWaitTime);
+                let toolResult = await Job.waitForContent(toolReq, expectedResults, toolWaitTime);
                 toolResult = toolsResultTemplate.replace("{{TOOL_RESULT}}", toolResult);
                 newContext += toolResult + "\n";
             }catch(e){