ESi dZddlmZddlmZddlmZmZddlm Z GddeZ eGdd Z Gd d Z y ) u<查询路由器 - 评估查询复杂度并决定处理模式) dataclass)Enum)DictTuple) get_settingsceZdZdZdZdZy)Modeu 处理模式syncasyncN)__name__ __module__ __qualname____doc__SYNCASYNC:/home/jang/Projects/ai-researcher/src/core/query_router.pyr r s D Err c0eZdZUdZeed<eed<eed<y)QueryRouterResultu 路由结果modescoreestimated_secondsN)r r rrr __annotations__floatintrrrrrs J LrrcteZdZdZdZdededeee ffdZ dede fdZ dede fdZ dede fd Z dede fd Zy ) QueryRouteru查询路由器c\t}|j|_|j|_yN)r!QUERY_ROUTER_COMPLEXITY_THRESHOLD thresholdQUERY_ROUTER_WEIGHTSweights)selfsettingss r__init__zQueryRouter.__init__s#>!CC44 rquerycontextreturnc|j||j||j||j|d}|d|jdz|d|jdzz|d|jdzz|d|jdzz}||j k\rt jnt j}||fS)u 评估查询复杂度 Args: query: 用户查询文本 context: 上下文信息 Returns: (处理模式, 复杂度评分) ) data_sources api_callsreasoning_depthhistorical_timer,r-r.r/) _score_data_sources_estimate_api_calls_estimate_reasoning_get_historical_avgr$r"r rr)r%r(r)scoresrrs rassess_complexityzQueryRouter.assess_complexitys!44W=11%8#77>#77>   > "T\\.%A A ; $,,{"; ; < $ % 5F(G G H $ % 5F(G G H #dnn4tzz$))U{rcV|jdg}t|}|dk(ry|dk(ryy)u根据数据源数量评分required_sources?皙?)getlen)r%r)sourcescounts rr0zQueryRouter._score_data_sources;s3++0"5G  A: aZrcNgd}tfd|D}|dk(ry|dkryy)u估计API调用次数)u对比u汇总u多个u所有u分别u各自c3,K|] }|vsd yw)r8Nr).0indr(s r z2QueryRouter._estimate_api_calls..Is>Ca>s rr9r:r;r<)sum)r%r( indicatorsmatchess ` rr1zQueryRouter._estimate_api_callsFs/Q >:>> a< \rcJgd}tfd|Drydvsdvryy)u估计推理复杂度)u 为什么u建议u分析u预测u原因u趋势c3&K|]}|v ywr r)rCkwr(s rrEz2QueryRouter._estimate_reasoning..Ts8rrU{8sg333333??u？333333?r9)any)r%r(reasoning_keywordss ` rr2zQueryRouter._estimate_reasoningQs.\ 8%78 8 E\Ue^rcy)u$获取历史平均处理时间评分rMr)r%r(s rr3zQueryRouter._get_historical_avgZsrN)r r rrr'strdictrr rr5r0r1r2r3rrrrrsy5 sTeD%K>P8 4 E    rrN) r dataclassesrenumrtypingrr src.configrr rrrrrrWsEB!#4   GGr