aipa-cli 0.1.0__py3-none-any.whl

aipriceaction_terminal/agents/personas.py

@@ -0,0 +1,175 @@
+"""Persona definitions and registry for multi-agent support."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from aipriceaction.system import get_system_prompt
+
+
+@dataclass
+class Persona:
+    """An agent persona with custom instructions."""
+
+    id: str
+    name: str
+    description: str
+    extra_instructions: str = ""
+    include_data_policy: bool = True
+    include_analysis_framework: bool = True
+    include_ma_score: bool = True
+    include_disclaimer: bool = True
+
+    def build_system_prompt(self, lang: str) -> str:
+        """Build the full system prompt for this persona."""
+        prompt = get_system_prompt(
+            lang,
+            include_data_policy=self.include_data_policy,
+            include_analysis_framework=self.include_analysis_framework,
+            include_ma_score=self.include_ma_score,
+            include_disclaimer=self.include_disclaimer,
+            include_persona=True,
+        )
+        if self.extra_instructions:
+            prompt += "\n\n" + self.extra_instructions
+        return prompt
+
+
+# -- Bilingual extra instructions --
+
+_GENERAL_INSTRUCTIONS = {
+    "en": """## Tool Usage
+
+You have tools to fetch OHLCV data and list available tickers:
+- `get_ohlcv_data`: Fetch price data for any ticker with MA indicators and scores.
+- `get_ticker_list`: Discover available tickers grouped by sector/industry.
+
+When the user asks about a specific ticker, market sector, or price-related question, you MUST call the relevant tools before answering. Do not answer from memory alone — always fetch fresh data.
+
+For non-market questions (greetings, general knowledge, etc.), respond naturally without calling tools.
+
+## Research Workflow (when analyzing tickers)
+1. Call `get_ohlcv_data` for each ticker mentioned by the user.
+2. If relevant, call `get_ticker_list` to discover related tickers in the same sector.
+3. Base your analysis ONLY on the tool results.
+4. Assess: trend direction, VPA signals, MA score momentum, volume confirmation.
+5. Structure your answer clearly with specific data points.
+6. Include the investment disclaimer at the end of any financial analysis.""",
+    "vn": """## Sử Dụng Công Cụ
+
+Bạn có các công cụ để lấy dữ liệu OHLCV và danh sách mã chứng khoán:
+- `get_ohlcv_data`: Lấy dữ liệu giá cho bất kỳ mã nào với chỉ báo MA và điểm số.
+- `get_ticker_list`: Khám phá các mã chứng khoán theo nhóm ngành.
+
+Khi người dùng hỏi về mã cụ thể, ngành, hoặc câu hỏi liên quan giá — bạn PHẢI gọi tools trước khi trả lời. Không trả lời từ trí nhớ — luôn lấy dữ liệu mới nhất.
+
+Với câu hỏi ngoài thị trường (chào hỏi, kiến thức chung), trả lời tự nhiên không cần gọi tools.
+
+## Quy Trình Nghiên Cứu (khi phân tích mã)
+1. Gọi `get_ohlcv_data` cho mỗi mã người dùng nhắc đến.
+2. Nếu cần, gọi `get_ticker_list` để tìm mã cùng ngành.
+3. Phân tích CHỈ dựa trên kết quả tools.
+4. Đánh giá: xu hướng, tín hiệu VPA, động lực MA score, khối lượng.
+5. Cấu trúc câu trả lời rõ ràng với số liệu cụ thể.
+6. Bao gồm tuyên bố miễn trách nhiệm đầu tư cuối phân tích.""",
+}
+
+_ANALYST_INSTRUCTIONS = {
+    "en": """## Tool Usage
+
+You have tools to fetch OHLCV data and list available tickers:
+- `get_ohlcv_data`: Fetch price data for any ticker with MA indicators and scores.
+- `get_ticker_list`: Discover available tickers grouped by sector/industry.
+
+## Research Workflow (MANDATORY)
+1. First, call `get_ohlcv_data` for EACH ticker explicitly mentioned in the user question.
+2. Then, call `get_ticker_list` to discover other tickers in the same sectors/industries.
+3. Call `get_ohlcv_data` for at least 2-3 additional tickers per sector to enable meaningful comparison. Do NOT skip this step.
+4. For each ticker, assess: trend direction, VPA signals (accumulation/distribution), MA score momentum across timeframes, volume confirmation, and support/resistance.
+5. Structure your final answer with:
+   - Per-ticker analysis with specific data points from the tool results
+   - Sector rotation observations (which sectors are leading/lagging)
+   - Multi-ticker ranking table
+6. Include the investment disclaimer at the end.
+
+FAILURE TO CALL TOOLS = INVALID RESPONSE.""",
+    "vn": """## Sử Dụng Công Cụ
+
+Bạn có các công cụ để lấy dữ liệu OHLCV và danh sách mã:
+- `get_ohlcv_data`: Lấy dữ liệu giá cho bất kỳ mã nào với chỉ báo MA.
+- `get_ticker_list`: Khám phá các mã theo nhóm ngành.
+
+## Quy Trình Nghiên Cứu (BẮT BUỘC)
+1. Gọi `get_ohlcv_data` cho MỖI mã được nhắc đến trong câu hỏi.
+2. Gọi `get_ticker_list` để tìm mã cùng ngành.
+3. Gọi `get_ohlcv_data` cho ít nhất 2-3 mã thêm mỗi ngành để so sánh. KHÔNG được bỏ qua bước này.
+4. Mỗi mã: đánh giá xu hướng, tín hiệu VPA, động lực MA score, khối lượng, hỗ trợ/kháng cự.
+5. Cấu trúc câu trả lời:
+   - Phân tích từng mã với số liệu cụ thể từ tools
+   - Quan sát luân chuyển ngành (ngành dẫn đầu/lagging)
+   - Bảng xếp hạng đa mã
+6. Bao gồm tuyên bố miễn trách nhiệm đầu tư ở cuối.
+
+KHÔNG GỌI TOOL = PHẢN HỒI KHÔNG HỢP LỆ.""",
+}
+
+
+def _bilingual(texts: dict[str, str], lang: str) -> str:
+    return texts.get(lang, texts["en"])
+
+
+class PersonaRegistry:
+    """Registry for agent personas."""
+
+    def __init__(self) -> None:
+        self._personas: dict[str, Persona] = {}
+        self._default_id: str = ""
+
+    def register(self, persona: Persona, *, is_default: bool = False) -> None:
+        self._personas[persona.id] = persona
+        if is_default or not self._default_id:
+            self._default_id = persona.id
+
+    def get(self, persona_id: str) -> Persona | None:
+        return self._personas.get(persona_id)
+
+    def list_personas(self) -> list[Persona]:
+        return list(self._personas.values())
+
+    @property
+    def default_id(self) -> str:
+        return self._default_id
+
+
+def get_default_personas(lang: str = "en") -> PersonaRegistry:
+    """Return a PersonaRegistry with the built-in personas."""
+    registry = PersonaRegistry()
+
+    registry.register(
+        Persona(
+            id="general",
+            name="General Advisor",
+            description="Handles both market and non-market questions. Auto-calls tools when needed.",
+            extra_instructions=_bilingual(_GENERAL_INSTRUCTIONS, lang),
+        ),
+        is_default=True,
+    )
+
+    registry.register(
+        Persona(
+            id="analyst",
+            name="Deep Analyst",
+            description="Deep multi-ticker specialist with mandatory research workflow.",
+            extra_instructions=_bilingual(_ANALYST_INSTRUCTIONS, lang),
+        ),
+    )
+
+    return registry
+
+
+def get_default_persona(lang: str = "en") -> Persona:
+    """Return the default (general) persona."""
+    registry = get_default_personas(lang)
+    persona = registry.get(registry.default_id)
+    assert persona is not None
+    return persona

aipriceaction_terminal/agents/tools.py

@@ -0,0 +1,152 @@
+"""Tool registry and built-in tools for the agent."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import TYPE_CHECKING
+
+from langchain_core.tools import tool
+
+if TYPE_CHECKING:
+    from aipriceaction import AIPriceAction, AIContextBuilder
+
+
+@dataclass
+class ToolDef:
+    """Wraps a LangChain BaseTool with metadata."""
+
+    tool: object  # BaseTool
+    name: str
+    description: str
+    category: str = "general"
+
+
+class ToolRegistry:
+    """Registry for agent tools."""
+
+    def __init__(self) -> None:
+        self._tools: list[ToolDef] = []
+
+    def register(self, tool_def: ToolDef) -> None:
+        self._tools.append(tool_def)
+
+    def unregister(self, name: str) -> None:
+        self._tools = [t for t in self._tools if t.name != name]
+
+    def get_tools(self, category: str | None = None) -> list[object]:
+        if category is None:
+            return [t.tool for t in self._tools]
+        return [t.tool for t in self._tools if t.category == category]
+
+    def get_tool_names(self, category: str | None = None) -> list[str]:
+        if category is None:
+            return [t.name for t in self._tools]
+        return [t.name for t in self._tools if t.category == category]
+
+
+# -- Lazy singletons (mirrors langchain_agent.py pattern) --
+
+_client: AIPriceAction | None = None
+_builder: AIContextBuilder | None = None
+
+
+def _ensure_clients(lang: str = "en") -> tuple[AIPriceAction, AIContextBuilder]:
+    global _client, _builder
+    if _client is None:
+        from aipriceaction import AIPriceAction
+        _client = AIPriceAction()
+    if _builder is None or _builder._lang != lang:
+        from aipriceaction import AIContextBuilder
+        _builder = AIContextBuilder(lang=lang)
+    return _client, _builder
+
+
+def _reset_clients() -> None:
+    global _client, _builder
+    _client = None
+    _builder = None
+
+
+# -- Tool factories --
+
+
+def create_ohlcv_tool(lang: str = "en") -> ToolDef:
+    """Factory: creates the get_ohlcv_data tool."""
+
+    @tool
+    def get_ohlcv_data(ticker: str, interval: str = "1D", limit: int = 5) -> str:
+        """Fetch OHLCV data for a ticker. Returns formatted context with MA indicators and scores.
+
+        Args:
+            ticker: Ticker symbol (e.g. VCB, FPT, BTCUSDT).
+            interval: Time interval — "1D" (default), "1h", or "1m".
+            limit: Number of bars to return (default 5).
+        """
+        _, builder = _ensure_clients(lang)
+        try:
+            ctx = builder.build(
+                ticker=ticker,
+                interval=interval,
+                limit=limit,
+                reference_ticker=None,
+                include_system_prompt=False,
+            )
+        except Exception as e:
+            return f"Error fetching {ticker}: {e}"
+        if not ctx.strip():
+            return f"No data found for {ticker} ({interval})."
+        return ctx
+
+    return ToolDef(
+        tool=get_ohlcv_data,
+        name="get_ohlcv_data",
+        description="Fetch OHLCV data for a ticker with MA indicators and scores.",
+        category="market_data",
+    )
+
+
+def create_ticker_list_tool(lang: str = "en") -> ToolDef:
+    """Factory: creates the get_ticker_list tool."""
+
+    @tool
+    def get_ticker_list(source: str | None = None) -> str:
+        """List available ticker symbols and metadata.
+
+        Args:
+            source: Filter by source — "vn", "yahoo", "crypto", "sjc". None = all.
+        """
+        client, _ = _ensure_clients(lang)
+        tickers = client.get_tickers(source=source)
+        if not tickers:
+            return "No tickers found."
+
+        from collections import Counter
+
+        source_counts = Counter(t.source for t in tickers)
+        group_counts = Counter(t.group for t in tickers if t.group)
+
+        lines = [f"Available tickers (source={source or 'all'}), total: {len(tickers)}"]
+        lines.append("Counts by source: " + ", ".join(f"{s}={c}" for s, c in source_counts.most_common()))
+        lines.append("Groups: " + ", ".join(f"{g}={c}" for g, c in group_counts.most_common(15)))
+        lines.append("")
+
+        # Symbols only, comma-separated for compactness
+        symbols = [t.ticker for t in tickers]
+        lines.append("Symbols: " + ", ".join(symbols))
+
+        return "\n".join(lines)
+
+    return ToolDef(
+        tool=get_ticker_list,
+        name="get_ticker_list",
+        description="List available ticker symbols and metadata.",
+        category="market_data",
+    )
+
+
+def get_default_tools(lang: str = "en") -> ToolRegistry:
+    """Return a ToolRegistry pre-loaded with the built-in market data tools."""
+    registry = ToolRegistry()
+    registry.register(create_ohlcv_tool(lang))
+    registry.register(create_ticker_list_tool(lang))
+    return registry

