scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-15 04:19:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 121

fix(staan): validate types for max_snippets and min_score in StaanSearchConfig

Browse source
This commit is contained in:
Tobias Goerke 2026-09-25 14:09:36 +02:00
parent f6227ebf8a
commit df15f099f3
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/staan/search/transformation.py
View file

@ -270,10 +270,12 @@ class StaanSearchConfig(BaseSearchConfig):
min_score: Final = optional_params.get("min_score")
full_content: Final = optional_params.get("full_content")
if should_fetch_snippets and max_snippets is not None and not 1 <= max_snippets <= 10:
raise ValueError("max_snippets must be between 1 and 10")
if should_fetch_snippets and min_score is not None and not 0 <= min_score <= 1:
raise ValueError("min_score must be between 0 and 1")
if should_fetch_snippets and max_snippets is not None:
if isinstance(max_snippets, bool) or not isinstance(max_snippets, int) or not 1 <= max_snippets <= 10:
raise ValueError("max_snippets must be between 1 and 10")
if should_fetch_snippets and min_score is not None:
if isinstance(min_score, bool) or not isinstance(min_score, (int, float)) or not 0 <= min_score <= 1:
raise ValueError("min_score must be between 0 and 1")
enrichment_entries: Final = (("extra_snippets", True),) if should_fetch_snippets else ()
max_snippet_entry: Final = (