From e2d3ca2feed4253d6c65b5e2ce96a32b85e36c66 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Wed, 24 Jun 2026 18:58:53 +0530 Subject: [PATCH] fix(data_sources): add direct name/client/embedding_model params to FactCheckDataSource and VectorStoreDataSource Co-Authored-By: Claude Sonnet 4.6 --- .../data_sources.py | 25 +++++++++++++------ 1 file changed, 18 insertions(+), 7 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/bias_hallucination_estimator/data_sources.py b/litellm/proxy/guardrails/guardrail_hooks/bias_hallucination_estimator/data_sources.py index 5249b7fddd3..5ed7530cbae 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/bias_hallucination_estimator/data_sources.py +++ b/litellm/proxy/guardrails/guardrail_hooks/bias_hallucination_estimator/data_sources.py @@ -251,6 +251,8 @@ class VectorStoreDataSource(DataSource): provider: str, source_config: DataSourceConfig | None = None, client_config: VectorStoreClientConfig | None = None, + client: object | None = None, + embedding_model: object | None = None, **config: str, ) -> None: _sc = source_config or DataSourceConfig() @@ -263,14 +265,22 @@ class VectorStoreDataSource(DataSource): self.config = config _client_config = client_config or VectorStoreClientConfig() self.client: object | None = ( - _client_config.client - if _client_config.client is not None - else self._initialize_client(provider, config) + client + if client is not None + else ( + _client_config.client + if _client_config.client is not None + else self._initialize_client(provider, config) + ) ) self.embedding_model: object | None = ( - _client_config.embedding_model - if _client_config.embedding_model is not None - else self._load_embedding_model() + embedding_model + if embedding_model is not None + else ( + _client_config.embedding_model + if _client_config.embedding_model is not None + else self._load_embedding_model() + ) ) @staticmethod @@ -414,11 +424,12 @@ class FactCheckDataSource(DataSource): provider: str = "snopes", source_config: DataSourceConfig | None = None, api_key: str | None = None, + name: str | None = None, **config: str, ) -> None: _sc = source_config or DataSourceConfig() super().__init__( - name=_sc.name or f"factcheck_{provider}", + name=name or _sc.name or f"factcheck_{provider}", enabled=_sc.enabled, priority=_sc.priority, )