inddlmZddlZddlmZGddeZGddeZGdd eZGd d eZ Gd d eZ GddeZ GddeZ GddeZ GddeZGddeZGddeZGddeZGddeZGddeZGd d!eZy)") annotationsN)LiteralcTeZdZUdZded<ded<ded< d fd ZxZS) APIStatusErrorz  #//"**..~>)rstrrrrz object | NonereturnNone)__name__ __module__ __qualname____doc____annotations__r __classcell__rs@rrrsBF??)7??L? ??rrc0eZdZdd dfdZxZS)APIConnectionErrorzConnection error.)rc*t|||dy)Nrrr)rrrrs rrzAPIConnectionError.__init__s '5r)rrr httpx.Requestrrrrrrr"r#s@rr%r%s( 366>K6 66rr%ceZdZUdZded<y)BadRequestErroriz Literal[400]r Nrrrr r!rrr+r+ #K#rr+ceZdZUdZded<y)AuthenticationErroriz Literal[401]r Nr,r-rrr0r0"r.rr0ceZdZUdZded<y)PermissionDeniedErroriz Literal[403]r Nr,r-rrr2r2&r.rr2ceZdZUdZded<y) NotFoundErroriz Literal[404]r Nr,r-rrr4r4*r.rr4ceZdZUdZded<y) ConflictErroriz Literal[409]r Nr,r-rrr6r6.r.rr6ceZdZUdZded<y)UnprocessableEntityErroriz Literal[422]r Nr,r-rrr8r82r.rr8ceZdZUdZded<y)RateLimitErroriz Literal[429]r Nr,r-rrr:r:6r.rr:c eZdZdfd ZxZS)APITimeoutErrorc(t|d|y)NzRequest timed out.)rrr')rrrs rrzAPITimeoutError.__init__;s !5wGr)rr(rrr)r#s@rr<r<:sHHrr<c&eZdZdZddfd ZxZS)StorageNotInitializedErrorzCRaised when storage operations are attempted before initialization.c*t||dy)Nz not initialized. Please ensure proper initialization: rag = LightRAG(...) await rag.initialize_storages() # Required - auto-initializes pipeline_status See: https://github.com/HKUDS/LightRAG#important-initialization-requirementsr')r storage_typers rrz#StorageNotInitializedError.__init__Bs! n[ \ r)Storage)rArrrrr rr"r#s@rr?r??sM  rr?c&eZdZdZddfd ZxZS)PipelineNotInitializedErrorz>Raised when pipeline status is accessed before initialization.c0d|d}t||y)NzPipeline namespace 'a2' not found. Pipeline status should be auto-initialized by initialize_storages(). If you see this error, please ensure: 1. You called await rag.initialize_storages() 2. For multi-workspace setups, each LightRAG instance was properly initialized Standard initialization: rag = LightRAG(workspace='your_workspace') await rag.initialize_storages() # Auto-initializes pipeline_status If you need manual control (advanced): from lightrag.kg.shared_storage import initialize_pipeline_status await initialize_pipeline_status(workspace='your_workspace')r')r namespacemsgrs rrz$PipelineNotInitializedError.__init__Ps*"9+.M N " r))rGrrCr#s@rrErEMsHrrEc&eZdZdZddfd ZxZS)PipelineCancelledExceptionz=Raised when pipeline processing is cancelled by user request.c2t||||_yNrrrrrrs rrz#PipelineCancelledException.__init__h ! r)zUser cancelledrrrCr#s@rrKrKesGrrKc6eZdZdZ d dfd ZxZS)ChunkTokenLimitExceededErrorz7Raised when a chunk exceeds the configured token limit.c|r|jnd}|r|ddnd}|rd|dnd}d|d|d|}t| |||_||_||_y)NPz Preview: ''rIzChunk token length z exceeds chunk_token_size .)striprr chunk_tokenschunk_token_limit chunk_preview) rrYrZr[previewtruncated_preview preview_noterrs rrz%ChunkTokenLimitExceededError.__init__ps ,9-%%'d,3GCRL=N%6$7q9TV !,/IJ[I\\]n   !(!2.rrM)rYr rZr r[r rrrCr#s@rrSrSms;A %) ///" /  //rrSc$eZdZdZdfd ZxZS)DataMigrationErrorz>Raised when data migration from legacy collection/table fails.c2t||||_yrMrNrOs rrzDataMigrationError.__init__rPrrQrCr#s@rr`r`sHrr`) __future__rhttpxtypingr Exceptionrr%r+r0r2r4r6r8r:r< RuntimeErrorr?KeyErrorrErK ValueErrorrSr`r-rrris"  ?Y ? 66$n$$.$$N$$N$$N$$~$$^$H(H    (0/:/,r