From a9851e3a357308a418c0ae82ce424e3bc5dba82b Mon Sep 17 00:00:00 2001 From: sujal Bhavsar Date: Thu, 7 May 2026 11:53:08 +0000 Subject: [PATCH] added pgvector support fix-1 --- litellm/rag/ingestion/openai_ingestion.py | 6 +++--- litellm/rag/main.py | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/rag/ingestion/openai_ingestion.py b/litellm/rag/ingestion/openai_ingestion.py index 891e3d0e914..811f35f0f55 100644 --- a/litellm/rag/ingestion/openai_ingestion.py +++ b/litellm/rag/ingestion/openai_ingestion.py @@ -89,7 +89,7 @@ class OpenAIRAGIngestion(BaseRAGIngestion): ) create_response = await vector_store_acreate( name=self.ingest_name or "litellm-rag-ingest", - custom_llm_provider="openai", + custom_llm_provider=self.custom_llm_provider, expires_after=expires_after, api_key=api_key, api_base=api_base, @@ -107,7 +107,7 @@ class OpenAIRAGIngestion(BaseRAGIngestion): content_type or "application/octet-stream", ), purpose="assistants", - custom_llm_provider="openai", + custom_llm_provider=self.custom_llm_provider, api_key=api_key, api_base=api_base, ) @@ -117,7 +117,7 @@ class OpenAIRAGIngestion(BaseRAGIngestion): await vector_store_file_acreate( vector_store_id=vector_store_id, file_id=result_file_id, - custom_llm_provider="openai", + custom_llm_provider=self.custom_llm_provider, chunking_strategy=cast( Optional[Dict[str, Any]], self.chunking_strategy ), diff --git a/litellm/rag/main.py b/litellm/rag/main.py index e3d354b6c33..e05982efe17 100644 --- a/litellm/rag/main.py +++ b/litellm/rag/main.py @@ -52,6 +52,7 @@ INGESTION_REGISTRY: Dict[str, Type[BaseRAGIngestion]] = { "gemini": GeminiRAGIngestion, "s3_vectors": S3VectorsRAGIngestion, "vertex_ai": VertexAIRAGIngestion, + "pg_vector": OpenAIRAGIngestion, }