tj6ddlmZmZmZmZmZddlZddlZ ddl m Z m Z deeej deeeeeeej ffdZdeeefde d eddfd Zdeeefd e ddfd Zde ded fdZd e dd fdZdS))ListDictAnyOptionalUnionN) QueryResult GetResult embeddingsreturncf|dSt|dtjr d|Dn|S)z Transform embeddings from numpy arrays to lists of floats. This is a shared helper function to avoid duplicating the transformation logic. Nrc6g|]}|S)tolist).0embs T/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/chromadb/utils/results.py z)_transform_embeddings..s ,,,#,,,) isinstancenpndarray)r s r_transform_embeddingsrsEt jmRZ 0 0 ,,,,,, r data_dict query_result query_idxc|dD]x}||}|_|d}|dkrt|}t|trt |dkr||}|||<ydS)a@ Helper function to add fields from a query result to a dictionary. Handles the nested array structure specific to query results. Args: data_dict: Dictionary to add the fields to query_result: QueryResult containing the data query_idx: Index of the current query being processed includedNsr r)getrstriprrlistlen)rrrfieldvaluekeys r_add_query_fieldsr&sj)##  ''  ,,s##C $$-e44%&& )3u::>>i("IcN##r get_resultc|dD]H}||}|/|d}|dkrt|}|||<IdS)z Helper function to add fields from a get result to a dictionary. Handles the flat array structure specific to get results. Args: data_dict: Dictionary to add the fields to get_result: GetResult containing the data rNrr )rr r)rr'r#r$r%s r_add_get_fieldsr)0slJ'##u%%  ,,s##C $$-e44"IcN ##rz pd.DataFramec^ ddl}n#t$rtdwxYwg}t|d}t|D]b}i}|d||d<t ||||j|}|dd||c|S)a> Function to convert QueryResult to list of DataFrames. Handles the nested array structure specific to query results. Column order is defined by the order of the fields in the QueryResult. Args: query_result: QueryResult to convert to DataFrames. Returns: List of DataFrames. rNz:pandas is required to convert query results to DataFrames.idsidTinplace)pandas ImportErrorr"ranger& DataFrame set_indexappend)rpddfs num_queriesi data_for_dfdfs rquery_result_to_dfsr;EsX XXXVWWWX Cl5)**K ;  &( (/2 D+|Q777 R\+ & & T4 ((( 2 J!c ddl}n#t$rtdwxYwi}|d|d<t|||j|}|dd|S)a# Function to convert GetResult to a DataFrame. Handles the flat array structure specific to get results. Column order is defined by the order of the fields in the GetResult. Args: get_result: GetResult to convert to a DataFrame. Returns: DataFrame. rNz9pandas is required to convert get results to a DataFrame.r+r,Tr-)r/r0r)r2r3)r'r5r9r:s rget_result_to_dfr>esW WWWUVVVW#%K"5)KK,,, k " "BLLtL$$$ Ir<)typingrrrrrnumpyrr/r5chromadb.api.typesrr rfloatrstrintr&r)r;r>rrrrEsx3333333333333355555555 bj)*  eDe%tBJ'7789     #CH~### ####2#CH~## ####*kd>6J@~r