AHjZJdZddlmZmZerddlmZdededfdZdefdZy ) ze RAG utility functions. Provides provider configuration utilities similar to ProviderConfigManager. ) TYPE_CHECKINGType)BaseRAGIngestioncustom_llm_providerreturnrc ddlm}ddlm}ddlm}|||d}|j |}|(td|dt|j|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)openaibedrock vertex_aiz*RAG ingestion not supported for provider: z. Supported providers: ) +litellm.llms.vertex_ai.rag_engine.ingestionr 'litellm.rag.ingestion.bedrock_ingestionr &litellm.rag.ingestion.openai_ingestionr get ValueErrorlistkeys)rr r r provider_mapingestion_classs I/root/tools/cai/cai_env/lib/python3.12/site-packages/litellm/rag/utils.pyget_rag_ingestion_classr swQKI%&)L #&&':;O89L8MN$$():):)<$=#> @  c|dk(rddlm}|Sy)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 rr)VertexAIRAGTransformationN)0litellm.llms.vertex_ai.rag_engine.transformationr)rrs rget_rag_transformation_classr.sk) )( rN) __doc__typingrr$litellm.rag.ingestion.base_ingestionrstrrrrrr$s; 'E>P9QBcr