@kuandotdev/indicator 0.1.1 → 0.1.3

package/project/src/tv_indicator/mcp/data_server.py

@@ -0,0 +1,239 @@
+"""MCP server: TradingView data and server-side model forecasts.
+
+Exposes the same functions as `training-view data ...` and
+`training-view model forecast ...` so any MCP-aware agent can call them.
+
+Register in claude_desktop_config.json:
+    {
+      "mcpServers": {
+        "training-view-data": {
+          "command": "training-view-mcp-data",
+          "env": { "TV_SESSIONID": "..." }
+        }
+      }
+    }
+"""
+from __future__ import annotations
+
+import asyncio
+import json
+from typing import Any
+
+from mcp.server import Server
+from mcp.server.stdio import stdio_server
+from mcp.types import TextContent, Tool
+
+server = Server("training-view-data")
+
+
+# ─── Tool definitions ──────────────────────────────────────────────────────
+
+
+@server.list_tools()
+async def list_tools() -> list[Tool]:
+    return [
+        Tool(
+            name="get_ohlcv",
+            description=(
+                "Fetch OHLCV candles for a TradingView symbol. "
+                "Returns JSON with `index` (timestamps) and column arrays."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "symbol": {"type": "string", "description": "e.g. 'NASDAQ:AAPL' or 'AAPL'"},
+                    "interval": {
+                        "type": "string",
+                        "enum": ["1", "3", "5", "15", "30", "45", "60", "120", "180", "240", "1D", "1W", "1M"],
+                        "default": "1D",
+                    },
+                    "n_bars": {"type": "integer", "default": 500, "minimum": 1, "maximum": 5000},
+                    "exchange": {"type": "string", "description": "Optional if not in symbol"},
+                    "extended_session": {"type": "boolean", "default": False},
+                },
+                "required": ["symbol"],
+            },
+        ),
+        Tool(
+            name="search_symbol",
+            description="Search TradingView symbols by ticker or company name.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {"type": "string"},
+                    "limit": {"type": "integer", "default": 20, "minimum": 1, "maximum": 100},
+                },
+                "required": ["query"],
+            },
+        ),
+        Tool(
+            name="screener",
+            description=(
+                "Run a TradingView screener query. Filters are tuples of (column, op, value). "
+                "Ops: >, <, >=, <=, =, !="
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "market": {"type": "string", "default": "america"},
+                    "columns": {"type": "array", "items": {"type": "string"}},
+                    "filters": {
+                        "type": "array",
+                        "items": {
+                            "type": "array",
+                            "prefixItems": [
+                                {"type": "string"},
+                                {"type": "string"},
+                                {},
+                            ],
+                            "minItems": 3,
+                            "maxItems": 3,
+                        },
+                    },
+                    "order_by": {
+                        "type": "array",
+                        "prefixItems": [{"type": "string"}, {"type": "boolean"}],
+                        "minItems": 2,
+                        "maxItems": 2,
+                    },
+                    "limit": {"type": "integer", "default": 50},
+                },
+            },
+        ),
+        Tool(
+            name="technical_rating",
+            description="Get TradingView's Strong Buy/Buy/Neutral/Sell/Strong Sell rating for a symbol.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "symbol": {"type": "string"},
+                    "interval": {"type": "string", "default": "1D"},
+                },
+                "required": ["symbol"],
+            },
+        ),
+        Tool(
+            name="model_forecast",
+            description="Fetch the server-side xs_range 20-trading-day forecast for a stock/ETF/index symbol.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "symbol": {"type": "string", "description": "e.g. 'NASDAQ:TSLA' or 'TSLA'"},
+                    "refresh": {"type": "boolean", "default": False},
+                    "period": {"type": "string", "default": "10y"},
+                },
+                "required": ["symbol"],
+            },
+        ),
+    ]
+
+
+# ─── Tool dispatch ─────────────────────────────────────────────────────────
+
+
+@server.call_tool()
+async def call_tool(name: str, arguments: dict[str, Any]) -> list[TextContent]:
+    if name == "get_ohlcv":
+        return await _get_ohlcv(arguments)
+    if name == "search_symbol":
+        return await _search_symbol(arguments)
+    if name == "screener":
+        return await _screener(arguments)
+    if name == "technical_rating":
+        return await _technical_rating(arguments)
+    if name == "model_forecast":
+        return await _model_forecast(arguments)
+    return [TextContent(type="text", text=f"Unknown tool: {name}")]
+
+
+async def _get_ohlcv(args: dict[str, Any]) -> list[TextContent]:
+    from ..data.client import get_ohlcv
+
+    df = await asyncio.to_thread(
+        get_ohlcv,
+        args["symbol"],
+        args.get("interval", "1D"),
+        args.get("n_bars", 500),
+        args.get("exchange"),
+        args.get("extended_session", False),
+    )
+    payload = {
+        "symbol": args["symbol"],
+        "interval": args.get("interval", "1D"),
+        "rows": len(df),
+        "index": [str(i) for i in df.index],
+        "columns": {c: df[c].tolist() for c in df.columns},
+    }
+    return [TextContent(type="text", text=json.dumps(payload, default=str))]
+
+
+async def _search_symbol(args: dict[str, Any]) -> list[TextContent]:
+    from ..data.screener import search_symbol
+
+    results = await asyncio.to_thread(
+        search_symbol, args["query"], args.get("limit", 20)
+    )
+    return [TextContent(type="text", text=json.dumps(results, default=str))]
+
+
+async def _screener(args: dict[str, Any]) -> list[TextContent]:
+    from ..data.screener import screener
+
+    filters_raw = args.get("filters") or []
+    filters = [(f[0], f[1], f[2]) for f in filters_raw]
+    order_by_raw = args.get("order_by")
+    order_by = (order_by_raw[0], bool(order_by_raw[1])) if order_by_raw else None
+
+    df = await asyncio.to_thread(
+        screener,
+        args.get("market", "america"),
+        args.get("columns"),
+        filters or None,
+        order_by,
+        args.get("limit", 50),
+    )
+    payload = {
+        "rows": len(df),
+        "columns": list(df.columns),
+        "data": df.to_dict(orient="records"),
+    }
+    return [TextContent(type="text", text=json.dumps(payload, default=str))]
+
+
+async def _technical_rating(args: dict[str, Any]) -> list[TextContent]:
+    from ..data.screener import technical_rating
+
+    r = await asyncio.to_thread(
+        technical_rating, args["symbol"], args.get("interval", "1D")
+    )
+    return [TextContent(type="text", text=json.dumps(r, default=str))]
+
+
+async def _model_forecast(args: dict[str, Any]) -> list[TextContent]:
+    from ..model.forecast import forecast_for_symbol
+
+    payload = await asyncio.to_thread(
+        forecast_for_symbol,
+        args["symbol"],
+        refresh=bool(args.get("refresh", False)),
+        period=args.get("period", "10y"),
+    )
+    return [TextContent(type="text", text=json.dumps(payload, default=str))]
+
+
+# ─── Entry point ───────────────────────────────────────────────────────────
+
+
+async def _run() -> None:
+    async with stdio_server() as (read_stream, write_stream):
+        await server.run(
+            read_stream, write_stream, server.create_initialization_options()
+        )
+
+
+def main() -> None:
+    asyncio.run(_run())
+
+
+if __name__ == "__main__":
+    main()

package/project/src/tv_indicator/model/__init__.py

@@ -0,0 +1,19 @@
+"""Server-side stock-forecast model API helpers."""
+
+from .forecast import (
+    DEFAULT_MODEL_ROOT,
+    DEFAULT_MODEL_REPO,
+    MODEL_UNIVERSE,
+    forecast_for_symbol,
+    model_status,
+    setup_model_env,
+)
+
+__all__ = [
+    "DEFAULT_MODEL_ROOT",
+    "DEFAULT_MODEL_REPO",
+    "MODEL_UNIVERSE",
+    "forecast_for_symbol",
+    "model_status",
+    "setup_model_env",
+]