Skip to content

Commit 7cd4a3e

Browse files
authored
Update RAG embedding model filter tag to 'vector-rag' (#1854)
* Update RAG embedding model filter tag from 'embedding' to 'vector-rag' * Update huggingface.loc.sample: use 'vector-rag' free_tag for RAG embedding models
1 parent c90b170 commit 7cd4a3e

2 files changed

Lines changed: 5 additions & 4 deletions

File tree

tools/rag/rag_retriever.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ python '$__tool_directory__/rag_retriever.py' '$context_files' '$question' '$MOD
5858
<options from_data_table="huggingface">
5959
<column name="name" index="1" />
6060
<column name="value" index="6" />
61-
<filter type="static_value" value="embedding" column="4" />
61+
<filter type="static_value" value="vector-rag" column="4" />
6262
<filter type="static_value" value="1" column="5" />
6363
</options>
6464
<validator type="no_options"

tools/rag/tool-data/huggingface.loc.sample

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,10 @@
44
# <unique_build_id> <display_name> <pipeline_tag> <domain> <free_tag> <version> <folder_base_path>
55
#
66
# For this tool, embedding models can be registered here and selected via the existing
7-
# Galaxy "huggingface" data table.
7+
# Galaxy "huggingface" data table. Use free_tag=vector-rag to make models visible to
8+
# the RAG retriever tool.
89
#
910
# Example entries for embedding models:
1011
#
11-
# sentence-transformers/all-MiniLM-L6-v2 all-MiniLM-L6-v2 feature-extraction text embedding 1 /path/to/huggingface_models/sentence-transformers/all-MiniLM-L6-v2
12-
# BAAI/bge-small-en BAAI bge-small-en feature-extraction text embedding 1 /path/to/huggingface_models/BAAI/bge-small-en
12+
# sentence-transformers/all-MiniLM-L6-v2 all-MiniLM-L6-v2 feature-extraction text vector-rag 1 /path/to/huggingface_models/sentence-transformers/all-MiniLM-L6-v2
13+
# BAAI/bge-small-en BAAI bge-small-en feature-extraction text vector-rag 1 /path/to/huggingface_models/BAAI/bge-small-en

0 commit comments

Comments
 (0)