@liquidmetal-ai/raindrop 0.4.0 → 0.4.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (71) hide show
  1. package/README.md +184 -51
  2. package/dist/base-command.d.ts +7 -0
  3. package/dist/base-command.d.ts.map +1 -1
  4. package/dist/base-command.js +25 -1
  5. package/dist/codegen.d.ts.map +1 -1
  6. package/dist/codegen.js +9 -0
  7. package/dist/commands/annotation/get.d.ts +22 -0
  8. package/dist/commands/annotation/get.d.ts.map +1 -0
  9. package/dist/commands/annotation/get.js +142 -0
  10. package/dist/commands/annotation/list.d.ts +23 -0
  11. package/dist/commands/annotation/list.d.ts.map +1 -0
  12. package/dist/commands/annotation/list.js +223 -0
  13. package/dist/commands/annotation/put.d.ts +23 -0
  14. package/dist/commands/annotation/put.d.ts.map +1 -0
  15. package/dist/commands/annotation/put.js +172 -0
  16. package/dist/commands/build/branch.d.ts +1 -0
  17. package/dist/commands/build/branch.d.ts.map +1 -1
  18. package/dist/commands/build/checkout.d.ts +1 -0
  19. package/dist/commands/build/checkout.d.ts.map +1 -1
  20. package/dist/commands/build/delete.d.ts +1 -0
  21. package/dist/commands/build/delete.d.ts.map +1 -1
  22. package/dist/commands/build/deploy.d.ts +1 -0
  23. package/dist/commands/build/deploy.d.ts.map +1 -1
  24. package/dist/commands/build/env/get.d.ts +1 -0
  25. package/dist/commands/build/env/get.d.ts.map +1 -1
  26. package/dist/commands/build/env/set.d.ts +1 -0
  27. package/dist/commands/build/env/set.d.ts.map +1 -1
  28. package/dist/commands/build/find.d.ts +1 -0
  29. package/dist/commands/build/find.d.ts.map +1 -1
  30. package/dist/commands/build/list.d.ts +1 -0
  31. package/dist/commands/build/list.d.ts.map +1 -1
  32. package/dist/commands/build/sandbox.d.ts +1 -0
  33. package/dist/commands/build/sandbox.d.ts.map +1 -1
  34. package/dist/commands/build/start.d.ts +1 -0
  35. package/dist/commands/build/start.d.ts.map +1 -1
  36. package/dist/commands/build/status.d.ts +1 -0
  37. package/dist/commands/build/status.d.ts.map +1 -1
  38. package/dist/commands/build/stop.d.ts +1 -0
  39. package/dist/commands/build/stop.d.ts.map +1 -1
  40. package/dist/commands/build/unsandbox.d.ts +1 -0
  41. package/dist/commands/build/unsandbox.d.ts.map +1 -1
  42. package/dist/commands/build/upload.d.ts +1 -0
  43. package/dist/commands/build/upload.d.ts.map +1 -1
  44. package/dist/commands/object/delete.d.ts +1 -0
  45. package/dist/commands/object/delete.d.ts.map +1 -1
  46. package/dist/commands/object/delete.js +5 -2
  47. package/dist/commands/object/get.d.ts +1 -0
  48. package/dist/commands/object/get.d.ts.map +1 -1
  49. package/dist/commands/object/get.js +5 -2
  50. package/dist/commands/object/list.d.ts +1 -0
  51. package/dist/commands/object/list.d.ts.map +1 -1
  52. package/dist/commands/object/list.js +5 -2
  53. package/dist/commands/object/put.d.ts +1 -0
  54. package/dist/commands/object/put.d.ts.map +1 -1
  55. package/dist/commands/object/put.js +5 -2
  56. package/dist/commands/query/chunk-search.d.ts +1 -0
  57. package/dist/commands/query/chunk-search.d.ts.map +1 -1
  58. package/dist/commands/query/chunk-search.js +4 -4
  59. package/dist/commands/query/document.d.ts +26 -0
  60. package/dist/commands/query/document.d.ts.map +1 -0
  61. package/dist/commands/query/document.js +117 -0
  62. package/dist/commands/query/search.d.ts +1 -0
  63. package/dist/commands/query/search.d.ts.map +1 -1
  64. package/dist/commands/query/search.js +3 -3
  65. package/dist/index.d.ts +7 -2
  66. package/dist/index.d.ts.map +1 -1
  67. package/dist/index.js +10 -4
  68. package/dist/tsconfig.tsbuildinfo +1 -1
  69. package/oclif.manifest.json +1534 -293
  70. package/package.json +2 -2
  71. package/templates/db/package.json +4 -3
