@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.
- package/README.md +184 -51
- package/dist/base-command.d.ts +7 -0
- package/dist/base-command.d.ts.map +1 -1
- package/dist/base-command.js +25 -1
- package/dist/codegen.d.ts.map +1 -1
- package/dist/codegen.js +9 -0
- package/dist/commands/annotation/get.d.ts +22 -0
- package/dist/commands/annotation/get.d.ts.map +1 -0
- package/dist/commands/annotation/get.js +142 -0
- package/dist/commands/annotation/list.d.ts +23 -0
- package/dist/commands/annotation/list.d.ts.map +1 -0
- package/dist/commands/annotation/list.js +223 -0
- package/dist/commands/annotation/put.d.ts +23 -0
- package/dist/commands/annotation/put.d.ts.map +1 -0
- package/dist/commands/annotation/put.js +172 -0
- package/dist/commands/build/branch.d.ts +1 -0
- package/dist/commands/build/branch.d.ts.map +1 -1
- package/dist/commands/build/checkout.d.ts +1 -0
- package/dist/commands/build/checkout.d.ts.map +1 -1
- package/dist/commands/build/delete.d.ts +1 -0
- package/dist/commands/build/delete.d.ts.map +1 -1
- package/dist/commands/build/deploy.d.ts +1 -0
- package/dist/commands/build/deploy.d.ts.map +1 -1
- package/dist/commands/build/env/get.d.ts +1 -0
- package/dist/commands/build/env/get.d.ts.map +1 -1
- package/dist/commands/build/env/set.d.ts +1 -0
- package/dist/commands/build/env/set.d.ts.map +1 -1
- package/dist/commands/build/find.d.ts +1 -0
- package/dist/commands/build/find.d.ts.map +1 -1
- package/dist/commands/build/list.d.ts +1 -0
- package/dist/commands/build/list.d.ts.map +1 -1
- package/dist/commands/build/sandbox.d.ts +1 -0
- package/dist/commands/build/sandbox.d.ts.map +1 -1
- package/dist/commands/build/start.d.ts +1 -0
- package/dist/commands/build/start.d.ts.map +1 -1
- package/dist/commands/build/status.d.ts +1 -0
- package/dist/commands/build/status.d.ts.map +1 -1
- package/dist/commands/build/stop.d.ts +1 -0
- package/dist/commands/build/stop.d.ts.map +1 -1
- package/dist/commands/build/unsandbox.d.ts +1 -0
- package/dist/commands/build/unsandbox.d.ts.map +1 -1
- package/dist/commands/build/upload.d.ts +1 -0
- package/dist/commands/build/upload.d.ts.map +1 -1
- package/dist/commands/object/delete.d.ts +1 -0
- package/dist/commands/object/delete.d.ts.map +1 -1
- package/dist/commands/object/delete.js +5 -2
- package/dist/commands/object/get.d.ts +1 -0
- package/dist/commands/object/get.d.ts.map +1 -1
- package/dist/commands/object/get.js +5 -2
- package/dist/commands/object/list.d.ts +1 -0
- package/dist/commands/object/list.d.ts.map +1 -1
- package/dist/commands/object/list.js +5 -2
- package/dist/commands/object/put.d.ts +1 -0
- package/dist/commands/object/put.d.ts.map +1 -1
- package/dist/commands/object/put.js +5 -2
- package/dist/commands/query/chunk-search.d.ts +1 -0
- package/dist/commands/query/chunk-search.d.ts.map +1 -1
- package/dist/commands/query/chunk-search.js +4 -4
- package/dist/commands/query/document.d.ts +26 -0
- package/dist/commands/query/document.d.ts.map +1 -0
- package/dist/commands/query/document.js +117 -0
- package/dist/commands/query/search.d.ts +1 -0
- package/dist/commands/query/search.d.ts.map +1 -1
- package/dist/commands/query/search.js +3 -3
- package/dist/index.d.ts +7 -2
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +10 -4
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/oclif.manifest.json +1534 -293
- package/package.json +2 -2
- 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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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(
|
|
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
|
package/dist/index.d.ts.map
CHANGED
|
@@ -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,
|
|
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
|
-
|
|
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
|
|
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"}
|