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.3.0.tar.gz.

File metadata

File hashes

Hashes for agentforge_reranker_sentence_transformers-0.3.0.tar.gz
Algorithm Hash digest
SHA256 df6c1b6f568c7af33761386e0da5b8f89f8f8b0ce6bcaf26777a9f7cbcb944a8
MD5 b6b026d83f80d10ca840182e3b6d4ba3
BLAKE2b-256 c15207692f513f376492f9dda9124a23598a6bd8f8992698c25565c8f61de02f

See more details on using hashes here.

Provenance

The following attestation bundles were made for agentforge_reranker_sentence_transformers-0.3.0.tar.gz:

Publisher: release.yml on Scaffoldic/agentforge-py

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

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

File metadata

File hashes

Hashes for agentforge_reranker_sentence_transformers-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 f16582143cdcb67c8ad890ee895d440a7ddf6a250655a7c9cd8c7b0b1339b507
MD5 11d988164c0fe307a55d70fca9b4d5ba
BLAKE2b-256 2761027818a58a952fec99822a840a3faf20b599622e9a24ecbc77e659f627c5

See more details on using hashes here.

Provenance

The following attestation bundles were made for agentforge_reranker_sentence_transformers-0.3.0-py3-none-any.whl:

Publisher: release.yml on Scaffoldic/agentforge-py

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

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