github/ChatDev
1
0
Fork 0
mirror of https://github.com/OpenBMB/ChatDev.git synced 2026-07-29 01:16:02 +00:00
Code Issues 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 camel.typing import ModelType
from chatdev.statistics import prompt_cost from chatdev.statistics import prompt_cost
from chatdev.utils import log_and_print_online from chatdev.utils import log_visualize
class ModelBackend(ABC): class ModelBackend(ABC):
@ -79,7 +79,7 @@ class OpenAIModel(ModelBackend):
num_completion_tokens=response["usage"]["completion_tokens"] 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( "**[OpenAI_Usage_Info Receive]**\nprompt_tokens: {}\ncompletion_tokens: {}\ntotal_tokens: {}\ncost: ${:.6f}\n".format(
response["usage"]["prompt_tokens"], response["usage"]["completion_tokens"], response["usage"]["prompt_tokens"], response["usage"]["completion_tokens"],
response["usage"]["total_tokens"], cost)) response["usage"]["total_tokens"], cost))
@ -131,6 +131,6 @@ class ModelFactory:
if model_type is None: if model_type is None:
model_type = default_model_type 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) inst = model_class(model_type, model_config_dict)
return inst return inst