Spaces:
Running
Running
File size: 2,081 Bytes
5e1a30c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
"""
Base interface for Fusion Strategy sub-components.
This module defines the abstract base class for all fusion strategy implementations
in the modular retriever architecture.
"""
from abc import ABC, abstractmethod
from typing import List, Dict, Any, Tuple
class FusionStrategy(ABC):
"""
Abstract base class for fusion strategy implementations.
This interface defines the contract for all fusion strategy sub-components
in the modular retriever architecture. All implementations are direct
as they implement pure algorithms without external dependencies.
"""
@abstractmethod
def __init__(self, config: Dict[str, Any]):
"""
Initialize the fusion strategy.
Args:
config: Configuration dictionary specific to the fusion strategy
"""
pass
@abstractmethod
def fuse_results(
self,
dense_results: List[Tuple[int, float]],
sparse_results: List[Tuple[int, float]]
) -> List[Tuple[int, float]]:
"""
Fuse dense and sparse retrieval results.
Args:
dense_results: List of (document_index, score) from dense retrieval
sparse_results: List of (document_index, score) from sparse retrieval
Returns:
List of (document_index, fused_score) tuples sorted by score
"""
pass
@abstractmethod
def get_strategy_info(self) -> Dict[str, Any]:
"""
Get information about the fusion strategy.
Returns:
Dictionary with strategy configuration and statistics
"""
pass
def get_component_info(self) -> Dict[str, Any]:
"""
Get component information for logging and debugging.
Returns:
Dictionary with component details
"""
return {
"type": "fusion_strategy",
"class": self.__class__.__name__,
"module": self.__class__.__module__,
**self.get_strategy_info()
} |