refactor: add fallback LLM configuration options to settings

This commit is contained in:
Siddhant Rai
2025-06-06 17:05:15 +05:30
parent 5f5c31cd5b
commit 35f4b13237
2 changed files with 3 additions and 1 deletions

View File

@@ -35,6 +35,9 @@ class Settings(BaseSettings):
)
RETRIEVERS_ENABLED: list = ["classic_rag", "duckduck_search"] # also brave_search
AGENT_NAME: str = "classic"
FALLBACK_LLM_PROVIDER: Optional[str] = None # provider for fallback llm
FALLBACK_LLM_NAME: Optional[str] = None # model name for fallback llm
FALLBACK_LLM_API_KEY: Optional[str] = None # api key for fallback llm
# LLM Cache
CACHE_REDIS_URL: str = "redis://localhost:6379/2"

View File

@@ -72,7 +72,6 @@ class BaseLLM(ABC):
logger.warning(
f"Falling back to {self.fallback_provider}/{self.fallback_model_name}. Error: {str(e)}"
)
# Retry with fallback (without decorators for accurate token tracking)
fallback_method = getattr(
self.fallback_llm, method_name.replace("_raw_", "")