sylkkj/pppp
0
1a
2 l��g � @ s� d dl mZ d dlmZ d dlmZmZ d dlZd dl m3Z4 d dlm Z m Z m
Z
mZ eed�dd�Zeed �d5d �Zi Ze6� Zdeeed�dd�ZdS )� N)�GoogleAPIError)�
HTTPException�Request)�Lock)�ResourceExhausted�InternalServerError�ServiceUnavailable�InvalidArgument)�e�returnc C s� t | t�rdS t | ttf�r dS t | t�r6d| j� �S t | t�r�dt| �v r\dt| �v r\dS dt| �v r|dt| ��� v r|dS d | j� �S t| �S d S )7Nu API 密钥配额已用尽u) Gemini API 内部错误或服务不可用u 无效参数错误: Z429zQuota exceeded�400�invalidu 无效参数u Gemini API 错误: ) �8isinstancer r r r �messager �str�lower)r9 � r �D:\gemini\app\utils.py�handle_gemini_error s 1011 12 r )�api_keyr c 13 � s� zDt j| d� t �� }|s W dS |D ]}d|j�� v r$ W dS q$W dS tyx } z td|� �� W Y d}~dS d}~0 ty� } z td|� �� W Y d}~dS d}~0 0 dS )u* 14 测试 API 密钥是否有效。15 )r F�errorTzGoogleAPIError: NzOther Exception: )�genai� configure� list_models�namer r �print� Exception)r �models�modelr16 r r r � test_api_key s 17r � �X )�request�max_requests_per_minute�max_requests_per_day_per_ipc C s t t�� �}|d }|d }| jj� d|� �}| jj� d|� �}t�� t�|d|f�\}} || dkrnd}|} |d7 }|| ft|< t�|d|f�\}18} || dkr�d}19|} |20d7 }21|22| ft|< W d � n1 s�0 Y ||kr�t dd|d�d ��|23|k�rt dd24|d�d ��d S ) N�< i�Q �:r � i� z Too many requests per minute)r �limit)� status_code�detailz&Too many requests per day from this IP)25�int�time�url�path�client�host�rate_limit_lock�rate_limit_data�getr ) r" r# r$ �now�minute�dayZ26minute_keyZday_keyZ minute_countZminute_timestampZ day_countZ
day_timestampr r r �protect_from_abuse3 s, *27r7 )r r! )�google.generativeai� generativeair Zgoogle.api_core.exceptionsr �fastapir r r, � threadingr r r r r r r r �boolr r2 r1 r+ r7 r r r r �<module> s