From 644c5128246af22cbaf913ea2bc39e4007563e4d Mon Sep 17 00:00:00 2001 From: Boris Date: Sat, 17 Feb 2024 17:46:03 +0100 Subject: [PATCH] fix: replace logging in classify_summary.py Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> --- cognitive_architecture/classifiers/classify_summary.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognitive_architecture/classifiers/classify_summary.py b/cognitive_architecture/classifiers/classify_summary.py index 45dfe0c32..d437c658f 100644 --- a/cognitive_architecture/classifiers/classify_summary.py +++ b/cognitive_architecture/classifiers/classify_summary.py @@ -52,11 +52,11 @@ async def classify_summary(query, document_summaries): {"query": query, "document_summaries": document_summaries} ) arguments_str = classifier_output.additional_kwargs["function_call"]["arguments"] - print("This is the arguments string", arguments_str) + logging.info("This is the arguments string %s", arguments_str) arguments_dict = json.loads(arguments_str) logging.info("Relevant summary is %s", arguments_dict.get("DocumentSummary", None)) classfier_id = arguments_dict.get("d_id", None) - print("This is the classifier id ", classfier_id) + logging.info("This is the classifier id %s", classfier_id) return classfier_id \ No newline at end of file