@@ -20,7 +20,7 @@ Run a RAG search query against a Smart Bucket.
20
20
  ...BaseCommand.HIDDEN_FLAGS,
21
21
  buckets: Flags.string({
22
22
  char: 'b',
23
- description: 'Bucket names to search in, version can be specified with @, e.g. my-bucket@versionId',
23
+ description: 'Bucket names to search in, version can be specified with #, e.g. my-bucket#versionId',
24
24
  multiple: true,
25
25
  required: false,
26
26
  }),
@@ -62,8 +62,8 @@ Run a RAG search query against a Smart Bucket.
62
62
  organizationId,
63
63
  bucketLocations: [
64
64
  ...(this.flags.buckets || []).map((bucket) => {
65
- //string split on @ to get bucket name and version
66
- const [bucketName, version] = bucket.split('@');
65
+ //string split on # to get bucket name and version
66
+ const [bucketName, version] = bucket.split('#');
67
67
  return {
68
68
  bucketLocation: {
69
69
  case: 'bucket',
@@ -95,7 +95,7 @@ Run a RAG search query against a Smart Bucket.
95
95
  console.log('----------------------------------------');
96
96
  console.log(`Type: ${result.type || 'N/A'}`);
97
97
  console.log(`Score: ${result.score || 0}`);
98
- console.log(`Source: ${result.source ? `${result.source.bucket}/${result.source.object}` : 'N/A'}`);
98
+ console.log(`Source: ${result.source ? `${result.source.bucket?.bucketName}/${result.source.object}` : 'N/A'}`);
99
99
  console.log(`Content: ${result.text || 'N/A'}`);
100
100
  console.log('----------------------------------------\n');
101
101
  }
@@ -0,0 +1,26 @@
1
+ import { BaseCommand } from '../../base-command.js';
2
+ export default class DocumentChat extends BaseCommand<typeof DocumentChat> {
3
+ static args: {
4
+ query: import("@oclif/core/interfaces").Arg<string, Record<string, unknown>>;
5
+ };
6
+ static description: string;
7
+ static examples: string[];
8
+ static flags: {
9
+ bucket: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
10
+ moduleId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
11
+ objectId: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
12
+ output: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
13
+ impersonate: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
14
+ manifest: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
15
+ config: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
16
+ rainbowAuthService: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
17
+ raindropCatalogService: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
18
+ rainbowAuthToken: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
19
+ rainbowOrganizationId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
20
+ rainbowUserId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
21
+ sendVersionMetadata: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
22
+ };
23
+ documentChat(query: string): Promise<void>;
24
+ run(): Promise<void>;
25
+ }
26
+ //# sourceMappingURL=document.d.ts.map
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,117 @@
1
+ import { Args, Flags } from '@oclif/core';
2
+ import { BaseCommand } from '../../base-command.js';
3
+ import { ulid } from '@liquidmetal-ai/drizzle/ulid';
4
+ import { toJsonString } from '@bufbuild/protobuf';
5
+ import { DocumentChatResponseSchema } from '@liquidmetal-ai/drizzle/liquidmetal/v1alpha1/search_agent_pb';
6
+ export default class DocumentChat extends BaseCommand {
7
+ static args = {
8
+ query: Args.string({
9
+ description: 'question or query about the document',
10
+ required: true,
11
+ }),
12
+ };
13
+ static description = 'Chat with a document stored in a Smart Bucket';
14
+ static examples = [
15
+ `<%= config.bin %> query document "What is the main topic of this document?" -b my-bucket -o document.pdf
16
+ Chat with a document in a Smart Bucket using bucket name.
17
+ `,
18
+ `<%= config.bin %> query document "What is the main topic of this document?" -m module-id -o document.pdf
19
+ Chat with a document in a Smart Bucket using module ID.
20
+ `,
21
+ ];
22
+ static flags = {
23
+ ...BaseCommand.HIDDEN_FLAGS,
24
+ bucket: Flags.string({
25
+ char: 'b',
26
+ description: 'Bucket name containing the document, version can be specified with #, e.g. my-bucket#versionId',
27
+ required: false,
28
+ }),
29
+ moduleId: Flags.string({
30
+ char: 'm',
31
+ description: 'Module ID containing the document',
32
+ required: false,
33
+ }),
34
+ objectId: Flags.string({
35
+ char: 'o',
36
+ description: 'Object ID of the document to chat with',
37
+ required: true,
38
+ }),
39
+ output: Flags.string({
40
+ char: 'O',
41
+ description: 'output format',
42
+ default: 'text',
43
+ options: ['text', 'json'],
44
+ }),
45
+ impersonate: Flags.string({
46
+ char: 'i',
47
+ description: 'impersonate organization',
48
+ required: false,
49
+ hidden: true,
50
+ }),
51
+ manifest: Flags.string({
52
+ char: 'M',
53
+ description: 'project manifest',
54
+ required: false,
55
+ default: 'raindrop.manifest',
56
+ hidden: true,
57
+ }),
58
+ };
59
+ async documentChat(query) {
60
+ // Check if both bucket and moduleId flags are set
61
+ if (this.flags.bucket && this.flags.moduleId) {
62
+ this.error('Cannot specify both --bucket and --moduleId flags. Please use only one type of location.');
63
+ }
64
+ // Check if at least one location is specified
65
+ if (!this.flags.bucket && !this.flags.moduleId) {
66
+ this.error('Must specify either --bucket or --moduleId flag to identify the document location.');
67
+ }
68
+ const { client: searchAgentService, userId, organizationId } = await this.searchAgentService();
69
+ let bucketLocation;
70
+ // Create bucket location based on which flag is used
71
+ if (this.flags.bucket) {
72
+ // Parse bucket name and version if provided
73
+ const [bucketName, version] = this.flags.bucket.split('#');
74
+ bucketLocation = {
75
+ bucketLocation: {
76
+ case: 'bucket',
77
+ value: {
78
+ name: bucketName,
79
+ version: version || '',
80
+ },
81
+ },
82
+ };
83
+ }
84
+ else {
85
+ // Use moduleId
86
+ bucketLocation = {
87
+ bucketLocation: {
88
+ case: 'moduleId',
89
+ value: this.flags.moduleId,
90
+ },
91
+ };
92
+ }
93
+ // Run the document chat
94
+ const response = await searchAgentService.documentChat({
95
+ userId,
96
+ organizationId,
97
+ bucketLocation,
98
+ objectId: this.flags.objectId,
99
+ requestId: ulid(),
100
+ input: query,
101
+ });
102
+ // Output results based on format flag
103
+ if (this.flags.output === 'json') {
104
+ console.log(toJsonString(DocumentChatResponseSchema, response, { prettySpaces: 2 }));
105
+ }
106
+ else {
107
+ console.log('\nDocument Chat Response:\n');
108
+ console.log('----------------------------------------');
109
+ console.log(response.answer);
110
+ console.log('----------------------------------------\n');
111
+ }
112
+ }
113
+ async run() {
114
+ const { args } = await this.parse(DocumentChat);
115
+ await this.documentChat(args.query);
116
+ }
117
+ }
@@ -19,6 +19,7 @@ export default class AgentSearch extends BaseCommand<typeof AgentSearch> {
19
19
  rainbowAuthToken: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
20
20
  rainbowOrganizationId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
21
21
  rainbowUserId: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
22
+ sendVersionMetadata: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
22
23
  };
23
24
  run(): Promise<void>;
24
25
  }
@@ -1 +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;;;;;;;;;;;;;;MA0CV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAsG3B"}
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;;;;;;;;;;;;;;;MA0CV;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAsG3B"}
@@ -21,7 +21,7 @@ Get page 2 of previous search results.
21
21
  ...BaseCommand.HIDDEN_FLAGS,
22
22
  buckets: Flags.string({
23
23
  char: 'b',
24
- description: 'Bucket names to search in, version can be specified with @, e.g. my-bucket@versionId',
24
+ description: 'Bucket names to search in, version can be specified with #, e.g. my-bucket#versionId',
25
25
  multiple: true,
26
26
  required: false,
27
27
  }),
@@ -100,8 +100,8 @@ Get page 2 of previous search results.
100
100
  input: args.query,
101
101
  bucketLocations: [
102
102
  ...(this.flags.buckets || []).map((bucket) => {
103
- //string split on @ to get bucket name and version
104
- const [bucketName, version] = bucket.split('@');
103
+ //string split on # to get bucket name and version
104
+ const [bucketName, version] = bucket.split('#');
105
105
  return {
106
106
  bucketLocation: {
107
107
  case: 'bucket',
package/dist/index.d.ts CHANGED
@@ -7,6 +7,7 @@ import { RainbowAuthService } from '@liquidmetal-ai/drizzle/liquidmetal/v1alpha1
7
7
  import { type RaindropState } from '@liquidmetal-ai/drizzle/liquidmetal/v1alpha1/raindrop_pb';
8
8
  import { SearchAgentService } from '@liquidmetal-ai/drizzle/liquidmetal/v1alpha1/search_agent_pb';
9
9
  import { StrictClient } from './strict-client.js';
10
+ import { AnnotationService } from '@liquidmetal-ai/protogen2/annotation_pb';
10
11
  export { run } from '@oclif/core';
11
12
  export declare const EPOCH_TS: import("@bufbuild/protobuf/wkt").Timestamp;
12
13
  export declare function configFromAppFile(appFile: string): Promise<Application[]>;
@@ -21,7 +22,9 @@ export declare function createAuthenticateInterceptor(configDir: string, rainbow
21
22
  authenticate: Interceptor;
22
23
  };
23
24
  export declare function createTraceInterceptor(): Interceptor;
24
- export declare function versionInterceptor(logger?: {
25
+ export declare function versionInterceptor({ enableSendVersion }: {
26
+ enableSendVersion: boolean;
27
+ }, logger?: {
25
28
  warn: (input: Error | string) => void;
26
29
  }): Interceptor;
27
30
  export declare function readState(configDir: string): Promise<RaindropState>;
@@ -33,11 +36,13 @@ export declare function rainbowAuthService(baseUrl: string): Promise<{
33
36
  export declare function catalogService(opts: ServiceClientOptions): Promise<ServiceClient<typeof CatalogService>>;
34
37
  export declare function searchAgentService(opts: ServiceClientOptions): Promise<ServiceClient<typeof SearchAgentService>>;
35
38
  export declare function objectService(opts: ServiceClientOptions): Promise<ServiceClient<typeof ObjectService>>;
39
+ export declare function annotationService(opts: ServiceClientOptions): Promise<ServiceClient<typeof AnnotationService>>;
36
40
  export type ServiceClientOptions = {
37
41
  rainbowAuth: StrictClient<typeof RainbowAuthService>;
38
42
  configDir: string;
39
43
  identity: ServiceIdentity;
40
44
  overrideAuthToken?: string;
45
+ enableSendVersion: boolean;
41
46
  logger?: {
42
47
  warn(input: Error | string): void;
43
48
  error(input: Error | string): void;
@@ -48,7 +53,7 @@ export type ServiceClient<T extends DescService> = {
48
53
  userId: string;
49
54
  organizationId: string;
50
55
  };
51
- export declare function serviceClient<T extends DescService>(service: T, { rainbowAuth, configDir, identity, overrideAuthToken, logger }: ServiceClientOptions): Promise<ServiceClient<T>>;
56
+ export declare function serviceClient<T extends DescService>(service: T, { rainbowAuth, configDir, identity, overrideAuthToken, logger, enableSendVersion }: ServiceClientOptions): Promise<ServiceClient<T>>;
52
57
  /**
53
58
  * Gets the current version of the raindrop package
54
59
  * @returns The package version string or 'unknown' if it cannot be determined
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAwC,KAAK,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAE5F,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,WAAW,EAAE,MAAM,sCAAsC,CAAC;AAEnE,OAAO,EAAE,cAAc,EAAE,MAAM,yDAAyD,CAAC;AACzF,OAAO,EAAE,aAAa,EAAE,MAAM,wDAAwD,CAAC;AACvF,OAAO,EAEL,kBAAkB,EAEnB,MAAM,8DAA8D,CAAC;AACtE,OAAO,EAAuB,KAAK,aAAa,EAAE,MAAM,0DAA0D,CAAC;AACnH,OAAO,EAAE,kBAAkB,EAAE,MAAM,8DAA8D,CAAC;AAQlG,OAAO,EAAsB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEtE,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAElC,eAAO,MAAM,QAAQ,4CAAiC,CAAC;AAEvD,wBAAsB,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAG/E;AAGD,wBAAgB,QAAQ,CAAC,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,CAEnD;AAkED,MAAM,MAAM,eAAe,GAAG;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,MAAM,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,wBAAsB,oBAAoB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAgBtF;AAKD,wBAAgB,6BAA6B,CAC3C,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,YAAY,CAAC,OAAO,kBAAkB,CAAC,EACpD,iBAAiB,CAAC,EAAE,MAAM,GACzB;IAAE,YAAY,EAAE,WAAW,CAAA;CAAE,CAa/B;AAED,wBAAgB,sBAAsB,IAAI,WAAW,CAiBpD;AAGD,wBAAgB,kBAAkB,CAAC,MAAM,CAAC,EAAE;IAAE,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,KAAK,IAAI,CAAA;CAAE,GAAG,WAAW,CAalG;AAID,wBAAsB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAUzE;AAED,wBAAsB,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CASzF;AAGD,wBAAsB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAUhE;AAED,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;IACjE,MAAM,EAAE,YAAY,CAAC,OAAO,kBAAkB,CAAC,CAAC;CACjD,CAAC,CAUD;AAED,wBAAsB,cAAc,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,aAAa,CAAC,OAAO,cAAc,CAAC,CAAC,CAE9G;AAED,wBAAsB,kBAAkB,CACtC,IAAI,EAAE,oBAAoB,GACzB,OAAO,CAAC,aAAa,CAAC,OAAO,kBAAkB,CAAC,CAAC,CAEnD;AAED,wBAAsB,aAAa,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,aAAa,CAAC,OAAO,aAAa,CAAC,CAAC,CAE5G;AAED,MAAM,MAAM,oBAAoB,GAAG;IACjC,WAAW,EAAE,YAAY,CAAC,OAAO,kBAAkB,CAAC,CAAC;IACrD,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,eAAe,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,MAAM,CAAC,EAAE;QACP,IAAI,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,IAAI,CAAC;QAClC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,IAAI,CAAC;KACpC,CAAC;CACH,CAAC;AACF,MAAM,MAAM,aAAa,CAAC,CAAC,SAAS,WAAW,IAAI;IACjD,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,wBAAsB,aAAa,CAAC,CAAC,SAAS,WAAW,EACvD,OAAO,EAAE,CAAC,EACV,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,EAAE,EAAE,oBAAoB,GACpF,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAe3B;AAED;;;GAGG;AACH,wBAAsB,iBAAiB,IAAI,OAAO,CAAC,MAAM,CAAC,CAkBzD"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAwC,KAAK,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAE5F,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,WAAW,EAAE,MAAM,sCAAsC,CAAC;AAEnE,OAAO,EAAE,cAAc,EAAE,MAAM,yDAAyD,CAAC;AACzF,OAAO,EAAE,aAAa,EAAE,MAAM,wDAAwD,CAAC;AACvF,OAAO,EAEL,kBAAkB,EAEnB,MAAM,8DAA8D,CAAC;AACtE,OAAO,EAAuB,KAAK,aAAa,EAAE,MAAM,0DAA0D,CAAC;AACnH,OAAO,EAAE,kBAAkB,EAAE,MAAM,8DAA8D,CAAC;AAQlG,OAAO,EAAsB,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAElC,eAAO,MAAM,QAAQ,4CAAiC,CAAC;AAEvD,wBAAsB,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAG/E;AAGD,wBAAgB,QAAQ,CAAC,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,CAEnD;AAkED,MAAM,MAAM,eAAe,GAAG;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,MAAM,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,wBAAsB,oBAAoB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAgBtF;AAKD,wBAAgB,6BAA6B,CAC3C,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,YAAY,CAAC,OAAO,kBAAkB,CAAC,EACpD,iBAAiB,CAAC,EAAE,MAAM,GACzB;IAAE,YAAY,EAAE,WAAW,CAAA;CAAE,CAa/B;AAED,wBAAgB,sBAAsB,IAAI,WAAW,CAiBpD;AAGD,wBAAgB,kBAAkB,CAChC,EAAE,iBAAwB,EAAE,EAAE;IAAE,iBAAiB,EAAE,OAAO,CAAA;CAAE,EAC5D,MAAM,CAAC,EAAE;IAAE,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,KAAK,IAAI,CAAA;CAAE,GACjD,WAAW,CAiBb;AAID,wBAAsB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAUzE;AAED,wBAAsB,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CASzF;AAGD,wBAAsB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAUhE;AAED,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;IACjE,MAAM,EAAE,YAAY,CAAC,OAAO,kBAAkB,CAAC,CAAC;CACjD,CAAC,CAUD;AAED,wBAAsB,cAAc,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,aAAa,CAAC,OAAO,cAAc,CAAC,CAAC,CAE9G;AAED,wBAAsB,kBAAkB,CACtC,IAAI,EAAE,oBAAoB,GACzB,OAAO,CAAC,aAAa,CAAC,OAAO,kBAAkB,CAAC,CAAC,CAEnD;AAED,wBAAsB,aAAa,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,aAAa,CAAC,OAAO,aAAa,CAAC,CAAC,CAE5G;AACD,wBAAsB,iBAAiB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,aAAa,CAAC,OAAO,iBAAiB,CAAC,CAAC,CAEpH;AAED,MAAM,MAAM,oBAAoB,GAAG;IACjC,WAAW,EAAE,YAAY,CAAC,OAAO,kBAAkB,CAAC,CAAC;IACrD,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,eAAe,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,EAAE,OAAO,CAAC;IAC3B,MAAM,CAAC,EAAE;QACP,IAAI,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,IAAI,CAAC;QAClC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,IAAI,CAAC;KACpC,CAAC;CACH,CAAC;AACF,MAAM,MAAM,aAAa,CAAC,CAAC,SAAS,WAAW,IAAI;IACjD,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;CACxB,CAAC;AAEF,wBAAsB,aAAa,CAAC,CAAC,SAAS,WAAW,EACvD,OAAO,EAAE,CAAC,EACV,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,EAAE,iBAAiB,EAAE,EAAE,oBAAoB,GACvG,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAe3B;AAED;;;GAGG;AACH,wBAAsB,iBAAiB,IAAI,OAAO,CAAC,MAAM,CAAC,CAkBzD"}
package/dist/index.js CHANGED
@@ -16,6 +16,7 @@ import { fileURLToPath } from 'node:url';
16
16
  import semver from 'semver';
17
17
  import { createStrictClient } from './strict-client.js';
18
18
  import { traceStore } from './trace.js';
19
+ import { AnnotationService } from '@liquidmetal-ai/protogen2/annotation_pb';
19
20
  export { run } from '@oclif/core';
20
21
  export const EPOCH_TS = timestampFromDate(new Date(0));
21
22
  export async function configFromAppFile(appFile) {
@@ -122,10 +123,12 @@ export function createTraceInterceptor() {
122
123
  };
123
124
  }
124
125
  // Creates an interceptor that adds the current CLI version to request headers
125
- export function versionInterceptor(logger) {
126
+ export function versionInterceptor({ enableSendVersion = true }, logger) {
126
127
  return (next) => async (req) => {
127
128
  const version = await getCurrentVersion();
128
- req.header.set(RAINDROP_CLI_VERSION_HEADER, version);
129
+ if (enableSendVersion) {
130
+ req.header.set(RAINDROP_CLI_VERSION_HEADER, version);
131
+ }
129
132
  const response = await next(req);
130
133
  const raindropServiceVersion = response.header.get(RAINDROP_SERVICE_VERSION_HEADER) ?? '0.3.4';
131
134
  if (semver.lt(version, raindropServiceVersion)) {
@@ -188,13 +191,16 @@ export async function searchAgentService(opts) {
188
191
  export async function objectService(opts) {
189
192
  return serviceClient(ObjectService, opts);
190
193
  }
191
- export async function serviceClient(service, { rainbowAuth, configDir, identity, overrideAuthToken, logger }) {
194
+ export async function annotationService(opts) {
195
+ return serviceClient(AnnotationService, opts);
196
+ }
197
+ export async function serviceClient(service, { rainbowAuth, configDir, identity, overrideAuthToken, logger, enableSendVersion }) {
192
198
  const { baseUrl, organizationId, userId } = identity;
193
199
  const { authenticate } = createAuthenticateInterceptor(configDir, rainbowAuth, overrideAuthToken);
194
200
  return {
195
201
  client: createStrictClient(service, createConnectTransport({
196
202
  baseUrl,
197
- interceptors: [authenticate, versionInterceptor(logger), createTraceInterceptor()],
203
+ interceptors: [authenticate, versionInterceptor({ enableSendVersion }, logger), createTraceInterceptor()],
198
204
  })),
199
205
  userId,
200
206
  organizationId,
@@ -1 +1 @@
1
- {"root":["../src/base-command.ts","../src/build.test.ts","../src/build.ts","../src/codegen.test.ts","../src/codegen.ts","../src/config.test.ts","../src/config.ts","../src/deploy.ts","../src/index.test.ts","../src/index.ts","../src/strict-client.ts","../src/trace.ts","../src/commands/tail.ts","../src/commands/auth/list.ts","../src/commands/auth/login.ts","../src/commands/auth/logout.ts","../src/commands/auth/select.ts","../src/commands/build/branch.ts","../src/commands/build/checkout.ts","../src/commands/build/clone.ts","../src/commands/build/delete.ts","../src/commands/build/deploy.ts","../src/commands/build/find.ts","../src/commands/build/generate.ts","../src/commands/build/init.ts","../src/commands/build/list.ts","../src/commands/build/sandbox.ts","../src/commands/build/start.ts","../src/commands/build/status.ts","../src/commands/build/stop.ts","../src/commands/build/unsandbox.ts","../src/commands/build/upload.ts","../src/commands/build/validate.ts","../src/commands/build/env/get.ts","../src/commands/build/env/set.ts","../src/commands/build/tools/check.ts","../src/commands/build/tools/fmt.ts","../src/commands/object/delete.ts","../src/commands/object/get.ts","../src/commands/object/list.ts","../src/commands/object/put.ts","../src/commands/query/chunk-search.ts","../src/commands/query/search.ts"],"version":"5.8.2"}
1
+ {"root":["../src/base-command.ts","../src/build.test.ts","../src/build.ts","../src/codegen.test.ts","../src/codegen.ts","../src/config.test.ts","../src/config.ts","../src/deploy.ts","../src/index.test.ts","../src/index.ts","../src/strict-client.ts","../src/trace.ts","../src/commands/tail.ts","../src/commands/annotation/get.ts","../src/commands/annotation/list.ts","../src/commands/annotation/put.ts","../src/commands/auth/list.ts","../src/commands/auth/login.ts","../src/commands/auth/logout.ts","../src/commands/auth/select.ts","../src/commands/build/branch.ts","../src/commands/build/checkout.ts","../src/commands/build/clone.ts","../src/commands/build/delete.ts","../src/commands/build/deploy.ts","../src/commands/build/find.ts","../src/commands/build/generate.ts","../src/commands/build/init.ts","../src/commands/build/list.ts","../src/commands/build/sandbox.ts","../src/commands/build/start.ts","../src/commands/build/status.ts","../src/commands/build/stop.ts","../src/commands/build/unsandbox.ts","../src/commands/build/upload.ts","../src/commands/build/validate.ts","../src/commands/build/env/get.ts","../src/commands/build/env/set.ts","../src/commands/build/tools/check.ts","../src/commands/build/tools/fmt.ts","../src/commands/object/delete.ts","../src/commands/object/get.ts","../src/commands/object/list.ts","../src/commands/object/put.ts","../src/commands/query/chunk-search.ts","../src/commands/query/document.ts","../src/commands/query/search.ts"],"version":"5.8.2"}