Paperlib MCP

Instructions

Implementation Reference

• src/paperlib_mcp/tools/writing.py:643-774 (handler)

  The core handler function for the 'draft_section' tool. It generates a specific section of a literature review using keyword matching from an evidence pack, formats content with citations, and returns structured output.

  @mcp.tool()
  def draft_section(
      pack_id: int,
      section: str,
      outline_style: str = "econ_finance_canonical",
  ) -> dict[str, Any]:
      """生成综述特定章节
      
      基于证据包，只生成指定章节的内容。适合迭代写作某个特定部分。
      
      Args:
          pack_id: 证据包 ID
          section: 章节类型，如 "methodology"、"findings"、"gaps" 等
          outline_style: 大纲样式，默认 "econ_finance_canonical"
          
      Returns:
          章节内容和引用列表
      """
      try:
          # 获取证据包
          pack = get_evidence_pack(pack_id)
          if not pack:
              return {
                  "error": f"Evidence pack not found: {pack_id}",
                  "pack_id": pack_id,
              }
          
          # 获取模板
          template = OUTLINE_TEMPLATES.get(outline_style, OUTLINE_TEMPLATES["general"])
          
          # 找到对应章节
          section_template = None
          for s in template["sections"]:
              if s["id"] == section:
                  section_template = s
                  break
          
          if not section_template:
              available_sections = [s["id"] for s in template["sections"]]
              return {
                  "error": f"Section '{section}' not found. Available: {available_sections}",
                  "pack_id": pack_id,
                  "section": section,
              }
          
          # 获取文档元数据
          doc_ids = list(set(item.doc_id for item in pack.items))
          doc_metadata: dict[str, dict] = {}
          for doc_id in doc_ids:
              doc = query_one(
                  "SELECT doc_id, title, authors, year FROM documents WHERE doc_id = %s",
                  (doc_id,)
              )
              if doc:
                  doc_metadata[doc_id] = {
                      "doc_id": doc["doc_id"],
                      "title": doc["title"] or "Untitled",
                      "authors": doc["authors"] or "Unknown",
                      "year": doc["year"],
                  }
          
          # 筛选与章节相关的证据
          keywords = section_template.get("keywords", [])
          relevant_items = []
          
          for item in pack.items:
              text_lower = item.text.lower()
              match_count = sum(1 for kw in keywords if kw.lower() in text_lower)
              if match_count > 0:
                  relevant_items.append((item, match_count))
          
          # 按匹配数排序
          relevant_items.sort(key=lambda x: x[1], reverse=True)
          
          # 构建章节内容
          content_parts = []
          citations = []
          
          content_parts.append(f"# {section_template['title']}\n")
          content_parts.append(f"**{section_template['description']}**\n")
          
          for item, match_count in relevant_items[:15]:  # 最多 15 条
              meta = doc_metadata.get(item.doc_id, {"title": "Unknown", "authors": "Unknown", "year": None})
              
              citation = {
                  "doc_id": item.doc_id,
                  "title": meta["title"],
                  "authors": meta["authors"],
                  "year": meta["year"],
                  "page_start": item.page_start,
                  "page_end": item.page_end,
                  "chunk_id": item.chunk_id,
                  "relevance": match_count,
              }
              citations.append(citation)
              
              year_str = str(meta["year"]) if meta["year"] else "n.d."
              cite_key = f"[{meta['authors']}, {year_str}: p.{item.page_start}-{item.page_end}]"
              
              text = item.text
              snippet = text[:400] + "..." if len(text) > 400 else text
              
              content_parts.append(f"- {snippet} {cite_key}")
          
          if not relevant_items:
              content_parts.append("（该章节暂无匹配的相关内容）")
          
          # 去重引用
          unique_citations = []
          seen_docs = set()
          for cite in citations:
              if cite["doc_id"] not in seen_docs:
                  seen_docs.add(cite["doc_id"])
                  unique_citations.append(cite)
          
          return {
              "pack_id": pack_id,
              "section_id": section,
              "title": section_template["title"],
              "content": "\n\n".join(content_parts),
              "citations": citations,
              "unique_documents": len(unique_citations),
              "total_evidence": len(relevant_items),
          }
          
      except Exception as e:
          return {
              "error": str(e),
              "pack_id": pack_id,
              "section": section,
          }

• src/paperlib_mcp/server.py:37-37 (registration)

  Invocation of register_writing_tools(mcp) which registers the 'draft_section' tool (via @mcp.tool() decorator in writing.py) to the FastMCP server instance.

  register_writing_tools(mcp)