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 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 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"], }, } ]