Skip to content

Commit 162e27f

Browse files
authored
Updated the site search to use mixed bread as the embeddings model (#1493)
1 parent 44d135b commit 162e27f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pgml-dashboard/src/utils/markdown.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1252,7 +1252,7 @@ pub struct SiteSearch {
12521252
impl SiteSearch {
12531253
pub async fn new() -> anyhow::Result<Self> {
12541254
let collection = pgml::Collection::new(
1255-
env!("CMS_HASH"),
1255+
&format!("{}-1", env!("CMS_HASH")),
12561256
Some(
12571257
std::env::var("SITE_SEARCH_DATABASE_URL")
12581258
.context("Please set the `SITE_SEARCH_DATABASE_URL` environment variable")?,
@@ -1267,7 +1267,7 @@ impl SiteSearch {
12671267
"configuration": "english"
12681268
},
12691269
"semantic_search": {
1270-
"model": "Alibaba-NLP/gte-base-en-v1.5",
1270+
"model": "mixedbread-ai/mxbai-embed-large-v1",
12711271
}
12721272
},
12731273
"contents": {
@@ -1278,7 +1278,7 @@ impl SiteSearch {
12781278
"configuration": "english"
12791279
},
12801280
"semantic_search": {
1281-
"model": "Alibaba-NLP/gte-base-en-v1.5",
1281+
"model": "mixedbread-ai/mxbai-embed-large-v1",
12821282
}
12831283
}
12841284
})
@@ -1319,14 +1319,14 @@ impl SiteSearch {
13191319
"title": {
13201320
"query": query,
13211321
"parameters": {
1322-
"instruction": "Represent the Wikipedia question for retrieving supporting documents: "
1322+
"prompt": "Represent this sentence for searching relevant passages: "
13231323
},
13241324
"boost": 10.0
13251325
},
13261326
"contents": {
13271327
"query": query,
13281328
"parameters": {
1329-
"instruction": "Represent the Wikipedia question for retrieving supporting documents: "
1329+
"prompt": "Represent this sentence for searching relevant passages: "
13301330
},
13311331
"boost": 1.0
13321332
}

0 commit comments

Comments
 (0)