run-llama/LlamaIndexTS
1
0
Fork 0
mirror of https://github.com/run-llama/LlamaIndexTS.git synced 2026-07-21 06:45:25 -04:00
Code Issues 138 Packages Projects Releases 10 Wiki Activity

feat(cloud): support pass project and org id to llama parse reader (#1594)

Browse Source
This commit is contained in:
Alex Yang
2025-01-08 11:23:48 -08:00
committed by GitHub
parent e3a77044d5
commit d6c270ec7a
2 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.changeset/sharp-walls-marry.md
+5
View File
@@ -0,0 +1,5 @@
---
"@llamaindex/cloud": patch
---
feat: support pass project and org id to llama parse reader
packages/cloud/src/reader.ts
+26
View File
@@ -31,6 +31,8 @@ var process: any;
* See https://github.com/run-llama/llama_parse
*/
export class LlamaParseReader extends FileReader {
project_id?: string | undefined;
organization_id?: string | undefined;
// The API key for the LlamaParse API. Can be set as an environment variable: LLAMA_CLOUD_API_KEY
apiKey: string;
// The base URL of the Llama Cloud Platform.
@@ -259,6 +261,10 @@ export class LlamaParseReader extends FileReader {
const response = await uploadFileApiV1ParsingUploadPost({
client: this.#client,
throwOnError: true,
query: {
project_id: this.project_id ?? null,
organization_id: this.organization_id ?? null,
},
signal: AbortSignal.timeout(this.maxTimeout * 1000),
body,
});
@@ -284,6 +290,10 @@ export class LlamaParseReader extends FileReader {
path: {
job_id: jobId,
},
query: {
project_id: this.project_id ?? null,
organization_id: this.organization_id ?? null,
},
signal,
});
const { data } = result;
@@ -300,6 +310,10 @@ export class LlamaParseReader extends FileReader {
path: {
job_id: jobId,
},
query: {
project_id: this.project_id ?? null,
organization_id: this.organization_id ?? null,
},
signal,
});
break;
@@ -311,6 +325,10 @@ export class LlamaParseReader extends FileReader {
path: {
job_id: jobId,
},
query: {
project_id: this.project_id ?? null,
organization_id: this.organization_id ?? null,
},
signal,
});
break;
@@ -322,6 +340,10 @@ export class LlamaParseReader extends FileReader {
path: {
job_id: jobId,
},
query: {
project_id: this.project_id ?? null,
organization_id: this.organization_id ?? null,
},
signal,
});
break;
@@ -511,6 +533,10 @@ export class LlamaParseReader extends FileReader {
job_id: jobId,
name: imageName,
},
query: {
project_id: this.project_id ?? null,
organization_id: this.organization_id ?? null,
},
});
if (response.error) {
throw new Error(`Failed to download image: ${response.error.detail}`);