🐛 fix(custom_component.py): improve error message formatting for type hint error in build method

🐛 fix(logger.py): fix conditional statement for log_level configuration to properly handle None value
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-27 08:54:11 -03:00
commit ede745c1dd
2 changed files with 11 additions and 4 deletions

View file

@ -102,7 +102,10 @@ class CustomComponent(Component, extra=Extra.allow):
status_code=400,
detail={
"error": "Type hint Error",
"traceback": "Prompt type is not supported in the build method. Try using PromptTemplate instead.",
"traceback": (
"Prompt type is not supported in the build method."
" Try using PromptTemplate instead."
),
},
)
return args

View file

@ -1,3 +1,4 @@
from math import log
from typing import Optional
from loguru import logger
from pathlib import Path
@ -25,12 +26,15 @@ def patching(record):
def configure(log_level: Optional[str] = None, log_file: Optional[Path] = None):
if os.getenv("LANGFLOW_LOG_LEVEL") in VALID_LOG_LEVELS and not log_level:
if os.getenv("LANGFLOW_LOG_LEVEL") in VALID_LOG_LEVELS and log_level is None:
log_level = os.getenv("LANGFLOW_LOG_LEVEL")
elif not log_level:
if log_level is None:
log_level = "INFO"
# Human-readable
log_format = "<green>{time:YYYY-MM-DD HH:mm:ss}</green> - <level>{level: <8}</level> - {module} - <level>{message}</level>"
log_format = (
"<green>{time:YYYY-MM-DD HH:mm:ss}</green> - <level>"
"{level: <8}</level> - {module} - <level>{message}</level>"
)
# log_format = log_format_dev if log_level.upper() == "DEBUG" else log_format_prod
logger.remove() # Remove default handlers