weave-typescript 0.28.0 → 0.29.0
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/dist/weaveapi/ingestion/v1/ingestion.pb.d.ts +204 -0
- package/dist/weaveapi/ingestion/v1/ingestion.pb.js +3830 -970
- package/dist/weaveapi/ingestion/v1/service.pb.d.ts +113 -1
- package/dist/weaveapi/ingestion/v1/service.pb.js +679 -40
- package/dist/weaveapi/retrieval/v1/retrieval.pb.d.ts +45 -1
- package/dist/weaveapi/retrieval/v1/retrieval.pb.js +729 -1
- package/dist/weaveapi/retrieval/v1/service.pb.d.ts +15 -1
- package/dist/weaveapi/retrieval/v1/service.pb.js +79 -0
- package/dist/weavesql/weavedb/ingestion_sql.d.ts +151 -0
- package/dist/weavesql/weavedb/ingestion_sql.js +423 -2
- package/dist/weavesql/weavedb/retrieval_sql.d.ts +31 -0
- package/dist/weavesql/weavedb/retrieval_sql.js +112 -1
- package/package.json +1 -1
|
@@ -1,7 +1,8 @@
|
|
|
1
|
-
import { ExplainRetrievalRequest, ExplainRetrievalResponse, RetrieveRequest, RetrieveResponse } from "./retrieval.pb";
|
|
1
|
+
import { ExplainRetrievalRequest, ExplainRetrievalResponse, RetrieveRequest, RetrieveResponse, SearchDocumentsRequest, SearchDocumentsResponse } from "./retrieval.pb";
|
|
2
2
|
export declare const protobufPackage = "weaveapi.retrieval.v1";
|
|
3
3
|
export interface RetrievalService {
|
|
4
4
|
Retrieve(request: RetrieveRequest): Promise<RetrieveResponse>;
|
|
5
|
+
SearchDocuments(request: SearchDocumentsRequest): Promise<SearchDocumentsResponse>;
|
|
5
6
|
ExplainRetrieval(request: ExplainRetrievalRequest): Promise<ExplainRetrievalResponse>;
|
|
6
7
|
}
|
|
7
8
|
export declare const RetrievalServiceServiceName = "weaveapi.retrieval.v1.RetrievalService";
|
|
@@ -12,6 +13,7 @@ export declare class RetrievalServiceClientImpl implements RetrievalService {
|
|
|
12
13
|
service?: string;
|
|
13
14
|
});
|
|
14
15
|
Retrieve(request: RetrieveRequest): Promise<RetrieveResponse>;
|
|
16
|
+
SearchDocuments(request: SearchDocumentsRequest): Promise<SearchDocumentsResponse>;
|
|
15
17
|
ExplainRetrieval(request: ExplainRetrievalRequest): Promise<ExplainRetrievalResponse>;
|
|
16
18
|
}
|
|
17
19
|
export type RetrievalServiceDefinition = typeof RetrievalServiceDefinition;
|
|
@@ -31,6 +33,18 @@ export declare const RetrievalServiceDefinition: {
|
|
|
31
33
|
};
|
|
32
34
|
};
|
|
33
35
|
};
|
|
36
|
+
readonly searchDocuments: {
|
|
37
|
+
readonly name: "SearchDocuments";
|
|
38
|
+
readonly requestType: typeof SearchDocumentsRequest;
|
|
39
|
+
readonly requestStream: false;
|
|
40
|
+
readonly responseType: typeof SearchDocumentsResponse;
|
|
41
|
+
readonly responseStream: false;
|
|
42
|
+
readonly options: {
|
|
43
|
+
readonly _unknownFields: {
|
|
44
|
+
readonly 578365826: readonly [Uint8Array];
|
|
45
|
+
};
|
|
46
|
+
};
|
|
47
|
+
};
|
|
34
48
|
readonly explainRetrieval: {
|
|
35
49
|
readonly name: "ExplainRetrieval";
|
|
36
50
|
readonly requestType: typeof ExplainRetrievalRequest;
|
|
@@ -16,6 +16,7 @@ class RetrievalServiceClientImpl {
|
|
|
16
16
|
this.service = (opts === null || opts === void 0 ? void 0 : opts.service) || exports.RetrievalServiceServiceName;
|
|
17
17
|
this.rpc = rpc;
|
|
18
18
|
this.Retrieve = this.Retrieve.bind(this);
|
|
19
|
+
this.SearchDocuments = this.SearchDocuments.bind(this);
|
|
19
20
|
this.ExplainRetrieval = this.ExplainRetrieval.bind(this);
|
|
20
21
|
}
|
|
21
22
|
Retrieve(request) {
|
|
@@ -23,6 +24,11 @@ class RetrievalServiceClientImpl {
|
|
|
23
24
|
const promise = this.rpc.request(this.service, "Retrieve", data);
|
|
24
25
|
return promise.then((data) => retrieval_pb_1.RetrieveResponse.decode(new wire_1.BinaryReader(data)));
|
|
25
26
|
}
|
|
27
|
+
SearchDocuments(request) {
|
|
28
|
+
const data = retrieval_pb_1.SearchDocumentsRequest.encode(request).finish();
|
|
29
|
+
const promise = this.rpc.request(this.service, "SearchDocuments", data);
|
|
30
|
+
return promise.then((data) => retrieval_pb_1.SearchDocumentsResponse.decode(new wire_1.BinaryReader(data)));
|
|
31
|
+
}
|
|
26
32
|
ExplainRetrieval(request) {
|
|
27
33
|
const data = retrieval_pb_1.ExplainRetrievalRequest.encode(request).finish();
|
|
28
34
|
const promise = this.rpc.request(this.service, "ExplainRetrieval", data);
|
|
@@ -96,6 +102,79 @@ exports.RetrievalServiceDefinition = {
|
|
|
96
102
|
},
|
|
97
103
|
},
|
|
98
104
|
},
|
|
105
|
+
searchDocuments: {
|
|
106
|
+
name: "SearchDocuments",
|
|
107
|
+
requestType: retrieval_pb_1.SearchDocumentsRequest,
|
|
108
|
+
requestStream: false,
|
|
109
|
+
responseType: retrieval_pb_1.SearchDocumentsResponse,
|
|
110
|
+
responseStream: false,
|
|
111
|
+
options: {
|
|
112
|
+
_unknownFields: {
|
|
113
|
+
578365826: [
|
|
114
|
+
new Uint8Array([
|
|
115
|
+
57,
|
|
116
|
+
58,
|
|
117
|
+
1,
|
|
118
|
+
42,
|
|
119
|
+
34,
|
|
120
|
+
52,
|
|
121
|
+
47,
|
|
122
|
+
118,
|
|
123
|
+
49,
|
|
124
|
+
47,
|
|
125
|
+
111,
|
|
126
|
+
114,
|
|
127
|
+
103,
|
|
128
|
+
47,
|
|
129
|
+
123,
|
|
130
|
+
111,
|
|
131
|
+
114,
|
|
132
|
+
103,
|
|
133
|
+
97,
|
|
134
|
+
110,
|
|
135
|
+
105,
|
|
136
|
+
122,
|
|
137
|
+
97,
|
|
138
|
+
116,
|
|
139
|
+
105,
|
|
140
|
+
111,
|
|
141
|
+
110,
|
|
142
|
+
95,
|
|
143
|
+
105,
|
|
144
|
+
100,
|
|
145
|
+
125,
|
|
146
|
+
47,
|
|
147
|
+
114,
|
|
148
|
+
101,
|
|
149
|
+
116,
|
|
150
|
+
114,
|
|
151
|
+
105,
|
|
152
|
+
101,
|
|
153
|
+
118,
|
|
154
|
+
97,
|
|
155
|
+
108,
|
|
156
|
+
47,
|
|
157
|
+
100,
|
|
158
|
+
111,
|
|
159
|
+
99,
|
|
160
|
+
117,
|
|
161
|
+
109,
|
|
162
|
+
101,
|
|
163
|
+
110,
|
|
164
|
+
116,
|
|
165
|
+
115,
|
|
166
|
+
47,
|
|
167
|
+
115,
|
|
168
|
+
101,
|
|
169
|
+
97,
|
|
170
|
+
114,
|
|
171
|
+
99,
|
|
172
|
+
104,
|
|
173
|
+
]),
|
|
174
|
+
],
|
|
175
|
+
},
|
|
176
|
+
},
|
|
177
|
+
},
|
|
99
178
|
explainRetrieval: {
|
|
100
179
|
name: "ExplainRetrieval",
|
|
101
180
|
requestType: retrieval_pb_1.ExplainRetrievalRequest,
|
|
@@ -161,6 +161,76 @@ export interface ListDocumentsByOrganizationRow {
|
|
|
161
161
|
metadata: any;
|
|
162
162
|
}
|
|
163
163
|
export declare function listDocumentsByOrganization(client: Client, args: ListDocumentsByOrganizationArgs): Promise<ListDocumentsByOrganizationRow[]>;
|
|
164
|
+
export declare const listAccessibleDocumentSummariesQuery = "-- name: ListAccessibleDocumentSummaries :many\nWITH latest_runs AS (\n SELECT DISTINCT ON (r.document_id)\n r.document_id,\n r.id AS latest_ingestion_run_id,\n r.status AS latest_ingestion_status\n FROM weave.ingestion_runs r\n WHERE r.organization_id = $1\n ORDER BY r.document_id, r.started_at DESC\n),\nthread_counts AS (\n SELECT\n t.document_id,\n COUNT(*)::int AS thread_count,\n COUNT(*) FILTER (\n WHERE t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_PENDING'\n )::int AS pending_count,\n COUNT(*) FILTER (\n WHERE t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n )::int AS approved_count,\n COUNT(*) FILTER (\n WHERE t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_REJECTED'\n )::int AS rejected_count\n FROM weave.threads t\n WHERE t.organization_id = $1\n GROUP BY t.document_id\n),\ntag_data AS (\n SELECT\n dt.document_id,\n array_agg(dt.tag ORDER BY dt.tag ASC) AS tags,\n array_agg(dt.confidence ORDER BY dt.tag ASC) AS tag_confidences\n FROM weave.document_tags dt\n JOIN weave.documents d ON d.id = dt.document_id\n WHERE d.organization_id = $1\n GROUP BY dt.document_id\n)\nSELECT\n d.id,\n d.organization_id,\n d.filename,\n d.mime_type,\n d.parser_name,\n d.status,\n d.scope,\n d.sensitivity_tags,\n COALESCE(tag_data.tags, '{}'::text[]) AS tags,\n COALESCE(tag_data.tag_confidences, '{}'::real[]) AS tag_confidences,\n d.uploaded_by_user_id,\n d.uploaded_at,\n d.updated_at,\n d.page_count,\n d.char_count,\n d.metadata,\n COALESCE(latest_runs.latest_ingestion_run_id, '00000000-0000-0000-0000-000000000000'::uuid) AS latest_ingestion_run_id,\n COALESCE(latest_runs.latest_ingestion_status, 'INGESTION_RUN_STATUS_UNSPECIFIED') AS latest_ingestion_status,\n COALESCE(thread_counts.thread_count, 0)::int AS thread_count,\n COALESCE(thread_counts.pending_count, 0)::int AS pending_count,\n COALESCE(thread_counts.approved_count, 0)::int AS approved_count,\n COALESCE(thread_counts.rejected_count, 0)::int AS rejected_count\nFROM weave.documents d\nLEFT JOIN latest_runs ON latest_runs.document_id = d.id\nLEFT JOIN thread_counts ON thread_counts.document_id = d.id\nLEFT JOIN tag_data ON tag_data.document_id = d.id\nWHERE d.organization_id = $1\n AND (\n d.scope = 'DOCUMENT_SCOPE_ORG'\n OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n AND d.uploaded_by_user_id = $2)\n OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n AND EXISTS (\n SELECT 1\n FROM weave.user_sensitivity_clearances c\n WHERE c.organization_id = $1\n AND c.user_id = $2\n AND cardinality(d.sensitivity_tags) = 1\n AND c.sensitivity_tag_slug = d.sensitivity_tags[1]\n ))\n )\n AND (cardinality($3::text[]) = 0 OR d.status = ANY($3::text[]))\n AND (cardinality($4::text[]) = 0 OR latest_runs.latest_ingestion_status = ANY($4::text[]))\n AND (cardinality($5::text[]) = 0 OR d.scope = ANY($5::text[]))\n AND (cardinality($6::text[]) = 0 OR tag_data.tags && $6::text[] OR d.sensitivity_tags && $6::text[])\n AND (cardinality($7::text[]) = 0 OR d.sensitivity_tags && $7::text[])\n AND ($8::timestamptz IS NULL OR d.uploaded_at >= $8::timestamptz)\n AND ($9::timestamptz IS NULL OR d.uploaded_at <= $9::timestamptz)\nORDER BY d.uploaded_at DESC, d.id DESC\nLIMIT $11 OFFSET $10";
|
|
165
|
+
export interface ListAccessibleDocumentSummariesArgs {
|
|
166
|
+
organizationId: string;
|
|
167
|
+
actingUserId: string;
|
|
168
|
+
documentStatuses: string[];
|
|
169
|
+
ingestionStatuses: string[];
|
|
170
|
+
scopes: string[];
|
|
171
|
+
tags: string[];
|
|
172
|
+
restrictedTags: string[];
|
|
173
|
+
uploadedAfter: Date | null;
|
|
174
|
+
uploadedBefore: Date | null;
|
|
175
|
+
pageOffset: string;
|
|
176
|
+
pageSize: string;
|
|
177
|
+
}
|
|
178
|
+
export interface ListAccessibleDocumentSummariesRow {
|
|
179
|
+
id: string;
|
|
180
|
+
organizationId: string;
|
|
181
|
+
filename: string;
|
|
182
|
+
mimeType: string;
|
|
183
|
+
parserName: string;
|
|
184
|
+
status: string;
|
|
185
|
+
scope: string;
|
|
186
|
+
sensitivityTags: string[];
|
|
187
|
+
tags: string;
|
|
188
|
+
tagConfidences: string;
|
|
189
|
+
uploadedByUserId: string;
|
|
190
|
+
uploadedAt: Date;
|
|
191
|
+
updatedAt: Date;
|
|
192
|
+
pageCount: number | null;
|
|
193
|
+
charCount: number | null;
|
|
194
|
+
metadata: any;
|
|
195
|
+
latestIngestionRunId: string;
|
|
196
|
+
latestIngestionStatus: string;
|
|
197
|
+
threadCount: number;
|
|
198
|
+
pendingCount: number;
|
|
199
|
+
approvedCount: number;
|
|
200
|
+
rejectedCount: number;
|
|
201
|
+
}
|
|
202
|
+
export declare function listAccessibleDocumentSummaries(client: Client, args: ListAccessibleDocumentSummariesArgs): Promise<ListAccessibleDocumentSummariesRow[]>;
|
|
203
|
+
export declare const getAccessibleDocumentSummaryQuery = "-- name: GetAccessibleDocumentSummary :one\nWITH latest_runs AS (\n SELECT DISTINCT ON (r.document_id)\n r.document_id,\n r.id AS latest_ingestion_run_id,\n r.status AS latest_ingestion_status\n FROM weave.ingestion_runs r\n WHERE r.organization_id = $1\n AND r.document_id = $2\n ORDER BY r.document_id, r.started_at DESC\n),\nthread_counts AS (\n SELECT\n t.document_id,\n COUNT(*)::int AS thread_count,\n COUNT(*) FILTER (\n WHERE t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_PENDING'\n )::int AS pending_count,\n COUNT(*) FILTER (\n WHERE t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n )::int AS approved_count,\n COUNT(*) FILTER (\n WHERE t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_REJECTED'\n )::int AS rejected_count\n FROM weave.threads t\n WHERE t.organization_id = $1\n AND t.document_id = $2\n GROUP BY t.document_id\n),\ntag_data AS (\n SELECT\n dt.document_id,\n array_agg(dt.tag ORDER BY dt.tag ASC) AS tags,\n array_agg(dt.confidence ORDER BY dt.tag ASC) AS tag_confidences\n FROM weave.document_tags dt\n JOIN weave.documents d ON d.id = dt.document_id\n WHERE d.organization_id = $1\n AND dt.document_id = $2\n GROUP BY dt.document_id\n)\nSELECT\n d.id,\n d.organization_id,\n d.filename,\n d.mime_type,\n d.parser_name,\n d.status,\n d.scope,\n d.sensitivity_tags,\n COALESCE(tag_data.tags, '{}'::text[]) AS tags,\n COALESCE(tag_data.tag_confidences, '{}'::real[]) AS tag_confidences,\n d.uploaded_by_user_id,\n d.uploaded_at,\n d.updated_at,\n d.page_count,\n d.char_count,\n d.metadata,\n COALESCE(latest_runs.latest_ingestion_run_id, '00000000-0000-0000-0000-000000000000'::uuid) AS latest_ingestion_run_id,\n COALESCE(latest_runs.latest_ingestion_status, 'INGESTION_RUN_STATUS_UNSPECIFIED') AS latest_ingestion_status,\n COALESCE(thread_counts.thread_count, 0)::int AS thread_count,\n COALESCE(thread_counts.pending_count, 0)::int AS pending_count,\n COALESCE(thread_counts.approved_count, 0)::int AS approved_count,\n COALESCE(thread_counts.rejected_count, 0)::int AS rejected_count\nFROM weave.documents d\nLEFT JOIN latest_runs ON latest_runs.document_id = d.id\nLEFT JOIN thread_counts ON thread_counts.document_id = d.id\nLEFT JOIN tag_data ON tag_data.document_id = d.id\nWHERE d.organization_id = $1\n AND d.id = $2\n AND (\n d.scope = 'DOCUMENT_SCOPE_ORG'\n OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n AND d.uploaded_by_user_id = $3)\n OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n AND EXISTS (\n SELECT 1\n FROM weave.user_sensitivity_clearances c\n WHERE c.organization_id = $1\n AND c.user_id = $3\n AND cardinality(d.sensitivity_tags) = 1\n AND c.sensitivity_tag_slug = d.sensitivity_tags[1]\n ))\n )";
|
|
204
|
+
export interface GetAccessibleDocumentSummaryArgs {
|
|
205
|
+
organizationId: string;
|
|
206
|
+
documentId: string;
|
|
207
|
+
actingUserId: string;
|
|
208
|
+
}
|
|
209
|
+
export interface GetAccessibleDocumentSummaryRow {
|
|
210
|
+
id: string;
|
|
211
|
+
organizationId: string;
|
|
212
|
+
filename: string;
|
|
213
|
+
mimeType: string;
|
|
214
|
+
parserName: string;
|
|
215
|
+
status: string;
|
|
216
|
+
scope: string;
|
|
217
|
+
sensitivityTags: string[];
|
|
218
|
+
tags: string;
|
|
219
|
+
tagConfidences: string;
|
|
220
|
+
uploadedByUserId: string;
|
|
221
|
+
uploadedAt: Date;
|
|
222
|
+
updatedAt: Date;
|
|
223
|
+
pageCount: number | null;
|
|
224
|
+
charCount: number | null;
|
|
225
|
+
metadata: any;
|
|
226
|
+
latestIngestionRunId: string;
|
|
227
|
+
latestIngestionStatus: string;
|
|
228
|
+
threadCount: number;
|
|
229
|
+
pendingCount: number;
|
|
230
|
+
approvedCount: number;
|
|
231
|
+
rejectedCount: number;
|
|
232
|
+
}
|
|
233
|
+
export declare function getAccessibleDocumentSummary(client: Client, args: GetAccessibleDocumentSummaryArgs): Promise<GetAccessibleDocumentSummaryRow | null>;
|
|
164
234
|
export declare const getDatabaseConfigQuery = "-- name: GetDatabaseConfig :one\nSELECT\n id,\n embedding_dimension,\n deployment_mode,\n created_at,\n updated_at\nFROM weave.database_config\nWHERE id = 1";
|
|
165
235
|
export interface GetDatabaseConfigRow {
|
|
166
236
|
id: number;
|
|
@@ -345,6 +415,12 @@ export interface DeleteDocumentTagsByDocumentIDArgs {
|
|
|
345
415
|
documentId: string;
|
|
346
416
|
}
|
|
347
417
|
export declare function deleteDocumentTagsByDocumentID(client: Client, args: DeleteDocumentTagsByDocumentIDArgs): Promise<void>;
|
|
418
|
+
export declare const deleteDocumentTagByDocumentIDAndTagQuery = "-- name: DeleteDocumentTagByDocumentIDAndTag :execrows\nDELETE FROM weave.document_tags dt\nUSING weave.documents d\nWHERE dt.document_id = d.id\n AND d.organization_id = $1\n AND dt.document_id = $2\n AND dt.tag = $3";
|
|
419
|
+
export interface DeleteDocumentTagByDocumentIDAndTagArgs {
|
|
420
|
+
organizationId: string;
|
|
421
|
+
documentId: string;
|
|
422
|
+
tag: string;
|
|
423
|
+
}
|
|
348
424
|
export declare const createDocumentTagQuery = "-- name: CreateDocumentTag :one\nINSERT INTO weave.document_tags (\n document_id,\n tag,\n confidence\n) VALUES (\n $1,\n $2,\n $3\n)\nRETURNING document_id, tag, confidence";
|
|
349
425
|
export interface CreateDocumentTagArgs {
|
|
350
426
|
documentId: string;
|
|
@@ -368,6 +444,11 @@ export interface ListDocumentTagsByDocumentIDRow {
|
|
|
368
444
|
confidence: number;
|
|
369
445
|
}
|
|
370
446
|
export declare function listDocumentTagsByDocumentID(client: Client, args: ListDocumentTagsByDocumentIDArgs): Promise<ListDocumentTagsByDocumentIDRow[]>;
|
|
447
|
+
export declare const deleteDocumentByOrganizationAndIDQuery = "-- name: DeleteDocumentByOrganizationAndID :execrows\nDELETE FROM weave.documents\nWHERE organization_id = $1\n AND id = $2";
|
|
448
|
+
export interface DeleteDocumentByOrganizationAndIDArgs {
|
|
449
|
+
organizationId: string;
|
|
450
|
+
documentId: string;
|
|
451
|
+
}
|
|
371
452
|
export declare const createThreadQuery = "-- name: CreateThread :one\nINSERT INTO weave.threads (\n id,\n organization_id,\n document_id,\n ingestion_run_id,\n text,\n knowledge_type,\n facet_fields,\n confidence,\n temporal_validity,\n classification,\n retrieval_status,\n source_location,\n metadata\n) VALUES (\n $1,\n $2,\n $3,\n $4,\n $5,\n $6,\n $7,\n $8,\n $9,\n $10,\n $11,\n $12,\n $13\n)\nRETURNING\n id,\n organization_id,\n document_id,\n ingestion_run_id,\n text,\n text_original,\n knowledge_type,\n facet_fields,\n confidence,\n temporal_validity,\n classification,\n retrieval_status,\n source_location,\n superseded_by,\n supersedes,\n cluster_id,\n extracted_at,\n approved_at,\n approved_by_user_id,\n edited,\n metadata";
|
|
372
453
|
export interface CreateThreadArgs {
|
|
373
454
|
id: string;
|
|
@@ -468,6 +549,36 @@ export interface ListThreadsByIngestionRunRow {
|
|
|
468
549
|
metadata: any;
|
|
469
550
|
}
|
|
470
551
|
export declare function listThreadsByIngestionRun(client: Client, args: ListThreadsByIngestionRunArgs): Promise<ListThreadsByIngestionRunRow[]>;
|
|
552
|
+
export declare const listThreadsByDocumentQuery = "-- name: ListThreadsByDocument :many\nSELECT\n t.id,\n t.organization_id,\n t.document_id,\n d.filename AS document_name,\n t.ingestion_run_id,\n t.text,\n t.text_original,\n t.knowledge_type,\n t.facet_fields,\n t.confidence,\n t.temporal_validity,\n t.classification,\n t.retrieval_status,\n t.source_location,\n t.superseded_by,\n t.supersedes,\n t.cluster_id,\n t.extracted_at,\n t.approved_at,\n t.approved_by_user_id,\n t.edited,\n t.metadata\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nWHERE t.organization_id = $1\n AND t.document_id = $2\nORDER BY t.extracted_at ASC, t.id ASC";
|
|
553
|
+
export interface ListThreadsByDocumentArgs {
|
|
554
|
+
organizationId: string;
|
|
555
|
+
documentId: string;
|
|
556
|
+
}
|
|
557
|
+
export interface ListThreadsByDocumentRow {
|
|
558
|
+
id: string;
|
|
559
|
+
organizationId: string;
|
|
560
|
+
documentId: string;
|
|
561
|
+
documentName: string;
|
|
562
|
+
ingestionRunId: string;
|
|
563
|
+
text: string;
|
|
564
|
+
textOriginal: string | null;
|
|
565
|
+
knowledgeType: string;
|
|
566
|
+
facetFields: any;
|
|
567
|
+
confidence: number;
|
|
568
|
+
temporalValidity: any | null;
|
|
569
|
+
classification: string;
|
|
570
|
+
retrievalStatus: string;
|
|
571
|
+
sourceLocation: any;
|
|
572
|
+
supersededBy: string | null;
|
|
573
|
+
supersedes: string[];
|
|
574
|
+
clusterId: string | null;
|
|
575
|
+
extractedAt: Date;
|
|
576
|
+
approvedAt: Date | null;
|
|
577
|
+
approvedByUserId: string | null;
|
|
578
|
+
edited: boolean;
|
|
579
|
+
metadata: any;
|
|
580
|
+
}
|
|
581
|
+
export declare function listThreadsByDocument(client: Client, args: ListThreadsByDocumentArgs): Promise<ListThreadsByDocumentRow[]>;
|
|
471
582
|
export declare const listPendingThreadsByRunQuery = "-- name: ListPendingThreadsByRun :many\nSELECT\n id,\n organization_id,\n document_id,\n ingestion_run_id,\n text,\n text_original,\n knowledge_type,\n facet_fields,\n confidence,\n temporal_validity,\n classification,\n retrieval_status,\n source_location,\n superseded_by,\n supersedes,\n cluster_id,\n extracted_at,\n approved_at,\n approved_by_user_id,\n edited,\n metadata\nFROM weave.threads\nWHERE organization_id = $1\n AND ingestion_run_id = $2\n AND retrieval_status = 'THREAD_RETRIEVAL_STATUS_PENDING'\nORDER BY extracted_at ASC, id ASC";
|
|
472
583
|
export interface ListPendingThreadsByRunArgs {
|
|
473
584
|
organizationId: string;
|
|
@@ -653,6 +764,46 @@ export interface EditPendingThreadRow {
|
|
|
653
764
|
metadata: any;
|
|
654
765
|
}
|
|
655
766
|
export declare function editPendingThread(client: Client, args: EditPendingThreadArgs): Promise<EditPendingThreadRow | null>;
|
|
767
|
+
export declare const editDocumentThreadQuery = "-- name: EditDocumentThread :one\nWITH updated AS (\n UPDATE weave.threads AS t\n SET\n text_original = CASE\n WHEN t.text_original IS NULL OR t.text_original = '' THEN t.text\n ELSE t.text_original\n END,\n text = $1,\n source_location = COALESCE($2::jsonb, t.source_location),\n metadata = t.metadata || COALESCE($3::jsonb, '{}'::jsonb),\n edited = TRUE\n WHERE t.id = $4\n AND t.organization_id = $5\n AND t.document_id = $6\n AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_PENDING'\n RETURNING\n id,\n organization_id,\n document_id,\n ingestion_run_id,\n text,\n text_original,\n knowledge_type,\n facet_fields,\n confidence,\n temporal_validity,\n classification,\n retrieval_status,\n source_location,\n superseded_by,\n supersedes,\n cluster_id,\n extracted_at,\n approved_at,\n approved_by_user_id,\n edited,\n metadata\n)\nSELECT\n updated.id,\n updated.organization_id,\n updated.document_id,\n d.filename AS document_name,\n updated.ingestion_run_id,\n updated.text,\n updated.text_original,\n updated.knowledge_type,\n updated.facet_fields,\n updated.confidence,\n updated.temporal_validity,\n updated.classification,\n updated.retrieval_status,\n updated.source_location,\n updated.superseded_by,\n updated.supersedes,\n updated.cluster_id,\n updated.extracted_at,\n updated.approved_at,\n updated.approved_by_user_id,\n updated.edited,\n updated.metadata\nFROM updated\nJOIN weave.documents d ON d.id = updated.document_id";
|
|
768
|
+
export interface EditDocumentThreadArgs {
|
|
769
|
+
text: string;
|
|
770
|
+
sourceLocation: any | null;
|
|
771
|
+
metadata: any | null;
|
|
772
|
+
id: string;
|
|
773
|
+
organizationId: string;
|
|
774
|
+
documentId: string;
|
|
775
|
+
}
|
|
776
|
+
export interface EditDocumentThreadRow {
|
|
777
|
+
id: string;
|
|
778
|
+
organizationId: string;
|
|
779
|
+
documentId: string;
|
|
780
|
+
documentName: string;
|
|
781
|
+
ingestionRunId: string;
|
|
782
|
+
text: string;
|
|
783
|
+
textOriginal: string | null;
|
|
784
|
+
knowledgeType: string;
|
|
785
|
+
facetFields: any;
|
|
786
|
+
confidence: number;
|
|
787
|
+
temporalValidity: any | null;
|
|
788
|
+
classification: string;
|
|
789
|
+
retrievalStatus: string;
|
|
790
|
+
sourceLocation: any;
|
|
791
|
+
supersededBy: string | null;
|
|
792
|
+
supersedes: string[];
|
|
793
|
+
clusterId: string | null;
|
|
794
|
+
extractedAt: Date;
|
|
795
|
+
approvedAt: Date | null;
|
|
796
|
+
approvedByUserId: string | null;
|
|
797
|
+
edited: boolean;
|
|
798
|
+
metadata: any;
|
|
799
|
+
}
|
|
800
|
+
export declare function editDocumentThread(client: Client, args: EditDocumentThreadArgs): Promise<EditDocumentThreadRow | null>;
|
|
801
|
+
export declare const deleteDocumentThreadByOrganizationAndIDQuery = "-- name: DeleteDocumentThreadByOrganizationAndID :execrows\nDELETE FROM weave.threads\nWHERE organization_id = $1\n AND document_id = $2\n AND id = $3\n AND retrieval_status = 'THREAD_RETRIEVAL_STATUS_PENDING'";
|
|
802
|
+
export interface DeleteDocumentThreadByOrganizationAndIDArgs {
|
|
803
|
+
organizationId: string;
|
|
804
|
+
documentId: string;
|
|
805
|
+
id: string;
|
|
806
|
+
}
|
|
656
807
|
export declare const approveThreadQuery = "-- name: ApproveThread :one\nUPDATE weave.threads\nSET\n retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED',\n approved_at = now(),\n approved_by_user_id = $1,\n embedding = COALESCE($2::vector, embedding),\n embedding_next = NULL\nWHERE id = $3\n AND organization_id = $4\n AND retrieval_status = 'THREAD_RETRIEVAL_STATUS_PENDING'\nRETURNING\n id,\n organization_id,\n document_id,\n ingestion_run_id,\n text,\n text_original,\n knowledge_type,\n facet_fields,\n confidence,\n temporal_validity,\n classification,\n retrieval_status,\n source_location,\n superseded_by,\n supersedes,\n cluster_id,\n extracted_at,\n approved_at,\n approved_by_user_id,\n edited,\n metadata";
|
|
657
808
|
export interface ApproveThreadArgs {
|
|
658
809
|
approvedByUserId: string | null;
|