Yash030/claude-code-proxy
2
1+
2 fb j � �� a � 0 t $ R t^ RIHt ^ RIHt ^ RIHt ]' d ^ RIH t ]! RRR7 ! R R4 4 t3R ]4! R5R R RRRR R
^PR7 R]6! RRRRRRR R
^FR7 R]7! RRRRRRR R^ZR7 R]8! RRRRRRR R^UR7 R]9! RRRRRRR R^KR7 R]10! RRRRRRR R^FR7 R ]11! RR R RRRR R^AR7 R ]12! RR R RRRR R^<R7 /t R ] R!& R" R# lt
R$ R% ltR+R' R( lltR) R* ltR&# ),z2Model capability registry for intelligent routing.)� annotations)� dataclass)�
TYPE_CHECKING)�SequenceT)�frozen�slotsc � � ] tR t^ t$ RtR]R& R]R& R]R& RtR]R& RtR]R & RtR]R13& R t R]R & Rt14R]R
& Rt R]R& Rt R]R& Rt
R]R& ^dtR]R& RtR# )�ModelCapabilitiesz7Capabilities of a specific model for routing decisions.�str� provider_id�model_id� model_refF�bool�vision�coding� reasoningT� general_text�multimodal_input�multimodal_outputi �int�15max_tokens�medium�speed�priority� N)�__name__�16__module__� __qualname__�__firstlineno__�__doc__�__annotations__r r r r r r r r r �__static_attributes__r � �?C:\Users\yashw\.claude\temp-hf-clone\core\model_capabilities.pyr r sq � �A����M��N��F�D���F�D���I�t�� �L�$� � "��d�"� #��t�#��J����E�3���H�c�r" r zzen/minimax-m2.5-free�zenzminimax-m2.5-freei } �fast) r r r
r r r r r r z$nvidia_nim/stepfun-ai/step-3.5-flash�17nvidia_nimzstep-3.5-flashz.nvidia_nim/qwen/qwen3-coder-480b-a35b-instructz qwen3-coder-480b-a35b-instruct�slowz7nvidia_nim/mistralai/mistral-large-3-675b-instruct-2512z"mistral-large-3-675b-instruct-2512z3nvidia_nim/abacusai/dracarys-llama-3.1-70b-instructz dracarys-llama-3.1-70b-instructr znvidia_nim/z-ai/glm4.7zglm4.7z*nvidia_nim/bytedance/seed-oss-36b-instructzseed-oss-36b-instructz%nvidia_nim/mistralai/mistral-nemotronzmistral-nemotronz dict[str, ModelCapabilities]�MODEL_CAPABILITIESc � � V ^8� d QhRRRR/# )� r
r18 �return�ModelCapabilities | Noner )�formats "r# � __annotate__r. ~ s � � -� -�c� -�.F� -r" c �, � \ P V 4 # )z0Get capabilities for a specific model reference.)r( �get)r
s &r# �get_model_capabilitiesr1 ~ s � � � !� !�)� ,�,r" c � � V ^8� d QhRRRR/# )r* �19capabilityr20 r+ zlist[ModelCapabilities]r )r- s "r# r. r. � s � � � �C� �4K� r" c � � . p\ P 4 F( p\ W R4 ' g K VP V4 K* VP R RR7 V# )z0Find all models that have a specific capability.Fc � � V P # �N�r ��xs &r# �<lambda>�-find_models_with_capability.<locals>.<lambda>� s � �q�z�zr" T��key�reverse)r( �values�getattr�append�sort)r3 �results�capss & r# �find_models_with_capabilityrE � sK � ��G�"�)�)�+�� �4�U� +� +� �N�N�4� � ,� �L�L�)�4�L�8� �Nr" Nc �$ � V ^8� d QhRRRRRR/# )r* �required_capabilities�set[str]�available_modelszSequence[str] | Noner+ r, r )r- s "r# r. r. � s& � � � �#� �*� � � r" c � a� . pV' d* V Uu. uF q3\ 9 g K \ V, NK upM \ \ P 4 4 pV FV o\ ;QJ d V3R lV 4 F ' d K RM RM! V3R lV 4 4 ' g KE VP S4 KX V' g R# VP R RR7 V^ , # u upi )a Find the best model matching required capabilities.21 22Args:23 required_capabilities: Set of capability names needed (e.g., {"coding", "vision"})24 available_models: Optional list of model refs to filter by25 26Returns:27 Best matching ModelCapabilities or None28c 3 �> <" � T F p\ SVR 4 x � K R# 5i�FN�r@ )�.0�caprD s & �r# � <genexpr>�+find_best_model_for_task.<locals>.<genexpr>� s �� � �J�4I�S�w�t�S�%�(�(�4I�s � FTNc � � V P # r6 r7 r8 s &r# r: �*find_best_model_for_task.<locals>.<lambda>� s � �!�*�*r" r< )r( �listr? �allrA rB )rG rI �29candidates�m�models_to_checkrD s && @r# �find_best_model_for_taskrY � s� �� � �J� � )9�T�(8�1�AS�<S� � �A� � �(8�T�
�$�+�+�-�
.� � �� �3�J�4I�J�3�3�3�J�4I�J� J� J� � � �d� #� �30 �� �O�O�,�d�O�;� �a�=��� Us31 � C�Cc �$ � V ^8� d QhRRRRRR/# )r* �32model_capsr �requiredrH r+ ztuple[int, int]r )r- s "r# r. r. � s& � � *� *�!� *�� *� � *r" c �J a � \ V 3R lV 4 4 pVS P 3# )zSCalculate match score for routing.33 34Returns (matched_count, priority) for sorting.35c 3 �R <" � T F p\ SVR 4 ' g K ^x � K R# 5irL rM )rN rO r[ s & �r# rP �-get_capability_match_score.<locals>.<genexpr>� s �� � �K�x��7�:�s�E�+J�!�!�x�s �'�36')�sumr )r[ r\ �matcheds f& r# �get_capability_match_scorerb � s& �� � �K�x�K�K�G� �Z�(�(� )�)r" r6 )�__conditional_annotations__r �37__future__r � dataclassesr �typingr �collections.abcr r r( r r1 rE rY rb )rc s @r# �<module>rh s� ��� 8� "� !� ��(� �$�d�#�� � $��* � .��$�)�������38 � +�,=� �!�8�������39-� 5�6G� �1�B�������407� >�?P� �5�K�������41@� :�;L� �2�G�������42<� � /� ��*�������43 � 1�2C� �(�>�������443� ,�->� �#�9�������45.�a[4� �0� [�|-�46� �D *r"