sa-assistant 0.1.1__py3-none-any.whl

cli/components.py

@@ -0,0 +1,346 @@
+"""SA Assistant CLI Components - UI 컴포넌트
+
+Rich 기반 UI 컴포넌트들을 제공합니다.
+"""
+
+from typing import Optional, Dict, Any
+from rich.console import Console, Group
+from rich.panel import Panel
+from rich.table import Table
+from rich.text import Text
+from rich.markdown import Markdown
+from rich.syntax import Syntax
+from rich.live import Live
+from rich.spinner import Spinner
+from rich.progress import Progress, SpinnerColumn, TextColumn
+
+from .console import get_console, SA_THEME
+
+
+# =============================================================================
+# 배너 및 헬프
+# =============================================================================
+
+BANNER = """
+[aws.orange]╔══════════════════════════════════════════════════════════════════════╗[/]
+[aws.orange]║[/]                                                                      [aws.orange]║[/]
+[aws.orange]║[/]   [bold white]AWS SA Assistant[/]                                                  [aws.orange]║[/]
+[aws.orange]║[/]   [dim]Solutions Architect Professional Agent[/]                            [aws.orange]║[/]
+[aws.orange]║[/]                                                                      [aws.orange]║[/]
+[aws.orange]║[/]   [cyan]Powered by Claude with Extended Thinking[/]                          [aws.orange]║[/]
+[aws.orange]║[/]   [cyan]+ Guru Sub-agents (Bezos, Vogels, Naval, Feynman)[/]                  [aws.orange]║[/]
+[aws.orange]║[/]   [cyan]+ Specialist Agents (Explorer, Researcher, Reviewer)[/]               [aws.orange]║[/]
+[aws.orange]║[/]                                                                      [aws.orange]║[/]
+[aws.orange]╚══════════════════════════════════════════════════════════════════════╝[/]
+"""
+
+
+def print_banner() -> None:
+    """SA Assistant 배너를 출력합니다."""
+    console = get_console()
+    console.print(BANNER)
+
+
+def print_help() -> None:
+    """도움말을 출력합니다."""
+    console = get_console()
+
+    help_table = Table(show_header=False, box=None, padding=(0, 2))
+    help_table.add_column("Command", style="cyan")
+    help_table.add_column("Description", style="dim")
+
+    help_table.add_row("Enter", "메시지 전송")
+    help_table.add_row("Esc+Enter / Alt+Enter", "줄바꿈")
+    help_table.add_row("↑ / ↓", "히스토리 탐색")
+    help_table.add_row("Ctrl+C", "입력 취소")
+    help_table.add_row("exit / quit", "종료")
+
+    console.print()
+    console.print(
+        Panel(
+            help_table,
+            title="[bold]키보드 단축키[/]",
+            border_style="dim",
+            padding=(0, 1),
+        )
+    )
+
+
+def print_specialists_info(specialists: list[str]) -> None:
+    """사용 가능한 Specialist 정보를 출력합니다."""
+    console = get_console()
+
+    specialist_descriptions = {
+        "explorer": ("🔍", "아키텍처 분석, 다이어그램 해석, 코드 탐색"),
+        "researcher": ("📚", "AWS 문서 검색, Best Practice, 가격 정보"),
+        "reviewer": ("✅", "Well-Architected 검토, 보안 분석, 개선 권장"),
+    }
+
+    guru_descriptions = {
+        "bezos": ("👔", "고객 중심 사고, 장기 비전"),
+        "vogels": ("🔧", "분산 시스템, 확장성"),
+        "naval": ("💡", "레버리지, ROI 분석"),
+        "feynman": ("🎓", "개념 단순화, 설명"),
+    }
+
+    table = Table(show_header=True, header_style="bold", box=None)
+    table.add_column("Agent", style="cyan")
+    table.add_column("Type", style="dim")
+    table.add_column("전문 분야")
+
+    for name, (emoji, desc) in specialist_descriptions.items():
+        table.add_row(f"{emoji} {name}", "Specialist", desc)
+
+    for name, (emoji, desc) in guru_descriptions.items():
+        table.add_row(f"{emoji} {name}", "Guru", desc)
+
+    console.print()
+    console.print(
+        Panel(
+            table,
+            title="[bold]사용 가능한 에이전트[/]",
+            border_style="cyan",
+            padding=(0, 1),
+        )
+    )
+
+
+# =============================================================================
+# 에이전트 패널
+# =============================================================================
+
+
+class AgentPanel:
+    """에이전트 실행 상태를 표시하는 패널"""
+
+    AGENT_COLORS = {
+        "orchestrator": "aws.orange",
+        "guru": "agent.guru",
+        "specialist": "agent.specialist",
+        "bezos": "agent.guru",
+        "vogels": "agent.guru",
+        "naval": "agent.guru",
+        "feynman": "agent.guru",
+        "explorer": "agent.explorer",
+        "researcher": "agent.researcher",
+        "reviewer": "agent.reviewer",
+    }
+
+    AGENT_EMOJIS = {
+        "orchestrator": "🎯",
+        "bezos": "👔",
+        "vogels": "🔧",
+        "naval": "💡",
+        "feynman": "🎓",
+        "explorer": "🔍",
+        "researcher": "📚",
+        "reviewer": "✅",
+    }
+
+    def __init__(self, agent_name: str, agent_type: str = "specialist"):
+        self.agent_name = agent_name
+        self.agent_type = agent_type
+        self.console = get_console()
+
+    def print_start(self, task: str = None) -> None:
+        """에이전트 시작을 표시합니다."""
+        emoji = self.AGENT_EMOJIS.get(self.agent_name.lower(), "🤖")
+        color = self.AGENT_COLORS.get(self.agent_name.lower(), "cyan")
+
+        text = f"{emoji} [{color}]{self.agent_name.upper()}[/] 에이전트 호출 중..."
+        if task:
+            text += f" ({task[:50]}...)" if len(task) > 50 else f" ({task})"
+
+        self.console.print(text)
+
+    def print_result(self, success: bool = True, message: str = None) -> None:
+        """에이전트 결과를 표시합니다."""
+        emoji = self.AGENT_EMOJIS.get(self.agent_name.lower(), "🤖")
+        color = self.AGENT_COLORS.get(self.agent_name.lower(), "cyan")
+
+        if success:
+            status = "[success]완료[/]"
+        else:
+            status = "[error]실패[/]"
+
+        text = f"{emoji} [{color}]{self.agent_name.upper()}[/] {status}"
+        if message:
+            text += f": {message}"
+
+        self.console.print(text)
+
+
+# =============================================================================
+# 도구 상태
+# =============================================================================
+
+
+class ToolStatus:
+    """도구 실행 상태를 관리합니다."""
+
+    TOOL_EMOJIS = {
+        "consult_guru": "🧙",
+        "consult_specialist": "🔬",
+        "file_read": "📄",
+        "file_write": "💾",
+        "shell": "🖥️",
+        "image_reader": "🖼️",
+        "code_interpreter": "⚡",
+        "skill": "📚",
+        "list_gurus": "📋",
+        "list_specialists": "📋",
+    }
+
+    def __init__(self):
+        self.console = get_console()
+        self._active_tools: Dict[str, str] = {}  # tool_id -> tool_name
+
+    def tool_start(
+        self, tool_id: str, tool_name: str, args: Dict[str, Any] = None
+    ) -> None:
+        """도구 시작을 표시합니다."""
+        if tool_id in self._active_tools:
+            return  # 이미 표시됨
+
+        self._active_tools[tool_id] = tool_name
+        emoji = self.TOOL_EMOJIS.get(tool_name, "🔧")
+
+        # 특별한 도구들에 대한 상세 정보
+        detail = ""
+        if args:
+            if tool_name == "consult_guru":
+                guru_name = args.get("guru_name", "")
+                detail = f" → {guru_name.upper()}"
+            elif tool_name == "consult_specialist":
+                specialist_name = args.get("specialist_name", "")
+                detail = f" → {specialist_name.upper()}"
+            elif tool_name == "skill":
+                skill_name = args.get("skill_name", "")
+                detail = f" → {skill_name}"
+            elif tool_name in ("file_read", "file_write"):
+                path = args.get("path", args.get("file_path", ""))
+                if path:
+                    # 경로가 길면 축약
+                    if len(path) > 40:
+                        path = "..." + path[-37:]
+                    detail = f" → {path}"
+
+        self.console.print(f"[tool.running]{emoji} {tool_name}{detail}[/]")
+
+    def tool_end(self, tool_id: str, success: bool = True, error: str = None) -> None:
+        """도구 종료를 표시합니다. (선택적)"""
+        if tool_id in self._active_tools:
+            del self._active_tools[tool_id]
+
+        # 에러가 있을 때만 표시
+        if not success and error:
+            self.console.print(f"[tool.error]✗ 오류: {error}[/]")
+
+
+# =============================================================================
+# 응답 렌더러
+# =============================================================================
+
+
+class ResponseRenderer:
+    """에이전트 응답을 렌더링합니다."""
+
+    def __init__(self):
+        self.console = get_console()
+        self._buffer = ""
+
+    def stream_text(self, text: str) -> None:
+        """스트리밍 텍스트를 출력합니다."""
+        print(text, end="", flush=True)
+        self._buffer += text
+
+    def render_markdown(self, text: str) -> None:
+        """마크다운을 렌더링합니다."""
+        md = Markdown(text)
+        self.console.print(md)
+
+    def render_code(self, code: str, language: str = "python") -> None:
+        """코드를 구문 강조하여 렌더링합니다."""
+        syntax = Syntax(code, language, theme="monokai", line_numbers=True)
+        self.console.print(syntax)
+
+    def render_thinking(self, text: str) -> None:
+        """Thinking 텍스트를 렌더링합니다."""
+        self.console.print(
+            Panel(
+                Text(text, style="text.thinking"),
+                title="[dim]💭 Thinking[/]",
+                border_style="dim",
+            )
+        )
+
+    def clear_buffer(self) -> str:
+        """버퍼를 비우고 내용을 반환합니다."""
+        content = self._buffer
+        self._buffer = ""
+        return content
+
+
+# =============================================================================
+# 프로그레스 및 스피너
+# =============================================================================
+
+
+def create_spinner(text: str = "처리 중...") -> Progress:
+    """스피너 Progress 인스턴스를 생성합니다."""
+    return Progress(
+        SpinnerColumn(),
+        TextColumn("[progress.description]{task.description}"),
+        console=get_console(),
+        transient=True,
+    )
+
+
+class ThinkingSpinner:
+    """Thinking 상태를 표시하는 스피너"""
+
+    def __init__(self):
+        self.console = get_console()
+        self._live: Optional[Live] = None
+
+    def start(self) -> None:
+        """스피너를 시작합니다."""
+        spinner = Spinner("dots", text="[text.thinking]생각 중...[/]")
+        self._live = Live(spinner, console=self.console, transient=True)
+        self._live.start()
+
+    def update(self, text: str) -> None:
+        """스피너 텍스트를 업데이트합니다."""
+        if self._live:
+            spinner = Spinner("dots", text=f"[text.thinking]{text}[/]")
+            self._live.update(spinner)
+
+    def stop(self) -> None:
+        """스피너를 중지합니다."""
+        if self._live:
+            self._live.stop()
+            self._live = None
+
+
+# =============================================================================
+# 유틸리티
+# =============================================================================
+
+
+def format_duration(seconds: float) -> str:
+    """시간을 포맷팅합니다."""
+    if seconds < 1:
+        return f"{seconds * 1000:.0f}ms"
+    elif seconds < 60:
+        return f"{seconds:.1f}s"
+    else:
+        minutes = int(seconds // 60)
+        secs = seconds % 60
+        return f"{minutes}m {secs:.0f}s"
+
+
+def truncate_text(text: str, max_length: int = 100) -> str:
+    """텍스트를 축약합니다."""
+    if len(text) <= max_length:
+        return text
+    return text[: max_length - 3] + "..."

cli/console.py

@@ -0,0 +1,106 @@
+"""SA Assistant Console - Rich 기반 콘솔 래퍼
+
+Rich 라이브러리를 사용하여 터미널 UI를 개선합니다.
+"""
+
+from typing import Optional
+from rich.console import Console
+from rich.theme import Theme
+from rich.style import Style
+
+# SA Assistant 커스텀 테마 - AWS 오렌지 기반
+SA_THEME = Theme(
+    {
+        # AWS 컬러 스킴
+        "aws.orange": "bold #FF9900",
+        "aws.dark": "#232F3E",
+        "aws.blue": "#1A73E8",
+        # 에이전트 타입별 색상
+        "agent.orchestrator": "bold #FF9900",
+        "agent.guru": "bold magenta",
+        "agent.specialist": "bold cyan",
+        "agent.explorer": "bold green",
+        "agent.researcher": "bold blue",
+        "agent.reviewer": "bold yellow",
+        # 도구 상태 색상
+        "tool.running": "cyan",
+        "tool.success": "green",
+        "tool.error": "red",
+        "tool.pending": "dim",
+        # 텍스트 스타일
+        "text.user": "bold green",
+        "text.assistant": "bold #FF9900",
+        "text.thinking": "dim italic",
+        "text.error": "bold red",
+        "text.warning": "yellow",
+        "text.info": "cyan",
+        "text.dim": "dim",
+        # 특수 용도
+        "highlight": "bold #FF9900",
+        "success": "bold green",
+        "error": "bold red",
+        "warning": "yellow",
+    }
+)
+
+
+class SAConsole:
+    """SA Assistant 전용 Rich Console 래퍼
+
+    싱글톤 패턴으로 전역 콘솔 인스턴스를 관리합니다.
+    """
+
+    _instance: Optional[Console] = None
+
+    @classmethod
+    def get_console(cls) -> Console:
+        """전역 Console 인스턴스를 반환합니다."""
+        if cls._instance is None:
+            cls._instance = Console(
+                theme=SA_THEME,
+                highlight=True,
+                markup=True,
+                emoji=True,
+            )
+        return cls._instance
+
+    @classmethod
+    def reset(cls) -> None:
+        """Console 인스턴스를 리셋합니다. (테스트용)"""
+        cls._instance = None
+
+
+# 편의 함수들
+def get_console() -> Console:
+    """전역 Console 인스턴스를 반환합니다."""
+    return SAConsole.get_console()
+
+
+def print_styled(text: str, style: str = None, **kwargs) -> None:
+    """스타일이 적용된 텍스트를 출력합니다."""
+    console = get_console()
+    console.print(text, style=style, **kwargs)
+
+
+def print_error(text: str) -> None:
+    """에러 메시지를 출력합니다."""
+    console = get_console()
+    console.print(f"[text.error]✗ {text}[/]")
+
+
+def print_warning(text: str) -> None:
+    """경고 메시지를 출력합니다."""
+    console = get_console()
+    console.print(f"[text.warning]⚠ {text}[/]")
+
+
+def print_info(text: str) -> None:
+    """정보 메시지를 출력합니다."""
+    console = get_console()
+    console.print(f"[text.info]ℹ {text}[/]")
+
+
+def print_success(text: str) -> None:
+    """성공 메시지를 출력합니다."""
+    console = get_console()
+    console.print(f"[success]✓ {text}[/]")

cli/mdstream.py

@@ -0,0 +1,236 @@
+"""SA Assistant Markdown Stream - Streaming markdown renderer with Rich Live display."""
+
+import io
+import time
+from typing import Optional
+
+from rich import box
+from rich.console import Console
+from rich.live import Live
+from rich.markdown import Markdown, CodeBlock, Heading
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.text import Text
+
+from .console import get_console
+
+
+class SACodeBlock(CodeBlock):
+    """Code block with minimal padding and monokai theme."""
+
+    def __rich_console__(self, console, options):
+        code = str(self.text).rstrip()
+        syntax = Syntax(
+            code,
+            self.lexer_name,
+            theme="monokai",
+            word_wrap=True,
+            padding=(1, 2),
+            background_color="#1a1a1a",
+        )
+        yield syntax
+
+
+class SAHeading(Heading):
+    """Left-aligned heading with AWS color scheme."""
+
+    def __rich_console__(self, console, options):
+        text = self.text
+        text.justify = "left"
+
+        if self.tag == "h1":
+            yield Panel(
+                text,
+                box=box.HEAVY,
+                style="bold #FF9900",
+                border_style="#FF9900",
+            )
+        elif self.tag == "h2":
+            yield Text("")
+            text.stylize("bold cyan")
+            yield text
+        else:
+            text.stylize("bold")
+            yield text
+
+
+class SAMarkdown(Markdown):
+    """Custom markdown renderer with SA Assistant styling."""
+
+    elements = {
+        **Markdown.elements,
+        "fence": SACodeBlock,
+        "code_block": SACodeBlock,
+        "heading_open": SAHeading,
+    }
+
+
+class MarkdownStream:
+    """Streaming markdown renderer using Rich Live display.
+
+    Renders LLM responses in real-time with proper markdown formatting.
+    Uses sliding window approach: stable lines go to console,
+    unstable lines stay in Live area for smooth updates.
+
+    Usage:
+        stream = MarkdownStream()
+        for chunk in llm_response:
+            stream.update(accumulated_text)
+        stream.update(final_text, final=True)
+    """
+
+    MIN_FPS = 20
+    MAX_DELAY = 2.0
+
+    def __init__(
+        self,
+        console: Optional[Console] = None,
+        live_window: int = 6,
+        min_delay: float = 1.0 / 20,
+    ):
+        self.console = console or get_console()
+        self.live_window = live_window
+        self.min_delay = min_delay
+
+        self.live: Optional[Live] = None
+        self.when: float = 0
+        self.printed: list = []
+        self._live_started: bool = False
+        self._text_buffer: str = ""
+
+    def _render_markdown_to_lines(self, text: str) -> list:
+        string_io = io.StringIO()
+        temp_console = Console(
+            file=string_io, force_terminal=True, width=self.console.width
+        )
+
+        markdown = SAMarkdown(text)
+        temp_console.print(markdown)
+        output = string_io.getvalue()
+
+        return output.splitlines(keepends=True)
+
+    def update(self, text: str, final: bool = False) -> None:
+        """Update displayed markdown content."""
+        self._text_buffer = text
+
+        if not self._live_started:
+            self.live = Live(
+                Text(""),
+                console=self.console,
+                refresh_per_second=1.0 / self.min_delay,
+                transient=True,
+            )
+            self.live.start()
+            self._live_started = True
+
+        now = time.time()
+        if not final and now - self.when < self.min_delay:
+            return
+        self.when = now
+
+        start = time.time()
+        lines = self._render_markdown_to_lines(text)
+        render_time = time.time() - start
+
+        self.min_delay = min(max(render_time * 10, 1.0 / self.MIN_FPS), self.MAX_DELAY)
+
+        num_lines = len(lines)
+
+        if not final:
+            num_lines -= self.live_window
+
+        if final or num_lines > 0:
+            num_printed = len(self.printed)
+            show_count = num_lines - num_printed
+
+            if show_count > 0 and self.live is not None:
+                show = lines[num_printed:num_lines]
+                show_text = "".join(show)
+                show_rich = Text.from_ansi(show_text)
+                self.live.console.print(show_rich, end="")
+
+                self.printed = lines[:num_lines]
+
+        if final:
+            if self.live:
+                self.live.update(Text(""))
+                self.live.stop()
+                self.live = None
+            return
+
+        rest = lines[num_lines:]
+        rest_text = "".join(rest)
+        rest_rich = Text.from_ansi(rest_text)
+        if self.live:
+            self.live.update(rest_rich)
+
+    def finish(self) -> None:
+        """Finalize stream and output remaining content."""
+        if self._text_buffer:
+            self.update(self._text_buffer, final=True)
+        elif self.live:
+            self.live.stop()
+            self.live = None
+
+    def __del__(self):
+        if self.live:
+            try:
+                self.live.stop()
+            except Exception:
+                pass
+
+
+class SimpleMarkdownStream:
+    """Simple markdown stream - renders final result only.
+
+    Outputs plain text during streaming, renders full markdown on completion.
+    """
+
+    def __init__(self, console: Optional[Console] = None):
+        self.console = console or get_console()
+        self._buffer = ""
+        self._last_printed_len = 0
+
+    def update(self, text: str, final: bool = False) -> None:
+        """Update text content."""
+        if final:
+            print()
+            self.console.print(SAMarkdown(text))
+        else:
+            new_text = text[self._last_printed_len :]
+            if new_text:
+                print(new_text, end="", flush=True)
+                self._last_printed_len = len(text)
+
+        self._buffer = text
+
+    def finish(self) -> None:
+        """Finalize stream."""
+        if self._buffer:
+            self.update(self._buffer, final=True)
+
+
+def render_markdown(text: str, console: Optional[Console] = None) -> None:
+    """Render markdown text to console."""
+    console = console or get_console()
+    console.print(SAMarkdown(text))
+
+
+def render_code(
+    code: str,
+    language: str = "python",
+    console: Optional[Console] = None,
+    line_numbers: bool = True,
+) -> None:
+    """Render code with syntax highlighting."""
+    console = console or get_console()
+    syntax = Syntax(
+        code,
+        language,
+        theme="monokai",
+        line_numbers=line_numbers,
+        word_wrap=True,
+        background_color="#1a1a1a",
+    )
+    console.print(syntax)

mcp_client/client.py

@@ -0,0 +1,12 @@
+from mcp.client.streamable_http import streamablehttp_client
+from strands.tools.mcp.mcp_client import MCPClient
+
+# ExaAI provides information about code through web searches, crawling and code context searches through their platform. Requires no authentication
+EXAMPLE_MCP_ENDPOINT = "https://mcp.exa.ai/mcp"
+
+def get_streamable_http_mcp_client() -> MCPClient:
+    """
+    Returns an MCP Client compatible with Strands
+    """
+    # to use an MCP server that supports bearer authentication, add headers={"Authorization": f"Bearer {access_token}"}
+    return MCPClient(lambda: streamablehttp_client(EXAMPLE_MCP_ENDPOINT))