j|ddlmZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZGdd Zd S) )copy)Iterableget_args) BaseModel)model_fields_set)INFERENCE_OBJECT_TYPES)ModelSchemaParser) convert_paths FieldPathc eZdZdZddedzddfdZdeeezdee fdZ dded ee dzd e dzdee fd Z dd ed e dee d e dzdee f dZ dS)InspectorEmbedzInspector which collects paths to objects requiring inference in the received models Attributes: parser: ModelSchemaParser instance Nparserreturnc4|tn||_dSN)r r)selfrs QC:\Projects\DSCons\.venv\Lib\site-packages\qdrant_client\embed\embed_inspector.py__init__zInspectorEmbed.__init__s-3^'))) pointscg}t|trH|j|j|||nt|tr@|D]*}|| |+nvt|tra|D]^}t|trG|j|j|||_tt|}t|S)zLooks for all the paths to objects requiring inference in the received models Args: points: models to inspect Returns: list of FieldPath objects ) isinstancerr parse_model __class__extend_inspect_modeldictvaluesinspectrsortedsetr )rrpathsvaluepoints rrzInspectorEmbed.inspects3 fi ( ( = K # #F$4 5 5 5 LL,,V44 5 5 5 5  % % = 2 2 T\\%001111 2  ) ) = = =eY//=K++EO<<<LL!4!4U!;!;<<<s5zz""U###rmodr"accumc |*|jj|jjgn|}g}|D]@}||||j|jr|jng|A|S)aLooks for all the paths to objects requiring inference in the received model Args: mod: model to inspect paths: list of paths to the fields possibly containing objects for inference accum: accumulator for the path. Path is a dot separated string of field names which we assemble recursively Returns: list of paths to the model fields containing objects for inference ) r path_cachegetr__name__r_inspect_inner_modelscurrenttail)rr%r"r& found_pathspaths rrzInspectorEmbed._inspect_model1sKP- &**3=+A2FFF]b   D   **DI'Etyy2u     roriginal_model current_pathr-c g}||}n|d|z }dtdtdttffd }t||d}|gSt |t t r|gSt |trd|||||D]@}|||j |j r|j ng|A|St |tr|D]n} t | tst | t t r| |||| |o|D]E}|D]@} || |j |j r|j ng|AF|St |trg}| D]\} } t | ts| gn| } | D]n} t | tst | t t r| |||| |o|D]E}| D]@} || |j |j r|j ng|AF|S)aLooks for all the paths to objects requiring inference in the received model Args: original_model: model to inspect current_path: the field to inspect on the current iteration tail: list of FieldPath objects to the fields possibly containing objects for inference accum: accumulator for the path. Path is a dot separated string of field names which we assemble recursively Returns: list of paths to the model fields containing objects for inference N.member accumulatorrcg}t|D]G}|jjvr7jj|}|jj|H|t ||S)aUIterates over the set model fields, expand recursive ones and find paths to objects requiring inference Args: member: currently inspected model, which may or may not contain recursive fields accumulator: accumulator for the path, which is a dot separated string assembled recursively )rrname_recursive_ref_mappingrr(rr)r4r5recursive_pathsfield mapped_fieldrs rinspect_recursivez?InspectorEmbed._inspect_inner_models..inspect_recursiveas!O)&11 Q QDKBBB#';#I%#PL#**4;+A,+OPPP&&vtO/D/DkRR Rr)rstrlistgetattrrrrrr+r,r-appendritems) rr0r1r-r&r.r;model next_path current_modelkeyrs ` rr+z$InspectorEmbed._inspect_inner_modelsIs$ = EE ''' 'E Si Sc Sd3i S S S S S S d;; =I eX&<== > > 7N eY ' '8    00>> ? ? ?!   ""..y0IN2Z)..XZ\a   t $ $, !& L L !-;;mX6L-M-MNN.&&u---""#4#4]E#J#JKKKK!   %*M&&22)%-.7nDINN"!   t $ $ K${{}}   V)3FD)A)AM&v%+PPM%mY??! !-:P1Q1QRR2#**5111&&'8'8'N'NOOOO!%  I)/ #** 66 - ) 12;. H b %  rr)NN)r* __module__ __qualname____doc__r rrrr=r rr<rr+rrr r s: HH047H4HHHH$hy1I=$$y/$$$$6Y]%))_t%;KNQU: c:! gg!gg9o g Tz g c ggggggrr N)rtypingrrpydanticrqdrant_client._pydantic_compatrqdrant_client.embed.commonr!qdrant_client.embed.schema_parserr qdrant_client.embed.utilsr r r rHrrrOs%%%%%%%%;;;;;;======??????>>>>>>>>ccccccccccr