method_cache.cpython-311.pyc•7.19 kB
�
�L�g� � � � d dl Z d dlZd dlmZ d dlmZmZmZ d dlZddl m
Z
G d� d� � Z G d� d � � Z G d
� d � � Z
dS ) � N)�Path)�Dict�Optional�Tuple� )�FunctionDefinitionc �V � e Zd Zdefd�Zdedefd�Zdedee fd�Z dedefd�Z
d S )
� MethodCache� cache_dirc � � || _ | j � d�� � t j t � � | _ d S )NT)�exist_ok)r �mkdir�logging� getLogger�__name__�logger)�selfr s �8/Users/arjun/repos/sparkmango/mcp_server/method_cache.py�__init__zMethodCache.__init__ s8 � �"��� �� � �d� �+�+�+� �'��1�1�� � � � �function�returnc � � t j |j d� |j D � � d� |j D � � |j j d�d�� � }t j |� � � � � �
� � S )z+Generate a unique cache key for a function.c �* � g | ]}|j |j f��S � ��name�type��.0�ps r �
<listcomp>z.MethodCache._get_cache_key.<locals>.<listcomp> s! � �A�A�A�A�����'�A�A�Ar c �* � g | ]}|j |j f��S r r r s r r" z.MethodCache._get_cache_key.<locals>.<listcomp> s! � �C�C�C�Q�����(�C�C�Cr )r �inputs�outputs�state_mutabilityT)� sort_keys) �json�dumpsr r$ r% r& �value�hashlib�sha256�encode� hexdigest)r r �func_strs r �_get_cache_keyzMethodCache._get_cache_key s� � � �:� �M�A�A�� �A�A�A�C�C�(�2B�C�C�C� (� 9� ?�
�
�
� � � �� �~�h�o�o� /� /�0�0�:�:�<�<�<r c �" � | � |� � }| j |� d�z }|� � � r6| j � d|j � �� � |� � � S | j � d|j � �� � dS )z'Get cached implementation if it exists.�.pyzCache hit for function zCache miss for function N)r0 r �existsr �infor � read_text)r r � cache_key�
cache_files r �get_cached_implementationz%MethodCache.get_cached_implementation s� � ��'�'��1�1� ��^��&7�&7�&7�7�
� � � � � � *� �K� � � F�x�}� F� F� G� G� G� �'�'�)�)� )� � ���C�H�M�C�C�D�D�D��tr �implementationc � � | � |� � }| j |� d�z }|� |� � | j � d|j � �� � dS )z!Cache a generated implementation.r2 z#Cached implementation for function N)r0 r �
write_textr r4 r )r r r9 r6 r7 s r �cache_implementationz MethodCache.cache_implementation& se � ��'�'��1�1� ��^��&7�&7�&7�7�
�� � �n�-�-�-� � ���N�x�}�N�N�O�O�O�O�Or N) r �
__module__� __qualname__r r r �strr0 r r8 r< r r r r
r
s� � � � � � �2�$� 2� 2� 2� 2�
=�'9�
=�c�
=�
=�
=�
=�
�2D�
��RU� �
�
�
�
�P�-?� P�QT� P� P� P� P� P� Pr r
c �8 � e Zd Zd� Zdededeeef fd�ZdS )�MethodValidatorc �B � t j t � � | _ d S )N)r r r r �r s r r zMethodValidator.__init__/ s � � �'��1�1�� � � r r r9 r c �v � g d�}|j j dk r|� d� � n|� d� � |D ]}||vr dd|� �fc S �d|j � d�}|� |� � sd S |j D ] }|j � d
�|vrdd |j � �fc S � |j j dk rd |vrd
S n
d|vrd|vrdS dS )z`
Validate a generated implementation.
Returns (is_valid, error_message)
)z async defzweb3.eth.contractztry:zexcept Exception as e:�view�build_transactionzcall()Fz Missing required component: z
async def z
(state: State)FzInvalid function signature�:zMissing parameter: zreturn {)Fz&View function must return a dictionaryz "type": "transaction_to_sign"z 'type': 'transaction_to_sign')Fz7State-changing function must return transaction_to_sign)T� )r& r* �appendr �
startswithr$ )r r r9 �required_components� component�expected_signature�params r �validate_implementationz'MethodValidator.validate_implementation2 s\ � �
�
�
�� � $� *�f� 4� 4� � &� &�':� ;� ;� ;� ;� � &� &�x� 0� 0� 0� -� I� I�I���.�.� � H�Y� H� H�H�H�H�H� /� G�(�-� F� F� F�� �(�(�);�<�<� 7�6�6� �_� A� A�E��*� � � �~�5�5� � @�E�J� @� @�@�@�@�@� 6� � $� *�f� 4� 4�� �/�/�F�F� 0� /�n�D�D�Ih�p~�I~�I~�W�W��xr N) r r= r> r r r? r �boolrO r r r rA rA . s[ � � � � � �2� 2� 2�+�0B� +�TW� +�\a�bf�hk�bk�\l� +� +� +� +� +� +r rA c �, � e Zd Zd� Zdefd�Zdefd�ZdS )�LLMMeterc �^ � d| _ d| _ t j t � � | _ d S )Nr )� total_tokens�total_requestsr r r r rC s r r zLLMMeter.__init__` s( � � ��� ��� �'��1�1�� � � r �tokensc � � | xj |z
c_ | xj dz
c_ | j � d|� d| j � d| j � d�� � dS )zRecord LLM usage.r z LLM Usage: z tokens (Total: z tokens, z
requests)N)rT rU r r4 )r rV s r � record_usagezLLMMeter.record_usagee si � � ���V�#��� ���q� ��� � ���z�v�z�z�t�?P�z�z�[_�[n�z�z�z�{�{�{�{�{r r c �V � | j | j | j dk r| j | j z ndd�S )zGet usage statistics.r )rT rU �average_tokens_per_request)rT rU rC s r �get_usage_statszLLMMeter.get_usage_statsk sA � � !� -� "� 1�UY�Uh�kl�Ul�Ul�$�*;�d�>Q�*Q�*Q�rs�
�
�
r N)r r= r> r �intrX r r[ r r r rR rR _ s^ � � � � � �2� 2� 2�
|�3� |� |� |� |�
��
�
�
�
�
�
r rR )r( r+ �pathlibr �typingr r r r � abi_analyzerr r
rA rR r r r �<module>r` s� �� � � � � ���� � � � � � � (� (� (� (� (� (� (� (� (� (� ���� ,� ,� ,� ,� ,� ,�$P� $P� $P� $P� $P� $P� $P� $P�L/� /� /� /� /� /� /� /�b
�
�
�
�
�
�
�
�
�
r