q 9i"$ dZddlmZmZmZmZmZddlmZddl m Z m Z m Z ddl mZer ddlmZddlmZdd d efd Zdd d ed d fdZddZ ddd ded efdZ ddd d edee ed eeeffdZy)aUtility functions for managing collection statistics. This module provides standalone functions for enabling, disabling, and retrieving statistics for ChromaDB collections. These functions work with the attached function system to automatically compute metadata value frequencies. Example: >>> from chromadb.utils.statistics import attach_statistics_function, get_statistics >>> import chromadb >>> >>> client = chromadb.Client() >>> collection = client.get_or_create_collection("my_collection") >>> >>> # Attach statistics function with output collection name >>> attach_statistics_function(collection, "my_collection_statistics") >>> >>> # Add some data >>> collection.add( ... ids=["id1", "id2"], ... documents=["doc1", "doc2"], ... metadatas=[{"category": "A"}, {"category": "B"}] ... ) >>> >>> # Get statistics from the named output collection >>> stats = get_statistics(collection, "my_collection_statistics") >>> print(stats) ) TYPE_CHECKINGOptionalDictAnycast) defaultdict) OneOrManyWheremaybe_cast_one_to_many)STATISTICS_FUNCTION Collection)AttachedFunction collectionrreturnc |jdS)zGenerate the default name for the statistics attached function. Args: collection: The collection to generate the name for Returns: str: The statistics function name _stats)name)rs b/var/www/html/leadgen/airagagent/rag_env/lib/python3.12/site-packages/chromadb/utils/statistics.pyget_statistics_fn_namer'soo f %%stats_collection_namercF|jtt||dS)aAttach statistics collection function to a collection. This attaches the statistics function which will automatically compute and update metadata value frequencies whenever records are added, updated, or deleted. Args: collection: The collection to enable statistics for stats_collection_name: Name of the collection where statistics will be stored. Returns: AttachedFunction: The attached statistics function Example: >>> attach_statistics_function(collection, "my_collection_statistics") >>> collection.add(ids=["id1"], documents=["doc1"], metadatas=[{"key": "value"}]) >>> # Statistics are automatically computed >>> stats = get_statistics(collection, "my_collection_statistics") N)functionroutput_collectionparams)attach_functionr r)rrs rattach_statistics_functionr3s-,  % %$ #J // & rcf|jt|}|jdk(sJd|S)a\Get the statistics attached function for a collection. Args: collection: The collection to get the statistics function for Returns: AttachedFunction: The statistics function Raises: NotFoundError: If statistics are not enabled AssertionError: If the attached function is not a statistics function statisticsz.Attached function is not a statistics function)get_attached_functionr function_name)rafs rget_statistics_fnr$Qs=  ) )*@*L MB L(878( Irdelete_stats_collectioncRt|}|j|j|S)aDetach statistics collection function from a collection. Args: collection: The collection to disable statistics for delete_stats_collection: If True, also delete the statistics output collection. Defaults to False. Returns: bool: True if successful Example: >>> detach_statistics_function(collection, delete_stats_collection=True) )delete_output_collection)r$detach_functionr)rr% attached_fns rdetach_statistics_functionr*es2 $J/K  % %3J & rNkeysct|}d}|)t||kDrtdt|d|dddlm}|j j ||j|j}||j |dd }td }i} d} |ttd d |d gziind} |jdg| | } | jdxsg} | sn| D]}||jd }|jd}|jd}|jd}|jd}|^|a|d|g|d k(r |dk(sr|| d<x||n|}t|tsJt|tsJt|tsJ||||d<| t| z } dt!|i}| r| |d <|S)aTGet the current statistics for a collection. Statistics include frequency counts for all metadata key-value pairs, as well as a summary with the total record count. Args: collection: The collection to get statistics for stats_collection_name: Name of the statistics collection to read from. keys: Optional metadata key(s) to filter statistics for. Can be a single key string or a list of keys. If provided, only returns statistics for those specific keys. Returns: Dict[str, Any]: A dictionary with the structure: { "statistics": { "key1": { "value1": {"count": count, ...}, "value2": {"count": count, ...} }, "key2": {...}, ... }, "summary": { "total_count": count } } Example: >>> attach_statistics_function(collection, "my_collection_statistics") >>> collection.add( ... ids=["id1", "id2"], ... documents=["doc1", "doc2"], ... metadatas=[{"category": "A", "score": 10}, {"category": "B", "score": 10}] ... ) >>> # Wait for statistics to be computed >>> stats = get_statistics(collection, "my_collection_statistics") >>> print(stats) { "statistics": { "category": { "A": {"count": 1}, "B": {"count": 1} }, "score": { "10": {"count": 2} } }, "summary": { "total_count": 2 } } Raises: ValueError: If more than 30 keys are provided in the keys filter. NzToo many keys provided: z. Maximum allowed is z[ keys per request. Consider calling get_statistics multiple times with smaller key batches.rr )rtenantdatabase)clientmodelembedding_function data_loaderc ttSN)rdictrrz get_statistics..s kRVFWrkeyz$insummary metadatas)includeoffsetwherevalue value_labeltypecount total_countr )r len ValueErrorchromadb.api.models.Collectionr_clientget_collectionr.r/rrr get isinstancestrintr6)rrr+ keys_listMAX_KEYSrstats_collection_modelstats_collectionstatsr:r= where_filterpager;metadatameta_keyr?r@ value_typerB stats_keyresults rget_statisticsrY{sM|'t,IHY(!:&s9~&67""*,W W  :(//>> "  $$?"!!$ 3>>W2XE G F   UUUI $;<=>  ## M& $ HH[)/R  ! @H||E*HLL)E",,}5K!f-JLL)E$%*%y( -16 .0;/F EI%h444%i555%eS111:?E(OI.w73 @8 #i. K NDK (F#y Mr)rrrr)Fr5)__doc__typingrrrrr collectionsrchromadb.api.typesr r r chromadb.api.functionsr rFr$chromadb.api.models.AttachedFunctionrrKrrr$boolr*rYr7rrras8<;#GG69E &| & &58<*?D7; 2&*RRR 9S> "R #s(^ Rr