@liquidmetal-ai/raindrop 0.3.5 → 0.4.1

package/dist/commands/object/get.js

@@ -26,8 +26,13 @@ Download file with key 'my-key' from my-bucket to output.txt
         ...BaseCommand.HIDDEN_FLAGS,
         bucket: Flags.string({
             char: 'b',
-            description: 'bucket ID',
-            required: true,
+            description: 'bucket name version can be specified with #, e.g. my-bucket#versionId',
+            required: false,
+        }),
+        moduleId: Flags.string({
+            char: 'm',
+            description: 'moduleId',
+            required: false,
         }),
         format: Flags.string({
             char: 'f',
@@ -42,7 +47,7 @@ Download file with key 'my-key' from my-bucket to output.txt
             hidden: true,
         }),
         manifest: Flags.string({
-            char: 'm',
+            char: 'M',
             description: 'project manifest',
             required: false,
             default: 'raindrop.manifest',
@@ -50,12 +55,51 @@ Download file with key 'my-key' from my-bucket to output.txt
         }),
     };
     async run() {
+        if (this.flags.bucket && this.flags.moduleId) {
+            this.error('Cannot specify both --bucket and --moduleId flags. Please use only one type of filter.');
+        }
+        if (!this.flags.bucket && !this.flags.moduleId) {
+            this.error('Either --bucket or --moduleId flag must be specified.');
+        }
         const { args, flags } = await this.parse(ObjectGet);
-        const { client: objectService, userId, organizationId } = await this.objectService();
+        let moduleId = this.flags.moduleId || '';
+        let bucket;
+        const { client: catalogService, userId, organizationId } = await this.catalogService();
+        if (!moduleId && this.flags.bucket) {
+            // If moduleId is not provided and bucket is specified, we need to fetch the moduleId from the catalog service
+            //string split on # to get bucket name and version
+            const [bucketName, version] = this.flags.bucket.split('#');
+            bucket = (await catalogService.resolveBucketLocationToModuleId({
+                userId,
+                organizationId,
+                bucketLocation: {
+                    bucketLocation: {
+                        case: 'bucket',
+                        value: {
+                            name: bucketName,
+                            version: version || '',
+                        },
+                    },
+                },
+            })).bucket;
+            if (!bucket || !bucket.moduleId) {
+                this.error(`Bucket "${this.flags.bucket}" not found or does not belong to any module.`);
+            }
+            moduleId = bucket.moduleId;
+        }
+        else {
+            // query for the bucket if moduleId is provided
+            bucket = (await catalogService.resolveModuleIdToBucketLocation({
+                userId,
+                organizationId,
+                moduleId,
+            })).bucket;
+        }
+        const { client: objectService } = await this.objectService(moduleId);
         const response = await objectService.getObject({
             userId,
             organizationId,
-            bucketId: flags.bucket,
+            moduleId,
             key: args.key,
         });
         if (!response.content) {

package/dist/commands/object/list.d.ts

@@ -3,7 +3,8 @@ export default class ObjectList extends BaseCommand<typeof ObjectList> {
     static description: string;
     static examples: string[];
     static flags: {
-        bucket: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        bucket: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        moduleId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         output: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
         impersonate: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         manifest: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
@@ -13,6 +14,7 @@ export default class ObjectList extends BaseCommand<typeof ObjectList> {
         rainbowAuthToken: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowOrganizationId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowUserId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        sendVersionMetadata: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
     };
     run(): Promise<void>;
 }

package/dist/commands/object/list.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"list.d.ts","sourceRoot":"","sources":["../../../src/commands/object/list.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAKpD,MAAM,CAAC,OAAO,OAAO,UAAW,SAAQ,WAAW,CAAC,OAAO,UAAU,CAAC;IACpE,MAAM,CAAC,WAAW,SAA8B;IAEhD,MAAM,CAAC,QAAQ,WAIb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;MA0BV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAkD3B"}
+{"version":3,"file":"list.d.ts","sourceRoot":"","sources":["../../../src/commands/object/list.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAMpD,MAAM,CAAC,OAAO,OAAO,UAAW,SAAQ,WAAW,CAAC,OAAO,UAAU,CAAC;IACpE,MAAM,CAAC,WAAW,SAA8B;IAEhD,MAAM,CAAC,QAAQ,WAIb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;;;MA+BV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CA4F3B"}

package/dist/commands/object/list.js

@@ -14,8 +14,13 @@ List all objects in my-bucket
         ...BaseCommand.HIDDEN_FLAGS,
         bucket: Flags.string({
             char: 'b',
-            description: 'bucket ID',
-            required: true,
+            description: 'bucket name version can be specified with #, e.g. my-bucket#versionId',
+            required: false,
+        }),
+        moduleId: Flags.string({
+            char: 'm',
+            description: 'moduleId',
+            required: false,
         }),
         output: Flags.string({
             char: 'o',
@@ -30,7 +35,7 @@ List all objects in my-bucket
             hidden: true,
         }),
         manifest: Flags.string({
-            char: 'm',
+            char: 'M',
             description: 'project manifest',
             required: false,
             default: 'raindrop.manifest',
@@ -38,19 +43,58 @@ List all objects in my-bucket
         }),
     };
     async run() {
+        if (this.flags.bucket && this.flags.moduleId) {
+            this.error('Cannot specify both --bucket and --moduleId flags. Please use only one type of filter.');
+        }
+        if (!this.flags.bucket && !this.flags.moduleId) {
+            this.error('Either --bucket or --moduleId flag must be specified.');
+        }
         const { flags } = await this.parse(ObjectList);
-        const { client: objectService, userId, organizationId } = await this.objectService();
+        let moduleId = this.flags.moduleId || '';
+        let bucket;
+        const { client: catalogService, userId, organizationId } = await this.catalogService();
+        if (!moduleId && this.flags.bucket) {
+            // If moduleId is not provided and bucket is specified, we need to fetch the moduleId from the catalog service
+            //string split on # to get bucket name and version
+            const [bucketName, version] = this.flags.bucket.split('#');
+            bucket = (await catalogService.resolveBucketLocationToModuleId({
+                userId,
+                organizationId,
+                bucketLocation: {
+                    bucketLocation: {
+                        case: 'bucket',
+                        value: {
+                            name: bucketName,
+                            version: version || '',
+                        },
+                    },
+                },
+            })).bucket;
+            if (!bucket || !bucket.moduleId) {
+                this.error(`Bucket "${this.flags.bucket}" not found or does not belong to any module.`);
+            }
+            moduleId = bucket.moduleId;
+        }
+        else {
+            // query for the bucket if moduleId is provided
+            bucket = (await catalogService.resolveModuleIdToBucketLocation({
+                userId,
+                organizationId,
+                moduleId,
+            })).bucket;
+        }
+        const { client: objectService } = await this.objectService(moduleId);
         let response;
         try {
             response = await objectService.listObjects({
                 userId,
                 organizationId,
-                bucketId: flags.bucket,
+                moduleId,
             });
         }
         catch (error) {
             const err = error;
-            this.error(`Failed to list objects in bucket ${flags.bucket}: ${err.message}`);
+            this.error(`Failed to list objects in bucket ${bucket.bucketName}: ${err.message}`);
         }
         if (!response.objects?.length) {
             console.log('No objects found in bucket');
@@ -69,7 +113,7 @@ List all objects in my-bucket
                 })));
                 break;
             default:
-                console.log(`Objects in bucket ${flags.bucket}:`);
+                console.log(`Objects in bucket ${bucket.bucketName}:`);
                 for (const obj of response.objects) {
                     const lastModified = obj.lastModified ? timestampDate(obj.lastModified).toUTCString() : 'N/A';
                     console.log(`Key: ${obj.key}`);

package/dist/commands/object/put.d.ts

@@ -9,7 +9,8 @@ export default class ObjectPut extends BaseCommand<typeof ObjectPut> {
     static description: string;
     static examples: string[];
     static flags: {
-        bucket: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        bucket: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        moduleId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         contentType: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         output: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
         impersonate: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
@@ -20,6 +21,7 @@ export default class ObjectPut extends BaseCommand<typeof ObjectPut> {
         rainbowAuthToken: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowOrganizationId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowUserId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        sendVersionMetadata: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
     };
     run(): Promise<void>;
     private getContentType;

package/dist/commands/object/put.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"put.d.ts","sourceRoot":"","sources":["../../../src/commands/object/put.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAMpD,MAAM,CAAC,OAAO,OAAO,SAAU,SAAQ,WAAW,CAAC,OAAO,SAAS,CAAC;IAClE,MAAM,CAAC,IAAI;;;;;MAUT;IAEF,MAAM,CAAC,WAAW,SAAyC;IAE3D,MAAM,CAAC,QAAQ,WAIb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;;MA+BV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;IA6B1B,OAAO,CAAC,cAAc;CAavB"}
+{"version":3,"file":"put.d.ts","sourceRoot":"","sources":["../../../src/commands/object/put.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAOpD,MAAM,CAAC,OAAO,OAAO,SAAU,SAAQ,WAAW,CAAC,OAAO,SAAS,CAAC;IAClE,MAAM,CAAC,IAAI;;;;;MAUT;IAEF,MAAM,CAAC,WAAW,SAAyC;IAE3D,MAAM,CAAC,QAAQ,WAIb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;;;;MAoCV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;IAyE1B,OAAO,CAAC,cAAc;CAavB"}

package/dist/commands/object/put.js

@@ -26,8 +26,13 @@ Upload myfile.txt to my-bucket with key 'my-key'
         ...BaseCommand.HIDDEN_FLAGS,
         bucket: Flags.string({
             char: 'b',
-            description: 'bucket ID',
-            required: true,
+            description: 'bucket name version can be specified with #, e.g. my-bucket#versionId',
+            required: false,
+        }),
+        moduleId: Flags.string({
+            char: 'm',
+            description: 'moduleId',
+            required: false,
         }),
         contentType: Flags.string({
             char: 't',
@@ -47,7 +52,7 @@ Upload myfile.txt to my-bucket with key 'my-key'
             hidden: true,
         }),
         manifest: Flags.string({
-            char: 'm',
+            char: 'M',
             description: 'project manifest',
             required: false,
             default: 'raindrop.manifest',
@@ -55,8 +60,48 @@ Upload myfile.txt to my-bucket with key 'my-key'
         }),
     };
     async run() {
+        if (this.flags.bucket && this.flags.moduleId) {
+            this.error('Cannot specify both --bucket and --moduleId flags. Please use only one type of filter.');
+        }
+        if (!this.flags.bucket && !this.flags.moduleId) {
+            this.error('Either --bucket or --moduleId flag must be specified.');
+        }
         const { args, flags } = await this.parse(ObjectPut);
-        const { client: objectService, userId, organizationId } = await this.objectService();
+        let moduleId = this.flags.moduleId || '';
+        let bucket;
+        const { client: catalogService, userId, organizationId } = await this.catalogService();
+        if (!moduleId && this.flags.bucket) {
+            // If moduleId is not provided and bucket is specified, we need to fetch the moduleId from the catalog service
+            //string split on # to get bucket name and version
+            const [bucketName, version] = this.flags.bucket.split('#');
+            bucket = (await catalogService.resolveBucketLocationToModuleId({
+                userId,
+                organizationId,
+                bucketLocation: {
+                    bucketLocation: {
+                        case: 'bucket',
+                        value: {
+                            name: bucketName,
+                            version: version || '',
+                        },
+                    },
+                },
+            })).bucket;
+            if (!bucket || !bucket.moduleId) {
+                this.error(`Bucket "${this.flags.bucket}" not found or does not belong to any module.`);
+            }
+            moduleId = bucket.moduleId;
+        }
+        else {
+            // query for the bucket if moduleId is provided
+            bucket = (await catalogService.resolveModuleIdToBucketLocation({
+                userId,
+                organizationId,
+                moduleId,
+            })).bucket;
+        }
+        console.log(`Using moduleId: ${moduleId}`);
+        const { client: objectService } = await this.objectService(moduleId);
         // Read the file
         const fileContent = await fs.readFile(args.file);
         // Get content type if not specified
@@ -64,17 +109,19 @@ Upload myfile.txt to my-bucket with key 'my-key'
         const response = await objectService.putObject({
             userId,
             organizationId,
-            bucketId: flags.bucket,
+            moduleId,
             key: args.key,
             content: fileContent,
             contentType,
         });
+        //attach the bucket to the response
+        response.bucket = bucket;
         if (flags.output === 'json') {
             console.log(toJsonString(PutObjectResponseSchema, response, { prettySpaces: 2 }));
         }
         else {
             console.log('Successfully uploaded file:');
-            console.log(`Bucket: ${response.bucketId}`);
+            console.log(`Bucket: ${response.bucket.bucketName}`);
             console.log(`Key: ${response.key}`);
         }
     }

package/dist/commands/query/chunk-search.d.ts

@@ -6,6 +6,8 @@ export default class RagSearch extends BaseCommand<typeof RagSearch> {
     static description: string;
     static examples: string[];
     static flags: {
+        buckets: import("@oclif/core/interfaces").OptionFlag<string[] | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        moduleIds: import("@oclif/core/interfaces").OptionFlag<string[] | undefined, import("@oclif/core/interfaces").CustomOptions>;
         output: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
         impersonate: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         manifest: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
@@ -15,6 +17,7 @@ export default class RagSearch extends BaseCommand<typeof RagSearch> {
         rainbowAuthToken: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowOrganizationId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowUserId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        sendVersionMetadata: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
     };
     searchAgent(query: string): Promise<void>;
     run(): Promise<void>;

package/dist/commands/query/chunk-search.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chunk-search.d.ts","sourceRoot":"","sources":["../../../src/commands/query/chunk-search.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAKpD,MAAM,CAAC,OAAO,OAAO,SAAU,SAAQ,WAAW,CAAC,OAAO,SAAS,CAAC;IAClE,MAAM,CAAC,IAAI;;MAKT;IAEF,MAAM,CAAC,WAAW,SAAoF;IAEtG,MAAM,CAAC,QAAQ,WAIb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;MAqBV;IAEI,WAAW,CAAC,KAAK,EAAE,MAAM;IAkCzB,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAI3B"}
+{"version":3,"file":"chunk-search.d.ts","sourceRoot":"","sources":["../../../src/commands/query/chunk-search.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAKpD,MAAM,CAAC,OAAO,OAAO,SAAU,SAAQ,WAAW,CAAC,OAAO,SAAS,CAAC;IAClE,MAAM,CAAC,IAAI;;MAKT;IAEF,MAAM,CAAC,WAAW,SAAoF;IAEtG,MAAM,CAAC,QAAQ,WAIb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;;;MAiCV;IAEI,WAAW,CAAC,KAAK,EAAE,MAAM;IA0DzB,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAI3B"}

package/dist/commands/query/chunk-search.js

@@ -18,6 +18,18 @@ Run a RAG search query against a Smart Bucket.
     ];
     static flags = {
         ...BaseCommand.HIDDEN_FLAGS,
+        buckets: Flags.string({
+            char: 'b',
+            description: 'Bucket names to search in, version can be specified with #, e.g. my-bucket#versionId',
+            multiple: true,
+            required: false,
+        }),
+        moduleIds: Flags.string({
+            char: 'm',
+            description: 'Module IDs to search in',
+            multiple: true,
+            required: false,
+        }),
         output: Flags.string({
             char: 'o',
             description: 'output format',
@@ -31,7 +43,7 @@ Run a RAG search query against a Smart Bucket.
             hidden: true,
         }),
         manifest: Flags.string({
-            char: 'm',
+            char: 'M',
             description: 'project manifest',
             required: false,
             default: 'raindrop.manifest',
@@ -39,12 +51,36 @@ Run a RAG search query against a Smart Bucket.
         }),
     };
     async searchAgent(query) {
+        // Check if both buckets and moduleIds flags are set
+        if (this.flags.buckets?.length && this.flags.moduleIds?.length) {
+            this.error('Cannot specify both --buckets and --moduleIds flags. Please use only one type of filter.');
+        }
         const { client: searchAgentService, userId, organizationId } = await this.searchAgentService();
         // Run the RAG search
         const response = await searchAgentService.ragSearch({
             userId,
             organizationId,
-            bucketIds: [],
+            bucketLocations: [
+                ...(this.flags.buckets || []).map((bucket) => {
+                    //string split on # to get bucket name and version
+                    const [bucketName, version] = bucket.split('#');
+                    return {
+                        bucketLocation: {
+                            case: 'bucket',
+                            value: {
+                                name: bucketName,
+                                version: version || '',
+                            },
+                        },
+                    };
+                }),
+                ...(this.flags.moduleIds || []).map((moduleId) => ({
+                    bucketLocation: {
+                        case: 'moduleId',
+                        value: moduleId,
+                    },
+                })),
+            ],
             requestId: ulid(),
             input: query,
         });
@@ -59,7 +95,7 @@ Run a RAG search query against a Smart Bucket.
                 console.log('----------------------------------------');
                 console.log(`Type: ${result.type || 'N/A'}`);
                 console.log(`Score: ${result.score || 0}`);
-                console.log(`Source: ${result.source ? `${result.source.bucket}/${result.source.object}` : 'N/A'}`);
+                console.log(`Source: ${result.source ? `${result.source.bucket?.bucketName}/${result.source.object}` : 'N/A'}`);
                 console.log(`Content: ${result.text || 'N/A'}`);
                 console.log('----------------------------------------\n');
             }

package/dist/commands/query/document.d.ts

@@ -0,0 +1,26 @@
+import { BaseCommand } from '../../base-command.js';
+export default class DocumentChat extends BaseCommand<typeof DocumentChat> {
+    static args: {
+        query: import("@oclif/core/interfaces").Arg<string, Record<string, unknown>>;
+    };
+    static description: string;
+    static examples: string[];
+    static flags: {
+        bucket: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        moduleId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        objectId: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        output: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        impersonate: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        manifest: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        config: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        rainbowAuthService: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        raindropCatalogService: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        rainbowAuthToken: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        rainbowOrganizationId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        rainbowUserId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        sendVersionMetadata: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+    };
+    documentChat(query: string): Promise<void>;
+    run(): Promise<void>;
+}
+//# sourceMappingURL=document.d.ts.map

package/dist/commands/query/document.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"document.d.ts","sourceRoot":"","sources":["../../../src/commands/query/document.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAKpD,MAAM,CAAC,OAAO,OAAO,YAAa,SAAQ,WAAW,CAAC,OAAO,YAAY,CAAC;IACxE,MAAM,CAAC,IAAI;;MAKT;IAEF,MAAM,CAAC,WAAW,SAAmD;IAErE,MAAM,CAAC,QAAQ,WAOb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;;;;MAoCV;IAEI,YAAY,CAAC,KAAK,EAAE,MAAM;IA2D1B,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAI3B"}

package/dist/commands/query/document.js

@@ -0,0 +1,117 @@
+import { Args, Flags } from '@oclif/core';
+import { BaseCommand } from '../../base-command.js';
+import { ulid } from '@liquidmetal-ai/drizzle/ulid';
+import { toJsonString } from '@bufbuild/protobuf';
+import { DocumentChatResponseSchema } from '@liquidmetal-ai/drizzle/liquidmetal/v1alpha1/search_agent_pb';
+export default class DocumentChat extends BaseCommand {
+    static args = {
+        query: Args.string({
+            description: 'question or query about the document',
+            required: true,
+        }),
+    };
+    static description = 'Chat with a document stored in a Smart Bucket';
+    static examples = [
+        `<%= config.bin %> query document "What is the main topic of this document?" -b my-bucket -o document.pdf
+Chat with a document in a Smart Bucket using bucket name.
+`,
+        `<%= config.bin %> query document "What is the main topic of this document?" -m module-id -o document.pdf
+Chat with a document in a Smart Bucket using module ID.
+`,
+    ];
+    static flags = {
+        ...BaseCommand.HIDDEN_FLAGS,
+        bucket: Flags.string({
+            char: 'b',
+            description: 'Bucket name containing the document, version can be specified with #, e.g. my-bucket#versionId',
+            required: false,
+        }),
+        moduleId: Flags.string({
+            char: 'm',
+            description: 'Module ID containing the document',
+            required: false,
+        }),
+        objectId: Flags.string({
+            char: 'o',
+            description: 'Object ID of the document to chat with',
+            required: true,
+        }),
+        output: Flags.string({
+            char: 'O',
+            description: 'output format',
+            default: 'text',
+            options: ['text', 'json'],
+        }),
+        impersonate: Flags.string({
+            char: 'i',
+            description: 'impersonate organization',
+            required: false,
+            hidden: true,
+        }),
+        manifest: Flags.string({
+            char: 'M',
+            description: 'project manifest',
+            required: false,
+            default: 'raindrop.manifest',
+            hidden: true,
+        }),
+    };
+    async documentChat(query) {
+        // Check if both bucket and moduleId flags are set
+        if (this.flags.bucket && this.flags.moduleId) {
+            this.error('Cannot specify both --bucket and --moduleId flags. Please use only one type of location.');
+        }
+        // Check if at least one location is specified
+        if (!this.flags.bucket && !this.flags.moduleId) {
+            this.error('Must specify either --bucket or --moduleId flag to identify the document location.');
+        }
+        const { client: searchAgentService, userId, organizationId } = await this.searchAgentService();
+        let bucketLocation;
+        // Create bucket location based on which flag is used
+        if (this.flags.bucket) {
+            // Parse bucket name and version if provided
+            const [bucketName, version] = this.flags.bucket.split('#');
+            bucketLocation = {
+                bucketLocation: {
+                    case: 'bucket',
+                    value: {
+                        name: bucketName,
+                        version: version || '',
+                    },
+                },
+            };
+        }
+        else {
+            // Use moduleId
+            bucketLocation = {
+                bucketLocation: {
+                    case: 'moduleId',
+                    value: this.flags.moduleId,
+                },
+            };
+        }
+        // Run the document chat
+        const response = await searchAgentService.documentChat({
+            userId,
+            organizationId,
+            bucketLocation,
+            objectId: this.flags.objectId,
+            requestId: ulid(),
+            input: query,
+        });
+        // Output results based on format flag
+        if (this.flags.output === 'json') {
+            console.log(toJsonString(DocumentChatResponseSchema, response, { prettySpaces: 2 }));
+        }
+        else {
+            console.log('\nDocument Chat Response:\n');
+            console.log('----------------------------------------');
+            console.log(response.answer);
+            console.log('----------------------------------------\n');
+        }
+    }
+    async run() {
+        const { args } = await this.parse(DocumentChat);
+        await this.documentChat(args.query);
+    }
+}

package/dist/commands/query/search.d.ts

@@ -7,6 +7,7 @@ export default class AgentSearch extends BaseCommand<typeof AgentSearch> {
     static examples: string[];
     static flags: {
         buckets: import("@oclif/core/interfaces").OptionFlag<string[] | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        moduleIds: import("@oclif/core/interfaces").OptionFlag<string[] | undefined, import("@oclif/core/interfaces").CustomOptions>;
         requestId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         page: import("@oclif/core/interfaces").OptionFlag<number, import("@oclif/core/interfaces").CustomOptions>;
         output: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
@@ -18,6 +19,7 @@ export default class AgentSearch extends BaseCommand<typeof AgentSearch> {
         rainbowAuthToken: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowOrganizationId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
         rainbowUserId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        sendVersionMetadata: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
     };
     run(): Promise<void>;
 }

package/dist/commands/query/search.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../../src/commands/query/search.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAGpD,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,WAAW,CAAC,OAAO,WAAW,CAAC;IACtE,MAAM,CAAC,IAAI;;MAKT;IAEF,MAAM,CAAC,WAAW,SAAsE;IAExF,MAAM,CAAC,QAAQ,WAOb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;;;MAoCV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAoF3B"}
+{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../../src/commands/query/search.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAGpD,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,WAAW,CAAC,OAAO,WAAW,CAAC;IACtE,MAAM,CAAC,IAAI;;MAKT;IAEF,MAAM,CAAC,WAAW,SAAsE;IAExF,MAAM,CAAC,QAAQ,WAOb;IAEF,MAAM,CAAC,KAAK;;;;;;;;;;;;;;;MA0CV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAsG3B"}