Skip to main content

SentenceTransformers cross-encoder reranker for AgentForge

Project description

agentforge-reranker-sentence-transformers

SentenceTransformers cross-encoder reranker for the AgentForge framework (feat-021 default concrete impl).

Registers as agentforge.rerankers:sentence-transformers and implements the Reranker ABC from agentforge-core.

Installation

pip install agentforge-reranker-sentence-transformers[sentence-transformers]

The [sentence-transformers] extra pulls in sentence-transformers>=3.0 (~500MB for base + one model). Without it, the production factory raises ModuleError with pip remediation.

Usage

from agentforge import Retriever
from agentforge_reranker_sentence_transformers import (
    SentenceTransformersReranker,
)

reranker = SentenceTransformersReranker.from_config(
    model="cross-encoder/ms-marco-MiniLM-L-6-v2",
)
retriever = Retriever(
    store=vector_store,
    embedder=embedding_client,
    reranker=reranker,
    over_fetch_factor=3,
)
results = await retriever.retrieve("how do I deploy?", top_k=5)

Score normalisation

The cross-encoder returns raw logits (typically -10..+10). The reranker applies a sigmoid (1 / (1 + exp(-x))) so returned scores are normalised to (0, 1) — fitting the VectorMatch.score contract from agentforge-core.

License

Apache-2.0.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

File details

Details for the file agentforge_reranker_sentence_transformers-0.2.3.tar.gz.

File metadata

  • Download URL: agentforge_reranker_sentence_transformers-0.2.3.tar.gz
  • Upload date:
  • Size: 10.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.11.12 {"installer":{"name":"uv","version":"0.11.12","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for agentforge_reranker_sentence_transformers-0.2.3.tar.gz
Algorithm Hash digest
SHA256 80212d4de0d5c69b624d02c2288150d0d94de952e6fc8ff88ff6ae409cdf61c3
MD5 0338a2961be885e2132612510fe3b2d1
BLAKE2b-256 85cda5c5287419f6f6f50446846bf432898bba3b40789ff2d0d268b4673b1153

See more details on using hashes here.

File details

Details for the file agentforge_reranker_sentence_transformers-0.2.3-py3-none-any.whl.

File metadata

  • Download URL: agentforge_reranker_sentence_transformers-0.2.3-py3-none-any.whl
  • Upload date:
  • Size: 10.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.11.12 {"installer":{"name":"uv","version":"0.11.12","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for agentforge_reranker_sentence_transformers-0.2.3-py3-none-any.whl
Algorithm Hash digest
SHA256 424c31b86f9e7fdc57cbe04ea7a6b99c23e884f2a29bb221af4e95a68fefa57a
MD5 5f63ba7721f1852dc7cb7630f31bc01b
BLAKE2b-256 80719cba27bbeb870cbca5712b535b91501a72c372465322b7634f4b7f6ce5ce

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page