Skip to content

Commit a0251fd

Browse files
committed
add search templates
1 parent 8442706 commit a0251fd

File tree

3 files changed

+120
-0
lines changed

3 files changed

+120
-0
lines changed
Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
PUT _scripts/google-template-knn-multioutput
2+
{
3+
"script": {
4+
"lang": "mustache",
5+
"source": {
6+
"_source": {
7+
"excludes": [ "title_embedding", "description_embedding", "images"]
8+
},
9+
"size": "{{num_hits}}",
10+
"knn" : [
11+
{
12+
"field": "description_embedding",
13+
"k": 5,
14+
"num_candidates": 10,
15+
"query_vector_builder": {
16+
"text_embedding": {
17+
"model_id": "googlevertexai_embeddings_004",
18+
"model_text": "{{query}}"
19+
}
20+
},
21+
"boost": 0.4
22+
},
23+
{
24+
"field": "title_embedding",
25+
"k": 5,
26+
"num_candidates": 10,
27+
"query_vector_builder": {
28+
"text_embedding": {
29+
"model_id": "googlevertexai_embeddings_004",
30+
"model_text": "{{query}}"
31+
}
32+
},
33+
"boost": 0.6
34+
}
35+
]
36+
}
37+
}
38+
}
Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
#knn template
2+
PUT _scripts/google-template-knn
3+
{
4+
"script": {
5+
"lang": "mustache",
6+
"source": {
7+
"_source": false,
8+
"size": "{{num_hits}}",
9+
"query": {
10+
"nested": {
11+
"path": "document_text.inference.chunks",
12+
"query":{
13+
"knn": {
14+
"field": "document_text.inference.chunks.embeddings",
15+
"query_vector_builder": {
16+
"text_embedding": {
17+
"model_id": ".multilingual-e5-small_linux-x86_64",
18+
"model_text": "{{query}}"
19+
}
20+
}
21+
}
22+
},
23+
"inner_hits": {
24+
"size": 1,
25+
"name": "{{index_name}}.document_text",
26+
"_source": "*.text"
27+
}
28+
}
29+
}
30+
}
31+
}
32+
}
33+
34+
#Run the knn templated search.
35+
GET nqdataset-simplified-clean-semantictext-e5/_search/template
36+
{
37+
"id": "google-template-knn",
38+
"params": {
39+
"query": "italian food",
40+
"index_name": "nqdataset-simplified-clean-semantictext-e5",
41+
"num_hits": 1
42+
}
43+
}
44+
Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
PUT _scripts/google-template-knn-multioutput
2+
{
3+
"script": {
4+
"lang": "mustache",
5+
"source": {
6+
"_source": {
7+
"excludes": [ "title_embedding", "description_embedding", "images"]
8+
},
9+
"size": "{{num_hits}}",
10+
"knn" : [
11+
{
12+
"field": "description_embedding",
13+
"k": 5,
14+
"num_candidates": 10,
15+
"query_vector_builder": {
16+
"text_embedding": {
17+
"model_id": "googlevertexai_embeddings_004",
18+
"model_text": "{{query}}"
19+
}
20+
},
21+
"boost": 0.4
22+
},
23+
{
24+
"field": "title_embedding",
25+
"k": 5,
26+
"num_candidates": 10,
27+
"query_vector_builder": {
28+
"text_embedding": {
29+
"model_id": "googlevertexai_embeddings_004",
30+
"model_text": "{{query}}"
31+
}
32+
},
33+
"boost": 0.6
34+
}
35+
]
36+
}
37+
}
38+
}

0 commit comments

Comments
 (0)