AGENT SKILL

pretrained-indexing-search

Use RAGatouille RAGPretrainedModel for pretrained checkpoint loading, persisted index creation/loading/search, document IDs/metadata, and experimental index CRUD.

License 许可
Apache-2.0
最近更新
2026-09-03
Stars
★ 351
Forks
27

01 前言 Front Matter

name
pretrained-indexing-search
license
Apache 2.0
metadata
{ "disco-role": "operating" }
description
Use RAGatouille RAGPretrainedModel for pretrained checkpoint loading, persisted index creation/loading/search, document IDs/metadata, and experimental index CRUD.
disable-model-invocation
true

02 资源文件 4

03 SKILL.md 正文

Pretrained Indexing Search

Use this sub-skill when a task needs RAGatouille's RAGPretrainedModel for persisted ColBERT/PLAID indexes: loading a pretrained checkpoint, loading an existing index, building an index from documents, searching it, validating result dictionaries, or doing experimental add/delete updates.

Route Here For

  • Loading models with RAGPretrainedModel.from_pretrained(...) or existing persisted indexes with RAGPretrainedModel.from_index(...).
  • Creating disk-backed indexes with RAG.index(...), including document_ids, document_metadatas, splitting, preprocessing, index_root, index_name, overwrite_index, bsize, and use_faiss choices.
  • Searching persisted indexes with RAG.search(...), including single-query vs multi-query result shape, k, doc_ids, force_fast, and zero_index_ranks.
  • Updating persisted indexes with RAG.add_to_index(...) or RAG.delete_from_index(...), while treating CRUD support as experimental.
  • Diagnosing input-shape, ID, metadata, index-name, loading, CPU/GPU/FAISS, and dependency compatibility problems before running expensive model work.

Route Elsewhere

  • Use ../index-free-reranking/SKILL.md for rerank, encode, search_encoded_docs, or small transient in-memory document sets.
  • Use ../training-data-finetuning/SKILL.md for RAGTrainer, data prep, hard negatives, training, or fine-tuning.
  • Use ../integrations-export/SKILL.md for LangChain retrievers/compressors, LlamaIndex integration examples, Hugging Face export, or Vespa ONNX export.

Start Here

  1. Read references/api-reference.md for verified signatures, defaults, persisted index layout, and result schemas.
  2. Read references/workflows.md for copyable indexing, existing-index search, metadata, filtering, and CRUD recipes.
  3. Read references/troubleshooting.md before changing dependencies, running downloads, indexing large corpora, using Windows/WSL, or debugging ID/metadata/search failures.
  4. Run scripts/validate_index_inputs.py --help when you need an offline sanity check for collection, document ID, and metadata consistency before model downloads or indexing.

Safety Notes

  • from_pretrained can download model weights; do not run it in lightweight validation unless downloads are allowed.
  • Indexing/search can be slow and may use GPU, FAISS, or PyTorch k-means paths; prefer offline validation and small examples before expensive runs.
  • Keep scripts that use RAGatouille inside if __name__ == "__main__": when running as standalone programs.
  • For RAGatouille 0.0.9post2, avoid latest LangChain 1.x if import ragatouille fails on langchain.retrievers.document_compressors.base.

04 同仓库更多 Skills

VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-triton-opt

Make a single PassNet Triton kernel fast AND numerically faithful: performance model (when a replacement can win at all), block/grid/warp tuning, autotune policy, launch overhead, and per-op numeric recipes to pass the dtype baseline tolerances. Use when a pass matches and is correct but speedup below expected, or when correctness fails by small numeric margins.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-solve

END-TO-END playbook for solving one PassNet sample: analyze the computation graphs, decide the optimization strategy (what to fuse, what to replace, what to leave alone), drive the iteration loop, and maximize the sample score. This is the ENTRY skill — invoke it first for any PassNet optimization task; it tells you when to use passnet-pattern-fusion, passnet-triton-opt and passnet-feedback.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-skill

PassNet GPU kernel optimization via compiler passes. Design and implement Triton-based optimization passes, create pass files under ./pass_dir/, self-evaluate with pass_evaluator, and iterate to maximize GPU speedup.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-pattern-fusion

Author PassNet pass files: write patterns that actually MATCH the FX graph, pick fusion regions (which ops to absorb into one Triton kernel), and structure multi-pass submissions with the mandatory shared-dispatch architecture. Use when creating or fixing pattern()/replacement_args()/replacement_func() files, when a pass "failed to match", or when deciding how to fuse multiple kernels/ops.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-orchestrate

Round/batch planning and decision support for PassNet graph optimization. Use for multi-sample triage, worker allocation, eval-budget policy, and post-eval keep/revert decisions. For solving one concrete sample end to end, use passnet-solve as the entry skill; consult this skill only when you need broader planning or a specific decision gate.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-feedback

Fast, deterministic feedback for PassNet work: pre-flight pattern/match verification WITHOUT burning a GPU evaluation, per-node bottleneck analysis of a sample's graphs, and eval-log parsing into per-variant status + estimated score + failure classification. Use BEFORE every GPU evaluation (check_pattern), at the START of a sample (analyze_graph), and AFTER every evaluation (parse_eval_log).

★ 351