feat: make embedding model configurable for semantic search

- Add SemanticSearchConfig to settings.py with embedding model options
- Update WorldMapper to accept embedding_model parameter
- Update main.py to pass model from config to WorldMapper
- Add documentation for available embedding models:
  - sentence-transformers/all-MiniLM-L6-v2 (default, fast)
  - sentence-transformers/all-mpnet-base-v2 (better quality)
  - google/embeddinggemma-300m (state-of-the-art)
- Update config.yaml and config.example.yaml with new options

Users can now choose the best embedding model for their needs.
This commit is contained in:
truetype committed 2026-02-09 11:01:29 +01:00
1 parent b9f21c7199
commit 9a113002d4
5 files changed
+44 -26

No files matched your search

+6 -1
View File
@@ -71,7 +71,12 @@ mapper:
codec: "h264"
semantic_search:
model: "all-MiniLM-L6-v2" # Sentence transformer model
# Embedding model for semantic room search
# Available models:
# - "sentence-transformers/all-MiniLM-L6-v2" (default, fast, 22MB)
# - "sentence-transformers/all-mpnet-base-v2" (better quality, 418MB)
# - "google/embeddinggemma-300m" (state-of-the-art, requires transformers>=4.40)
model: "sentence-transformers/all-MiniLM-L6-v2"
top_k: 5 # Number of search results
similarity_threshold: 0.7 # Minimum similarity score
```