16 lines
660 B
Python
16 lines
660 B
Python
from typing import Optional
|
|
from pydantic import Field
|
|
|
|
from mem0.configs.rerankers.base import BaseRerankerConfig
|
|
|
|
|
|
class CohereRerankerConfig(BaseRerankerConfig):
|
|
"""
|
|
Configuration class for Cohere reranker-specific parameters.
|
|
Inherits from BaseRerankerConfig and adds Cohere-specific settings.
|
|
"""
|
|
|
|
model: Optional[str] = Field(default="rerank-english-v3.0", description="The Cohere rerank model to use")
|
|
return_documents: bool = Field(default=False, description="Whether to return the document texts in the response")
|
|
max_chunks_per_doc: Optional[int] = Field(default=None, description="Maximum number of chunks per document")
|