intentkit 1.2.10__tar.gz → 1.2.11__tar.gz

{intentkit-1.2.10 → intentkit-1.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 1.2.10
+Version: 1.2.11
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestalnetwork/intentkit
 Project-URL: Repository, https://github.com/crestalnetwork/intentkit
@@ -40,6 +40,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: ==3.13.*
 Requires-Dist: aiohttp>=3.11.16
 Requires-Dist: aiosqlite>=0.21.0
+Requires-Dist: akshare>=1.16.0
 Requires-Dist: asyncpg>=0.30.0
 Requires-Dist: aws-secretsmanager-caching>=1.1.3
 Requires-Dist: boto3<2.0.0,>=1.37.23

{intentkit-1.2.10 → intentkit-1.2.11}/intentkit/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "1.2.10"
+__version__ = "1.2.11"
 __author__ = "Ruihua"
 __email__ = "ruihua@crestal.network"
 

{intentkit-1.2.10 → intentkit-1.2.11}/intentkit/core/manager/engine.py

@@ -28,6 +28,7 @@ from intentkit.core.manager.skills import (
 from intentkit.models.agent import Agent
 from intentkit.models.agent_data import AgentData
 from intentkit.models.chat import ChatMessage, ChatMessageCreate
+from intentkit.models.llm_picker import pick_default_model
 
 logger = logging.getLogger(__name__)
 
@@ -141,7 +142,7 @@ def _build_manager_agent(agent_id: str, user_id: str) -> Agent:
             "5. When updating a draft, try to select the right skills. Don't pick too many, just enough to meet user needs.\n"
             "6. Update skill is override update, you must put the whole fields to input data, not only changed fields."
         ),
-        "model": "grok-code-fast-1",
+        "model": pick_default_model(),
         "prompt": prompt,
         "prompt_append": None,
         "temperature": 0.2,

{intentkit-1.2.10 → intentkit-1.2.11}/intentkit/models/llm.csv

@@ -30,10 +30,8 @@ deepseek-v4-flash,Deepseek V4 Flash,deepseek,TRUE,0.14,0.028,0.28,1,1048576,3200
 deepseek/deepseek-v4-flash,Deepseek V4 Flash,openrouter,TRUE,0.14,0.028,0.28,1,1048576,32000,4,4,FALSE,FALSE,FALSE,FALSE,none,TRUE,TRUE,TRUE,300
 deepseek-v4-pro,Deepseek V4 Pro,deepseek,TRUE,1.74,0.145,3.48,3,1048576,64000,5,2,FALSE,FALSE,FALSE,FALSE,high,TRUE,TRUE,TRUE,300
 deepseek/deepseek-v4-pro,Deepseek V4 Pro,openrouter,TRUE,1.74,0.145,3.48,3,1048576,64000,5,2,FALSE,FALSE,FALSE,FALSE,high,TRUE,TRUE,TRUE,300
-grok-4.20-0309-reasoning,Grok 4.20,xai,TRUE,2,0.2,6,3,2000000,4096,5,3,TRUE,FALSE,FALSE,TRUE,high,TRUE,FALSE,FALSE,180
-x-ai/grok-4.20,Grok 4.20,openrouter,TRUE,2,0.2,6,3,2000000,4096,5,3,TRUE,FALSE,FALSE,TRUE,high,TRUE,TRUE,TRUE,180
-grok-4-1-fast-non-reasoning,Grok 4.1 Fast,xai,TRUE,0.2,0.05,0.5,2,2000000,4096,3,4,TRUE,FALSE,FALSE,TRUE,none,TRUE,FALSE,FALSE,180
-x-ai/grok-4.1-fast,Grok 4.1 Fast,openrouter,TRUE,0.2,0.05,0.5,2,2000000,4096,3,4,TRUE,FALSE,FALSE,TRUE,none,TRUE,TRUE,TRUE,180
-grok-4-1-fast-reasoning,Grok 4.1 Fast Reasoning,xai,TRUE,0.2,0.05,0.5,2,2000000,4096,3,4,TRUE,FALSE,FALSE,TRUE,high,TRUE,FALSE,FALSE,180
-x-ai/grok-4.1-fast-reasoning,Grok 4.1 Fast Reasoning,openrouter,TRUE,0.2,0.05,0.5,2,2000000,4096,3,4,TRUE,FALSE,FALSE,TRUE,high,TRUE,TRUE,TRUE,180
+grok-4.3,Grok 4.3,xai,TRUE,1.25,0.125,2.5,2,1000000,128000,5,4,TRUE,FALSE,FALSE,TRUE,high,TRUE,FALSE,FALSE,180
+x-ai/grok-4.3,Grok 4.3,openrouter,TRUE,1.25,0.125,2.5,2,1000000,128000,5,4,TRUE,FALSE,FALSE,TRUE,high,TRUE,TRUE,TRUE,180
+grok-4.20-non-reasoning,Grok 4.20,xai,TRUE,1.25,0.125,2.5,2,2000000,128000,4,4,TRUE,FALSE,FALSE,TRUE,none,TRUE,FALSE,FALSE,180
+x-ai/grok-4.20,Grok 4.20,openrouter,TRUE,1.25,0.125,2.5,2,2000000,128000,4,4,TRUE,FALSE,FALSE,TRUE,none,TRUE,TRUE,TRUE,180
 MiniMax-M2.7,MiniMax M2.7,minimax,TRUE,0.1,0.01,0.4,1,204800,131072,5,3,FALSE,FALSE,FALSE,FALSE,high,TRUE,TRUE,TRUE,300

{intentkit-1.2.10 → intentkit-1.2.11}/intentkit/models/llm_picker.py

@@ -14,7 +14,7 @@ def pick_summarize_model() -> str:
         ("gemini-3.1-flash-lite-preview", LLMProvider.GOOGLE),
         ("deepseek/deepseek-v4-flash", LLMProvider.OPENROUTER),
         ("gpt-5.4-mini", LLMProvider.OPENAI),
-        ("grok-4-1-fast-non-reasoning", LLMProvider.XAI),
+        ("grok-4.20-non-reasoning", LLMProvider.XAI),
         ("deepseek-v4-flash", LLMProvider.DEEPSEEK),
         ("MiniMax-M2.7", LLMProvider.MINIMAX),
         ("mimo-v2.5", LLMProvider.MIMO_PLAN),
@@ -51,7 +51,7 @@ def pick_default_model() -> str:
         ("MiniMax-M2.7", LLMProvider.MINIMAX),
         ("minimax/minimax-m2.7", LLMProvider.OPENROUTER),
         ("gpt-5.4-mini", LLMProvider.OPENAI),
-        ("grok-4-1-fast-non-reasoning", LLMProvider.XAI),
+        ("grok-4.20-non-reasoning", LLMProvider.XAI),
         ("deepseek-v4-flash", LLMProvider.DEEPSEEK),
         ("mimo-v2.5", LLMProvider.MIMO_PLAN),
     ]
