jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source 
🐛 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
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/interface/custom/custom_component.py
View file

@ -102,7 +102,10 @@ class CustomComponent(Component, extra=Extra.allow):
status_code=400, status_code=400,
detail={ detail={
"error": "Type hint Error", "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 return args

10
src/backend/langflow/utils/logger.py
View file

@ -1,3 +1,4 @@
from math import log
from typing import Optional from typing import Optional
from loguru import logger from loguru import logger
from pathlib import Path from pathlib import Path
@ -25,12 +26,15 @@ def patching(record):
def configure(log_level: Optional[str] = None, log_file: Optional[Path] = None): 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") log_level = os.getenv("LANGFLOW_LOG_LEVEL")
elif not log_level: if log_level is None:
log_level = "INFO" log_level = "INFO"
# Human-readable # 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 # log_format = log_format_dev if log_level.upper() == "DEBUG" else log_format_prod
logger.remove() # Remove default handlers logger.remove() # Remove default handlers