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

added pgvector support fix-1

Browse source
This commit is contained in:
sujal Bhavsar 2026-05-07 11:53:08 +00:00
parent a67b7a7e87
commit a9851e3a35
2 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/rag/ingestion/openai_ingestion.py
View file

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

1
litellm/rag/main.py
View file

@ -52,6 +52,7 @@ INGESTION_REGISTRY: Dict[str, Type[BaseRAGIngestion]] = {
"gemini": GeminiRAGIngestion,
"s3_vectors": S3VectorsRAGIngestion,
"vertex_ai": VertexAIRAGIngestion,
"pg_vector": OpenAIRAGIngestion,
}