aipriceaction_terminal/app.py

@@ -0,0 +1,97 @@
+"""Main application: TabbedContent shell with shared state."""
+
+import asyncio
+
+from textual import work
+from textual.app import App, ComposeResult
+from textual.reactive import reactive
+from textual.widgets import TabbedContent, TabPane, Header, Footer, Input, Select
+
+from .bindings import BINDINGS
+from .actions import AppActions
+from .theme import AI_GREEN, SCREEN_CSS
+from .chat import ChatTab
+from .workflows import WorkflowsTab
+from .ticker_data import TickerDataTab
+from .settings_tab import SettingsTab
+from .user_settings import load_settings
+
+
+class AIPriceActionApp(AppActions, App):
+    """AIPriceAction Terminal TUI."""
+
+    TITLE = "AIPriceAction Terminal"
+    SUB_TITLE = "AI-powered ticker analysis"
+    CSS = SCREEN_CSS
+    BINDINGS = BINDINGS
+
+    # Reactive state shared across tabs
+    ticker: reactive[str] = reactive("VNINDEX")
+    interval: reactive[str] = reactive("1D")
+    language: reactive[str] = reactive("en")
+    ticker_options: reactive[list[tuple[str, str]]] = reactive(list)
+
+    def on_mount(self) -> None:
+        self.register_theme(AI_GREEN)
+        self.theme = "ai-green"
+        saved = load_settings()
+        self.ticker = saved["ticker"]
+        self.interval = saved["interval"]
+        self.language = saved["language"]
+        from aipriceaction import AIContextBuilder
+        from aipriceaction import AIPriceAction as AAPClient
+        self.builder = AIContextBuilder(lang=self.language)
+        self.client = AAPClient()
+        from .agents import AgentSession, AgentConfig
+        self.agent = AgentSession(AgentConfig(lang=self.language))
+        self._load_ticker_options()
+        # Populate SettingsTab widgets with loaded values
+        self.query_one("#setting-ticker", Input).value = self.ticker
+        self.query_one("#setting-interval", Select).value = self.interval
+        self.query_one("#setting-language", Select).value = self.language
+        self.query_one("#chat-input-field", Input).focus()
+
+    @work(exclusive=True)
+    async def _load_ticker_options(self) -> None:
+        """Load ticker list from SDK and populate ticker_options reactive."""
+        try:
+            tickers = await asyncio.to_thread(self.client.get_tickers)
+            source_tags = {"vn": "[VN]", "crypto": "[CRYPTO]", "yahoo": "[YH]", "sjc": "[SJC]"}
+            options = []
+            for t in tickers:
+                tag = source_tags.get(t.source, f"[{t.source.upper()}]")
+                label = f"{tag} {t.ticker}"
+                if t.name:
+                    label += f" - {t.name}"
+                options.append((label, t.ticker))
+            options.sort(key=lambda x: x[0])
+            self.ticker_options = options
+        except Exception as e:
+            self.notify(f"Failed to load tickers: {e}", severity="error")
+
+    def compose(self) -> ComposeResult:
+        yield Header(show_clock=True)
+        with TabbedContent(initial="chat"):
+            with TabPane("Chat", id="chat"):
+                yield ChatTab()
+            with TabPane("Vietnam", id="tickers-vn"):
+                yield TickerDataTab(mode="vn")
+            with TabPane("Crypto", id="tickers-crypto"):
+                yield TickerDataTab(mode="crypto")
+            with TabPane("Global", id="tickers-global"):
+                yield TickerDataTab(mode="global")
+            with TabPane("Workflows", id="workflows"):
+                yield WorkflowsTab()
+            with TabPane("Settings", id="settings"):
+                yield SettingsTab()
+        yield Footer()
+
+
+def main():
+    """Entry point."""
+    app = AIPriceActionApp()
+    app.run()
+
+
+if __name__ == "__main__":
+    main()

