`gj. dZddlmZmZmZddlZddlmZidddddddddd d d d d d ddd ddd ddddd dddddddddd ddddddd d d!d"ddd#d#d$d d%d&dddddddd d'd(d)did d*d+ddd,d d-d.dddd/d d0d1d d dd2d d3d4d d#d5d6d d7d8dddd9d d:d;dd?d@ddAd dBdCd?d?d?dDdddEd dFdGdHdHdHdIdddJd ZdKe dLeee e ffdMZ dLee effdNZ dOe dLee fdPZ dOe dLefdQZdOe dLdfdRZedSk(rZ edTedUedVedWe j'D]e\ZZedXedYedZed[d\j-ed]j'Dcgc] \}}|d^|d_c}}Zed`egedaedWgdbZeD]ZedcedYgZeddD]!Ze eZej=ee#edeededfedgedhediedjedkedledmednedWed edyycc}}w)oa Toolset Distributions Module This module defines distributions of toolsets for data generation runs. Each distribution specifies which toolsets should be used and their probability of being selected for any given prompt during the batch processing. A distribution is a dictionary mapping toolset names to their selection probability (%). Probabilities should sum to 100, but the system will normalize if they don't. Usage: from toolset_distributions import get_distribution, list_distributions # Get a specific distribution dist = get_distribution("image_gen") # List all available distributions all_dists = list_distributions() )DictListOptionalN)validate_toolsetdefaultz!All available tools, all the timed)webvision image_genterminalfilebrowser) descriptiontoolsetsr z;Heavy focus on image generation with vision and web supportZ7-)r r r r researchz/Web research with vision analysis and reasoningF2 )r rr r sciencezFScientific research with web, terminal, file, and browser capabilities^A)r r r r rr developmentz>Terminal, file tools, and reasoning with occasional web lookupP)r r r r safez$All tools except terminal for safety<)r rr r balancedz!Equal probability of all toolsetsminimalz!Only web tools for basic researchr terminal_onlyz0Terminal and file tools for code execution tasks)r r terminal_webz@Terminal and file tools with web search for documentation lookup)r r r creativez*Image generation and vision analysis focus)r r r reasoningz>Heavy research/reasoning distribution with minimal other tools)r r r browser_usezHFull browser-based web interaction with search, vision, and page control)rr r browser_onlyzt|djdd }t|r|j ||S) a Sample toolsets based on a distribution's probabilities. Each toolset in the distribution has a % chance of being included. This allows multiple toolsets to be active simultaneously. Args: distribution_name (str): Name of the distribution to sample from Returns: List[str]: List of sampled toolset names Raises: ValueError: If distribution name is not found zUnknown distribution: ru⚠️ Warning: Toolset 'z' in distribution 'z' is not validrc |dSNr<xs r7z3sample_toolsets_from_distribution..s 1Q4r9)keyr)r8 ValueErroritemsrprintrandomappendmax)r>distselected_toolsets toolset_name probabilityhighest_prob_toolsets r7!sample_toolsets_from_distributionrRs - .D 12C1DEFF%)*%5%;%;%=3! k - .|ns r7validate_distributionrTs  --r9ct|}|std|ytd|td|dtdt|djdd D]\}}td |d d |ddy)z| Print detailed information about a distribution. Args: distribution_name (str): Distribution name u❌ Unknown distribution: Nu 📊 Distribution: z Description: rz Toolsets:rc |dSrAr<rCs r7rEz)print_distribution_info..9s !r9T)rFreverseu • 15z : 3z% chance)r8rIsortedrH)r>rMtoolsetprobs r7print_distribution_infor]*s - .D  *+<*=>? !"3!4 56 T-01 23 .Z 0 6 6 8nVZ[;  '"Sa9:;r9__main__u📊 Toolset Distributions Demoz<============================================================u 📋 Available Distributions:z(----------------------------------------z :z rz, r(z%)z Toolsets: u 🎲 Sampling Examples:)r rr!r z Sample 1: z Sample 2: rBz Sample 3: z Sample 4: z Sample 5: u" 📊 Detailed Distribution Info:) __doc__typingrrrrJrrr5stranyr8r=rRboolrTr]__name__rIrHr2rMjoin toolset_listtest_distributions dist_namesamplesrange_sampledrKrZ)tsps00r7rvs((' % y :   yT  y4H  5yJ_   KydW  eyz =   {yP:   Qyj: 3 kyzI {yLY My`C aytW uyJa Ky^U s _ynY oyBs   Cy\p   ]y x #3 #8DcN#; # DdO ))c)X .S .T .;s;t;& z +, (O +, (O(*002/ d TF!n T-()*+yyT*=M=S=S=U!VEBRD!B-!VW  |n-. / '( (OI' +  9+Q q ,A7 BG NN6'? + ,  WQZL)* WQZL)* WQZL)* WQZL)* WQZL)* + 01 (OK(J'Q"WsI(