From 58f3dcfa0ffdd1b0f958e34b6cf3aa919d1b6f45 Mon Sep 17 00:00:00 2001 From: LuoChen Date: Mon, 5 Aug 2024 14:42:25 +0800 Subject: [PATCH] improve: [ChatBot] remove prints --- src/ai_powered/chat_bot.py | 5 +++-- src/ai_powered/tool_call.py | 4 +++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/src/ai_powered/chat_bot.py b/src/ai_powered/chat_bot.py index a4a6011..67fb0bd 100644 --- a/src/ai_powered/chat_bot.py +++ b/src/ai_powered/chat_bot.py @@ -6,7 +6,7 @@ from ai_powered.constants import DEBUG, OPENAI_API_KEY, OPENAI_BASE_URL, OPENAI_MODEL_NAME from ai_powered.llm_adapter.known_models import complete_model_config from ai_powered.llm_adapter.openai.param_types import ChatCompletionMessageParam -from ai_powered.tool_call import MakeTool +from ai_powered.tool_call import ChatCompletionToolParam, MakeTool default_client = openai.OpenAI(base_url=OPENAI_BASE_URL, api_key=OPENAI_API_KEY) model_config = complete_model_config(OPENAI_BASE_URL, OPENAI_MODEL_NAME) @@ -24,6 +24,7 @@ def __post_init__(self): if len(self.system_prompt) > 0: self.conversation.append({"role": "system", "content": self.system_prompt}) self._tool_dict = {tool.fn.__name__: tool for tool in self.tools} + self._tool_schemas : list[ChatCompletionToolParam] | openai.NotGiven = [ t.schema() for t in self.tools ] if len(self.tools) > 0 else openai.NOT_GIVEN def chat_continue(self) -> str: if DEBUG: @@ -32,7 +33,7 @@ def chat_continue(self) -> str: response = self.client.chat.completions.create( model = model_config.model_name, messages = self.conversation, - tools = [ t.schema() for t in self.tools ] if len(self.tools) > 0 else openai.NOT_GIVEN, + tools = self._tool_schemas, ) assistant_message = response.choices[0].message diff --git a/src/ai_powered/tool_call.py b/src/ai_powered/tool_call.py index 6424bc3..30c9b8d 100644 --- a/src/ai_powered/tool_call.py +++ b/src/ai_powered/tool_call.py @@ -58,7 +58,9 @@ def schema(self) -> ChatCompletionToolParam: raw_schema = msgspec.json.schema(self.struct_of_parameters()) parameters_schema = deref(raw_schema) - print(yellow(f"{parameters_schema =}")) + + if DEBUG: + print(yellow(f"{parameters_schema =}")) return { "type": "function",