ashnaali22/phase-3-h2
0
1�
2 ��Ji
� �v � S r SSKrSSKrSSKJr SSKJr SSKJr \R " \35 r " S S\5 r g)z�4ChatKit Adapter Middleware5 6Translates ChatKit protocol requests to custom endpoint format.7 8Converts:9 {"type": "threads.create", "params": {"input": {"content": "Hello"}}}10To:11 {"message": "Hello", "conversation_id": null}12� N��Request)�BaseHTTPMiddleware)�ASGIAppc �B ^ � \ rS rSrSrS\4U 4S jjrS\4S jrSr U =r13$ )�ChatKitAdapterMiddleware� a$ 14Middleware to adapt ChatKit protocol requests to custom format.15 16ChatKit sends requests in its protocol format with types like:17- threads.create18- threads.add_user_message19- threads.list20etc.21 22This middleware translates those to the custom {"message": "...", "conversation_id": ...} format.23�appc �$ >� [ TU ] U5 g )N)�super�__init__)�selfr24 � __class__s ��QD:\code\Q4\hackathon-2\todo-app\phase-3\backend\src\middleware\chatkit_adapter.pyr
�!ChatKitAdapterMiddleware.__init__" s �� �
���� � �requestc � �� ^# � UR R S:X Ga� UR S:X Ga� UR 5 I S h v�N n [ R25 " U5 nUR
S5 n[ R SU 35 US:X aU UR
S0 5 nUR
S0 5 nUR
SS 5 nS n [ R S26US S S 35 O�US
:X a� UR
S0 5 nUR
S0 5 nUR
SS 5 nUR
S5 n27U28( a! U29R 5 ( a [ U305 n OS n [ R SU31 SUS S S 35 O)[ R SU S35 U" U5 I S h v�N $ UU S.m[ R ST 35 U4S jn SSK J
n UR n
U4S jnX�l U" U5 I S h v�N nU$ GN� NZ! [ R a" n[ R! SU 35 S nANHS nAf[" a! n[ R! SU 3SS 9 S nANqS nAff = f Nm7f) Nz /api/v1/chat�POST�typez([ChatKitAdapter] Received request type: zthreads.create�params�input�content� z+[ChatKitAdapter] threads.create - message: �2 z...zthreads.add_user_message� thread_idz7[ChatKitAdapter] threads.add_user_message - thread_id: z , message: z+[ChatKitAdapter] Unsupported request type: z, passing through unchanged)�message�conversation_idz [ChatKitAdapter] Translated to: c � �Z ># � S[ R " T 5 R 5 S.$ 7f�Nz http.request)r �body��json�dumps�encode��32custom_reqs �r �receive�2ChatKitAdapterMiddleware.dispatch.<locals>.receiveZ �( �� � � .� $�33�34�:� 6� =� =� ?� � �� �(+r r c � �Z ># � S[ R " T 5 R 5 S.$ 7fr r" r& s �r �custom_receive�9ChatKitAdapterMiddleware.dispatch.<locals>.custom_receivef r* r+ z'[ChatKitAdapter] Failed to parse JSON: z+[ChatKitAdapter] Error processing request: T)�exc_info)�url�path�methodr! r# �loads�get�logger�info�isdigit�int�warning�fastapir �_receive�JSONDecodeError�error� Exception)r r � call_next�35body_bytes� chatkit_req�req_typer �36input_datar r r r( �FastAPIRequest�original_receiver- �e�responser' s @r �dispatch�!ChatKitAdapterMiddleware.dispatch% sQ �� � � �;�;� � �~� -�'�.�.�F�2J� &�|�|�~�-�J�G
_� "�j�j�� 4� �&�?�?�6�2��� � � F�x�j� Q�R� � /�/� (�_�_�X�r� :�F�!'���G�R�!8�J� (�n�n�Y�� ;�G�&*�O��K�K�"M�g�VY�WY�l�^�[^� _�`� �!;�;� (�_�_�X�r� :�F�!'���G�R�!8�J� (�n�n�Y�� ;�G� &�37�38�;� 7�I� !�Y�%6�%6�%8�%8�*-�i�.��*.���K�K�"Y�Zc�Yd�do�pw�x{�y{�p|�o}� ~A� !B� C� �N�N�%P�QY�PZ�Zu�#v�w�!*�7�!3�3�3� '�'6� �39�40 � � � >�z�l� K�L�� >� $+�#3�#3� �� $2� � #�7�+�+����] .�J 4��@ �'�'�
L�� � � F�q�c� J�K�K�� �
_�� � � J�1�#� N�Y]� �^��
_��41 ,�sz �A I �G6�I � EG; � G9� G; �"I �#AG; �$ I �/I �0I �9G; �;I�H,�'I �,
I�9I�I �I�I � ) �__name__�42__module__� __qualname__�__firstlineno__�__doc__r r
r rH �__static_attributes__�
__classcell__)r s @r r r s( �� �43� �G� �R�g� R� Rr r )
rO r# �loggingr: r �starlette.middleware.baser �starlette.typesr � getLoggerrK r5 r rJ r r �<module>rV s: �� � � � � 8� #� � � �8� $��b� 1� br