From c182fa32421857d630a8f3bd0004a59e99b497a6 Mon Sep 17 00:00:00 2001 From: LuoChen Date: Mon, 5 Aug 2024 14:42:25 +0800 Subject: [PATCH] improve: [ChatBot] tidy prints --- src/ai_powered/tool_call.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/src/ai_powered/tool_call.py b/src/ai_powered/tool_call.py index 30c9b8d..b100e44 100644 --- a/src/ai_powered/tool_call.py +++ b/src/ai_powered/tool_call.py @@ -48,19 +48,13 @@ def schema(self) -> ChatCompletionToolParam: docstring = inspect.getdoc(self.fn) if DEBUG: - print(f"{sig =}") - print(f"{docstring =}") - - for param in sig.parameters.values(): - print(f"{param.name}: {param.annotation}") - - print(f"{sig.return_annotation =}") + print(f"[MakeTool.schema()] {sig =}") raw_schema = msgspec.json.schema(self.struct_of_parameters()) parameters_schema = deref(raw_schema) if DEBUG: - print(yellow(f"{parameters_schema =}")) + print(yellow(f"[MakeTool.schema()] {parameters_schema =}")) return { "type": "function",