Fix logging level and version display

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-03 15:45:46 -03:00
commit d98410f90c
6 changed files with 27 additions and 19 deletions

View file

@ -149,13 +149,13 @@ def run(
Run the Langflow.
"""
configure(log_level=log_level, log_file=log_file)
set_var_for_macos_issue()
# override env variables with .env file
if env_file:
load_dotenv(env_file, override=True)
configure(log_level=log_level, log_file=log_file)
update_settings(
config,
dev=dev,
@ -246,10 +246,10 @@ def get_free_port(port):
def print_banner(host, port):
# console = Console()
from langflow.version import __version__
word = "Langflow"
colors = ["#3300cc"]
colors = ["#6e42f5"]
styled_word = ""
@ -259,7 +259,7 @@ def print_banner(host, port):
# Title with emojis and gradient text
title = (
f"[bold]Welcome to :chains: {styled_word} [/bold]\n\n"
f"[bold]Welcome to :chains: {styled_word} v{__version__}[/bold]\n"
f"Access [link=http://{host}:{port}]http://{host}:{port}[/link]"
)
info_text = (
@ -307,7 +307,7 @@ def run_langflow(host, port, log_level, options, app):
def superuser(
username: str = typer.Option(..., prompt=True, help="Username for the superuser."),
password: str = typer.Option(..., prompt=True, hide_input=True, help="Password for the superuser."),
log_level: str = typer.Option("critical", help="Logging level.", envvar="LANGFLOW_LOG_LEVEL"),
log_level: str = typer.Option("error", help="Logging level.", envvar="LANGFLOW_LOG_LEVEL"),
):
"""
Create a superuser.

View file

@ -1,4 +1,5 @@
import os
import warnings
from logging.config import fileConfig
from alembic import context
@ -82,11 +83,12 @@ def run_migrations_online() -> None:
prefix="sqlalchemy.",
poolclass=pool.NullPool,
)
with connectable.connect() as connection:
context.configure(connection=connection, target_metadata=target_metadata, render_as_batch=True)
with context.begin_transaction():
context.run_migrations()
with warnings.catch_warnings():
warnings.simplefilter("ignore")
with connectable.connect() as connection:
context.configure(connection=connection, target_metadata=target_metadata, render_as_batch=True)
with context.begin_transaction():
context.run_migrations()
if context.is_offline_mode():

View file

@ -1,3 +1,4 @@
import warnings
from typing import Callable, Dict, List, Optional
from langchain.agents import agent_toolkits
@ -29,13 +30,15 @@ class ToolkitCreator(LangChainTypeCreator):
@property
def type_to_loader_dict(self) -> Dict:
if self.type_dict is None:
settings_service = get_settings_service()
self.type_dict = {
toolkit_name: import_class(f"langchain.agents.agent_toolkits.{toolkit_name}")
# if toolkit_name is not lower case it is a class
for toolkit_name in agent_toolkits.__all__
if not toolkit_name.islower() and toolkit_name in settings_service.settings.TOOLKITS
}
with warnings.catch_warnings():
warnings.simplefilter("ignore")
settings_service = get_settings_service()
self.type_dict = {
toolkit_name: import_class(f"langchain.agents.agent_toolkits.{toolkit_name}")
# if toolkit_name is not lower case it is a class
for toolkit_name in agent_toolkits.__all__
if not toolkit_name.islower() and toolkit_name in settings_service.settings.TOOLKITS
}
return self.type_dict

View file

@ -130,7 +130,7 @@ if __name__ == "__main__":
host="127.0.0.1",
port=7860,
workers=get_number_of_workers(),
log_level="debug",
log_level="error",
reload=True,
loop="asyncio",
)

View file

@ -1,3 +1,5 @@
import os
from gunicorn.app.base import BaseApplication # type: ignore
from uvicorn.workers import UvicornWorker
@ -11,6 +13,7 @@ class LangflowApplication(BaseApplication):
self.options = options or {}
self.options["worker_class"] = "langflow.server.LangflowUvicornWorker"
self.options["loglevel"] = os.getenv("LANGFLOW_LOG_LEVEL", "error").lower()
self.application = app
super().__init__()

View file

@ -28,7 +28,7 @@ def configure(log_level: Optional[str] = None, log_file: Optional[Path] = None):
if os.getenv("LANGFLOW_LOG_LEVEL", "").upper() in VALID_LOG_LEVELS and log_level is None:
log_level = os.getenv("LANGFLOW_LOG_LEVEL")
if log_level is None:
log_level = "INFO"
log_level = "ERROR"
# Human-readable
log_format = (
"<green>{time:YYYY-MM-DD HH:mm:ss}</green> - <level>"