o ưi@stdZddlmZmZmZmZmZddlZddlZddl m Z ddl m Z ddl mZddlmZGdd d eZdS) z` Anthropic CountTokens API handler. Uses httpx for HTTP requests instead of the Anthropic SDK. )AnyDictListOptionalUnionN)verbose_logger)AnthropicError)AnthropicCountTokensConfig)get_async_httpx_clientc@s~eZdZdZ    d dedeeeefdedeedee e e j fdeeeeefd eed eeeffd d Z dS)AnthropicCountTokensHandlerz Handler for Anthropic CountTokens API requests. Uses httpx for HTTP requests, following the same pattern as BedrockCountTokensHandler. Nmodelmessagesapi_keyapi_basetimeouttoolssystemreturnc sz}|||td||j||||d}td||p&|} td| ||} ttjj d} |durA|ntj } | j | | || dIdH} td| j | j d krp| j }td |t| j |d | }td ||WStytjy}ztd t|t|jj |jj d d}~wty}ztdt|tddt|d d}~ww)a Handle a CountTokens request using httpx. Args: model: The model identifier (e.g., "claude-3-5-sonnet-20241022") messages: The messages to count tokens for api_key: The Anthropic API key api_base: Optional custom API base URL timeout: Optional timeout for the request (defaults to litellm.request_timeout) Returns: Dictionary containing token count response Raises: AnthropicError: If the API request fails z4Processing Anthropic CountTokens request for model: )r r rrzTransformed request: zMaking request to: )Z llm_providerN)headersjsonrzResponse status: zAnthropic API error: ) status_codemessagezAnthropic response: z#HTTP error in CountTokens handler: zError in CountTokens handler: izCountTokens processing error: )Zvalidate_requestrdebugZ!transform_request_to_count_tokensZ#get_anthropic_count_tokens_endpointZget_required_headersr litellmZ LlmProvidersZ ANTHROPICrequest_timeoutpostrtexterrorrrhttpxZHTTPStatusErrorstrresponse Exception)selfr r rrrrr request_bodyZ endpoint_urlrZ async_clientrr!Z error_textZanthropic_responseer&b/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/anthropic/count_tokens/handler.pyhandle_count_tokens_requestsl       z7AnthropicCountTokensHandler.handle_count_tokens_request)NNNN)__name__ __module__ __qualname____doc__r rrrrrfloatrTimeoutr(r&r&r&r'r s.  r )r,typingrrrrrrrZlitellm._loggingrZ#litellm.llms.anthropic.common_utilsrZ2litellm.llms.anthropic.count_tokens.transformationr Z&litellm.llms.custom_httpx.http_handlerr r r&r&r&r's