scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(data_sources): add direct name/client/embedding_model params to FactCheckDataSource and VectorStoreDataSource

Browse source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Sameer Kankute 2026-06-24 18:58:53 +05:30
parent 774c8e2e7d
commit e2d3ca2fee
No known key found for this signature in database
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
litellm/proxy/guardrails/guardrail_hooks/bias_hallucination_estimator/data_sources.py
View file

@ -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,
)