scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

refactor(vector-stores): avoid S3 ingest config mutation

Browse source
This commit is contained in:
Yujong Lee 2026-08-30 18:55:12 -07:00
parent dcf1744998
commit 396dfc7af2
2 changed files with 13 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -219,16 +219,18 @@ async def _save_vector_store_to_db_from_rag_ingest(
# Extract provider-specific params from vector_store_config to save as litellm_params
# This ensures params like aws_region_name, embedding_model, etc. are available for search
provider_specific_params: Final = {}
excluded_keys: Final = {"custom_llm_provider", "vector_store_id"}
for key, value in vector_store_config.items():
if key not in excluded_keys and value is not None:
provider_specific_params[key] = value
embedding_config: Final = ingest_options.get("embedding")
embedding_model: Final = embedding_config.get("model") if embedding_config else None
if custom_llm_provider == "s3_vectors" and isinstance(embedding_model, str) and embedding_model:
provider_specific_params["embedding_model"] = embedding_model
s3_embedding_params: Final = (
{"embedding_model": embedding_model}
if custom_llm_provider == "s3_vectors" and isinstance(embedding_model, str) and embedding_model
else {}
)
provider_specific_params: Final = {
**{key: value for key, value in vector_store_config.items() if key not in excluded_keys and value is not None},
**s3_embedding_params,
}
# Build file metadata entry using helper
file_entry: Final = _build_file_metadata_entry(

9
ui/litellm-dashboard/src/components/networking.tsx
View file

@ -5805,11 +5805,10 @@ export const ragIngestCall = async (
const formData = new FormData();
formData.append("file", file);
const vectorStoreParams = { ...(providerSpecificParams ?? {}) };
const s3EmbeddingModel = customLlmProvider === "s3_vectors" ? vectorStoreParams.embedding_model : undefined;
if (customLlmProvider === "s3_vectors") {
delete vectorStoreParams.embedding_model;
}
const { embedding_model: providerEmbeddingModel, ...paramsWithoutEmbeddingModel } = providerSpecificParams ?? {};
const s3EmbeddingModel = customLlmProvider === "s3_vectors" ? providerEmbeddingModel : undefined;
const vectorStoreParams =
customLlmProvider === "s3_vectors" ? paramsWithoutEmbeddingModel : providerSpecificParams ?? {};
const ingestOptions: any = {
ingest_options: {