mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-17 23:51:19 +00:00
5fc57e7f4e
- Add BaseClient, BaseFileStore, BaseFileWatcher, BaseJob, BaseService, and BaseStep classes - Move component initialization logic from ApplicationContext to Application class - Add logo printing and logging initialization in Application startup - Create client module with base client implementation - Add file store base class with embedding resolution and validation - Implement file watcher base class with watchfiles integration - Add job base class for sequential step execution orchestration - Create service base class for job exposure mechanisms - Refactor BaseStep with LLM workflow execution capabilities - Add case converter utility for naming convention transformations - Update import structure and module organization - Add proper type hints and docstrings across all components - Implement component registry integration for dynamic loading - Add error handling for missing backend configurations
91 lines
3.8 KiB
Python
91 lines
3.8 KiB
Python
"""Terminal branding and configuration display utilities."""
|
|
|
|
import importlib.metadata
|
|
from typing import TYPE_CHECKING
|
|
|
|
from rich.console import Console, Group
|
|
from rich.panel import Panel
|
|
from rich.table import Table
|
|
from rich.text import Text
|
|
|
|
if TYPE_CHECKING:
|
|
from ..schema import ApplicationConfig
|
|
|
|
|
|
def get_version(package_name: str) -> str:
|
|
"""Return the installed version of a package or 'unknown'."""
|
|
try:
|
|
return importlib.metadata.version(package_name)
|
|
except importlib.metadata.PackageNotFoundError:
|
|
return ""
|
|
|
|
|
|
def print_logo(app_config: "ApplicationConfig"):
|
|
"""Print a stylized ASCII logo and service metadata to the console."""
|
|
ascii_art = [
|
|
r" ██████╗ ███████╗ ███╗ ███╗ ███████╗ ",
|
|
r" ██╔══██╗ ██╔════╝ ████╗ ████║ ██╔════╝ ",
|
|
r" ██████╔╝ █████╗ ██╔████╔██║ █████╗ ",
|
|
r" ██╔══██╗ ██╔══╝ ██║╚██╔╝██║ ██╔══╝ ",
|
|
r" ██║ ██║ ███████╗ ██║ ╚═╝ ██║ ███████╗ ",
|
|
r" ╚═╝ ╚═╝ ╚══════╝ ╚═╝ ╚═╝ ╚══════╝ ",
|
|
]
|
|
|
|
start_color = (85, 239, 196)
|
|
end_color = (162, 155, 254)
|
|
|
|
logo_text = Text()
|
|
for line in ascii_art:
|
|
line_len = max(1, len(line) - 1)
|
|
for i, char in enumerate(line):
|
|
# Calculate gradient shift per character
|
|
ratio = i / line_len
|
|
rgb = tuple(int(s + (e - s) * ratio) for s, e in zip(start_color, end_color))
|
|
logo_text.append(char, style=f"bold rgb({rgb[0]},{rgb[1]},{rgb[2]})")
|
|
logo_text.append("\n")
|
|
|
|
# Layout configuration info
|
|
info_table = Table.grid(padding=(0, 1))
|
|
info_table.add_column(style="bold", justify="center")
|
|
info_table.add_column(style="bold cyan", justify="left")
|
|
info_table.add_column(style="white", justify="left")
|
|
|
|
# Get service config (ComponentConfig with extra="allow")
|
|
service = app_config.service
|
|
backend = service.backend
|
|
|
|
# Add core service info
|
|
info_table.add_row("📦", "Backend:", backend)
|
|
|
|
match backend:
|
|
case "http":
|
|
host = service.model_extra.get("host", "localhost") if service.model_extra else "localhost"
|
|
port = service.model_extra.get("port", 8000) if service.model_extra else 8000
|
|
info_table.add_row("🔗", "URL:", f"http://{host}:{port}")
|
|
info_table.add_row("📚", "FastAPI:", Text(get_version("fastapi"), style="dim"))
|
|
case "mcp":
|
|
transport = service.model_extra.get("transport", "stdio") if service.model_extra else "stdio"
|
|
info_table.add_row("🚌", "Transport:", transport)
|
|
if transport != "stdio":
|
|
host = service.model_extra.get("host", "localhost") if service.model_extra else "localhost"
|
|
port = service.model_extra.get("port", 8000) if service.model_extra else 8000
|
|
url = f"http://{host}:{port}"
|
|
if transport == "sse":
|
|
url += "/sse"
|
|
info_table.add_row("🔗", "URL:", url)
|
|
info_table.add_row("📚", "FastMCP:", Text(get_version("fastmcp"), style="dim"))
|
|
|
|
info_table.add_row("🚀", "ReMe:", Text(get_version("reme-ai"), style="dim"))
|
|
|
|
# Render layout within a panel
|
|
panel = Panel(
|
|
Group(logo_text, info_table),
|
|
title=app_config.app_name,
|
|
title_align="left",
|
|
border_style="dim",
|
|
padding=(1, 4),
|
|
expand=False,
|
|
)
|
|
|
|
# use justify="center" to adjust position
|
|
Console().print(Group("\n", panel, "\n"))
|