chore: switch to AllMiniLML6V2 for faster embeddings

Author: Anush008

README.md

@@ -21,7 +21,7 @@
 </div>
 
 ## 🔎 The Project
-RepoQuery is an early-beta project, that uses recursive [OpenAI function calling](https://platform.openai.com/docs/api-reference/chat/create#chat/create-functions) paired with semantic search using [BAAI/bge-small-en-v1.5](https://huggingface.co/BAAI/bge-small-en-v1.5) to index and answer user queries about public GitHub repositories.
+RepoQuery is an early-beta project, that uses recursive [OpenAI function calling](https://platform.openai.com/docs/api-reference/chat/create#chat/create-functions) paired with semantic search using [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) to index and answer user queries about public GitHub repositories.
 
 ## 📬 Service Endpoints
 
@@ -173,15 +173,16 @@ make up
 
 ## Attributions
 
-[baai.ac.cn](https://www.baai.ac.cn/english.html) for [BAAI/bge-small-en-v1.5](https://huggingface.co/BAAI/bge-small-en-v1.5).
+[sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2).
 ```
-@misc{bge_embedding,
-      title={C-Pack: Packaged Resources To Advance General Chinese Embedding}, 
-      author={Shitao Xiao and Zheng Liu and Peitian Zhang and Niklas Muennighoff},
-      year={2023},
-      eprint={2309.07597},
-      archivePrefix={arXiv},
-      primaryClass={cs.CL}
+@inproceedings{reimers-2019-sentence-bert,
+  title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
+  author = "Reimers, Nils and Gurevych, Iryna",
+  booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
+  month = "11",
+  year = "2019",
+  publisher = "Association for Computational Linguistics",
+  url = "https://arxiv.org/abs/1908.10084",
 }
 ```
 

src/embeddings/fastembed.rs

@@ -1,5 +1,5 @@
 use crate::prelude::*;
-use fastembed::{EmbeddingBase, FlagEmbedding};
+use fastembed::{EmbeddingBase, FlagEmbedding, InitOptions};
 
 use super::{Embeddings, EmbeddingsModel};
 
@@ -9,7 +9,10 @@ pub struct Fastembed {
 
 impl Fastembed {
     pub fn try_new() -> Result<Self> {
-        let model = FlagEmbedding::try_new(Default::default())?;
+        let model = FlagEmbedding::try_new(InitOptions {
+            model_name: fastembed::EmbeddingModel::AllMiniLML6V2,
+            ..Default::default()
+        })?;
         Ok(Self { model })
     }
 }