Remove Global Logging Config from All Production Files (#13321)

run-llama · May 7, 2024 · f94dfdf · f94dfdf
1 parent c26ee76
commit f94dfdf
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 12 deletions.
diff --git a/...ex-integrations/readers/llama-index-readers-openalex/llama_index/readers/openalex/base.py b/...ex-integrations/readers/llama-index-readers-openalex/llama_index/readers/openalex/base.py
@@ -9,7 +9,8 @@
 from llama_index.core.readers.base import BaseReader
 from llama_index.core.schema import Document
 
-logging.basicConfig(level=logging.ERROR)
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.ERROR)
 
 
 class OpenAlexReader(BaseReader):
@@ -46,11 +47,11 @@ def _search_openalex(self, query, fields):
                 raise ValueError(f"API returned error: {data['error']}")
             return data
         except requests.exceptions.HTTPError as http_error:
-            logging.error(f"HTTP error occurred: {http_error}")
+            logger.error(f"HTTP error occurred: {http_error}")
         except requests.exceptions.RequestException as request_error:
-            logging.error(f"Error occurred: {request_error}")
+            logger.error(f"Error occurred: {request_error}")
         except ValueError as value_error:
-            logging.error(value_error)
+            logger.error(value_error)
         return None
 
     def _fulltext_search_openalex(self, query, fields):
@@ -66,11 +67,11 @@ def _fulltext_search_openalex(self, query, fields):
                 raise ValueError(f"API returned error: {data['error']}")
             return data
         except requests.exceptions.HTTPError as http_error:
-            logging.error(f"HTTP error occurred: {http_error}")
+            logger.error(f"HTTP error occurred: {http_error}")
         except requests.exceptions.RequestException as request_error:
-            logging.error(f"Error occurred: {request_error}")
+            logger.error(f"Error occurred: {request_error}")
         except ValueError as value_error:
-            logging.error(value_error)
+            logger.error(value_error)
         return None
 
     def _invert_abstract(self, inv_index):

diff --git a/llama-index-integrations/readers/llama-index-readers-openalex/pyproject.toml b/llama-index-integrations/readers/llama-index-readers-openalex/pyproject.toml
@@ -29,7 +29,7 @@ license = "MIT"
 maintainers = ["shauryr"]
 name = "llama-index-readers-openalex"
 readme = "README.md"
-version = "0.1.3"
+version = "0.1.4"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"

diff --git a/llama-index-packs/llama-index-packs-raft-dataset/llama_index/packs/raft_dataset/base.py b/llama-index-packs/llama-index-packs-raft-dataset/llama_index/packs/raft_dataset/base.py
@@ -8,7 +8,8 @@
 from datasets import Dataset
 
 # Configure logging to output to the console, with messages of level DEBUG and above
-logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
 
 from llama_index.core.llama_pack.base import BaseLlamaPack
 from llama_index.core import SimpleDirectoryReader
@@ -199,14 +200,14 @@ def run(self) -> Any:
         """Run the pipeline."""
         chunks = self.get_chunks(self.file_path, self.chunk_size)
 
-        logging.info(f"Number of chunks created: {len(chunks)}")
+        logger.info(f"Number of chunks created: {len(chunks)}")
 
         self.num_distract_docs = (
             min(self.num_distract_docs, len(chunks)) - 1
         )  # should be less than number of chunks/ nodes created
 
         for index, chunk in enumerate(chunks):
-            logging.info(f"Processing chunk: {index}")
+            logger.info(f"Processing chunk: {index}")
             self.add_chunk_to_dataset(
                 chunks, chunk, self.num_questions_per_chunk, self.num_distract_docs
             )

diff --git a/llama-index-packs/llama-index-packs-raft-dataset/pyproject.toml b/llama-index-packs/llama-index-packs-raft-dataset/pyproject.toml
@@ -29,7 +29,7 @@ license = "MIT"
 maintainers = ["ravi-theja"]
 name = "llama-index-packs-raft-dataset"
 readme = "README.md"
-version = "0.1.4"
+version = "0.1.5"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"