From 0ccf40c13b35294ec04d122498b158edacadc17d Mon Sep 17 00:00:00 2001
From: CoreRasurae <luis.p.mendes@gmail.com>
Date: Mon, 5 Jan 2026 22:19:54 +0000
Subject: [PATCH] fix: Enable console logging

---
 .../cswarm_benchmark_client.py                 | 12 ++++++------
 .../cswarm_text_to_kg_client.py                | 18 +++++++++++-------
 2 files changed, 17 insertions(+), 13 deletions(-)

diff --git a/src/cleverswarm_python_client/cswarm_benchmark_client.py b/src/cleverswarm_python_client/cswarm_benchmark_client.py
index 708d8ef..ad6657b 100644
--- a/src/cleverswarm_python_client/cswarm_benchmark_client.py
+++ b/src/cleverswarm_python_client/cswarm_benchmark_client.py
@@ -19,6 +19,7 @@ CleverSwarm - text to knowledge graph extraction benchmark client.
 
 import argparse
 import logging
+import os
 import sys
 from pathlib import Path
 from typing import Dict, List, Set
@@ -282,13 +283,12 @@ class BenchmarkCLI(object):
 def main():
     """Main entry point for the benchmark client console script."""
 
-    handler = logging.StreamHandler(sys.stdout)
-    handler.setLevel(logging.DEBUG)
-    formatter = logging.Formatter(
-        "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    logging.basicConfig(
+        level=os.environ.get("PYTHON_LOGGING_LEVEL", logging.INFO),
+        stream=sys.stderr,
+        datefmt="%Y-%m-%dT%H:%M:%S",
+        format=("%(asctime)s - %(name)s - %(levelname)s - %(message)s"),
     )
-    handler.setFormatter(formatter)
-    logger.addHandler(handler)
 
     parser = argparse.ArgumentParser(
         description="CleverSwarm benchmark client for REST API."
diff --git a/src/cleverswarm_python_client/cswarm_text_to_kg_client.py b/src/cleverswarm_python_client/cswarm_text_to_kg_client.py
index 47a433c..2a498b1 100644
--- a/src/cleverswarm_python_client/cswarm_text_to_kg_client.py
+++ b/src/cleverswarm_python_client/cswarm_text_to_kg_client.py
@@ -19,6 +19,7 @@ CleverSwarm - unstructured text to knowledge graph triplets extractor client.
 
 import argparse
 import logging
+import os
 import sys
 from pathlib import Path
 from typing import Any, Dict
@@ -89,7 +90,11 @@ class TextoToKGCLI(object):
             )
 
         # If ontology_json was provided, validate that the file exists
-        if ontology_json and ontology_json_path and (not ontology_json_path.exists() or not ontology_json_path.is_file()):
+        if (
+            ontology_json
+            and ontology_json_path
+            and (not ontology_json_path.exists() or not ontology_json_path.is_file())
+        ):
             raise ClientException(
                 f"Specified optional Ontology JSON file does not exist or is not a file: {str(ontology_json_path)}"
             )
@@ -218,13 +223,12 @@ class TextoToKGCLI(object):
 def main():
     """Main entry point for the text-to-KG client console script."""
 
-    handler = logging.StreamHandler(sys.stdout)
-    handler.setLevel(logging.DEBUG)
-    formatter = logging.Formatter(
-        "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    logging.basicConfig(
+        level=os.environ.get("PYTHON_LOGGING_LEVEL", logging.INFO),
+        stream=sys.stderr,
+        datefmt="%Y-%m-%dT%H:%M:%S",
+        format=("%(asctime)s - %(name)s - %(levelname)s - %(message)s"),
     )
-    handler.setFormatter(formatter)
-    logger.addHandler(handler)
 
     parser = argparse.ArgumentParser(
         description="CleverSwarm unstructured text to Knowledge Graph triplets client for REST API."