"""
科研助手技能 - 每日论文收集与日报生成

使用方法：
- 发送指令："今天阅读文献" 或 "写个日报" 或 "科研日报"
- 技能会自动：访问RSS、搜索arXiv、筛选论文、生成日报
"""

from typing import Optional
import re
from datetime import datetime

def research_assistant(
    date: Optional[str] = None,
    include_rss: bool = True,
    include_arxiv: bool = True,
    categories: Optional[str] = None,
    max_results: int = 15
) -> str:
    """
    生成每日科研日报

    Args:
        date: 指定日期，格式YYYY-MM-DD，默认为今天
        include_rss: 是否包含RSS订阅论文
        include_arxiv: 是否包含arXiv论文
        categories: 指定arXiv分类，逗号分隔，如 "RAG,Spatial"
        max_results: 每个分类返回的最大论文数

    Returns:
        生成日报的指令字符串
    """

    # 默认日期为今天
    if date is None:
        date = datetime.now().strftime("%Y-%m-%d")

    # 构建完整的提示词
    prompt = f"""# 科研助手任务 - {date}

请作为我的科研助手，完成以下论文收集和日报生成任务：

## 我的研究方向

**核心领域**：数字火星平台构建（计算机与遥感结合）

1. **时空数据管理与检索**
   - 大规模遥感影像的并发时空范围查询
   - I/O优化
   - 时空索引

2. **RAG（检索增强生成）**
   - 空间数据RAG
   - 多源检索、幻觉消解
   - 多尺度融合、时空动态性
   - 拓扑关系编码

3. **大模型**
   - 地球科学/行星科学大模型
   - 多模态大模型
   - 知识融合

4. **KV Cache**
   - 大模型推理优化

## 任务清单

### 1. RSS订阅论文收集
{rss_task if include_rss else "（跳过RSS）"}

使用Chrome MCP访问：http://192.168.190.20:8080/
- 用户名：la
- 密码：longao2001
- 查看今天的论文（14篇遥感学报论文）
- 提取标题、摘要、作者信息

### 2. arXiv论文搜索
{arxiv_task if include_arxiv else "（跳过arXiv）"}

搜索以下方向的最新论文（最近7天）：

**方向1：RAG**
- 查询：`"retrieval augmented generation" OR RAG`
- 分类：cs.CL, cs.AI, cs.IR, cs.LG
- 重点：多源RAG、图RAG、幻觉消解

**方向2：空间数据与遥感**
- 查询：`"spatial data" OR "geospatial" OR "remote sensing" AND "deep learning"`
- 分类：cs.CV, cs.LG
- 重点：地理空间推理、遥感基础模型、行星科学

**方向3：高光谱图像**
- 查询：`"hyperspectral" OR "multispectral" AND ("classification" OR "unmixing")`
- 分类：cs.CV, cs.LG, eess.IV
- 重点：光谱-空间特征、Mamba网络

**方向4：KV Cache**
- 查询：`"KV cache" OR "attention cache" OR "LLM inference"`
- 分类：cs.CL, cs.AI, cs.LG
- 重点：压缩、共享、优化

每个方向返回最多 {max_results} 篇论文。

### 3. 论文筛选与分类

根据我的研究方向，将论文分为：

- ⭐⭐⭐⭐⭐ 高度相关（直接对应我的研究问题）
- ⭐⭐⭐⭐ 值得关注（方法可借鉴、相关领域）
- ⭐⭐⭐ 了解即可（领域前沿、扩展视野）

### 4. 生成日报

日报格式：

# 📅 科研日报 - {date}

## 📊 数据来源统计
- RSS订阅：X篇
- arXiv：X篇

## 🔥 重点关注（高度相关）
[论文列表，包含标题、作者、核心贡献、与研究关联]

## 📝 值得关注
[论文列表]

## 💡 研究启示与建议
[对我的Geo-MultiRAG、数字火星平台、KV Cache研究的启发]

## 📌 今日行动建议
[必读论文、深入阅读论文、思考问题]

### 5. 保存日报

将日报保存到：`C:\\Users\\islon\\OneDrive\\Desktop\\studio\\research-assistant\\daily\\{date}.md`

---

请开始执行任务。
"""

    # 处理RSS任务描述
    rss_task = """使用Chrome MCP：
1. 打开 http://192.168.190.20:8080/
2. 登录（la/longao2001）
3. 查看今天的未读论文
4. 逐篇阅读摘要，提取关键信息"""

    # 处理arXiv任务描述
    arxiv_task = f"""使用arXiv MCP搜索4个方向，每个方向最多{max_results}篇：
- RAG方向
- 空间数据/遥感方向
- 高光谱方向
- KV Cache方向"""

    prompt = prompt.replace("{rss_task}", rss_task).replace("{arxiv_task}", arxiv_task)

    return prompt


# 技能元数据
skill_metadata = {
    "name": "research-assistant",
    "description": "每日科研助手：收集arXiv和RSS论文，生成科研日报",
    "version": "1.0.0",
    "author": "Claude",
    "triggers": [
        "今天阅读文献",
        "写个日报",
        "科研日报",
        "收集论文",
        "查看今天论文",
        "文献日报"
    ],
    "parameters": {
        "date": {
            "type": "string",
            "description": "指定日期（YYYY-MM-DD），默认今天",
            "required": False
        },
        "include_rss": {
            "type": "boolean",
            "description": "是否包含RSS订阅",
            "required": False,
            "default": True
        },
        "include_arxiv": {
            "type": "boolean",
            "description": "是否包含arXiv论文",
            "required": False,
            "default": True
        },
        "categories": {
            "type": "string",
            "description": "指定分类（逗号分隔）",
            "required": False
        },
        "max_results": {
            "type": "integer",
            "description": "每个分类的最大论文数",
            "required": False,
            "default": 15
        }
    },
    "examples": [
        "今天阅读文献",
        "写个日报",
        "科研日报",
        "收集论文 max_results=20",
        "科研日报 categories=RAG,KV_Cache"
    ]
}