fix(reranker): support nested llm config in LLMReranker for non-OpenAI providers (#4405)

This commit is contained in:
Kartik
2026-03-19 15:26:00 +05:30
committed by GitHub
parent 0c4d0290cb
commit 348f44b632
6 changed files with 385 additions and 15 deletions
+7 -1
View File
@@ -1,4 +1,5 @@
from typing import Optional
from typing import Any, Dict, Optional
from pydantic import Field
from mem0.configs.rerankers.base import BaseRerankerConfig
@@ -46,3 +47,8 @@ class LLMRerankerConfig(BaseRerankerConfig):
default=None,
description="Custom prompt template for scoring documents"
)
llm: Optional[Dict[str, Any]] = Field(
default=None,
description="Nested LLM configuration with 'provider' and 'config' keys. "
"Overrides top-level provider/model/api_key when provided.",
)