@databricks/sdk-vectorsearch 0.0.0-dev → 0.1.0-dev.1

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.
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/v1/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAC,MAAM,EAAE,oBAAoB,EAAC,MAAM,UAAU,CAAC;AAEtD,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,sBAAsB,EACtB,eAAe,EACf,oBAAoB,GACrB,MAAM,SAAS,CAAC;AAEjB,YAAY,EACV,UAAU,EACV,qBAAqB,EACrB,wBAAwB,EACxB,SAAS,EACT,4BAA4B,EAC5B,6BAA6B,EAC7B,qBAAqB,EACrB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EACzB,wBAAwB,EACxB,+BAA+B,EAC/B,2BAA2B,EAC3B,qBAAqB,EACrB,qBAAqB,EACrB,QAAQ,EACR,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,qBAAqB,EACrB,mBAAmB,EACnB,oBAAoB,EACpB,SAAS,EACT,sBAAsB,EACtB,uBAAuB,EACvB,mBAAmB,EACnB,MAAM,EACN,WAAW,EACX,WAAW,EACX,YAAY,EACZ,eAAe,EACf,gCAAgC,EAChC,iCAAiC,EACjC,oBAAoB,EACpB,+BAA+B,EAC/B,uBAAuB,EACvB,wBAAwB,EACxB,cAAc,EACd,iCAAiC,EACjC,UAAU,EACV,cAAc,EACd,iCAAiC,EACjC,kCAAkC,EAClC,sBAAsB,EACtB,uBAAuB,EACvB,MAAM,EACN,sBAAsB,EACtB,uBAAuB,EACvB,+BAA+B,EAC/B,gCAAgC,EAChC,4BAA4B,EAC5B,6BAA6B,EAC7B,sBAAsB,EACtB,KAAK,EACL,WAAW,EACX,iBAAiB,GAClB,MAAM,SAAS,CAAC"}
@@ -0,0 +1,4 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+ export { Client, CreateEndpointWaiter } from './client';
3
+ export { EndpointType, IndexSubtype, PipelineType, ScalingChangeState, UpsertDeleteDataStatus, VectorIndexType, EndpointStatus_State, } from './model';
4
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/v1/index.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAE/E,OAAO,EAAC,MAAM,EAAE,oBAAoB,EAAC,MAAM,UAAU,CAAC;AAEtD,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,sBAAsB,EACtB,eAAe,EACf,oBAAoB,GACrB,MAAM,SAAS,CAAC"}
@@ -0,0 +1,661 @@
1
+ import type { Temporal } from '@js-temporal/polyfill';
2
+ import type { JsonValue } from '@databricks/sdk-core/wkt';
3
+ import { z } from 'zod';
4
+ /** Type of endpoint. */
5
+ export declare enum EndpointType {
6
+ STORAGE_OPTIMIZED = "STORAGE_OPTIMIZED",
7
+ STANDARD = "STANDARD"
8
+ }
9
+ /**
10
+ * The subtype of the AI Search index, determining the indexing and retrieval strategy.
11
+ * - `VECTOR`: Not supported. Use `HYBRID` instead.
12
+ * - `FULL_TEXT`: An index that uses full-text search without vector embeddings.
13
+ * - `HYBRID`: An index that uses vector embeddings for similarity search and hybrid search.
14
+ */
15
+ export declare enum IndexSubtype {
16
+ VECTOR = "VECTOR",
17
+ FULL_TEXT = "FULL_TEXT",
18
+ HYBRID = "HYBRID"
19
+ }
20
+ /**
21
+ * Pipeline execution mode.
22
+ * - `TRIGGERED`: If the pipeline uses the triggered execution mode, the system stops processing after successfully refreshing the source table in the pipeline once, ensuring the table is updated based on the data available when the update started.
23
+ * - `CONTINUOUS`: If the pipeline uses continuous execution, the pipeline processes new data as it arrives in the source table to keep vector index fresh.
24
+ */
25
+ export declare enum PipelineType {
26
+ TRIGGERED = "TRIGGERED",
27
+ CONTINUOUS = "CONTINUOUS"
28
+ }
29
+ export declare enum ScalingChangeState {
30
+ SCALING_CHANGE_UNSPECIFIED = "SCALING_CHANGE_UNSPECIFIED",
31
+ SCALING_CHANGE_APPLIED = "SCALING_CHANGE_APPLIED",
32
+ SCALING_CHANGE_IN_PROGRESS = "SCALING_CHANGE_IN_PROGRESS"
33
+ }
34
+ export declare enum UpsertDeleteDataStatus {
35
+ SUCCESS = "SUCCESS",
36
+ PARTIAL_SUCCESS = "PARTIAL_SUCCESS",
37
+ FAILURE = "FAILURE"
38
+ }
39
+ /**
40
+ * There are 2 types of AI Search indexes:
41
+ * - `DELTA_SYNC`: An index that automatically syncs with a source Delta Table, automatically and incrementally updating the index as the underlying data in the Delta Table changes.
42
+ * - `DIRECT_ACCESS`: An index that supports direct read and write of vectors and metadata through our REST and SDK APIs. With this model, the user manages index updates.
43
+ */
44
+ export declare enum VectorIndexType {
45
+ DELTA_SYNC = "DELTA_SYNC",
46
+ DIRECT_ACCESS = "DIRECT_ACCESS"
47
+ }
48
+ /** Current state of the endpoint */
49
+ export declare enum EndpointStatus_State {
50
+ PROVISIONING = "PROVISIONING",
51
+ ONLINE = "ONLINE",
52
+ OFFLINE = "OFFLINE",
53
+ /**
54
+ * After the endpoint is ready, it can be in one of the following states:
55
+ * - RED_STATE: The endpoint is unhealthy and needs to be investigated.
56
+ * - YELLOW_STATE: The endpoint is healthy but needs to be monitored.
57
+ * - ONLINE: The endpoint is healthy and ready to serve traffic.
58
+ */
59
+ RED_STATE = "RED_STATE",
60
+ YELLOW_STATE = "YELLOW_STATE",
61
+ /**
62
+ * The endpoint is being deleted or has been deleted. Associated resources
63
+ * are being cleaned up; once cleanup completes the endpoint will no longer
64
+ * be retrievable.
65
+ */
66
+ DELETED = "DELETED"
67
+ }
68
+ export interface ColumnInfo {
69
+ /** Name of the column. */
70
+ name?: string | undefined;
71
+ /** Data type of the column (e.g., "string", "int", "array<float>") */
72
+ typeText?: string | undefined;
73
+ }
74
+ export interface CreateEndpointRequest {
75
+ /** Name of the AI Search endpoint */
76
+ name?: string | undefined;
77
+ /** Type of endpoint */
78
+ endpointType?: EndpointType | undefined;
79
+ /** The budget policy id to be applied */
80
+ budgetPolicyId?: string | undefined;
81
+ /** The usage policy id to be applied once we've migrated to usage policies */
82
+ usagePolicyId?: string | undefined;
83
+ /**
84
+ * Target QPS for the endpoint. Mutually exclusive with num_replicas.
85
+ * The actual replica count is calculated at index creation/sync time based on this value.
86
+ * Best-effort target; the system does not guarantee this QPS will be achieved.
87
+ */
88
+ targetQps?: bigint | undefined;
89
+ }
90
+ export interface CreateVectorIndexRequest {
91
+ /** Name of the index */
92
+ name?: string | undefined;
93
+ /** Name of the endpoint to be used for serving the index */
94
+ endpointName?: string | undefined;
95
+ /** Primary key of the index */
96
+ primaryKey?: string | undefined;
97
+ indexType?: VectorIndexType | undefined;
98
+ indexSpec?: {
99
+ $case: 'directAccessIndexSpec';
100
+ /** Specification for Direct Vector Access Index. Required if `index_type` is `DIRECT_ACCESS`. */
101
+ directAccessIndexSpec: DirectAccessVectorIndexSpec;
102
+ } | {
103
+ $case: 'deltaSyncIndexSpec';
104
+ /** Specification for Delta Sync Index. Required if `index_type` is `DELTA_SYNC`. */
105
+ deltaSyncIndexSpec: DeltaSyncVectorIndexSpecRequest;
106
+ } | undefined;
107
+ /** The subtype of the index. Use `HYBRID` or `FULL_TEXT`. `VECTOR` is not supported. */
108
+ indexSubtype?: IndexSubtype | undefined;
109
+ }
110
+ export interface CustomTag {
111
+ /** Key field for an AI Search endpoint tag. */
112
+ key?: string | undefined;
113
+ /** [Optional] Value field for an AI Search endpoint tag. */
114
+ value?: string | undefined;
115
+ }
116
+ /** Request payload for deleting data from a vector index. */
117
+ export interface DeleteDataVectorIndexRequest {
118
+ /** Name of the vector index where data is to be deleted. Must be a Direct Vector Access Index. */
119
+ name?: string | undefined;
120
+ /** List of primary keys for the data to be deleted. */
121
+ primaryKeys?: string[] | undefined;
122
+ }
123
+ export interface DeleteDataVectorIndexResponse {
124
+ /** Status of the delete operation. */
125
+ status?: UpsertDeleteDataStatus | undefined;
126
+ /** Result of the upsert or delete operation. */
127
+ result?: UpsertDeleteDataResult | undefined;
128
+ }
129
+ export interface DeleteEndpointRequest {
130
+ /** Name of the AI Search endpoint */
131
+ name?: string | undefined;
132
+ }
133
+ export interface DeleteEndpointResponse {
134
+ }
135
+ export interface DeleteVectorIndexRequest {
136
+ /** Name of the index */
137
+ name?: string | undefined;
138
+ }
139
+ export interface DeleteVectorIndexResponse {
140
+ }
141
+ export interface DeltaSyncVectorIndexSpec {
142
+ /** The name of the source table. */
143
+ sourceTable?: string | undefined;
144
+ /** The columns that contain the embedding source. */
145
+ embeddingSourceColumns?: EmbeddingSourceColumn[] | undefined;
146
+ /** The columns that contain the embedding vectors. */
147
+ embeddingVectorColumns?: EmbeddingVectorColumn[] | undefined;
148
+ /**
149
+ * Pipeline execution mode.
150
+ * - `TRIGGERED`: If the pipeline uses the triggered execution mode, the system stops processing after successfully refreshing the source table in the pipeline once, ensuring the table is updated based on the data available when the update started.
151
+ * - `CONTINUOUS`: If the pipeline uses continuous execution, the pipeline processes new data as it arrives in the source table to keep vector index fresh.
152
+ */
153
+ pipelineType?: PipelineType | undefined;
154
+ /** The ID of the pipeline that is used to sync the index. */
155
+ pipelineId?: string | undefined;
156
+ /** [Optional] Name of the Delta table to sync the vector index contents and computed embeddings to. */
157
+ embeddingWritebackTable?: string | undefined;
158
+ /**
159
+ * [Optional] Select the columns to sync with the vector index. If you leave this field blank, all columns
160
+ * from the source table are synced with the index. The primary key column and embedding source column or
161
+ * embedding vector column are always synced.
162
+ */
163
+ columnsToSync?: string[] | undefined;
164
+ /**
165
+ * [Optional] Alias for columns_to_sync. Select the columns to include in the vector index.
166
+ * If you leave this field blank, all columns from the source table are included.
167
+ * The primary key column and embedding source column or embedding vector column are always included.
168
+ * Only one of columns_to_sync or columns_to_index may be specified.
169
+ */
170
+ columnsToIndex?: string[] | undefined;
171
+ }
172
+ export interface DeltaSyncVectorIndexSpecRequest {
173
+ /** The name of the source table. */
174
+ sourceTable?: string | undefined;
175
+ /** The columns that contain the embedding source. */
176
+ embeddingSourceColumns?: EmbeddingSourceColumn[] | undefined;
177
+ /** The columns that contain the embedding vectors. */
178
+ embeddingVectorColumns?: EmbeddingVectorColumn[] | undefined;
179
+ /**
180
+ * Pipeline execution mode.
181
+ * - `TRIGGERED`: If the pipeline uses the triggered execution mode, the system stops processing after successfully refreshing the source table in the pipeline once, ensuring the table is updated based on the data available when the update started.
182
+ * - `CONTINUOUS`: If the pipeline uses continuous execution, the pipeline processes new data as it arrives in the source table to keep vector index fresh.
183
+ */
184
+ pipelineType?: PipelineType | undefined;
185
+ /** The ID of the pipeline that is used to sync the index. */
186
+ pipelineId?: string | undefined;
187
+ /** [Optional] Name of the Delta table to sync the vector index contents and computed embeddings to. */
188
+ embeddingWritebackTable?: string | undefined;
189
+ /**
190
+ * [Optional] Select the columns to sync with the vector index. If you leave this field blank, all columns
191
+ * from the source table are synced with the index. The primary key column and embedding source column or
192
+ * embedding vector column are always synced.
193
+ */
194
+ columnsToSync?: string[] | undefined;
195
+ /**
196
+ * [Optional] Alias for columns_to_sync. Select the columns to include in the vector index.
197
+ * If you leave this field blank, all columns from the source table are included.
198
+ * The primary key column and embedding source column or embedding vector column are always included.
199
+ * Only one of columns_to_sync or columns_to_index may be specified.
200
+ */
201
+ columnsToIndex?: string[] | undefined;
202
+ }
203
+ export interface DirectAccessVectorIndexSpec {
204
+ /** The columns that contain the embedding vectors. The format should be array[double]. */
205
+ embeddingVectorColumns?: EmbeddingVectorColumn[] | undefined;
206
+ /**
207
+ * The schema of the index in JSON format.
208
+ * Supported types are `integer`, `long`, `float`, `double`, `boolean`, `string`, `date`, `timestamp`.
209
+ * Supported types for vector column: `array<float>`, `array<double>`,`.
210
+ */
211
+ schemaJson?: string | undefined;
212
+ /** The columns that contain the embedding source. The format should be array[double]. */
213
+ embeddingSourceColumns?: EmbeddingSourceColumn[] | undefined;
214
+ }
215
+ export interface EmbeddingSourceColumn {
216
+ /** Name of the column */
217
+ name?: string | undefined;
218
+ /** TODO: clean up ai gateway related code. It's deprecated on ModelServing side. */
219
+ embeddingConfig?: {
220
+ $case: 'embeddingModelEndpointName';
221
+ /** Name of the embedding model endpoint, used by default for both ingestion and querying. */
222
+ embeddingModelEndpointName: string;
223
+ } | undefined;
224
+ /** Name of the embedding model endpoint which, if specified, is used for querying (not ingestion). */
225
+ modelEndpointNameForQuery?: string | undefined;
226
+ }
227
+ export interface EmbeddingVectorColumn {
228
+ /** Name of the column */
229
+ name?: string | undefined;
230
+ /** Dimension of the embedding vector */
231
+ embeddingDimension?: number | undefined;
232
+ }
233
+ export interface Endpoint {
234
+ /** Name of the AI Search endpoint */
235
+ name?: string | undefined;
236
+ /** Creator of the endpoint */
237
+ creator?: string | undefined;
238
+ /** Timestamp of endpoint creation */
239
+ creationTimestamp?: bigint | undefined;
240
+ /** Timestamp of last update to the endpoint */
241
+ lastUpdatedTimestamp?: bigint | undefined;
242
+ /** Type of endpoint */
243
+ endpointType?: EndpointType | undefined;
244
+ /** User who last updated the endpoint */
245
+ lastUpdatedUser?: string | undefined;
246
+ /** Unique identifier of the endpoint */
247
+ id?: string | undefined;
248
+ /** Current status of the endpoint */
249
+ endpointStatus?: EndpointStatus | undefined;
250
+ /** Number of indexes on the endpoint */
251
+ numIndexes?: number | undefined;
252
+ /** The user-selected budget policy id for the endpoint. */
253
+ budgetPolicyId?: string | undefined;
254
+ /** The budget policy id applied to the endpoint */
255
+ effectiveBudgetPolicyId?: string | undefined;
256
+ /** The custom tags assigned to the endpoint */
257
+ customTags?: CustomTag[] | undefined;
258
+ /** Scaling information for the endpoint */
259
+ scalingInfo?: EndpointScalingInfo | undefined;
260
+ }
261
+ export interface EndpointScalingInfo {
262
+ /** The current state of the scaling change request. */
263
+ state?: ScalingChangeState | undefined;
264
+ /**
265
+ * The requested QPS target for the endpoint. Best-effort; the system does not
266
+ * guarantee this QPS will be achieved.
267
+ */
268
+ requestedTargetQps?: bigint | undefined;
269
+ }
270
+ /** Status information of an endpoint */
271
+ export interface EndpointStatus {
272
+ /** Current state of the endpoint */
273
+ state?: EndpointStatus_State | undefined;
274
+ /** Additional status message */
275
+ message?: string | undefined;
276
+ }
277
+ export interface GetEndpointRequest {
278
+ /** Name of the endpoint */
279
+ name?: string | undefined;
280
+ }
281
+ export interface GetVectorIndexRequest {
282
+ /** Name of the index */
283
+ name?: string | undefined;
284
+ /**
285
+ * If true, the URL returned for the index is guaranteed to be compatible with the reranker.
286
+ * Currently this means we return the CP URL regardless of how the index is being accessed.
287
+ * If not set or set to false, the URL may still be compatible with the reranker depending on
288
+ * what URL we return.
289
+ */
290
+ ensureRerankerCompatible?: boolean | undefined;
291
+ }
292
+ export interface ListEndpointRequest {
293
+ /** Token for pagination */
294
+ pageToken?: string | undefined;
295
+ }
296
+ export interface ListEndpointResponse {
297
+ /** An array of Endpoint objects */
298
+ endpoints?: Endpoint[] | undefined;
299
+ /** A token that can be used to get the next page of results. If not present, there are no more results to show. */
300
+ nextPageToken?: string | undefined;
301
+ }
302
+ export interface ListValue {
303
+ /** Repeated field of dynamically typed values. */
304
+ values?: Value[] | undefined;
305
+ }
306
+ export interface ListVectorIndexRequest {
307
+ /** Name of the endpoint */
308
+ endpointName?: string | undefined;
309
+ /** Token for pagination */
310
+ pageToken?: string | undefined;
311
+ }
312
+ export interface ListVectorIndexResponse {
313
+ vectorIndexes?: MiniVectorIndex[] | undefined;
314
+ /** A token that can be used to get the next page of results. If not present, there are no more results to show. */
315
+ nextPageToken?: string | undefined;
316
+ }
317
+ /** Key-value pair. */
318
+ export interface MapStringValueEntry {
319
+ /** Column name. */
320
+ key?: string | undefined;
321
+ /** Column value, nullable. */
322
+ value?: Value | undefined;
323
+ }
324
+ /** Metric specification */
325
+ export interface Metric {
326
+ /** Metric name */
327
+ name?: string | undefined;
328
+ /** Metric labels */
329
+ labels?: MetricLabel[] | undefined;
330
+ /** Percentile for the metric */
331
+ percentile?: number | undefined;
332
+ }
333
+ /** Label for a metric */
334
+ export interface MetricLabel {
335
+ /** Label name */
336
+ name?: string | undefined;
337
+ /** Label value */
338
+ value?: string | undefined;
339
+ }
340
+ /** Single metric value at a specific timestamp */
341
+ export interface MetricValue {
342
+ /** Timestamp of the metric value (milliseconds since epoch) */
343
+ timestamp?: bigint | undefined;
344
+ /** Metric value */
345
+ value?: number | undefined;
346
+ }
347
+ /** Collection of metric values for a specific metric */
348
+ export interface MetricValues {
349
+ /** Metric specification */
350
+ metric?: Metric | undefined;
351
+ /** Time series of metric values */
352
+ values?: MetricValue[] | undefined;
353
+ }
354
+ export interface MiniVectorIndex {
355
+ /** Name of the index */
356
+ name?: string | undefined;
357
+ /** Name of the endpoint associated with the index */
358
+ endpointName?: string | undefined;
359
+ /** Primary key of the index */
360
+ primaryKey?: string | undefined;
361
+ indexType?: VectorIndexType | undefined;
362
+ indexSpec?: {
363
+ $case: 'directAccessIndexSpec';
364
+ directAccessIndexSpec: DirectAccessVectorIndexSpec;
365
+ } | {
366
+ $case: 'deltaSyncIndexSpec';
367
+ deltaSyncIndexSpec: DeltaSyncVectorIndexSpec;
368
+ } | undefined;
369
+ status?: VectorIndexStatus | undefined;
370
+ /** The user who created the index. */
371
+ creator?: string | undefined;
372
+ /** The subtype of the index. */
373
+ indexSubtype?: IndexSubtype | undefined;
374
+ }
375
+ export interface PatchEndpointBudgetPolicyRequest {
376
+ /** Name of the AI Search endpoint */
377
+ name?: string | undefined;
378
+ /** The budget policy id to be applied */
379
+ budgetPolicyId?: string | undefined;
380
+ }
381
+ export interface PatchEndpointBudgetPolicyResponse {
382
+ budgetPolicyId?: string | undefined;
383
+ /** The budget policy applied to the AI Search endpoint. */
384
+ effectiveBudgetPolicyId?: string | undefined;
385
+ }
386
+ export interface PatchEndpointRequest {
387
+ /** Name of the AI Search endpoint */
388
+ name?: string | undefined;
389
+ /**
390
+ * Target QPS for the endpoint. Best-effort; the system does not guarantee this QPS
391
+ * will be achieved.
392
+ */
393
+ targetQps?: bigint | undefined;
394
+ }
395
+ /** Request payload for getting next page of results. */
396
+ export interface QueryVectorIndexNextPageRequest {
397
+ /** Name of the vector index to query. */
398
+ name?: string | undefined;
399
+ /** Name of the endpoint. */
400
+ endpointName?: string | undefined;
401
+ /** Page token returned from previous `QueryVectorIndex` or `QueryVectorIndexNextPage` API. */
402
+ pageToken?: string | undefined;
403
+ }
404
+ export interface QueryVectorIndexRequest {
405
+ /** Name of the vector index to query. */
406
+ name?: string | undefined;
407
+ /** Number of results to return. Defaults to 10. */
408
+ numResults?: number | undefined;
409
+ /** List of column names to include in the response. */
410
+ columns?: string[] | undefined;
411
+ /**
412
+ * JSON string representing query filters.
413
+ *
414
+ * Example filters:
415
+ *
416
+ * - `{"id <": 5}`: Filter for id less than 5.
417
+ * - `{"id >": 5}`: Filter for id greater than 5.
418
+ * - `{"id <=": 5}`: Filter for id less than equal to 5.
419
+ * - `{"id >=": 5}`: Filter for id greater than equal to 5.
420
+ * - `{"id": 5}`: Filter for id equal to 5.
421
+ */
422
+ filtersJson?: string | undefined;
423
+ /** Query vector. Required for Direct Vector Access Index and Delta Sync Index using self-managed vectors. */
424
+ queryVector?: number[] | undefined;
425
+ /** Query text. Required for Delta Sync Index using model endpoint. */
426
+ queryText?: string | undefined;
427
+ /** Threshold for the approximate nearest neighbor search. Defaults to 0.0. */
428
+ scoreThreshold?: number | undefined;
429
+ /** The query type to use. Choices are `ANN` and `HYBRID` and `FULL_TEXT`. Defaults to `ANN`. */
430
+ queryType?: string | undefined;
431
+ /** Column names used to retrieve data to send to the reranker. */
432
+ columnsToRerank?: string[] | undefined;
433
+ /**
434
+ * If set, the top 50 results are reranked with the Databricks Reranker model before returning the `num_results` results to the user.
435
+ * The setting `columns_to_rerank` selects which columns are used for reranking. For each datapoint, the columns selected are concatenated before
436
+ * being sent to the reranking model. See https://docs.databricks.com/aws/en/vector-search/query-vector-search#rerank for more information.
437
+ */
438
+ reranker?: RerankerConfig | undefined;
439
+ }
440
+ export interface QueryVectorIndexResponse {
441
+ /** Metadata about the result set. */
442
+ manifest?: ResultManifest | undefined;
443
+ /** Data returned in the query result. */
444
+ result?: ResultData | undefined;
445
+ /**
446
+ * [Optional] Token that can be used in `QueryVectorIndexNextPage` API to get next page of results.
447
+ * If more than 1000 results satisfy the query, they are returned in groups of 1000.
448
+ * Empty value means no more results. The maximum number of results that can be returned is 10,000.
449
+ */
450
+ nextPageToken?: string | undefined;
451
+ }
452
+ export interface RerankerConfig {
453
+ /**
454
+ * Reranker identifier:
455
+ * - When model_type=BASE/UNSPECIFIED: must be "databricks_reranker".
456
+ * - When model_type=FINETUNED: the Model Serving endpoint name hosting a finetuned reranker.
457
+ */
458
+ model?: string | undefined;
459
+ /** Parameters that control how the reranker processes the query results. */
460
+ parameters?: RerankerConfig_RerankerParameters | undefined;
461
+ }
462
+ export interface RerankerConfig_RerankerParameters {
463
+ columnsToRerank?: string[] | undefined;
464
+ }
465
+ /** Data returned in the query result. */
466
+ export interface ResultData {
467
+ /** Number of rows in the result set. */
468
+ rowCount?: number | undefined;
469
+ /** Data rows returned in the query. */
470
+ dataArray?: JsonValue[][] | undefined;
471
+ }
472
+ /** Metadata about the result set. */
473
+ export interface ResultManifest {
474
+ /** Number of columns in the result set. */
475
+ columnCount?: number | undefined;
476
+ /** Information about each column in the result set. */
477
+ columns?: ColumnInfo[] | undefined;
478
+ }
479
+ /** Request to retrieve user-visible metrics */
480
+ export interface RetrieveUserVisibleMetricsRequest {
481
+ /** AI Search endpoint name */
482
+ name?: string | undefined;
483
+ /** Start time for metrics query */
484
+ startTime?: Temporal.Instant | undefined;
485
+ /** End time for metrics query */
486
+ endTime?: Temporal.Instant | undefined;
487
+ /** Granularity in seconds */
488
+ granularityInSeconds?: number | undefined;
489
+ /** List of metrics to retrieve */
490
+ metrics?: Metric[] | undefined;
491
+ /** Token for pagination */
492
+ pageToken?: string | undefined;
493
+ }
494
+ /** Response containing user-visible metrics */
495
+ export interface RetrieveUserVisibleMetricsResponse {
496
+ /** Collection of metric values */
497
+ metricValues?: MetricValues[] | undefined;
498
+ /** A token that can be used to get the next page of results. If not present, there are no more results to show. */
499
+ nextPageToken?: string | undefined;
500
+ }
501
+ export interface ScanVectorIndexRequest {
502
+ /** Name of the vector index to scan. */
503
+ name?: string | undefined;
504
+ /** Number of results to return. Defaults to 10. */
505
+ numResults?: number | undefined;
506
+ /** Primary key of the last entry returned in the previous scan. */
507
+ lastPrimaryKey?: string | undefined;
508
+ }
509
+ /** Response to a scan vector index request. */
510
+ export interface ScanVectorIndexResponse {
511
+ /** List of data entries */
512
+ data?: Struct[] | undefined;
513
+ /** Primary key of the last entry. */
514
+ lastPrimaryKey?: string | undefined;
515
+ }
516
+ export interface Struct {
517
+ /** Data entry, corresponding to a row in a vector index. */
518
+ fields?: MapStringValueEntry[] | undefined;
519
+ }
520
+ export interface SyncVectorIndexRequest {
521
+ /** Name of the vector index to synchronize. Must be a Delta Sync Index. */
522
+ name?: string | undefined;
523
+ }
524
+ export interface SyncVectorIndexResponse {
525
+ }
526
+ export interface UpdateEndpointCustomTagsRequest {
527
+ /** Name of the AI Search endpoint */
528
+ name?: string | undefined;
529
+ /** The new custom tags for the AI Search endpoint */
530
+ customTags?: CustomTag[] | undefined;
531
+ }
532
+ export interface UpdateEndpointCustomTagsResponse {
533
+ /** The name of the AI Search endpoint whose custom tags were updated. */
534
+ name?: string | undefined;
535
+ /** All the custom tags that are applied to the AI Search endpoint. */
536
+ customTags?: CustomTag[] | undefined;
537
+ }
538
+ export interface UpsertDataVectorIndexRequest {
539
+ /** Name of the vector index where data is to be upserted. Must be a Direct Vector Access Index. */
540
+ name?: string | undefined;
541
+ /** JSON string representing the data to be upserted. */
542
+ inputsJson?: string | undefined;
543
+ }
544
+ export interface UpsertDataVectorIndexResponse {
545
+ /** Status of the upsert operation. */
546
+ status?: UpsertDeleteDataStatus | undefined;
547
+ /** Result of the upsert or delete operation. */
548
+ result?: UpsertDeleteDataResult | undefined;
549
+ }
550
+ export interface UpsertDeleteDataResult {
551
+ /** Count of successfully processed rows. */
552
+ successRowCount?: bigint | undefined;
553
+ /** List of primary keys for rows that failed to process. */
554
+ failedPrimaryKeys?: string[] | undefined;
555
+ }
556
+ export interface Value {
557
+ /** (--The kind of value.--) */
558
+ kind?: {
559
+ $case: 'numberValue';
560
+ numberValue: number;
561
+ } | {
562
+ $case: 'stringValue';
563
+ stringValue: string;
564
+ } | {
565
+ $case: 'boolValue';
566
+ boolValue: boolean;
567
+ } | {
568
+ $case: 'structValue';
569
+ structValue: Struct;
570
+ } | {
571
+ $case: 'listValue';
572
+ listValue: ListValue;
573
+ } | undefined;
574
+ }
575
+ export interface VectorIndex {
576
+ /** Name of the index */
577
+ name?: string | undefined;
578
+ /** Name of the endpoint associated with the index */
579
+ endpointName?: string | undefined;
580
+ /** Primary key of the index */
581
+ primaryKey?: string | undefined;
582
+ indexType?: VectorIndexType | undefined;
583
+ indexSpec?: {
584
+ $case: 'directAccessIndexSpec';
585
+ directAccessIndexSpec: DirectAccessVectorIndexSpec;
586
+ } | {
587
+ $case: 'deltaSyncIndexSpec';
588
+ deltaSyncIndexSpec: DeltaSyncVectorIndexSpec;
589
+ } | undefined;
590
+ status?: VectorIndexStatus | undefined;
591
+ /** The user who created the index. */
592
+ creator?: string | undefined;
593
+ /** The subtype of the index. */
594
+ indexSubtype?: IndexSubtype | undefined;
595
+ }
596
+ export interface VectorIndexStatus {
597
+ /** Message associated with the index status */
598
+ message?: string | undefined;
599
+ /** Number of rows indexed */
600
+ indexedRowCount?: bigint | undefined;
601
+ /** Whether the index is ready for search */
602
+ ready?: boolean | undefined;
603
+ /** Index API Url to be used to perform operations on the index */
604
+ indexUrl?: string | undefined;
605
+ }
606
+ export declare const unmarshalColumnInfoSchema: z.ZodType<ColumnInfo>;
607
+ export declare const unmarshalCustomTagSchema: z.ZodType<CustomTag>;
608
+ export declare const unmarshalDeleteDataVectorIndexResponseSchema: z.ZodType<DeleteDataVectorIndexResponse>;
609
+ export declare const unmarshalDeleteEndpointResponseSchema: z.ZodType<DeleteEndpointResponse>;
610
+ export declare const unmarshalDeleteVectorIndexResponseSchema: z.ZodType<DeleteVectorIndexResponse>;
611
+ export declare const unmarshalDeltaSyncVectorIndexSpecSchema: z.ZodType<DeltaSyncVectorIndexSpec>;
612
+ export declare const unmarshalDirectAccessVectorIndexSpecSchema: z.ZodType<DirectAccessVectorIndexSpec>;
613
+ export declare const unmarshalEmbeddingSourceColumnSchema: z.ZodType<EmbeddingSourceColumn>;
614
+ export declare const unmarshalEmbeddingVectorColumnSchema: z.ZodType<EmbeddingVectorColumn>;
615
+ export declare const unmarshalEndpointSchema: z.ZodType<Endpoint>;
616
+ export declare const unmarshalEndpointScalingInfoSchema: z.ZodType<EndpointScalingInfo>;
617
+ export declare const unmarshalEndpointStatusSchema: z.ZodType<EndpointStatus>;
618
+ export declare const unmarshalListEndpointResponseSchema: z.ZodType<ListEndpointResponse>;
619
+ export declare const unmarshalListValueSchema: z.ZodType<ListValue>;
620
+ export declare const unmarshalListVectorIndexResponseSchema: z.ZodType<ListVectorIndexResponse>;
621
+ export declare const unmarshalMapStringValueEntrySchema: z.ZodType<MapStringValueEntry>;
622
+ export declare const unmarshalMetricSchema: z.ZodType<Metric>;
623
+ export declare const unmarshalMetricLabelSchema: z.ZodType<MetricLabel>;
624
+ export declare const unmarshalMetricValueSchema: z.ZodType<MetricValue>;
625
+ export declare const unmarshalMetricValuesSchema: z.ZodType<MetricValues>;
626
+ export declare const unmarshalMiniVectorIndexSchema: z.ZodType<MiniVectorIndex>;
627
+ export declare const unmarshalPatchEndpointBudgetPolicyResponseSchema: z.ZodType<PatchEndpointBudgetPolicyResponse>;
628
+ export declare const unmarshalQueryVectorIndexResponseSchema: z.ZodType<QueryVectorIndexResponse>;
629
+ export declare const unmarshalResultDataSchema: z.ZodType<ResultData>;
630
+ export declare const unmarshalResultManifestSchema: z.ZodType<ResultManifest>;
631
+ export declare const unmarshalRetrieveUserVisibleMetricsResponseSchema: z.ZodType<RetrieveUserVisibleMetricsResponse>;
632
+ export declare const unmarshalScanVectorIndexResponseSchema: z.ZodType<ScanVectorIndexResponse>;
633
+ export declare const unmarshalStructSchema: z.ZodType<Struct>;
634
+ export declare const unmarshalSyncVectorIndexResponseSchema: z.ZodType<SyncVectorIndexResponse>;
635
+ export declare const unmarshalUpdateEndpointCustomTagsResponseSchema: z.ZodType<UpdateEndpointCustomTagsResponse>;
636
+ export declare const unmarshalUpsertDataVectorIndexResponseSchema: z.ZodType<UpsertDataVectorIndexResponse>;
637
+ export declare const unmarshalUpsertDeleteDataResultSchema: z.ZodType<UpsertDeleteDataResult>;
638
+ export declare const unmarshalValueSchema: z.ZodType<Value>;
639
+ export declare const unmarshalVectorIndexSchema: z.ZodType<VectorIndex>;
640
+ export declare const unmarshalVectorIndexStatusSchema: z.ZodType<VectorIndexStatus>;
641
+ export declare const marshalCreateEndpointRequestSchema: z.ZodType;
642
+ export declare const marshalCreateVectorIndexRequestSchema: z.ZodType;
643
+ export declare const marshalCustomTagSchema: z.ZodType;
644
+ export declare const marshalDeltaSyncVectorIndexSpecRequestSchema: z.ZodType;
645
+ export declare const marshalDirectAccessVectorIndexSpecSchema: z.ZodType;
646
+ export declare const marshalEmbeddingSourceColumnSchema: z.ZodType;
647
+ export declare const marshalEmbeddingVectorColumnSchema: z.ZodType;
648
+ export declare const marshalMetricSchema: z.ZodType;
649
+ export declare const marshalMetricLabelSchema: z.ZodType;
650
+ export declare const marshalPatchEndpointBudgetPolicyRequestSchema: z.ZodType;
651
+ export declare const marshalPatchEndpointRequestSchema: z.ZodType;
652
+ export declare const marshalQueryVectorIndexNextPageRequestSchema: z.ZodType;
653
+ export declare const marshalQueryVectorIndexRequestSchema: z.ZodType;
654
+ export declare const marshalRerankerConfigSchema: z.ZodType;
655
+ export declare const marshalRerankerConfig_RerankerParametersSchema: z.ZodType;
656
+ export declare const marshalRetrieveUserVisibleMetricsRequestSchema: z.ZodType;
657
+ export declare const marshalScanVectorIndexRequestSchema: z.ZodType;
658
+ export declare const marshalSyncVectorIndexRequestSchema: z.ZodType;
659
+ export declare const marshalUpdateEndpointCustomTagsRequestSchema: z.ZodType;
660
+ export declare const marshalUpsertDataVectorIndexRequestSchema: z.ZodType;
661
+ //# sourceMappingURL=model.d.ts.map