diff --git a/src/seclab_taskflow_agent/__main__.py b/src/seclab_taskflow_agent/__main__.py index b3a0fc51..1cbee0cc 100644 --- a/src/seclab_taskflow_agent/__main__.py +++ b/src/seclab_taskflow_agent/__main__.py @@ -27,6 +27,7 @@ from .agent import DEFAULT_MODEL, TaskAgent, TaskAgentHooks, TaskRunHooks from .available_tools import AvailableTools +from .banner import get_banner from .capi import get_AI_token, list_tool_call_models from .env_utils import TmpEnv from .mcp_utils import ( @@ -677,4 +678,5 @@ async def _deploy_task_agents(resolved_agents, prompt): print(help_msg) sys.exit(1) + print(get_banner()) # print banner only before starting main event loop asyncio.run(main(available_tools, p, t, cli_globals, user_prompt), debug=True) diff --git a/src/seclab_taskflow_agent/banner.py b/src/seclab_taskflow_agent/banner.py new file mode 100644 index 00000000..668e1410 --- /dev/null +++ b/src/seclab_taskflow_agent/banner.py @@ -0,0 +1,38 @@ +# SPDX-FileCopyrightText: 2026 GitHub +# SPDX-License-Identifier: MIT + +from .capi import get_AI_endpoint + +def get_banner(): + api_endpoint = get_AI_endpoint() + banner = f""" + ╔══════════════════════════════════════════════════════════════════╗ + ║ ║ + ║ ██████╗ ██╗████████╗██╗ ██╗██╗ ██╗██████╗ ║ + ║ ██╔════╝ ██║╚══██╔══╝██║ ██║██║ ██║██╔══██╗ ║ + ║ ██║ ███╗██║ ██║ ███████║██║ ██║██████╔╝ ║ + ║ ██║ ██║██║ ██║ ██╔══██║██║ ██║██╔══██╗ ║ + ║ ╚██████╔╝██║ ██║ ██║ ██║╚██████╔╝██████╔╝ ║ + ║ ╚═════╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═════╝ ║ + ║ ║ + ║ ███████╗███████╗ ██████╗██╗ ██╗██████╗ ██╗████████╗██╗ ██╗ ║ + ║ ██╔════╝██╔════╝██╔════╝██║ ██║██╔══██╗██║╚══██╔══╝╚██╗ ██╔╝ ║ + ║ ███████╗█████╗ ██║ ██║ ██║██████╔╝██║ ██║ ╚████╔╝ ║ + ║ ╚════██║██╔══╝ ██║ ██║ ██║██╔══██╗██║ ██║ ╚██╔╝ ║ + ║ ███████║███████╗╚██████╗╚██████╔╝██║ ██║██║ ██║ ██║ ║ + ║ ╚══════╝╚══════╝ ╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚═╝ ╚═╝ ╚═╝ ║ + ║ ║ + ║ ██╗ █████╗ ██████╗ ║ + ║ ██║ ██╔══██╗██╔══██╗ ║ + ║ ██║ ███████║██████╔╝ ║ + ║ ██║ ██╔══██║██╔══██╗ ║ + ║ ███████╗██║ ██║██████╔╝ ║ + ║ ╚══════╝╚═╝ ╚═╝╚═════╝ ║ + ║ ║ + ║ TASKFLOW AGENT ║ + ║ ║ + ╠══════════════════════════════════════════════════════════════════╣ + ║ AI API Endpoint: {api_endpoint:<48}║ + ╚══════════════════════════════════════════════════════════════════╝ + """ + return banner