| from typing import Any, cast |
|
|
| import httpx |
|
|
| from ...config.logfire_config import get_logger |
|
|
| logger = get_logger(__name__) |
|
|
|
|
| async def execute_remote_rerank( |
| agents_url: str, query: str, results: list[dict], content_key: str, top_k: int, fallback_strategy: Any = None |
| ) -> list[dict]: |
| """Performs reranking via remote agents service (Phase 4.6.28).""" |
| try: |
| async with httpx.AsyncClient(timeout=30.0) as client: |
| response = await client.post( |
| f"{agents_url}/ml/rerank", |
| json={ |
| "query": query, |
| "results": results, |
| "content_key": content_key, |
| "top_k": top_k, |
| }, |
| ) |
| if response.status_code == 200: |
| data = response.json() |
| if data.get("success"): |
| logger.info("Remote reranking successful via Agents service.") |
| return cast(list[dict[str, Any]], data.get("results", [])) |
|
|
| logger.warning(f"Remote rerank failed (Status {response.status_code}). Falling back...") |
| except Exception as e: |
| logger.warning(f"Remote rerank connection error: {e}. Falling back...") |
|
|
| |
| if fallback_strategy: |
| fallback_results = await fallback_strategy.rerank_results(query, results, top_k=top_k, content_key=content_key) |
| return cast(list[dict], fallback_results) |
| return results[:top_k] |
|
|
