o ưi@s.UdZddlmZddlmZmZmZddlmZm Z m Z Gddde eZ gdZ ee ed<gd Zee ed <gd Zee ed <gd Zee ed<ddddddddZee efed<ddddZee efed<dddZee efed<d d!d"d"d#Zee e fed$<Gd%d&d&eZeZd'S)(z Configuration for the Complexity Router. Contains default keyword lists, weights, tier boundaries, and configuration classes. All values are configurable via proxy config.yaml. )Enum)DictListOptional) BaseModel ConfigDictFieldc@s eZdZdZdZdZdZdZdS)ComplexityTierz'Complexity tiers for routing decisions.SIMPLEMEDIUMCOMPLEX REASONINGN)__name__ __module__ __qualname____doc__r r r r rrg/home/app/Keep/.python/lib/python3.10/site-packages/litellm/router_strategy/complexity_router/config.pyr s r )-functionclassdefconstZletvarimportZexportreturnasyncawaittryZcatch exceptionerrordebugapiZendpointrequestresponseZdatabasesqlqueryZschema algorithmZ implementZrefactoroptimizepython javascript typescriptjavarustgolangZreactZvueangularnodedockerZ kubernetesgitcommitmergebranchz pull requestDEFAULT_CODE_KEYWORDS)z step by stepz think throughz let's thinkzreason throughz analyze thisz break downzexplain your reasoningzshow your workzchain of thoughtzthink carefullyz consider allevaluatez pros and conszcompare and contrastzweigh the optionsZlogicalZdeduceZinferZconcludeDEFAULT_REASONING_KEYWORDS) architecture distributedZscalableZ microservicezmachine learningzneural networkz deep learningZ encryptionZauthentication authorizationZ performanceZlatencyZ throughputZ benchmarkZ concurrencyZparallel threadingZmemorycpuZgpu optimizationprotocolZtcphttpZgrpcZ websocket containerZ orchestrationDEFAULT_TECHNICAL_KEYWORDS)zwhat iszwhat'sdefinez definition ofzwho iszwho waszwhen didzwhen waszwhere isz where waszhow manyzhow muchz yes or noz true or falsesimpleZbriefZshortZquickZhellohiZheyZthanksz thank youZgoodbyeZbyeokayDEFAULT_SIMPLE_KEYWORDSg?g333333?g?g?gQ?g{Gz?)Z tokenCountZ codePresenceZreasoningMarkersZtechnicalTermsZsimpleIndicatorsZmultiStepPatternsZquestionComplexityDEFAULT_DIMENSION_WEIGHTSg333333?gffffff?g333333?)Z simple_mediumZmedium_complexZcomplex_reasoningDEFAULT_TIER_BOUNDARIESi)rCcomplexDEFAULT_TOKEN_THRESHOLDSz gpt-4o-minizgpt-4ozclaude-sonnet-4-20250514)r r r r DEFAULT_TIER_MODELSc@s$eZdZUdZeddddZeeefed<eddddZ eee fed <ed dd dZ eee fed <ed dddZ eee fed<edddZeeeed<edddZeeeed<edddZeeeed<edddZeeeed<edddZeeed<eddZdS)ComplexityRouterConfigz'Configuration for the ComplexityRouter.cCtSN)rLcopyrrrruzComplexityRouterConfig.z*Mapping of complexity tiers to model names)default_factory descriptiontierscCrNrO)rHrPrrrrrQ{rRzScore boundaries between tierstier_boundariescCrNrO)rKrPrrrrrQrRz8Token count thresholds for simple/complex classificationtoken_thresholdscCrNrO)rGrPrrrrrQrRz"Weights for each scoring dimensiondimension_weightsNz(Keywords indicating code-related content)defaultrT code_keywordsz.Keywords indicating reasoning-required contentreasoning_keywordsz%Keywords indicating technical contenttechnical_keywordsz(Keywords indicating simple/basic queriessimple_keywordsz1Default model to use if tier cannot be determined default_modelZallow)extra)rrrrrrUrstr__annotations__rVfloatrWintrXrZrrr[r\r]r^rZ model_configrrrrrMpsN rMN)renumrtypingrrrZpydanticrrrr`r r5rar7rArFrGrbrHrKrcrLrMZDEFAULT_COMPLEXITY_CONFIGrrrrs>        7