OpenBMB/ChatDev
1
1
Fork 0
mirror of https://github.com/OpenBMB/ChatDev.git synced 2024-12-25 04:44:50 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Resolve #308: rename log_visualize in model_backend

Browse Source
This commit is contained in:
thinkwee 2023-12-19 08:07:29 +08:00
parent 259549fc4d
commit 49e38d7090
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
camel/model_backend.py
View File

@ -19,7 +19,7 @@ import tiktoken
from camel.typing import ModelType
from chatdev.statistics import prompt_cost
from chatdev.utils import log_and_print_online
from chatdev.utils import log_visualize
class ModelBackend(ABC):
@ -79,7 +79,7 @@ class OpenAIModel(ModelBackend):
num_completion_tokens=response["usage"]["completion_tokens"]
)
log_and_print_online(
log_visualize(
"**[OpenAI_Usage_Info Receive]**\nprompt_tokens: {}\ncompletion_tokens: {}\ntotal_tokens: {}\ncost: ${:.6f}\n".format(
response["usage"]["prompt_tokens"], response["usage"]["completion_tokens"],
response["usage"]["total_tokens"], cost))
@ -131,6 +131,6 @@ class ModelFactory:
if model_type is None:
model_type = default_model_type
# log_and_print_online("Model Type: {}".format(model_type))
# log_visualize("Model Type: {}".format(model_type))
inst = model_class(model_type, model_config_dict)
return inst