gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Set OpenAI logger level to INFO if VERBOSE_DEBUG is off

Browse Source
This commit is contained in:
yangdx
2025-02-17 12:20:47 +08:00
parent 9963fb7ff4
commit d3ff8c3537
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lightrag/llm/openai.py
View File

@@ -40,9 +40,10 @@ __version__ = "1.0.0"
__author__ = "lightrag Team" __author__ = "lightrag Team"
__status__ = "Production" __status__ = "Production"
from ..utils import verbose_debug from ..utils import verbose_debug, VERBOSE_DEBUG
import sys import sys
import os import os
import logging
if sys.version_info < (3, 9): if sys.version_info < (3, 9):
from typing import AsyncIterator from typing import AsyncIterator
@@ -110,6 +111,11 @@ async def openai_complete_if_cache(
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_8) LightRAG/{__api_version__}", "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_8) LightRAG/{__api_version__}",
"Content-Type": "application/json", "Content-Type": "application/json",
} }
# Set openai logger level to INFO when VERBOSE_DEBUG is off
if not VERBOSE_DEBUG and logger.level == logging.DEBUG:
logging.getLogger("openai").setLevel(logging.INFO)
openai_async_client = ( openai_async_client = (
AsyncOpenAI(default_headers=default_headers, api_key=api_key) AsyncOpenAI(default_headers=default_headers, api_key=api_key)
if base_url is None if base_url is None
@@ -125,8 +131,7 @@ async def openai_complete_if_cache(
messages.extend(history_messages) messages.extend(history_messages)
messages.append({"role": "user", "content": prompt}) messages.append({"role": "user", "content": prompt})
# 添加日志输出 logger.debug("===== Sending Query to LLM =====")
logger.debug("===== Query Input to LLM =====")
logger.debug(f"Model: {model} Base URL: {base_url}") logger.debug(f"Model: {model} Base URL: {base_url}")
logger.debug(f"Additional kwargs: {kwargs}") logger.debug(f"Additional kwargs: {kwargs}")
verbose_debug(f"Query: {prompt}") verbose_debug(f"Query: {prompt}")