aipriceaction_terminal/bindings.py

@@ -0,0 +1,25 @@
+"""Key bindings and tab shortcuts for AIPriceActionApp."""
+
+from textual.binding import Binding, BindingType
+
+_TAB_KEYS: dict[str, str] = {
+    "1": "chat",
+    "2": "tickers-vn",
+    "3": "tickers-crypto",
+    "4": "tickers-global",
+    "5": "workflows",
+    "6": "settings",
+}
+
+BINDINGS: list[BindingType] = [
+    Binding("ctrl+q", "confirm_quit", "Quit", key_display="ctrl+q", priority=True),
+    Binding("1", "switch_tab('chat')", "Chat", key_display="1"),
+    Binding("2", "switch_tab('tickers-vn')", "Vietnam", key_display="2"),
+    Binding("3", "switch_tab('tickers-crypto')", "Crypto", key_display="3"),
+    Binding("4", "switch_tab('tickers-global')", "Global", key_display="4"),
+    Binding("5", "switch_tab('workflows')", "Workflows", key_display="5"),
+    Binding("6", "switch_tab('settings')", "Settings", key_display="6"),
+    Binding("escape", "focus_none", "Back", key_display="esc", priority=True),
+    Binding("enter", "focus_first_input", "Focus input", key_display="enter"),
+    Binding("?", "show_help", "Help", key_display="?"),
+]