docs: message

run-llama · Aug 2, 2024 · f33aa09 · f33aa09
1 parent 7b0cfc5
commit f33aa09
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/templates/components/vectordbs/typescript/llamacloud/index.ts b/templates/components/vectordbs/typescript/llamacloud/index.ts
@@ -1,13 +1,20 @@
 import { LlamaCloudIndex } from "llamaindex/cloud/LlamaCloudIndex";
 
-export async function getDataSource(params?: any) {
+type LlamaCloudDataSourceParams = {
+  llamaCloudPipeline?: {
+    project: string;
+    pipeline: string;
+  };
+};
+
+export async function getDataSource(params?: LlamaCloudDataSourceParams) {
   const { project, pipeline } = params?.llamaCloudPipeline ?? {};
   const projectName = project ?? process.env.LLAMA_CLOUD_PROJECT_NAME;
   const pipelineName = pipeline ?? process.env.LLAMA_CLOUD_INDEX_NAME;
   const apiKey = process.env.LLAMA_CLOUD_API_KEY;
   if (!projectName || !pipelineName || !apiKey) {
     throw new Error(
-      "Set project, pipeline, and api key in the config file or as environment variables.",
+      "Set project, pipeline, and api key in the params or as environment variables.",
     );
   }
   const index = new LlamaCloudIndex({