Tribh/devops-copilot
0
1�
2 њi� � �� � d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl3m Z ddl m
Z
ddlmZ dd lZ ed4d d �
� � Z ej dd� � � dd� � Z e e�� � Ze� d� � d� � � Z G d� de� � Z G d� de� � Ze� d� � d� � � Ze� d� � d � � � Z e� d � � d e fd �� � Z e� d � � e� � fd e d!efd"�� � Z e� d#� � e� � fd e d!efd$�� � Z!d S )%u� 5api.py — FastAPI server exposing DevOps Copilot control plane.6 7Endpoints:8 POST /sessions/{session_id}/approve — Grant human approval (unblocks engine)9 POST /sessions/{session_id}/reject — Reject proposed action (marks false positive)10 GET /sessions/{session_id} — Inspect current session state11 GET /sessions — List all active sessions12 GET /health — Liveness probe13� )�FastAPI�
HTTPException)� JSONResponse)� BaseModel)�Optional)�PersistenceLayer)�FALSE_POSITIVE_TOTAL)�loggerNz AgentNexus DevOps Copilot APIuI Control plane for the AI DevOps Copilot — approve/reject agent actions.z1.0.0)�title� description�version� DATABASE_URLzagentnexus_state.dbz sqlite:///./� )�db_path�startupc � �p K � t � � � � d {V �� t j d� � d S )NzAPI server started.)� persistence�setupr14 �info� � �H/Users/tribhangind/Documents/GitHub/devops-copilot/devops_copilot/api.py�_startupr sA � � � �15�16�17�18 �19 � � � � � � � �20�K�%�&�&�&�&�&r c �* � e Zd ZU dZee ed< dS )�ApproveRequestzApproved via API�reasonN��__name__�21__module__� __qualname__r r �str�__annotations__r r r r r ' s& � � � � � � � .�F�H�S�M�.�.�.�.�.r r c �* � e Zd ZU dZee ed< dS )�
RejectRequestu Rejected — false positiver Nr r r r r$ r$ + s& � � � � � � � 9�F�H�S�M�9�9�9�9�9r r$ z/healthc � � K � ddiS )N�status�okr r r r �healthr( 1 s � � � � �d� �r z /sessionsc � �h K � t � � � � d {V ��} | t | � � d�S )N)�sessions�count)r �
list_sessions�len)r* s r r, r, 6 s= � � � � �.�.�0�0�0�0�0�0�0�0�H� �3�x�=�=� 9� 9�9r z/sessions/{session_id}�22session_idc � �x K � t � | � � � d {V ��}|�t dd| � d��� � �|S )N� � Session '� ' not found.�� status_code�detail)r � load_sessionr )r. �states r � get_sessionr8 < sT � � � � �*�*�:�6�6� 6� 6� 6� 6� 6� 6�E� �}���4X�23�4X�4X�4X�Y�Y�Y�Y� �Lr z /sessions/{session_id}/approve�bodyc � �� K � t � | � � � d{V �� n0# t $ r#}t dt |� � �� � �d}~ww xY wt j d| � d|j � �� � | d|j d�S )u� 24 Grant human approval for a PENDING_APPROVAL action.25 The engine polls session state — setting human_approved=True unblocks it.26 Nr0 r3 zApproved session �: �approved�r. r& r )r � set_approved�KeyErrorr r! r27 r r )r. r9 �es r �approve_actionrA D s� � � � � <��&�&�z�2�2�2�2�2�2�2�2�2�2�� � <� <� <���C��F�F�;�;�;�;�����<����28�K�?�J�?�?�$�+�?�?�@�@�@�$�29�d�k� R� R�Rs � % �30A� A
�
Az /sessions/{session_id}/rejectc � �x K � t � | � � � d{V ��}|�t dd| � d��� � �t j � � d|� di � � d< d |d d31< t � | |� � � d{V �� t j d | � d |j � �� � | d32|j d
�S )zQ33 Reject a pending action and track it as a false positive in Prometheus.34 Nr0 r1 r2 r3 F�metadata�human_approvedT�rejectedzRejected session r; r= )35r r6 r r �inc�36setdefault� save_sessionr37 �warningr )r. r9 r7 s r �
reject_actionrJ R s� � � � �38 �*�*�:�6�6� 6� 6� 6� 6� 6� 6�E� �}���4X�39�4X�4X�4X�Y�Y�Y�Y� � � � � � :?�E���Z��$�$�%5�6�$(�E�*��j�!�40�41"�42"�:�u�435�445�5�5�5�5�5�5�5�45�N�B�z�B�B�T�[�B�B�C�C�C�$�46�d�k� R� R�Rr )"�__doc__�fastapir r �fastapi.responsesr �pydanticr �typingr � devops_copilot.core.persistencer �!devops_copilot.core.observabilityr �devops_copilot.utils.loggerr47 �os�app�getenv�replace�_db_pathr �on_eventr r r$ �getr( r, r! r8 �postrA rJ r r r �<module>r[ s� �� � � +� *� *� *� *� *� *� *� *� *� *� *� *� *� � � � � � � � � � � � � <� <� <� <� <� <� B� B� B� B� B� B� .� .� .� .� .� .� � � � �
�g�48)�[� �� � �� �2�9�^�%:� ;� ;� C� C�N�TV� W� W�� � �x�0�0�0� � � � �i���'� '� ��'�/� /� /� /� /�Y� /� /� /�:� :� :� :� :�I� :� :� :� ������ � � �� � ������:� :� ��:�49 ��� !�"�"��#� � � � #�"�� ���50*�+�+�AO� �AQ�AQ� 51S� 52S�S� 53S�� 54S� 55S� 56S� ,�+�57S� ���58)�*�*�?L�}��� S� S�C� S�}� S� S� S� +�*�S� S� Sr