
    i|                         d dl mZ d dlmZ d dlmZmZmZmZm	Z	m
Z
 dedefdZdedefdZdedefdZdede
fd	ZddededefdZddedede	fdZdS )   )AsyncHttpClient)handle_response_error)ConcurrencyCheckCreditUsage
TokenUsageCreditUsageHistoricalResponseTokenUsageHistoricalResponseQueueStatusResponseclientreturnc           	        K   |                      d           d {V }|j        dk    rt          |d           |                                }|                     d          s#t	          |                     dd                    |                     d|          }t          |                     d          |                     d	|                     d
                              S )Nz/v2/concurrency-check  zget concurrencysuccesserrorUnknown errordataconcurrencymaxConcurrencymax_concurrency)r   r   )getstatus_coder   json	Exceptionr   r   respbodyr   s       h/home/agentuser/.hermes/hermes-agent/venv/lib/python3.11/site-packages/firecrawl/v2/methods/aio/usage.pyget_concurrencyr      s      344444444D3d$566699;;D88I </::;;;88FD!!DHH]++!1488<M3N3NOO       c                   K   |                      d           d {V }|j        dk    rt          |d           |                                }|                     d          s#t	          |                     dd                    |                     d|          }t          |                     d|                     d	d
                    |                     d|                     d                    |                     d|                     d                    |                     d|                     d                              S )Nz/v2/team/credit-usager   zget credit usager   r   r   r   remainingCreditsremaining_credits    planCreditsplan_creditsbillingPeriodStartbilling_period_startbillingPeriodEndbilling_period_end)r"   r%   r'   r)   )r   r   r   r   r   r   r   s       r   get_credit_usager*      s-     344444444D3d$677799;;D88I </::;;;88FD!!D((#5txx@SUV7W7WXXXXmTXXn-E-EFF!XX&:DHHE[<\<\]]88$6AU8V8VWW	   r   c                   K   |                      d           d {V }|j        dk    rt          |d           |                                }|                     d          s#t	          |                     dd                    |                     d|          }t          |                     d|                     d	d
                    |                     d|                     d                    |                     d|                     d                    |                     d|                     d                              S )Nz/v2/team/token-usager   zget token usager   r   r   r   remainingTokensremaining_tokensr#   
planTokensplan_tokensr&   r'   r(   r)   )r-   r/   r'   r)   )r   r   r   r   r   r   r   s       r   get_token_usager0   $   s-     233333333D3d$566699;;D88I </::;;;88FD!!D"3TXX>PRS5T5TUUHH\488M+B+BCC!XX&:DHHE[<\<\]]88$6AU8V8VWW	   r   c           
      &  K   |                      d           d {V }|j        dk    rt          |d           |                                }|                     d          s#t	          |                     dd                    |                     d|          }t          |                     dd	          |                     d
d	          |                     dd	          |                     dd	          |                     dd                     S )Nz/v2/team/queue-statusr   zget queue statusr   r   r   r   jobsInQueuer#   activeJobsInQueuewaitingJobsInQueuer   mostRecentSuccess)jobs_in_queueactive_jobs_in_queuewaiting_jobs_in_queuer   most_recent_success)r   r   r   r   r   r
   r   s       r   get_queue_statusr:   4   s	     344444444D3d$677799;;D88I </::;;;88FD!!Dhh}a00!XX&91=="hh';Q??!1155 HH%8$??   r   F
by_api_keyc                 4  K   |rdnd}|                      d|            d {V }|j        dk    rt          |d           |                                }|                     d          s#t	          |                     dd                    t          d	i |S )
N?byApiKey=true z /v2/team/credit-usage/historicalr   zget credit usage historicalr   r   r    )r   r   r   r   r   r   r   r;   queryr   r   s        r   get_credit_usage_historicalrB   E   s       *2EFuFFGGGGGGGGD3d$ABBB99;;D88I </::;;;(004000r   c                 4  K   |rdnd}|                      d|            d {V }|j        dk    rt          |d           |                                }|                     d          s#t	          |                     dd                    t          d	i |S )
Nr=   r>   z/v2/team/token-usage/historicalr   zget token usage historicalr   r   r   r?   )r   r   r   r   r   r	   r@   s        r   get_token_usage_historicalrD   P   s       *2EEeEEFFFFFFFFD3d$@AAA99;;D88I </::;;;'//$///r   N)F)utils.http_client_asyncr   utils.error_handlerr   typesr   r   r   r   r	   r
   r   r*   r0   r:   boolrB   rD   r?   r   r   <module>rI      s{   6 6 6 6 6 6 8 8 8 8 8 8 Q  Q  Q  Q  Q  Q  Q  Q  Q  Q  Q  Q  Q  Q  Q  Q/ 6F    ? {     / j     ? 7J    "1 1o 14 1\y 1 1 1 10 0_ 0$ 0[w 0 0 0 0 0 0r   