'Xh_|ddlmZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZGdd Zy ) )copy)UnionOptionalIterableget_args) BaseModel)model_fields_set)INFERENCE_OBJECT_TYPES)ModelSchemaParser) convert_paths FieldPathc eZdZdZddeeddfdZdeee e fde e fdZ dde d ee e d ee de e fd Z dd e d e de e d ee de e f dZy)InspectorEmbedzInspector which collects paths to objects requiring inference in the received models Attributes: parser: ModelSchemaParser instance Nparserreturnc6|t|_y||_yN)r r)selfrs M/RAG/venv/lib/python3.12/site-packages/qdrant_client/embed/embed_inspector.py__init__zInspectorEmbed.__init__s-3^')  pointscVg}t|trF|jj|j|j |j |nt|tr6|jD]"}|j |j|$nmt|tr]|D]X}t|ts|jj|j|j |j |Ztt|}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 rr!zInspectorEmbed.inspects fi ( KK # #F$4$4 5 LL,,V4 5  % T\\%01)  )eY/KK++EOO<LL!4!4U!;< s5z"U##rmodr$accumc |:|jjj|jjgn|}g}|D]G}|j |j ||j|jr |jng|I|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==+A+A2F]b D   **DIItyy2u  roriginal_model current_pathr/c g}||}n|d|z }dtdtdttffd }t||d}|gSt |t t r|gSt |trf|j||||D]G}|jj||j|jr |jng|I|St |tr|D]U} t | tst | t t r|j||j|| |W|D]N}|D]G} |jj| |j|jr |jng|IP|St |trg}|jD]\} } t | ts| gn| } | D]U} t | tst | t t r|j||j|| |W|D]N}| D]G} |jj| |j|jr |jng|IP|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]\}|jjvsjj|}|jjj|^j |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 )r rname_recursive_ref_mappingrr*rr)r6r7recursive_pathsfield mapped_fieldrs rinspect_recursivez?InspectorEmbed._inspect_inner_models..inspect_recursiveasx!O)&1DKKBBB#';;#I#I%#PL#**4;;+A+A,+OP2 &&vtO/DkR Rr)rstrlistgetattrrrr rr-r.r/appendritems) rr2r3r/r(r0r=model next_path current_modelkeyr s ` rr-z$InspectorEmbed._inspect_inner_modelsIsh$ = E q' 'E Si Sc Sd3i S d; =I eX&<= >7N eY '   0> ?! ""..y00INN)..XZ\a"  t $!& !-;mX6L-MN&&u-""#4]E#JK"'" %*M&&22)%--.7nnINN"! &+"  t $K${{} V)3FD)A&v%+M%mY? !-:P1QR#**51&&'8'NO&,"&I)/ #** 66 - ) 1 12;.. b % *0"& -*rr)NN)r, __module__ __qualname____doc__rr rrrrr?r r!r>rr-rrrr s Hx(9:HdH$eHY$7$BC$Y$6_c%-d9o%>NVWZm c: $ g!gg9o g } g c grrN)rtypingrrrrpydanticrqdrant_client._pydantic_compatr qdrant_client.embed.commonr !qdrant_client.embed.schema_parserr qdrant_client.embed.utilsr r rrJrrrQs)66;=?>ccr