From 89e1c4f29c4ce6e1fde7aa380be17072acfe49ad Mon Sep 17 00:00:00 2001 From: niliy01 Date: Fri, 30 Aug 2024 19:58:01 +0800 Subject: [PATCH] feat: add more embed models that Ollama supports recently (#3641) Signed-off-by: Yi Lin --- Makefile | 4 ++++ openhands/memory/memory.py | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/Makefile b/Makefile index ee1eb83dc2..c18f0d90b5 100644 --- a/Makefile +++ b/Makefile @@ -260,6 +260,10 @@ setup-config-prompts: echo " - nomic-embed-text"; \ echo " - all-minilm"; \ echo " - stable-code"; \ + echo " - bge-m3"; \ + echo " - bge-large"; \ + echo " - paraphrase-multilingual"; \ + echo " - snowflake-arctic-embed"; \ echo " - Leave blank to default to 'BAAI/bge-small-en-v1.5' via huggingface"; \ read -p "> " llm_embedding_model; \ echo "embedding_model=\"$$llm_embedding_model\"" >> $(CONFIG_FILE).tmp; \ diff --git a/openhands/memory/memory.py b/openhands/memory/memory.py index 67806fefc7..2eb2d166ad 100644 --- a/openhands/memory/memory.py +++ b/openhands/memory/memory.py @@ -73,6 +73,10 @@ if LLAMA_INDEX_AVAILABLE: 'nomic-embed-text', 'all-minilm', 'stable-code', + 'bge-m3', + 'bge-large', + 'paraphrase-multilingual', + 'snowflake-arctic-embed', ] if strategy in supported_ollama_embed_models: from llama_index.embeddings.ollama import OllamaEmbedding