From 77aea4d1a9d22681c2ac63eccd589d4160d06022 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 7 Dec 2023 14:55:51 -0300 Subject: [PATCH] Refactor code formatting and add version display option --- src/backend/langflow/__main__.py | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py index 5b10ba40f..885b10621 100644 --- a/src/backend/langflow/__main__.py +++ b/src/backend/langflow/__main__.py @@ -12,7 +12,8 @@ from dotenv import load_dotenv from langflow.main import setup_app from langflow.services.database.utils import session_getter from langflow.services.deps import get_db_service, get_settings_service -from langflow.services.utils import initialize_services, initialize_settings_service +from langflow.services.utils import (initialize_services, + initialize_settings_service) from langflow.utils.logger import configure, logger from multiprocess import Process, cpu_count # type: ignore from rich import box @@ -96,6 +97,27 @@ def update_settings( settings_service.settings.update_settings(STORE=False) + +def version_callback(value: bool): + """ + Show the version and exit. + """ + from langflow import __version__ + if value: + typer.echo(f"Langflow Version: {__version__}") + raise typer.Exit() + +@app.callback() +def main_entry_point( + version: bool = typer.Option(None, "--version", callback=version_callback, is_eager=True, help="Show the version and exit.") +): + """ + Main entry point for the Langflow CLI. + """ + pass + + + @app.command() def run( host: str = typer.Option("127.0.0.1", help="Host to bind the server to.", envvar="LANGFLOW_HOST"),