fix: fallback for result

run-llama · Nov 29, 2024 · 719ded9 · 719ded9
1 parent f658536
commit 719ded9
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/examples/vercel/llamacloud.ts b/examples/vercel/llamacloud.ts
@@ -14,9 +14,11 @@ async function main() {
     name: "test-pipeline",
     projectName: "Default",
     apiKey: process.env.LLAMA_CLOUD_API_KEY,
-    baseUrl: process.env.LLAMA_CLOUD_BASE_URL,
   });
-  const queryTool = await llamaindex({ index });
+  const queryTool = await llamaindex({
+    index,
+    description: "Search through Abramov's essay", // optional description
+  });
   console.log("Successfully created index and queryTool");
 
   streamText({

diff --git a/examples/vercel/tool.ts b/examples/vercel/tool.ts
@@ -19,7 +19,7 @@ export async function llamaindex({
     }),
     execute: async ({ query }) => {
       const result = await queryEngine?.query({ query });
-      return result?.message.content;
+      return result?.message.content ?? "No result found in documents.";
     },
   };
 }
diff --git a/examples/vercel/vector-store.ts b/examples/vercel/vector-store.ts
@@ -10,7 +10,10 @@ async function main() {
   const document = new Document({ text: essay, id_: path });
 
   const index = await VectorStoreIndex.fromDocuments([document]);
-  const queryTool = await llamaindex({ index });
+  const queryTool = await llamaindex({
+    index,
+    description: "Search through Abramov's essay", // optional description
+  });
   console.log("Successfully created index and queryTool");
 
   streamText({