@@ -106,7 +106,7 @@ def pick_long_context_model() -> str:
     # Priority order based on cost (cheapest first), one per provider:
     order: list[tuple[str, LLMProvider]] = [
         ("gemini-3.1-flash-lite-preview", LLMProvider.GOOGLE),
-        ("grok-4-1-fast-non-reasoning", LLMProvider.XAI),
+        ("grok-4.20-non-reasoning", LLMProvider.XAI),
         ("deepseek/deepseek-v4-flash", LLMProvider.OPENROUTER),
         ("gpt-5.4-nano", LLMProvider.OPENAI),
         ("deepseek-v4-flash", LLMProvider.DEEPSEEK),

intentkit-1.2.11/intentkit/skills/cn_stock/README.md

@@ -0,0 +1,34 @@
+# cn_stock — China A-Share Market Data
+
+Skill package providing real-time and historical market data for Chinese A-shares
+(Shanghai / Shenzhen / Beijing exchanges), backed by [akshare](https://akshare.akfamily.xyz/).
+
+## Tools
+
+| Skill | Purpose |
+|---|---|
+| `get_quote` | Real-time spot quote for one or more A-shares (price, change %, volume, P/E, P/B, market cap). |
+| `get_kline` | Daily / weekly / monthly OHLCV bars for a single A-share, with forward/back adjustment. |
+| `get_index` | Spot value (and optional 30-day history) for major Chinese stock indices (上证, 深证, 创业板, 沪深300, ...). |
+| `get_board` | Industry / concept board snapshot ranked by intraday % change. |
+| `get_capital_flow` | Net capital inflow/outflow for an individual stock or the whole market. |
+| `get_news` | Recent news for a specific A-share, or top macro financial headlines. |
+| `get_announcement` | Listed-company announcements (公告) for a given trading day. |
+| `get_financials` | Key financial metrics (EPS, ROE, revenue, margins) by reporting period. |
+| `is_trading_day` | Whether a given calendar date is an A-share trading day. Always call at the start of a scheduled task — cron triggers do not skip holidays. |
+
+## Operational notes
+
+- All akshare calls are blocking; the base class wraps them in `asyncio.to_thread`.
+- Results are short-TTL cached in Redis (5–60s for live quotes, longer for kline/financials)
+  to absorb retry storms — akshare scrapes free public endpoints which rate-limit by IP.
+- A category-level global rate limit of 60 calls/min protects shared infrastructure.
+- Stock codes are normalized: accepts `600519`, `sh600519`, `SH600519`, `600519.SH`.
+- Symbols not yet listed or recently delisted may return empty data; tools raise
+  `ToolException` with a clear message rather than silently returning empty results.
+
+## Composition
+
+These tools are designed to be used by **leaf** public agents (one per data domain),
+which a team-built monitor agent can orchestrate via `lead_call_agent`. See the
+public-agent YAMLs in `public_agents/base/cn-*.yaml` for working examples.

intentkit-1.2.11/intentkit/skills/cn_stock/__init__.py

@@ -0,0 +1,69 @@
+"""Chinese A-share market data skills (akshare-backed)."""
+
+import logging
+from typing import TypedDict
+
+from intentkit.skills.base import SkillConfig, SkillState
+from intentkit.skills.cn_stock.base import CNStockBaseTool
+from intentkit.skills.cn_stock.get_announcement import GetAnnouncement
+from intentkit.skills.cn_stock.get_board import GetBoard
+from intentkit.skills.cn_stock.get_capital_flow import GetCapitalFlow
+from intentkit.skills.cn_stock.get_financials import GetFinancials
+from intentkit.skills.cn_stock.get_index import GetIndex
+from intentkit.skills.cn_stock.get_kline import GetKLine
+from intentkit.skills.cn_stock.get_news import GetNews
+from intentkit.skills.cn_stock.get_quote import GetQuote
+from intentkit.skills.cn_stock.is_trading_day import IsTradingDay
+
+logger = logging.getLogger(__name__)
+
+
+class SkillStates(TypedDict, total=False):
+    get_quote: SkillState
+    get_kline: SkillState
+    get_index: SkillState
+    get_board: SkillState
+    get_capital_flow: SkillState
+    get_news: SkillState
+    get_announcement: SkillState
+    get_financials: SkillState
+    is_trading_day: SkillState
+
+
+class Config(SkillConfig):
+    """Configuration for cn_stock skills."""
+
+    states: SkillStates
+
+
+# Tool instances are stateless across calls; build once at import and reuse.
+_TOOLS: dict[str, CNStockBaseTool] = {
+    "get_quote": GetQuote(),
+    "get_kline": GetKLine(),
+    "get_index": GetIndex(),
+    "get_board": GetBoard(),
+    "get_capital_flow": GetCapitalFlow(),
+    "get_news": GetNews(),
+    "get_announcement": GetAnnouncement(),
+    "get_financials": GetFinancials(),
+    "is_trading_day": IsTradingDay(),
+}
+
+
+async def get_skills(
+    config: "Config",
+    is_private: bool,
+    **_,
+) -> list[CNStockBaseTool]:
+    available: list[CNStockBaseTool] = []
+    for name, state in config["states"].items():
+        if state == "disabled":
+            continue
+        if state == "private" and not is_private:
+            continue
+        tool = _TOOLS.get(name)
+        if tool is None:
+            logger.warning("Unknown cn_stock skill: %s", name)
+            continue
+        available.append(tool)
+    return available

intentkit-1.2.11/intentkit/skills/cn_stock/base.py

@@ -0,0 +1,107 @@
+"""Base class for cn_stock (Chinese A-share) tools backed by akshare."""
+
+import asyncio
+import json
+import logging
+from datetime import date, datetime
+from typing import Any, Callable
+
+import pytz
+from langchain_core.tools.base import ToolException
+
+from intentkit.config.redis import get_redis
+from intentkit.skills.base import IntentKitSkill
+
+logger = logging.getLogger(__name__)
+
+# A-share exchanges all run on Asia/Shanghai. Defaults must follow that wall clock,
+# regardless of the server's local timezone (typically UTC in production).
+_CN_TZ = pytz.timezone("Asia/Shanghai")
+
+
+def today_cn() -> date:
+    """Return today's date in the Asia/Shanghai timezone."""
+    return datetime.now(_CN_TZ).date()
+
+
+def normalize_a_share_symbol(symbol: str) -> str:
+    """Strip exchange prefix/suffix and return a bare 6-digit A-share code.
+
+    Accepts formats like "600519", "sh600519", "SH600519", "600519.SH", "600519.sh".
+    """
+    s = symbol.strip().upper()
+    for prefix in ("SH", "SZ", "BJ"):
+        if s.startswith(prefix):
+            s = s[len(prefix) :]
+            break
+    if "." in s:
+        s = s.split(".", 1)[0]
+    if len(s) != 6 or not s.isdigit():
+        raise ToolException(f"Invalid A-share code: {symbol!r} (expect 6 digits)")
+    return s
+
+
+def market_of(symbol: str) -> str:
+    """Return the exchange code ("sh"/"sz"/"bj") for a normalized 6-digit code."""
+    code = normalize_a_share_symbol(symbol)
+    # 920xxx is the BSE prefix introduced in 2025 for new listings; treat it as bj
+    # rather than letting all 9xxxxx through (900xxx are SSE B-shares, not A-shares).
+    if code.startswith("92"):
+        return "bj"
+    head = code[0]
+    if head == "6":
+        return "sh"
+    if head in ("0", "3"):
+        return "sz"
+    if head in ("4", "8"):
+        return "bj"
+    raise ToolException(f"Cannot infer market for code {code}")
+
+
+class CNStockBaseTool(IntentKitSkill):
+    category: str = "cn_stock"
+
+    async def run_blocking(
+        self,
+        cache_key: str | None,
+        cache_ttl: int,
+        func: Callable[..., Any],
+        *args: Any,
+        **kwargs: Any,
+    ) -> Any:
+        """Execute a blocking akshare call off the event loop with optional caching.
+
+        Pass cache_key=None to skip the cache for live data that must never be stale.
+        """
+        redis = get_redis() if cache_key is not None else None
+        if redis is not None:
+            cached = await redis.get(f"cn_stock:{cache_key}")
+            if cached:
+                try:
+                    return json.loads(cached)
+                except (TypeError, ValueError):
+                    logger.warning("Bad cache payload for %s, refetching", cache_key)
+
+        # akshare scrapes free public endpoints which rate-limit by IP; cap shared use.
+        await self.global_rate_limit_by_category(limit=60, seconds=60)
+
+        try:
+            result = await asyncio.to_thread(func, *args, **kwargs)
+        except Exception as e:
+            raise ToolException(f"akshare call failed: {e}") from e
+
+        if hasattr(result, "to_json"):
+            payload_str = result.to_json(
+                orient="records", date_format="iso", force_ascii=False
+            )
+            payload: Any = json.loads(payload_str)
+        elif isinstance(result, (str, int, float, bool, list, dict)) or result is None:
+            payload = result
+            payload_str = json.dumps(payload, ensure_ascii=False, default=str)
+        else:
+            payload = str(result)
+            payload_str = json.dumps(payload, ensure_ascii=False)
+
+        if redis is not None:
+            await redis.set(f"cn_stock:{cache_key}", payload_str, ex=cache_ttl)
+        return payload

intentkit-1.2.11/intentkit/skills/cn_stock/get_announcement.py

@@ -0,0 +1,47 @@
+"""Listed-company announcements (公告) via akshare."""
+
+from typing import Any
+
+import akshare as ak
+from langchain_core.tools import ArgsSchema
+from pydantic import BaseModel, Field
+
+from intentkit.skills.cn_stock.base import CNStockBaseTool, today_cn
+
+
+class GetAnnouncementInput(BaseModel):
+    on_date: str | None = Field(
+        default=None,
+        description="Date in YYYYMMDD; defaults to today. Announcements are released after market close.",
+    )
+    limit: int = Field(
+        default=20,
+        description="Maximum number of items to return (1-100).",
+        ge=1,
+        le=100,
+    )
+
+
+class GetAnnouncement(CNStockBaseTool):
+    name: str = "cn_stock_get_announcement"
+    description: str = (
+        "Get the day's listed-company announcements (财报、重大事项、增持减持、停复牌等). "
+        "Use to surface material disclosures that may move stock prices."
+    )
+    args_schema: ArgsSchema | None = GetAnnouncementInput
+
+    async def _arun(
+        self,
+        on_date: str | None = None,
+        limit: int = 20,
+        **_: Any,
+    ) -> list[dict[str, Any]]:
+        d = on_date or today_cn().strftime("%Y%m%d")
+        rows: list[dict[str, Any]] = await self.run_blocking(
+            f"announce:{d}",
+            900,
+            ak.stock_notice_report,
+            symbol="全部",
+            date=d,
+        )
+        return rows[:limit] if rows else []

intentkit-1.2.11/intentkit/skills/cn_stock/get_board.py

@@ -0,0 +1,63 @@
+"""Industry/concept board snapshot via akshare."""
+
+from typing import Any, Literal
+
+import akshare as ak
+from langchain_core.tools import ArgsSchema
+from pydantic import BaseModel, Field
+
+from intentkit.skills.cn_stock.base import CNStockBaseTool
+
+
+class GetBoardInput(BaseModel):
+    kind: Literal["industry", "concept"] = Field(
+        default="industry",
+        description="'industry' (行业板块) or 'concept' (概念板块).",
+    )
+    top: int = Field(
+        default=20,
+        description="Return top-N gainers and bottom-N losers by % change (1-50).",
+        ge=1,
+        le=50,
+    )
+
+
+class GetBoard(CNStockBaseTool):
+    name: str = "cn_stock_get_board"
+    description: str = (
+        "Get a snapshot of industry or concept boards, sorted by intraday percentage "
+        "change. Use to identify hot sectors and rotation. Returns top gainers and "
+        "top losers."
+    )
+    args_schema: ArgsSchema | None = GetBoardInput
+
+    async def _arun(
+        self,
+        kind: Literal["industry", "concept"] = "industry",
+        top: int = 20,
+        **_: Any,
+    ) -> dict[str, Any]:
+        fetcher = (
+            ak.stock_board_industry_name_em
+            if kind == "industry"
+            else ak.stock_board_concept_name_em
+        )
+        rows: list[dict[str, Any]] = await self.run_blocking(
+            f"board:{kind}",
+            60,
+            fetcher,
+        )
+
+        def pct(r: dict[str, Any]) -> float:
+            v = r.get("涨跌幅")
+            try:
+                return float(v) if v is not None else 0.0
+            except (TypeError, ValueError):
+                return 0.0
+
+        ranked = sorted(rows, key=pct, reverse=True)
+        return {
+            "kind": kind,
+            "gainers": ranked[:top],
+            "losers": ranked[-top:][::-1],
+        }

intentkit-1.2.11/intentkit/skills/cn_stock/get_capital_flow.py

@@ -0,0 +1,69 @@
+"""Capital flow (资金流向) for an individual stock or the whole market."""
+
+from typing import Any, Literal
+
+import akshare as ak
+from langchain_core.tools import ArgsSchema
+from langchain_core.tools.base import ToolException
+from pydantic import BaseModel, Field
+
+from intentkit.skills.cn_stock.base import (
+    CNStockBaseTool,
+    market_of,
+    normalize_a_share_symbol,
+)
+
+
+class GetCapitalFlowInput(BaseModel):
+    scope: Literal["stock", "market"] = Field(
+        ...,
+        description="'stock' for individual flow (requires symbol); 'market' for whole-market flow.",
+    )
+    symbol: str | None = Field(
+        default=None,
+        description="6-digit A-share code; required when scope='stock'.",
+    )
+    days: int = Field(
+        default=5,
+        description="Number of recent trading days to return (1-30).",
+        ge=1,
+        le=30,
+    )
+
+
+class GetCapitalFlow(CNStockBaseTool):
+    name: str = "cn_stock_get_capital_flow"
+    description: str = (
+        "Get net capital inflow / outflow data. For an individual stock returns recent "
+        "main / retail / institutional flows; for the whole market returns aggregate "
+        "north-bound and main-fund flows. Use to gauge smart-money positioning."
+    )
+    args_schema: ArgsSchema | None = GetCapitalFlowInput
+
+    async def _arun(
+        self,
+        scope: Literal["stock", "market"],
+        symbol: str | None = None,
+        days: int = 5,
+        **_: Any,
+    ) -> list[dict[str, Any]]:
+        if scope == "stock":
+            if not symbol:
+                raise ToolException("symbol is required when scope='stock'")
+            code = normalize_a_share_symbol(symbol)
+            mkt = market_of(symbol)
+            rows: list[dict[str, Any]] = await self.run_blocking(
+                f"flow:stock:{code}",
+                300,
+                ak.stock_individual_fund_flow,
+                stock=code,
+                market=mkt,
+            )
+            return rows[-days:] if rows else []
+
+        rows = await self.run_blocking(
+            "flow:market",
+            300,
+            ak.stock_market_fund_flow,
+        )
+        return rows[-days:] if rows else []

intentkit-1.2.11/intentkit/skills/cn_stock/get_financials.py

@@ -0,0 +1,52 @@
+"""Fundamental / financial metrics for an A-share via akshare."""
+
+from typing import Any, Literal
+
+import akshare as ak
+from langchain_core.tools import ArgsSchema
+from langchain_core.tools.base import ToolException
+from pydantic import BaseModel, Field
+
+from intentkit.skills.cn_stock.base import CNStockBaseTool, normalize_a_share_symbol
+
+
+class GetFinancialsInput(BaseModel):
+    symbol: str = Field(..., description="6-digit A-share code, e.g. '600519'.")
+    indicator: Literal["按报告期", "按年度", "按单季度"] = Field(
+        default="按报告期",
+        description="Reporting cadence to fetch.",
+    )
+    limit: int = Field(
+        default=8,
+        description="Maximum number of recent reporting periods to return (1-40).",
+        ge=1,
+        le=40,
+    )
+
+
+class GetFinancials(CNStockBaseTool):
+    name: str = "cn_stock_get_financials"
+    description: str = (
+        "Get key financial metrics (EPS, ROE, profit, revenue, margins, growth) for "
+        "an A-share by reporting period. Use for fundamental analysis and trend tracking."
+    )
+    args_schema: ArgsSchema | None = GetFinancialsInput
+
+    async def _arun(
+        self,
+        symbol: str,
+        indicator: Literal["按报告期", "按年度", "按单季度"] = "按报告期",
+        limit: int = 8,
+        **_: Any,
+    ) -> list[dict[str, Any]]:
+        code = normalize_a_share_symbol(symbol)
+        rows: list[dict[str, Any]] = await self.run_blocking(
+            f"financials:{code}:{indicator}",
+            21600,
+            ak.stock_financial_abstract_ths,
+            symbol=code,
+            indicator=indicator,
+        )
+        if not rows:
+            raise ToolException(f"No financials returned for {code}")
+        return rows[:limit]

intentkit-1.2.11/intentkit/skills/cn_stock/get_index.py

@@ -0,0 +1,96 @@
+"""Major Chinese market index spot and history via akshare."""
+
+import asyncio
+from datetime import timedelta
+from typing import Any, Literal
+
+import akshare as ak
+from langchain_core.tools import ArgsSchema
+from langchain_core.tools.base import ToolException
+from pydantic import BaseModel, Field
+
+from intentkit.skills.cn_stock.base import CNStockBaseTool, today_cn
+
+# Friendly names to akshare codes used by index_zh_a_hist.
+INDEX_CODES: dict[str, str] = {
+    "上证指数": "000001",
+    "沪深300": "000300",
+    "中证500": "000905",
+    "中证1000": "000852",
+    "深证成指": "399001",
+    "创业板指": "399006",
+    "科创50": "000688",
+}
+
+DEFAULT_INDICES = ["上证指数", "深证成指", "创业板指", "沪深300"]
+
+
+class GetIndexInput(BaseModel):
+    indices: list[str] = Field(
+        default_factory=lambda: list(DEFAULT_INDICES),
+        description=(
+            "Index names. Allowed: " + ", ".join(INDEX_CODES) + ". "
+            "Defaults to the four headline indices."
+        ),
+    )
+    history: Literal["spot", "30d"] = Field(
+        default="spot",
+        description="'spot' returns latest snapshot only; '30d' adds 30 daily bars.",
+    )
+
+
+class GetIndex(CNStockBaseTool):
+    name: str = "cn_stock_get_index"
+    description: str = (
+        "Get spot value (and optional 30-day history) for major Chinese stock indices "
+        "such as 上证指数, 深证成指, 创业板指, 沪深300. Use to gauge overall market direction."
+    )
+    args_schema: ArgsSchema | None = GetIndexInput
+
+    async def _arun(
+        self,
+        indices: list[str] | None = None,
+        history: Literal["spot", "30d"] = "spot",
+        **_: Any,
+    ) -> dict[str, Any]:
+        names = indices if indices is not None else list(DEFAULT_INDICES)
+        unknown = [n for n in names if n not in INDEX_CODES]
+        if unknown:
+            raise ToolException(
+                f"Unknown index names: {unknown}. Allowed: {list(INDEX_CODES)}"
+            )
+
+        spot_rows: list[dict[str, Any]] = await self.run_blocking(
+            "index_spot_em",
+            30,
+            ak.stock_zh_index_spot_em,
+            symbol="沪深重要指数",
+        )
+        spot_by_name = {r.get("名称"): r for r in spot_rows}
+        result: dict[str, Any] = {
+            "spot": [spot_by_name[n] for n in names if n in spot_by_name],
+        }
+
+        if history == "30d":
+            end = today_cn()
+            start = end - timedelta(days=45)
+            hist_results = await asyncio.gather(
+                *(
+                    self.run_blocking(
+                        f"index_hist:{INDEX_CODES[name]}:{start:%Y%m%d}:{end:%Y%m%d}",
+                        1800,
+                        ak.index_zh_a_hist,
+                        symbol=INDEX_CODES[name],
+                        period="daily",
+                        start_date=start.strftime("%Y%m%d"),
+                        end_date=end.strftime("%Y%m%d"),
+                    )
+                    for name in names
+                )
+            )
+            result["history"] = {
+                name: (hist[-30:] if isinstance(hist, list) else [])
+                for name, hist in zip(names, hist_results)
+            }
+
+        return result