From 8e5d2dae9401df0cdc620b7c3a3e16b84b8925be Mon Sep 17 00:00:00 2001 From: Boris Date: Sat, 17 Feb 2024 17:44:45 +0100 Subject: [PATCH 1/3] fix: replace logging function in classify_documents.py Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> --- cognitive_architecture/classifiers/classify_documents.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cognitive_architecture/classifiers/classify_documents.py b/cognitive_architecture/classifiers/classify_documents.py index 07aff6d76..d97d8ee1f 100644 --- a/cognitive_architecture/classifiers/classify_documents.py +++ b/cognitive_architecture/classifiers/classify_documents.py @@ -59,6 +59,6 @@ async def classify_documents(query: str, document_id: str, content: str): {"query": query, "d_id": document_id, "context": str(document_context)} ) 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) return arguments_dict \ No newline at end of file From 644c5128246af22cbaf913ea2bc39e4007563e4d Mon Sep 17 00:00:00 2001 From: Boris Date: Sat, 17 Feb 2024 17:46:03 +0100 Subject: [PATCH 2/3] 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 From d8240b62686bd90d48f0e6a5a0ee97aaea161024 Mon Sep 17 00:00:00 2001 From: Boris Date: Sat, 17 Feb 2024 17:47:24 +0100 Subject: [PATCH 3/3] fix: fix typo in classify_user_query.py Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> --- cognitive_architecture/classifiers/classify_user_query.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cognitive_architecture/classifiers/classify_user_query.py b/cognitive_architecture/classifiers/classify_user_query.py index 1f4210028..bbef1b137 100644 --- a/cognitive_architecture/classifiers/classify_user_query.py +++ b/cognitive_architecture/classifiers/classify_user_query.py @@ -36,7 +36,7 @@ async def classify_user_query(query, context, document_types): "description": "The classification of documents in groups such as legal, medical, etc.", } }, - "required": ["UserQueryClassiffier"], + "required": ["UserQueryClassifier"], }, } ]