github-mirrors/kibana
1
0
Fork 0
mirror of https://github.com/elastic/kibana.git synced 2025-04-24 01:38:56 -04:00
Code Issues Projects Releases Packages Wiki Activity

Add retrievers autocomplete in dev console (#216827)

Browse source
This commit is contained in:
Ioana Tagirta 2025-04-08 17:55:14 +02:00 • committed by GitHub
parent 3aa036d515
commit 9d698eb36a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 153 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/platform/plugins/shared/console/server/lib/spec_definitions/js/index.ts
View file

@ -19,6 +19,7 @@ import { mappings } from './mappings';
import { settings } from './settings';
import { query } from './query';
import { reindex } from './reindex';
import { retriever } from './retriever';
import { search } from './search';
export const jsSpecLoaders: Array<(registry: SpecDefinitionsService) => void> = [
@ -32,5 +33,6 @@ export const jsSpecLoaders: Array<(registry: SpecDefinitionsService) => void> =
settings,
query,
reindex,
retriever,
search,
];

148
src/platform/plugins/shared/console/server/lib/spec_definitions/js/retriever.ts Normal file
View file

@ -0,0 +1,148 @@
/*
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
* or more contributor license agreements. Licensed under the "Elastic License
* 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
* Public License v 1"; you may not use this file except in compliance with, at
* your election, the "Elastic License 2.0", the "GNU Affero General Public
* License v3.0 only", or the "Server Side Public License, v 1".
*/
import { SpecDefinitionsService } from '../../../services';
export const retriever = (specService: SpecDefinitionsService) => {
specService.addGlobalAutocompleteRules('retriever', {
knn: {
__template: {
field: '',
query_vector_builder: {
text_embedding: {
model_id: '',
model_text: '',
},
},
k: 10,
num_candidates: 100,
},
filter: {
__scope_link: 'GLOBAL.query',
},
field: '{field}',
query_vector: [],
similarity: { __one_of: ['l2_norm', 'cosine', 'dot_product', 'max_inner_product'] },
rescore_vector: {
oversample: 1.5,
},
},
linear: {
__template: {
retrievers: [{}],
},
rank_window_size: 100,
filter: {
__scope_link: 'GLOBAL.query',
},
retrievers: {
__any_of: [
{
retriever: {
__scope_link: '.',
},
weight: 2,
normalizer: 'minmax',
},
],
},
},
rescorer: {
__template: {
rescore: {
query: {
rescore_query: {},
},
},
retriever: {},
},
filter: {
__scope_link: 'GLOBAL.query',
},
retriever: {
__scope_link: '.',
},
rescore: {
query: {
rescore_query: {
__scope_link: 'GLOBAL.query',
},
},
window_size: 50,
},
},
rrf: {
__template: {
retrievers: [{}],
},
retrievers: [
{
__scope_link: '.',
},
],
filter: {
__scope_link: 'GLOBAL.query',
},
rank_constant: 60,
rank_window_size: 100,
},
rule: {
__template: {
retriever: {},
ruleset_ids: [],
match_criteria: {},
},
retriever: {
__scope_link: '.',
},
ruleset_ids: [],
match_criteria: {},
rank_window_size: 10,
},
standard: {
__template: {
query: {},
},
query: {
__scope_link: 'GLOBAL.query',
},
filter: {
__scope_link: 'GLOBAL.query',
},
collapse: {
__template: {
field: 'FIELD',
},
},
min_score: 0,
search_after: [],
sort: {},
terminate_after: 10000,
},
text_similarity_reranker: {
__template: {
retriever: {},
inference_id: '',
inference_text: '',
field: '',
},
retriever: {
__scope_link: '.',
},
inference_id: '',
inference_text: '',
field: '{field}',
rank_window_size: 10,
min_score: 0,
filter: {
__scope_link: 'GLOBAL.query',
},
},
});
};

3
src/platform/plugins/shared/console/server/lib/spec_definitions/js/search.ts
View file

@ -271,6 +271,9 @@ export const search = (specService: SpecDefinitionsService) => {
],
],
},
retriever: {
// populated by a global rule
},
},
});