Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions src/seclab_taskflow_agent/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@

from .agent import DEFAULT_MODEL, TaskAgent, TaskAgentHooks, TaskRunHooks
Comment thread
p- marked this conversation as resolved.
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 (
Expand Down Expand Up @@ -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)
35 changes: 35 additions & 0 deletions src/seclab_taskflow_agent/banner.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
from seclab_taskflow_agent.capi import get_AI_endpoint
Comment thread
p- marked this conversation as resolved.
Outdated
Comment thread
p- marked this conversation as resolved.
Outdated

def get_banner():
api_endpoint = get_AI_endpoint()
banner = f"""
╔══════════════════════════════════════════════════════════════════╗
║ ║
║ ██████╗ ██╗████████╗██╗ ██╗██╗ ██╗██████╗ ║
║ ██╔════╝ ██║╚══██╔══╝██║ ██║██║ ██║██╔══██╗ ║
║ ██║ ███╗██║ ██║ ███████║██║ ██║██████╔╝ ║
║ ██║ ██║██║ ██║ ██╔══██║██║ ██║██╔══██╗ ║
║ ╚██████╔╝██║ ██║ ██║ ██║╚██████╔╝██████╔╝ ║
║ ╚═════╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═════╝ ║
║ ║
║ ███████╗███████╗ ██████╗██╗ ██╗██████╗ ██╗████████╗██╗ ██╗ ║
║ ██╔════╝██╔════╝██╔════╝██║ ██║██╔══██╗██║╚══██╔══╝╚██╗ ██╔╝ ║
║ ███████╗█████╗ ██║ ██║ ██║██████╔╝██║ ██║ ╚████╔╝ ║
║ ╚════██║██╔══╝ ██║ ██║ ██║██╔══██╗██║ ██║ ╚██╔╝ ║
║ ███████║███████╗╚██████╗╚██████╔╝██║ ██║██║ ██║ ██║ ║
║ ╚══════╝╚══════╝ ╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚═╝ ╚═╝ ╚═╝ ║
║ ║
║ ██╗ █████╗ ██████╗ ║
║ ██║ ██╔══██╗██╔══██╗ ║
║ ██║ ███████║██████╔╝ ║
║ ██║ ██╔══██║██╔══██╗ ║
║ ███████╗██║ ██║██████╔╝ ║
║ ╚══════╝╚═╝ ╚═╝╚═════╝ ║
║ ║
║ TASKFLOW AGENT ║
║ ║
╠══════════════════════════════════════════════════════════════════╣
║ AI API Endpoint: {api_endpoint:<48}║
╚══════════════════════════════════════════════════════════════════╝
"""
return banner