Ji]ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZmZddlmZmZmZmZddlmZede ZGd d ej2ZGd d ee eZy)N)abstractmethod) copy_context)partial)AnyCallableDictGenericListOptionalTupleTypeVar) BaseModel ConfigDictField PrivateAttr)BaseSpanT)boundceZdZdZ d dddeedeedefdeedeedfd ee eefd ee d dffd Z xZ S)Threadzm A wrapper for threading.Thread that copies the current context and uses the copy to run the target. N)daemongrouptarget.nameargskwargsrreturnc|#t|g|it|tr|nif}nd}t||t j |||y)N)rrrrr)r isinstancedictsuper__init__rrun)selfrrrrrr __class__s /var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/llama_index_instrumentation/span_handlers/base.pyr#zThread.__init__s`  VVJvt4L&RTVDD >%%  )NNNrN) __name__ __module__ __qualname____doc__r rrstrr rboolr# __classcell__r&s@r'rrs $/3" "+/  "& } #s(+, sm  CHo  c3h(      r(rc~eZdZUedZeedZee e fe d<ee dZ ee e d<ee dZee e d<eid Zeeee fe d <eZeej,e d <iggifdee e fdee dee d eee fffd Zede fdZedej,fdZ d!de dej:deedee deee efdeddfdZ d"de dej:deedeededdf dZ d"de dej:deedee deddf dZ!e" d!de dej:deedee deee efdedee fdZ#e" d"de dej:deedeededee f dZ$e" d"de dej:deedee dedee f d Z%xZ&S)#BaseSpanHandlerT)arbitrary_types_allowedzDictionary of open spans.)default_factory description open_spanszList of completed spans.completed_spans dropped_spansz&Id of current spans in a given thread.)defaultr5current_span_ids_lockc:t|||||d|_y)N)r6r7r8r:)r"r#r;)r%r6r7r8r:r&s r'r#zBaseSpanHandler.__init__>s, !+'-   r(rcy)z Class name.r2r)clss r' class_namezBaseSpanHandler.class_nameMs!r(cd|jtj|_|jS)N)r; threadingLock)r%s r'lockzBaseSpanHandler.lockRs$ :: ")DJzzr(Nid_ bound_argsinstance parent_idtagsrc ||jvry|j|||||}|r%|j5||j|<dddyy#1swYyxYw)zLogic for entering a span.)rDrErFparent_span_idrHN)r6new_spanrC)r%rDrErFrGrHrspans r' span_enterzBaseSpanHandler.span_enterXsn $// ! ==%!( !DYY0+/DOOC(0000s AAresultc |j||||}|r#|j5|j|=dddyy#1swYyxYw)zLogic for exiting a span.)rDrErFrNN)prepare_to_exit_spanrCr6)r%rDrErFrNrrLs r' span_exitzBaseSpanHandler.span_exitpsY(( Xf)   )OOC( ) )  ) ) <Aerrc |j||||}|r#|j5|j|=dddyy#1swYyxYw)z*Logic for dropping a span i.e. early exit.)rDrErFrSN)prepare_to_drop_spanrCr6)r%rDrErFrSrrLs r' span_dropzBaseSpanHandler.span_dropsY(( X3)   )OOC( ) )  ) )rRrJc y)z Create a span. Subclasses of BaseSpanHandler should create the respective span type T and return it. Only NullSpanHandler should return a None here. Nr)r%rDrErFrJrHrs r'rKzBaseSpanHandler.new_span r(c y)z Logic for preparing to exit a span. Subclasses of BaseSpanHandler should return back the specific span T that is to be exited. If None is returned, then the span won't actually be exited. Nr)r%rDrErFrNrs r'rPz$BaseSpanHandler.prepare_to_exit_spanrXr(c y)z Logic for preparing to drop a span. Subclasses of BaseSpanHandler should return back the specific span T that is to be dropped. If None is returned, then the span won't actually be dropped. Nr)r%rDrErFrSrs r'rUz$BaseSpanHandler.prepare_to_drop_spanrXr()NNN)NN)'r)r*r+r model_configrr!r6rr-r__annotations__listr7r r8r:rr rr;rArBr# classmethodr?propertyrCinspectBoundArgumentsrMrQ BaseExceptionrVrrKrPrUr/r0s@r'r2r2.sd;L$*E JS!V  %*D OT!W#*DM4727 H2d3 -.'2mE8INN #3$&#%!#+- aL a Aw  sCx. !3!!inn#'#')- 0 0**03- 0 C= 0 tCH~& 00 08#' $ ) )**)3- )  )  ) )(#''+ ) )**)3- ) m $ )  ) )  #'(,)-   ** 3-  !  tCH~&    !  " #' $   ** 3-       !  " #''+   ** 3-  m $     !  r(r2)r`rAabcr contextvarsr functoolsrtypingrrrr r r r r pydanticrrrr%llama_index_instrumentation.span.baserrrr2rr(r'risX$OOO>>: Cx  Y   >V iV r(