
    dhK                        S r SSKrSSKJrJr  SSKJrJrJr  SSK	J
r
  SSKJr  SSKJr  SSKJrJr  0 S	S
_SS
_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_S S_0 S!S_S"S#_S$S#_S%S&_S'S&_S(S)_S*S)_S+S,_S-S,_S.S/_S0S/_S1S2_S3S2_S4S)_S5S)_S6S7_S8S7_E0 S9S
_S:S
_S;S_S<S_S=S_S>S_S?S@_SAS@_SBSC_SDSC_SESF_SGSF_SHS@_SIS@_SJSK_SLSK_SMSF_E0 SNSF_SOS@_SPSK_SQS@_SRSK_SSSF_STSF_SUS,_SVS/_SWS,_SXS/_SYS2_SZS2_S[S\_S]S^_S_S`_SaS\_E0 SbS^_ScS\_SdS^_SeSf_SgS,_ShSf_SiSf_SjS\_SkS\_SlS\_SmSf_SnSf_SoSf_SpS`_SqS`_SrS`_SsS\_E0 StS\_SuS\_SvSw_SxSw_SySw_SzS{_S|S}_S~S{_SS}_SS_SS_SS{_SS{_SS_SS_SS_SS_E0 SS_SS_SS_SS_SS{_SS{_SS_SS_SS\_SS\_SSf_SSf_SS_SS_SS_SS_SS_E0 SS_SS_SS2_SS2_SS2_SSf_SSf_SSf_SSf_SSf_SSf_SS2_SS2_SS2_SS_SS_SS_E0 SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS_SS
_SS_SS
_E0 SS
_SS
_SS
_SS_SS_SS_SS_SS
_SS_SS_SS_SS_SS
_SS_SS
_SS
_SS
_E0 SS_SS_SS_SS_SS_SS_SS
_SS
_SSw_SSw_SSw_SS_SS_SS_SS_SS_SS_E0 SS{_SS_SS_SS_SS_SS_SS_SS_SS
_SS_SS_SS
_SS
_SS_SS_SS_SS_Er " S S\5      r GS\R(                  S.S\S\S\GS \4GS jjjr GS\R(                  S.S\GS\S\S\GS \4
GS jjjr " GS GS\5      rg(  z(Callback Handler that prints to std out.    N)Enumauto)AnyDictList)warn_deprecated)BaseCallbackHandler)	AIMessage)ChatGeneration	LLMResultgpt-4.1gMb`?zgpt-4.1-2025-04-14zgpt-4.1-cachedgMb@?zgpt-4.1-2025-04-14-cachedzgpt-4.1-completiongMb?zgpt-4.1-2025-04-14-completionzgpt-4.1-minig-C6:?zgpt-4.1-mini-2025-04-14zgpt-4.1-mini-cachedg-C6?zgpt-4.1-mini-2025-04-14-cachedzgpt-4.1-mini-completiong-C6Z?z"gpt-4.1-mini-2025-04-14-completionzgpt-4.1-nanozgpt-4.1-nano-2025-04-14zgpt-4.1-nano-cachedg-C6>zgpt-4.1-nano-2025-04-14-cachedzgpt-4.1-nano-completionz"gpt-4.1-nano-2025-04-14-completionzgpt-4.5-previewg333333?zgpt-4.5-preview-2025-02-27zgpt-4.5-preview-cachedg333333?z!gpt-4.5-preview-2025-02-27-cachedzgpt-4.5-preview-completiong333333?z%gpt-4.5-preview-2025-02-27-completiono1gQ?zo1-2024-12-17z	o1-cachedgQ~?zo1-2024-12-17-cachedzo1-completiongQ?zo1-2024-12-17-completionzo1-prozo1-pro-2025-03-19zo1-pro-completiong333333?zo1-pro-2025-03-19-completiono3zo3-2025-04-16z	o3-cachedzo3-2025-04-16-cachedzo3-completionzo3-2025-04-16-completionzo4-minig/nR?zo4-mini-2025-04-16zo4-mini-cachedg/n2?zo4-mini-2025-04-16-cachedzo4-mini-completiong/nr?zo4-mini-2025-04-16-completionzo3-minizo3-mini-2025-01-31zo3-mini-cachedg/nB?zo3-mini-2025-01-31-cachedzo3-mini-completionzo3-mini-2025-01-31-completionzo1-minizo1-mini-cachedzo1-mini-2024-09-12zo1-mini-2024-09-12-cachedzo1-mini-completionzo1-mini-2024-09-12-completionz
o1-previewzo1-preview-cachedzo1-preview-2024-09-12zo1-preview-2024-09-12-cachedzo1-preview-completionz o1-preview-2024-09-12-completiongpt-4og{Gzd?zgpt-4o-cachedg{GzT?gpt-4o-2024-05-13g{Gzt?zgpt-4o-2024-08-06zgpt-4o-2024-08-06-cachedzgpt-4o-2024-11-20zgpt-4o-2024-11-20-cachedzgpt-4o-completiong{Gz?zgpt-4o-2024-05-13-completionzgpt-4o-2024-08-06-completionzgpt-4o-2024-11-20-completionzgpt-4o-audio-previewzgpt-4o-audio-preview-2024-12-17zgpt-4o-audio-preview-2024-10-01zgpt-4o-audio-preview-completionz*gpt-4o-audio-preview-2024-12-17-completionz*gpt-4o-audio-preview-2024-10-01-completionzgpt-4o-realtime-previewz"gpt-4o-realtime-preview-2024-12-17z"gpt-4o-realtime-preview-2024-10-01zgpt-4o-realtime-preview-cachedz)gpt-4o-realtime-preview-2024-12-17-cachedz)gpt-4o-realtime-preview-2024-10-01-cachedz"gpt-4o-realtime-preview-completiong{Gz?z-gpt-4o-realtime-preview-2024-12-17-completionz-gpt-4o-realtime-preview-2024-10-01-completionzgpt-4o-miniga2U0*#?zgpt-4o-mini-cachedga2U0*?zgpt-4o-mini-2024-07-18zgpt-4o-mini-2024-07-18-cachedzgpt-4o-mini-completionga2U0*C?z!gpt-4o-mini-2024-07-18-completionzgpt-4o-mini-audio-previewz$gpt-4o-mini-audio-preview-2024-12-17z$gpt-4o-mini-audio-preview-completionz/gpt-4o-mini-audio-preview-2024-12-17-completionzgpt-4o-mini-realtime-previewz'gpt-4o-mini-realtime-preview-2024-12-17z#gpt-4o-mini-realtime-preview-cachedga2U0*3?z.gpt-4o-mini-realtime-preview-2024-12-17-cachedz'gpt-4o-mini-realtime-preview-completionga2U0*c?z2gpt-4o-mini-realtime-preview-2024-12-17-completionzgpt-4o-mini-search-previewz%gpt-4o-mini-search-preview-2025-03-11z%gpt-4o-mini-search-preview-completionz0gpt-4o-mini-search-preview-2025-03-11-completionzgpt-4o-search-previewz gpt-4o-search-preview-2025-03-11z gpt-4o-search-preview-completionz+gpt-4o-search-preview-2025-03-11-completionzcomputer-use-previewg~jth?zcomputer-use-preview-2025-03-11zcomputer-use-preview-completiong~jt?z*computer-use-preview-2025-03-11-completiongpt-4gQ?z
gpt-4-0314z
gpt-4-0613z	gpt-4-32kzgpt-4-32k-0314zgpt-4-32k-0613zgpt-4-vision-previewzgpt-4-1106-previewzgpt-4-0125-previewzgpt-4-turbo-previewzgpt-4-turbozgpt-4-turbo-2024-04-09zgpt-4-completionzgpt-4-0314-completionzgpt-4-0613-completionzgpt-4-32k-completiongQ?zgpt-4-32k-0314-completionzgpt-4-32k-0613-completionzgpt-4-vision-preview-completionzgpt-4-1106-preview-completionzgpt-4-0125-preview-completionzgpt-4-turbo-preview-completionzgpt-4-turbo-completionz!gpt-4-turbo-2024-04-09-completionzgpt-3.5-turbog~jtX?zgpt-3.5-turbo-0125zgpt-3.5-turbo-0301zgpt-3.5-turbo-0613zgpt-3.5-turbo-1106gMbP?zgpt-3.5-turbo-instructzgpt-3.5-turbo-16kzgpt-3.5-turbo-16k-0613zgpt-3.5-turbo-completionzgpt-3.5-turbo-0125-completionzgpt-3.5-turbo-0301-completionzgpt-3.5-turbo-0613-completionzgpt-3.5-turbo-1106-completionz!gpt-3.5-turbo-instruct-completionzgpt-3.5-turbo-16k-completiongMbp?z!gpt-3.5-turbo-16k-0613-completionzgpt-35-turbozgpt-35-turbo-0125zgpt-35-turbo-0301zgpt-35-turbo-0613zgpt-35-turbo-instructzgpt-35-turbo-16kzgpt-35-turbo-16k-0613zgpt-35-turbo-completionzgpt-35-turbo-0125-completionzgpt-35-turbo-0301-completionzgpt-35-turbo-0613-completionz gpt-35-turbo-instruct-completionzgpt-35-turbo-16k-completionz gpt-35-turbo-16k-0613-completionztext-ada-001adaztext-babbage-001babbageztext-curie-001curieztext-davinci-003ztext-davinci-002zcode-davinci-002zbabbage-002-finetunedzdavinci-002-finetunedzgpt-3.5-turbo-0613-finetunedzgpt-3.5-turbo-1106-finetunedzgpt-3.5-turbo-0125-finetunedz gpt-4o-mini-2024-07-18-finetunedz'gpt-4o-mini-2024-07-18-finetuned-cachedz babbage-002-finetuned-completionz davinci-002-finetuned-completionz'gpt-3.5-turbo-0613-finetuned-completiong~jtx?z'gpt-3.5-turbo-1106-finetuned-completionz'gpt-3.5-turbo-0125-finetuned-completionz+gpt-4o-mini-2024-07-18-finetuned-completionga2U0*S?zbabbage-002-azure-finetunedzdavinci-002-azure-finetunedz!gpt-35-turbo-0613-azure-finetunedz&babbage-002-azure-finetuned-completionz&davinci-002-azure-finetuned-completionz,gpt-35-turbo-0613-azure-finetuned-completionzada-finetuned-legacyzbabbage-finetuned-legacyzcurie-finetuned-legacyzdavinci-finetuned-legacyc                   B    \ rS rSrSr\" 5       r\" 5       r\" 5       rSr	g)	TokenTypei  zToken type enum. N)
__name__
__module____qualname____firstlineno____doc__r   PROMPTPROMPT_CACHED
COMPLETION__static_attributes__r       a/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/callbacks/openai_info.pyr   r     s    VFFMJr"   r   
token_type
model_nameis_completionr%   returnc                   U(       a  [        SSSS9  [        R                  nU R                  5       n SU ;   a  U R	                  S5      S   S-   n SU ;   a  U R	                  S	5      S   S
-   n SU ;   a  U R	                  S	5      S   S-   n U[        R                  :X  a  U R                  S5      (       dz  U R                  S5      (       dd  U R                  S5      (       dN  U R                  S5      (       d8  U R                  S5      (       d"  U R                  S5      (       d  SU ;   a  SU ;  a  U S-   $ U[        R                  :X  a  U R                  S5      (       dX  U R                  S5      (       dB  U R                  S5      (       d,  U R                  S5      (       d  U R                  S5      (       a  U R                  S5      (       d  U S-   $ U $ )aj  
Standardize the model name to a format that can be used in the OpenAI API.

Args:
    model_name: Model name to standardize.
    is_completion: Whether the model is used for completion or not.
        Defaults to False. Deprecated in favor of ``token_type``.
    token_type: Token type. Defaults to ``TokenType.PROMPT``.

Returns:
    Standardized model name.

0.3.13zis_completion is deprecated. Use token_type instead. Example:

from langchain_community.callbacks.openai_info import TokenType

standardize_model_name('gpt-4o', token_type=TokenType.COMPLETION)
1.0sincemessageremovalz.ft-r   z-azure-finetunedz:ft-:z-finetuned-legacyzft:   z
-finetunedr   zgpt-3.5zgpt-35zo1-zo3-zo4-	finetunedlegacyz-completionr   r   r   r   o4r   z-cached)r   r   r    lowersplit
startswithr   )r&   r'   r%   s      r#   standardize_model_namer8     s   & V 	
 ))
!!#J%%f-a03EE
%%c*1-0CC

%%c*1-<
Y)))g&&  ++  **  ''  ''  '':%(**DM))i---!!(++$$Y//$$T**$$T**$$T**&&':;;I%%r"   
num_tokensc                    U(       a  [        SSSS9  [        R                  n[        XS9n U [        ;  a3  [        SU  S3SR                  [        R                  5       5      -   5      e[        U    US	-  -  $ )
ag  
Get the cost in USD for a given model and number of tokens.

Args:
    model_name: Name of the model
    num_tokens: Number of tokens.
    is_completion: Whether the model is used for completion or not.
        Defaults to False. Deprecated in favor of ``token_type``.
    token_type: Token type. Defaults to ``TokenType.PROMPT``.

Returns:
    Cost in USD.
r*   zis_completion is deprecated. Use token_type instead. Example:

from langchain_community.callbacks.openai_info import TokenType

get_openai_token_cost_for_model('gpt-4o', 10, token_type=TokenType.COMPLETION)
r+   r,   r$   zUnknown model: z=. Please provide a valid OpenAI model name.Known models are: z, i  )r   r   r    r8   MODEL_COST_PER_1K_TOKENS
ValueErrorjoinkeys)r&   r9   r'   r%   s       r#   get_openai_token_cost_for_modelr?   [  s    ( c 	
 ))
'
JJ11j\ *! !#'99-E-J-J-L#MN
 	
 $J/:3DEEr"   c                   ,  ^  \ rS rSr% SrSr\\S'   Sr\\S'   Sr	\\S'   Sr
\\S'   Sr\\S'   Sr\\S	'   S
r\\S'   SU 4S jjrS\4S jr\S\4S j5       rS\\\4   S\\   S\SS4S jrS\S\SS4S jrS\S\SS4S jrSS jrS\SS 4S jrSrU =r$ )OpenAICallbackHandleri  z)Callback Handler that tracks OpenAI info.r   total_tokensprompt_tokensprompt_tokens_cachedcompletion_tokensreasoning_tokenssuccessful_requestsg        
total_costr(   Nc                 V   > [         TU ]  5         [        R                  " 5       U l        g )N)super__init__	threadingLock_lock)self	__class__s    r#   rK   OpenAICallbackHandler.__init__  s    ^^%
r"   c                     SU R                    SU R                   SU R                   SU R                   SU R                   SU R
                   SU R                   3$ )NzTokens Used: z
	Prompt Tokens: z
		Prompt Tokens Cached: z
	Completion Tokens: z
		Reasoning Tokens: z
Successful Requests: z
Total Cost (USD): $)rB   rC   rD   rE   rF   rG   rH   rO   s    r#   __repr__OpenAICallbackHandler.__repr__  s    D--. /  $ 2 23 4))-)B)B(C D$$($:$:#; <%%)%:%:$; <$$($<$<#= >""&//!24	
r"   c                     g)z;Whether to call verbose callbacks even if verbose is False.Tr   rS   s    r#   always_verbose$OpenAICallbackHandler.always_verbose  s     r"   
serializedpromptskwargsc                     g)zPrint out the prompts.Nr   )rO   rY   rZ   r[   s       r#   on_llm_start"OpenAICallbackHandler.on_llm_start  s     	r"   tokenc                     g)zPrint out the token.Nr   )rO   r_   r[   s      r#   on_llm_new_token&OpenAICallbackHandler.on_llm_new_token  s    r"   responsec                 ^    UR                   S   S   n[        U[        5      (       a@   UR                  n[        U[
        5      (       a  UR                  nUR                  nO	SnSnOSnSnSnSnU(       a  SUS   0n	US   n
US   nU=(       d    0 R                  S5      =n(       a  [        U5      nO5UR                  c  SnO%[        UR                  R                  SS5      5      nSUR                  S	0 5      ;   a  US	   S   nS
UR                  S0 5      ;   a  US   S
   nOUR                  c  gSUR                  ;  a+  U R                     U =R                  S-  sl        SSS5        gUR                  S   n	U	R                  SS5      n
U	R                  SS5      n[        UR                  R                  SS5      5      nU[        ;   aR  X-
  n[        X[         R"                  S9n[        X[         R$                  S9nUU-   n[        X[         R&                  S9nOSnSnU R                     U =R(                  UU-   -  sl        U =R*                  U	R                  SS5      -  sl        U =R,                  U-  sl        U =R.                  U-  sl        U =R0                  U
-  sl        U =R2                  U-  sl        U =R                  S-  sl        SSS5        g! [         a    Sn GNf = f! [         a    SnSn GNf = f! , (       d  f       g= f! , (       d  f       g= f)zCollect token usage.r   NrB   output_tokensinput_tokensr&    
cache_readinput_token_details	reasoningoutput_token_detailstoken_usager1   rE   rC   r$   )generations
IndexError
isinstancer   r.   r
   usage_metadataresponse_metadataAttributeErrorgetr8   
llm_outputrN   rG   r;   r?   r   r   r   r    rH   rB   rC   rD   rE   rF   )rO   rc   r[   
generationr.   rp   rq   rD   rF   rl   rE   rC   response_model_namer&   uncached_prompt_tokensuncached_prompt_costcached_prompt_costprompt_costcompletion_costs                      r#   
on_llm_end OpenAICallbackHandler.on_llm_end  s7   	!--a03J j.11
)$,,gy11%,%;%;N(/(A(A%%)N(,%
 "N $ )>.+IJK . ?*>:M'8'>B&C&CL&QQ"Q34GH
$$,
3''++L"=
 ~112GLL'56K'L ($ n001GLL#12H#I+#V ""*H$7$77ZZ,,1,   #--m<K +0CQ G'OOOQ?M/##''b9J 11%2%I"#By?O?O$  "AY=T=T" /1CCK=):N:NO  OK ZZOO{_<<O!CC-/%%)==%""&77"!!%55!$$)$ ZS  	J	 " )!%$(!)B  Z6 ZsA   K% 9K8 $K8 L5B'L%K54K58L
	L

L
L,c                     U $ )z&Return a copy of the callback handler.r   rS   s    r#   __copy__OpenAICallbackHandler.__copy__      r"   memoc                     U $ )z+Return a deep copy of the callback handler.r   )rO   r   s     r#   __deepcopy__"OpenAICallbackHandler.__deepcopy__  r   r"   )rN   )r(   N)r(   rA   ) r   r   r   r   r   rB   int__annotations__rC   rD   rE   rF   rG   rH   floatrK   strrT   propertyboolrW   r   r   r   r]   ra   r   r|   r   r   r!   __classcell__)rP   s   @r#   rA   rA     s   3L#M3 !#!sc  J&	
# 	
   sCx.379HK	c S T U*9 U* U* U*n )@  r"   rA   )F)r   rL   enumr   r   typingr   r   r   langchain_core._apir   langchain_core.callbacksr	   langchain_core.messagesr
   langchain_core.outputsr   r   r;   r   r   r   r   r8   r   r   r?   rA   r   r"   r#   <module>r      s   .   " " / 8 - <EuE %E f	E
  E %E $UE FE vE 6E %fE vE  )&!E$ F%E& v'E( 8)E* %h+E. v/E0 )&1E4 u5E6 !%7E8 f9E: (;E> !$?E@ ,TAED 	%EEF UGEH IEJ FKEN TOEP QET dUEV WEZ [E\ #C]E` 	%aEb UcEd eEf FgEj UkEl mEp vqEr &sEt huEv  wEz &{E| $V}E@ vAEB &CED gEEF  GEJ &KEL $VMEP vQER gSET &UEV  WEZ &[E\ $V]E` %aEb cEd UeEf #FgEj TkEl 'mEp fqEr WsEt uEv wEx yEz {E| }E@ AEB #ECED #DEEF #DGEJ FKEL &vMEN &vOER &tSET 1$UEV 1$WEZ u[E\ )%]E^ )%_E` %faEb 0cEd 0eEh )$iEj 4TkEl 4TmEp 7qEr (sEt guEv $XwEz f{E| (}E@  AEB +GCEF +FGEH 6vIEL #FMEN .vOEP *6QER 5fSEV .vWEX 9&YE\ !']E^ ,W_Eb ,VcEd 7eEh ViEj 'kEn 'oEp 24qEt EuEv &uwEz &u{E| 1%}E@ TAEB $CED $EEF GEH dIEJ dKEL DMEN $OEP $QER 4SET 4UEV dWEZ [E\ T]E^ T_E` DaEb  cEd  eEf &tgEh $TiEj $TkEl %dmEn doEp (qEx VyEz &{E| &}E~ &E@ %AEB fCED EEF eGEN OEP $VQER $USET $UUEV $UWEX (YEZ #E[E\ (]E` FaEb cEd eEf gEh ViEj kEl UmEp uqEr #FsEt #EuEv #EwEx 'yEz "5{E| '}E@ FAEB 
6CED EEF vGEH eIEJ UKEL MEN OEP QET VUEV UWEX #EYEZ #E[E\ #E]E^ '_E` .waEd 'eEf 'gEh .uiEj .ukEl .umEn 26oEr "6sEt "5uEv (wEz -f{E| -e}E~ 3EEB FCED EEF eGEH IE P   < &,,	<<< 	<
 	<D  %F
 &,,%F%F%F %F
 %F %FPF/ Fr"   