Fd xddlmZddlmZmZddlZddlmZddlmZddl m Z ddl m Z Gdd e Z dS) ) annotations)ListOptionalN)Model)CompletionModelInfo)logger) SingletoncFeZdZdZdZdZdZdZdZdZ dZ d d Z d S) ApiManagercLd|_d|_d|_d|_d|_dS)Nrtotal_prompt_tokenstotal_completion_tokens total_cost total_budgetmodelsselfs KC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\llm\api_manager.py__init__zApiManager.__init__s,#$ '($-1 cLd|_d|_d|_d|_d|_dS)Nrgr rs rresetzApiManager.resets,#$ '($ rcddlm}|dr |ddn|}||}|xj|z c_|xj|z c_|xj||jzdz z c_tt|tr|xj||j zdz z c_tj d|jddS) a' Update the total cost, prompt tokens, and completion tokens. Args: prompt_tokens (int): The number of tokens used in the prompt. completion_tokens (int): The number of tokens used in the completion. model (str): The model used for the API call. r)OPEN_AI_MODELSz-v2NizTotal running cost: $z.3f) autogpt.llm.providers.openairendswithrrrprompt_token_cost issubclasstypercompletion_token_costrdebug)r prompt_tokenscompletion_tokensmodelr model_infos r update_costzApiManager.update_costs @?????#nnU33>crc #E*    M1   $$(99$$ =:+GG$NN d:&&(; < <  OO!J$DDtK OO  BT_BBBCCCCCrc||_dS)z Sets the total user-defined budget for API calls. Args: total_budget (float): The total budget for API calls. Nr)rrs rset_total_budgetzApiManager.set_total_budget5s)rc|jS)zz Get the total number of prompt tokens. Returns: int: The total number of prompt tokens. )rrs rget_total_prompt_tokensz"ApiManager.get_total_prompt_tokens>s ''rc|jS)z Get the total number of completion tokens. Returns: int: The total number of completion tokens. )rrs rget_total_completion_tokensz&ApiManager.get_total_completion_tokensGs ++rc|jS)zp Get the total cost of API calls. Returns: float: The total cost of API calls. )rrs rget_total_costzApiManager.get_total_costPs rc|jS)z Get the total user-defined budget for API calls. Returns: float: The total budget for API calls. r*rs rget_total_budgetzApiManager.get_total_budgetYs   rreturn List[Model]c|j5tjd}d|D|_|jS)zr Get list of available GPT models. Returns: list: List of available GPT models. Ndatac&g|]}d|dv |S)gptid).0r&s r z)ApiManager.get_models..ls&QQQUEU4[rautogpt.llm.baser autogpt.logsrautogpt.singletonr r r;rrrKs""""""!!!!!!!! 000000''''''aaaaa9aaaaaar