| Server IP : 157.230.181.24 / Your IP : 216.73.217.11 Web Server : Apache/2.4.58 (Ubuntu) System : Linux conductive 6.8.0-117-generic #117-Ubuntu SMP PREEMPT_DYNAMIC Tue May 5 19:26:24 UTC 2026 x86_64 User : ( 1000) PHP Version : 8.3.31 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : OFF Directory : /proc/self/root/opt/aim-bots/__pycache__/ |
Upload File : |
�
��Yj% � �l � d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z ddl
m Z d
d�Z G d� d � Z
y) u9 Fuzzy retrieval over archived AIM user→assistant pairs.� )� annotationsN)�SequenceMatcher)�Path)� is_followupc � � t j dd| j � j � � } t j dd| � S )Nz\s+� z
[^\w\s!?=.\-]� )�re�sub�lower�strip)�texts � /opt/aim-bots/reply_retriever.py� normalizer s7 � �
�6�6�&�#�t�z�z�|� 1� 1� 3� 4�D�
�6�6�"�B�� -�-� c �R � e Zd ZdZdd d�Zd
d d�Zd d�Z d
dd�Zddd�Zy)�ReplyRetrieverzFMap incoming lines to archived replies when the match is close enough.c � � || _ |j � sg | _ g | _ y t j
|dd�� 5 }t
j |� | _ d d d � | j D �cg c]6 }|j d� s�|j d� s�'t |d � |f��8 c}| _ y # 1 sw Y �[xY wc c}w )N�rtzutf-8)�encoding�user� assistant)
� min_ratio�exists�pairs�index�gzip�open�json�load�getr )�self�
pairs_pathr �fh�ps r �__init__zReplyRetriever.__init__ s� � �"���� � �"�%'�D�J�13�D�J� �
�Y�Y�z�4�'�
:� '�b� ���2� �D�J� '�9=���l�A�q�u�u�V�}�YZ�Y^�Y^�_j�Yk�y��6��+�Q�/�l��
� '� '��ls �B4�0C �C �C �4B=c �v � t |� }|r | j sg S g }t |� }| j D ]k \ }}t t |� |z
� t d|dz � kD r�.t d ||� j
� }|| j k\ s�Y|j ||f� �m |j d� �� |d | S )N� � c � � | d S �Nr � ��xs r �<lambda>z&ReplyRetriever.match.<locals>.<lambda>, � � �1�Q�4�%� r ��key)
r r �len�abs�maxr �ratior �append�sort) r" �message�limit�q�scored�qlen�norm�pairr6 s r �matchzReplyRetriever.match s� � � �g� ����
�
��I�+-���1�v�� �*�*� -�J�D�$��3�t�9�t�#�$�s�2�t�a�x�'8�8��#�D�!�T�2�8�8�:�E�����&��
�
�u�d�m�,� -� � � �� �(��f�u�~� r c
�. � |rt |� dk ryd }t |d d � D ]8 }|j d� dk( s�t |d j d� d � } n |syt d |t |j dd� � � j
� d z S )
Nr) g ����roler �content�
r g{�G�z�?)r3 �reversedr! r �splitr r6 )r" �historyr? � prev_user�msgs r �_context_boostz ReplyRetriever._context_boost/ s� � ��#�g�,��*��� ��G�C�R�L�)� �C��w�w�v��&�(� %�c�)�n�&:�&:�4�&@��&D� E� �� � �� �t�Y� �$�(�(�6�2�:N�0O�P�V�V�X�[_�_�_r Nc � � t |� ry | j |d� }|sy g }|D ]. \ }}| j ||� }|j ||z ||f� �0 |j d� �� |d \ } }
} t |j
� � dk } |r | j nd}
|
dk\ r| d S | r,|
|
k\ r't |j
� � d k r|
d
k ry | d S |
d k\ rt |� dk\ r| d S |r|
| j k\ r| d S y ) N� c � � | d S r+ r, r- s r r/ z+ReplyRetriever.best_reply.<locals>.<lambda>N r0 r r1 r � g�(\���?g��Q� ��?r r) g)\��(�?g�������?)r r@ rK r7 r8 r3 rG r )r" r9 rH �relaxed�hitsr< r6 r? �boost�
best_total�
best_ratio�best�short� thresholds r �
best_replyzReplyRetriever.best_reply< s' � � �w� ���z�z�'�1�%������ � 8�K�E�4��'�'���6�E� �M�M�5�5�=�%��6� 7� 8� � � �� �(�'-�a�y�$�
�J���G�M�M�O�$��)��&-�D�N�N�4� � �� �� �$� $� �Z�9�,��7�=�=�?�#�q�(�Z�$�->��� �$� $� �� �#�g�,�!�"3�� �$� $� �z�T�^�^�3�� �$� $�r c �� � | j |d� }|r |d d dk ryg }|D ]3 \ }}|dk r n)|j d|d�d|d dd � d
|d dd � �� �5 |syd }|rft |� d
k\ rX|dd D �cg c]1 }|j d� dk( s�|d j d� d dd ��3 } }| rddj | dd � � �}d|� d�dj |dd � z S c c}w )zESimilar archived exchanges to steer the LLM without copying verbatim.rO r g�p=
ף�?Nz- (similarity z.0%z
) they said: r �P u → you said: r r r) ���rC rD rE rB �< z
Recent thread from Caleb: z | ���u^ Similar moments from your real AIM logs (match the rhythm and length — write something NEW):� )r@ r7 r3 r! rG �join)
r" r9 rH rQ �linesr6 r? �ctx�m�recents
r � hint_for_llmzReplyRetriever.hint_for_llm` s> � ��z�z�'�1�%���t�A�w�q�z�D�(���� � �K�E�4��t�|�� �L�L� ��s� �=��f� �c�r�9J�8K�?�[_�`k�[l�mp�np�[q�Zr�s�
� � ���� �s�7�|�q�(�AH��� �i�A�QR�QV�QV�W]�Q^�bh�Qh�a� �l�(�(��.�r�2�3�B�7�i�F�i��4�U�Z�Z��r�s� �5L�4M�N�� l��e�2�
� ���5��!�9�-�
.�
�� js �7C#� C#)g
ףp=
�?)r# r r �float�return�None)rO )r9 �strr: �intrf zlist[tuple[float, dict]])rH �list[dict] | Noner? �dictrf re )NF)r9 rh rH rj rP �boolrf �
str | None)N)r9 rh rH rj rf rm ) �__name__�
__module__� __qualname__�__doc__r& r@ rK rX rd r, r r r r sO � �P�m�
� `� &*� � "��"� #�"� � "�
� "�H
r r )r rh rf rh )rq �
__future__r r r r
�difflibr �pathlibr � coherencer r r r, r r �<module>rv s. �� ?� "� � � � #� � !�.�
c
� c
r