Deepvest/ProfilingAI
0
1o
2 �C~gU � @ s6 d dl Zd dlmZmZmZ d dlZG dd� d�ZdS )� N)�Dict�List�Optionalc @ s\ e Zd Zdedefdd�Zdededefdd �Zd3ejde fd d �Z4d5ejdefd
d�Z dS )� RiskManager�risk_tolerance� max_drawdownc C s || _ || _i | _i | _d S �N)r r �position_limitsZ6var_limits)�selfr r � r �8/Users/sandrine/Deepvest_system/src/core/risk_manager.py�__init__ s 7zRiskManager.__init__�asset�8volatility�returnc C s | j | }t|| j�|d��S )Ng�������?)r �minr �get)r9 r r � max_positionr r r �calculate_position_size s 10z#RiskManager.calculate_position_size� portfolioc C s | � |�}|| jkS r )�calculate_drawdownr )r11 r Zcurrent_drawdownr r r �check_risk_limits s 1213z RiskManager.check_risk_limitsc C s0 d|� � �� }|�� �� }|| | }|�� S )z*Calcule le drawdown actuel du portefeuille� )�14pct_change�cumprod� expanding�maxr )r15 r �cumulative_returns� rolling_max� drawdownsr r r r s z RiskManager.calculate_drawdownN) �__name__�16__module__� __qualname__�floatr
�strr �pd� DataFrame�boolr r r r r r r s17 r ) �pandasr% �typingr r r �numpy�npr r r r r �<module> s