Browse Source

feat: add more embed models that Ollama supports recently (#3641)

Signed-off-by: Yi Lin <teroincn@gmail.com>
niliy01 1 year ago
parent
commit
89e1c4f29c
2 changed files with 8 additions and 0 deletions
  1. 4 0
      Makefile
  2. 4 0
      openhands/memory/memory.py

+ 4 - 0
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; \

+ 4 - 0
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