add resolve package test for document generator tool

run-llama · Oct 1, 2024 · 51a6dac · 51a6dac
1 parent b210302
commit 51a6dac
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/e2e/python/resolve_dependencies.spec.ts b/e2e/python/resolve_dependencies.spec.ts
@@ -33,6 +33,7 @@ if (
   const toolOptions = [
     "wikipedia.WikipediaToolSpec",
     "google.GoogleSearchToolSpec",
+    "document_generator",
   ];
 
   const dataSources = [

diff --git a/e2e/utils.ts b/e2e/utils.ts
@@ -143,10 +143,10 @@ export async function runCreateLlama({
       externalPort,
     );
   } else if (postInstallAction === "dependencies") {
-    await waitForProcess(appProcess, 2000 * 60); // wait 2 min for dependencies to be resolved
+    await waitForProcess(appProcess, 1000 * 60); // wait 1 min for dependencies to be resolved
   } else {
     // wait 10 seconds for create-llama to exit
-    await waitForProcess(appProcess, 1000 * 20);
+    await waitForProcess(appProcess, 1000 * 10);
   }
 
   return {

diff --git a/helpers/tools.ts b/helpers/tools.ts
@@ -117,7 +117,7 @@ For better results, you can specify the region parameter to get results from a s
     dependencies: [
       {
         name: "xhtml2pdf",
-        version: "^0.2.16",
+        version: "^0.2.14",
       },
       {
         name: "markdown",