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(rag): validate ingest providers before execution

Browse source
This commit is contained in:
Yujong Lee 2026-08-30 17:55:58 -07:00
parent dfb6ed24ba
commit 8e63f45703
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/rag_endpoints/endpoints.py
View file

@ -37,6 +37,7 @@ from litellm.proxy.rag_endpoints.upload_security import (
from litellm.proxy.vector_store_endpoints.utils import (
assert_user_can_access_vector_store_id,
)
from litellm.rag.main import get_ingestion_class
from litellm.repositories.table_repositories import ManagedVectorStoresRepository
if TYPE_CHECKING:
@ -428,6 +429,16 @@ async def parse_rag_ingest_request(
},
)
provider: Final = vector_store_opts.get("custom_llm_provider", "openai")
if isinstance(provider, str):
try:
get_ingestion_class(provider)
except ValueError as error:
raise HTTPException(
status_code=400,
detail={"error": str(error)},
) from error
return ingest_options, secured_file_data, file_url, file_id