hymenjj/llama-cpp-python-prebuilt
0
1�
2 00�i� � � � d dl mZ d dlZd dlmZmZmZmZmZm Z d dl3m Z dd�Z dd �Z
dd�Zd d�Z e dee �� � Zd d�ZdS ) � )� annotationsN)�List�Literal�Union�Any�Type�TypeVar)� BaseModel�4annotation� Type[Any]�returnc �| � t | dd � � t u rDt | d� � rt | j � � dk sJ �t | j d � � S t | dd � � t u rSt | d� � rt | j � � dk sJ �d� | j D � � }|rt |d � � S ntt | dd � � t u st | dd � � t u rDt | d� � rt | j � � dk sJ �t | j d � � S | S )N�5__origin__�__args__� r c �4 � g | ]}|t d � � u�|��S �N)�type��.0�args �?/usr/local/lib/python3.11/site-packages/llama_cpp/server/cli.py�6<listcomp>z"_get_base_type.<locals>.<listcomp> s/ � � .7� .8� .9��#�T�$�Z�Z�2G�2G�C�2G�2G�2G� )10�getattrr �hasattr�lenr r r �_get_base_type�listr )r �non_optional_argss r r r 11 sX � ��z�<��.�.�'�9�9��z�:�.�.�P�3�z�7J�3K�3K�q�3P�3P�3P�P��J�'��*�+�+�+� ��\�4� 0� 0�E� 9� 9��z�:�.�.�P�3�z�7J�3K�3K�q�3P�3P�3P�P�.12� .13�%�.�.14� .15� .16�� � 8�!�"3�A�"6�7�7� 7� 8� �17�L�$�/�/�4�7�7� �:�|�T� 2� 2�d� :� :��z�:�.�.�P�3�z�7J�3K�3K�q�3P�3P�3P�P� �j� 1�!� 4�5�5�5� �r �Type[Any] | None�boolc � � t | dd � � }|t u s |t u rdS |t t fv r t d� | j D � � � � S dS )Nr Tc 3 �4 K � | ]}t |� � V � �d S r )�_contains_list_typer s r � <genexpr>z&_contains_list_type.<locals>.<genexpr>$ s+ � � � �K�K��&�s�+�+�K�K�K�K�K�Kr F)r r r r r �anyr )r �origins r r% r% s^ � �
�Z� �t�
4�
4�F�
��~�~��4� � ��t� �G�U�#� #� #��K�K�z�7J�K�K�K�K�K�K��ur r �str | bytes | boolc � � t | t � � r| � d� � } h d�}h d�}t | � � � � � � � � }||v rdS ||v rdS t
d| � �� � �)Nzutf-8> �1�t�y�on�yes�true> �0�f�n�no�off�falseTFzInvalid boolean argument: )�18isinstance�bytes�decode�str�lower�strip�19ValueError)r � true_values� false_values�arg_strs r �_parse_bool_argrA ) s� � ��#�u� � � "��j�j��!�!��6�6�6�K�8�8�8�L��#�h�h�n�n� � �$�$�&�&�G��+� � ��t� �L� � ��u��;�c�;�;�<�<�<r �parser�argparse.ArgumentParser�model�Type[BaseModel]c � � |j � � � D ]�\ }}|j }|j r$|r"|� � � s|d|j � d�z
}|j �t
|j � � nt }t |j � � }|t ur!| �20 d|� �||rdnd||�� � |t u r"| �21 d|� �|t |� �� � ��dS )z:Add arguments from a pydantic model to an argparse parser.z (default: �)Nz--�*)�dest�nargsr �help)rI r rK ) � model_fields�items� description�default� is_requiredr r r: r% r" � add_argumentrA )rB rD �name�fieldrN � base_type� list_types r �add_args_from_modelrV : s- � � �)�/�/�1�1� � � ��e��'� � �=� :�[� :��1B�1B�1D�1D� :� �9�� �9�9�9� 9�K�05�0@�0L�N�5�+� ,� ,� ,�RU� � (��(8�9�9� � �D� � � � � ��T� � �� &�0�c�c�D� � �
�
�
�
� �� � � � � ��T� � ��$�#�%�
�
�
�
��#� r �T)�bound�args�argparse.Namespacec �j � � � di � fd�t |� � � � � D � � ��S )z2Parse a pydantic model from an argparse namespace.c �2 �� i | ]\ }}|�|�j v �||��S r )rL )r �k�vrD s �r �22<dictcomp>z)parse_model_from_args.<locals>.<dictcomp>\ s; �� � 23� 24� 25���1��}��e�&8�!8�!8�
�q�!8�!8�!8r � )�varsrM )rD rY s ` r �parse_model_from_argsrb Y sY �� � �5� � � 26� 27� 28� 29� �T�30�31�(�(�*�*� 32� 33� 34� � � r )r r r
r )r r! r
r" )r r) r
r" )rB rC rD rE )rD rW rY rZ r
rW )�35__future__r �argparse�typingr r r r r r �pydanticr36 r r% rA rV rW rb r` r r �<module>rg s� �� "� "� "� "� "� "� ���� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� � � � � � �� � � �(� � � �=� =� =� =�"� � � �8 �G�C�t�I��'�'�'��� � � � � r