Rework logging (#3358)

Unified all log files into one.

### What problem does this PR solve?

Unified all log files into one.

### Type of change

- [x] Refactoring
This commit is contained in:
Zhichang Yu
2024-11-12 17:35:13 +08:00
committed by GitHub
parent 567a7563e7
commit a2a5631da4
75 changed files with 481 additions and 853 deletions

View File

@ -7,7 +7,6 @@ Reference:
import argparse
import json
import logging
import re
import traceback
from dataclasses import dataclass
@ -18,12 +17,12 @@ import tiktoken
from graphrag.claim_prompt import CLAIM_EXTRACTION_PROMPT, CONTINUE_PROMPT, LOOP_PROMPT
from rag.llm.chat_model import Base as CompletionLLM
from graphrag.utils import ErrorHandlerFn, perform_variable_replacements
from api.utils.log_utils import logger
DEFAULT_TUPLE_DELIMITER = "<|>"
DEFAULT_RECORD_DELIMITER = "##"
DEFAULT_COMPLETION_DELIMITER = "<|COMPLETE|>"
CLAIM_MAX_GLEANINGS = 1
log = logging.getLogger(__name__)
@dataclass
@ -127,7 +126,7 @@ class ClaimExtractor:
]
source_doc_map[document_id] = text
except Exception as e:
log.exception("error extracting claim")
logger.exception("error extracting claim")
self._on_error(
e,
traceback.format_exc(),
@ -266,4 +265,4 @@ if __name__ == "__main__":
"claim_description": ""
}
claim = ex(info)
print(json.dumps(claim.output, ensure_ascii=False, indent=2))
logger.info(json.dumps(claim.output, ensure_ascii=False, indent=2))