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

File metadata

File hashes

Hashes for agentforge_reranker_sentence_transformers-0.4.0.tar.gz
Algorithm Hash digest
SHA256 9a435ec9b4b8532aaf11e5f22e1190131a54f20bcabf37734ad2447b6cc638e0
MD5 a3e4aa81c2e0e40a8316023aaa20af0c
BLAKE2b-256 fa784a997fe593b6c9b3a32151f8ab62d94e410f8bb7bf76d4baeff392fb1fa0

See more details on using hashes here.

Provenance

The following attestation bundles were made for agentforge_reranker_sentence_transformers-0.4.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.4.0-py3-none-any.whl.

File metadata

File hashes

Hashes for agentforge_reranker_sentence_transformers-0.4.0-py3-none-any.whl
Algorithm Hash digest
SHA256 0e15cc78f73098d9f57d2e18e063f4108149c67b7ec2c1c07d6f37756a07af28
MD5 b2452552a81742fcf234221dd078df6f
BLAKE2b-256 6c76c340da3ce60ffc77a642f122853e18c390ee293fedd6856074774d4e16f4

See more details on using hashes here.

Provenance

The following attestation bundles were made for agentforge_reranker_sentence_transformers-0.4.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