jddlmZddlmZGddeeZGddeZGddeZGd d eZGd d eZ Gd deZ GddeZ dS))ABCwrapsc0eZdZUeed<edZdS) CustomError error_stringc@tfd}|S)z Provides a wrapper for a function that catches any exception and re-raises it as the customer error. If the exception itself is already an instance of the custom error, re-raises original error. c |i|S#t$rd}t|sNtt|s1jt ||d}~wwxYw)N) BaseException isinstance issubclasstyperformatstr)argskwargserrorclsfs c/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/unstructured/ingest/error.pywrapperz!CustomError.wrap..wrappers q$)&)))    !%--Njec6R6RN#c.55c%jjAABBM s A9AA44A9r)rrrs`` rwrapzCustomError.wraps; q       N)__name__ __module__ __qualname__r__annotations__ classmethodrrrrrs;[rrceZdZdZdS)SourceConnectionErrorz3Error in getting data from upstream data source: {}Nrrrrrrrr!r!sHLLLrr!ceZdZdZdS)SourceConnectionNetworkErrorz/Error in connecting to upstream data source: {}Nr"rrrr$r$ sDLLLrr$ceZdZdZdS)DestinationConnectionErrorz1Error in connecting to downstream data source: {}Nr"rrrr&r&$sFLLLrr&ceZdZdZdS)EmbeddingEncoderConnectionErrorz7Error in connecting to the embedding model provider: {}Nr"rrrr(r((sLLLLrr(ceZdZdZdS) WriteErrorz.Error in writing to downstream data source: {}Nr"rrrr*r*,sCLLLrr*ceZdZdZdS)PartitionErrorz!Error in partitioning content: {}Nr"rrrr,r,0s6LLLrr,N) abcr functoolsr Exceptionrr!r$r&r(r*r,rrrr0s)S.IIIIIKIIIEEEEE#8EEEGGGGGGGGMMMMMkMMMDDDDDDDD77777[77777r