first commit

configs/rerankers/cohere.py

@@ -0,0 +1,15 @@
+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")