o ưi[@sLdZddlmZmZerddlmZdededfddZdefd d Zd S) ze RAG utility functions. Provides provider configuration utilities similar to ProviderConfigManager. ) TYPE_CHECKINGType)BaseRAGIngestioncustom_llm_providerreturnrcCsbddlm}ddlm}ddlm}|||d}||}|dur/td|dt| |S) a% Get the appropriate RAG ingestion class for a provider. Args: custom_llm_provider: The LLM provider name (e.g., "openai", "bedrock", "vertex_ai") Returns: The ingestion class for the provider Raises: ValueError: If the provider is not supported r)VertexAIRAGIngestion)BedrockRAGIngestion)OpenAIRAGIngestion)ZopenaiZbedrock vertex_aiNz*RAG ingestion not supported for provider: z. Supported providers: ) Z+litellm.llms.vertex_ai.rag_engine.ingestionrZ'litellm.rag.ingestion.bedrock_ingestionrZ&litellm.rag.ingestion.openai_ingestionr get ValueErrorlistkeys)rrrr Z provider_mapZingestion_classrH/home/app/Keep/.python/lib/python3.10/site-packages/litellm/rag/utils.pyget_rag_ingestion_class s    rcCs|dkr ddlm}|SdS)z Get the appropriate RAG transformation class for a provider. Args: custom_llm_provider: The LLM provider name Returns: The transformation class for the provider, or None if not needed r r)VertexAIRAGTransformationN)Z0litellm.llms.vertex_ai.rag_engine.transformationr)rrrrrget_rag_transformation_class.s rN) __doc__typingrrZ$litellm.rag.ingestion.base_ingestionrstrrrrrrrs  !