From 31d1d7d8963e0cf631cfd00bddb4571ce7b83b8a Mon Sep 17 00:00:00 2001 From: SilasMarvin <19626586+SilasMarvin@users.noreply.github.com> Date: Fri, 31 May 2024 12:30:54 -0700 Subject: [PATCH] Updated the site search to use mixed bread as the embeddings model --- pgml-dashboard/src/utils/markdown.rs | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pgml-dashboard/src/utils/markdown.rs b/pgml-dashboard/src/utils/markdown.rs index 291e5839e..3863dae2e 100644 --- a/pgml-dashboard/src/utils/markdown.rs +++ b/pgml-dashboard/src/utils/markdown.rs @@ -1252,7 +1252,7 @@ pub struct SiteSearch { impl SiteSearch { pub async fn new() -> anyhow::Result { let collection = pgml::Collection::new( - env!("CMS_HASH"), + &format!("{}-1", env!("CMS_HASH")), Some( std::env::var("SITE_SEARCH_DATABASE_URL") .context("Please set the `SITE_SEARCH_DATABASE_URL` environment variable")?, @@ -1267,7 +1267,7 @@ impl SiteSearch { "configuration": "english" }, "semantic_search": { - "model": "Alibaba-NLP/gte-base-en-v1.5", + "model": "mixedbread-ai/mxbai-embed-large-v1", } }, "contents": { @@ -1278,7 +1278,7 @@ impl SiteSearch { "configuration": "english" }, "semantic_search": { - "model": "Alibaba-NLP/gte-base-en-v1.5", + "model": "mixedbread-ai/mxbai-embed-large-v1", } } }) @@ -1319,14 +1319,14 @@ impl SiteSearch { "title": { "query": query, "parameters": { - "instruction": "Represent the Wikipedia question for retrieving supporting documents: " + "prompt": "Represent this sentence for searching relevant passages: " }, "boost": 10.0 }, "contents": { "query": query, "parameters": { - "instruction": "Represent the Wikipedia question for retrieving supporting documents: " + "prompt": "Represent this sentence for searching relevant passages: " }, "boost": 1.0 } pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.

Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy