mcp-scholar 0.1.0__tar.gz

mcp_scholar-0.1.0/.python-version

@@ -0,0 +1 @@
+3.12

mcp_scholar-0.1.0/PKG-INFO

@@ -0,0 +1,74 @@
+Metadata-Version: 2.4
+Name: mcp_scholar
+Version: 0.1.0
+Summary: MCP Scholar for Google Scholar search
+Requires-Python: >=3.10
+Requires-Dist: beautifulsoup4>=4.13.3
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: lxml>=4.9.3
+Requires-Dist: mcp[cli]>=1.4.1
+Requires-Dist: python-dotenv>=1.0.1
+Requires-Dist: scholarly>=1.7.0
+Description-Content-Type: text/markdown
+
+# MCP Scholar
+
+基于MCP协议的谷歌学术搜索和分析服务。
+
+## 功能特点
+
+- 谷歌学术论文搜索：根据关键词搜索相关论文，并按引用量排序
+- 学者主页分析：分析谷歌学术个人主页，提取引用量最高的论文
+- 支持与Cherry Studio集成：可以作为插件在Cherry Studio中使用
+
+## 安装方法
+
+### 使用uv安装
+
+```bash
+# 克隆仓库
+git clone https://github.com/yourusername/mcp-scholar.git
+cd mcp-scholar
+
+# 使用uv安装依赖
+uv pip install -e .
+```
+
+### 使用pip安装
+
+```bash
+pip install mcp-scholar
+```
+
+## 使用方法
+
+### 启动服务器
+
+```bash
+# 使用uvx启动
+uvx mcp-scholar
+
+# 或者直接使用Python模块
+python -m mcp_scholar
+```
+
+### 在Cherry Studio中使用
+
+1. 在Cherry Studio中，点击「设置」>「插件」
+2. 添加新插件，输入MCP服务器地址：`http://localhost:8080`
+3. 保存设置后，即可在对话中使用学术搜索功能
+
+## 示例用法
+
+在Cherry Studio中，可以使用以下提示：
+
+- 「总结5篇关于人工智能的论文」
+- 「分析学者主页 https://scholar.google.com/citations?user=SJdlvnQAAAAJ&hl=en 的前10篇高引论文」
+
+## 开发说明
+
+本项目使用MCP协议开发，基于Python SDK实现。详细信息请参考[MCP Python SDK](https://github.com/modelcontextprotocol/python-sdk)。
+
+## 许可证
+
+MIT

mcp_scholar-0.1.0/README.md

@@ -0,0 +1,61 @@
+# MCP Scholar
+
+基于MCP协议的谷歌学术搜索和分析服务。
+
+## 功能特点
+
+- 谷歌学术论文搜索：根据关键词搜索相关论文，并按引用量排序
+- 学者主页分析：分析谷歌学术个人主页，提取引用量最高的论文
+- 支持与Cherry Studio集成：可以作为插件在Cherry Studio中使用
+
+## 安装方法
+
+### 使用uv安装
+
+```bash
+# 克隆仓库
+git clone https://github.com/yourusername/mcp-scholar.git
+cd mcp-scholar
+
+# 使用uv安装依赖
+uv pip install -e .
+```
+
+### 使用pip安装
+
+```bash
+pip install mcp-scholar
+```
+
+## 使用方法
+
+### 启动服务器
+
+```bash
+# 使用uvx启动
+uvx mcp-scholar
+
+# 或者直接使用Python模块
+python -m mcp_scholar
+```
+
+### 在Cherry Studio中使用
+
+1. 在Cherry Studio中，点击「设置」>「插件」
+2. 添加新插件，输入MCP服务器地址：`http://localhost:8080`
+3. 保存设置后，即可在对话中使用学术搜索功能
+
+## 示例用法
+
+在Cherry Studio中，可以使用以下提示：
+
+- 「总结5篇关于人工智能的论文」
+- 「分析学者主页 https://scholar.google.com/citations?user=SJdlvnQAAAAJ&hl=en 的前10篇高引论文」
+
+## 开发说明
+
+本项目使用MCP协议开发，基于Python SDK实现。详细信息请参考[MCP Python SDK](https://github.com/modelcontextprotocol/python-sdk)。
+
+## 许可证
+
+MIT

mcp_scholar-0.1.0/pyproject.toml

@@ -0,0 +1,30 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "mcp_scholar"
+version = "0.1.0"
+description = "MCP Scholar for Google Scholar search"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "beautifulsoup4>=4.13.3",
+    "httpx>=0.28.1",
+    "mcp[cli]>=1.4.1",
+    "python-dotenv>=1.0.1",
+    "lxml>=4.9.3",
+    "scholarly>=1.7.0",
+]
+
+[dependency-groups]
+dev = [
+    "black>=25.1.0",
+    "mypy>=1.15.0",
+]
+
+[project.scripts]
+mcp-scholar = "mcp_scholar:cli_main"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/mcp_scholar"]

mcp_scholar-0.1.0/src/mcp_scholar/__init__.py

@@ -0,0 +1,4 @@
+"""MCP Scholar package."""
+from mcp_scholar.server import cli_main
+
+__all__ = ["cli_main"]

mcp_scholar-0.1.0/src/mcp_scholar/scholar.py

@@ -0,0 +1,263 @@
+"""
+谷歌学术搜索和解析功能
+使用 scholarly 库与谷歌学术交互
+"""
+
+import re
+import httpx
+import asyncio
+from scholarly import scholarly
+from typing import List, Dict, Any, Optional
+
+# 配置 scholarly 的代理，如果需要的话
+# scholarly.use_proxy(http="http://127.0.0.1:7890", https="http://127.0.0.1:7890")
+
+
+async def search_scholar(query: str, count: int = 5) -> List[Dict[str, Any]]:
+    """
+    搜索谷歌学术论文
+
+    Args:
+        query: 搜索关键词
+        count: 返回结果数量
+
+    Returns:
+        List[Dict]: 论文信息列表，按引用量排序
+    """
+    results = []
+    try:
+        # 使用scholarly库进行搜索
+        search_query = scholarly.search_pubs(query)
+
+        for _ in range(count):
+            try:
+                pub = next(search_query)
+
+                # 提取论文信息
+                paper = {
+                    "title": pub.get("bib", {}).get("title", "未知标题"),
+                    "authors": ", ".join(pub.get("bib", {}).get("author", [])),
+                    "abstract": pub.get("bib", {}).get("abstract", "无摘要"),
+                    "citations": pub.get("num_citations", 0),
+                    "year": pub.get("bib", {}).get("pub_year", "未知年份"),
+                    "venue": pub.get("bib", {}).get("venue", ""),
+                }
+
+                # 提取论文ID
+                pub_url = pub.get("pub_url", "")
+                if "citation_for_view=" in pub_url:
+                    paper["paper_id"] = pub_url.split("citation_for_view=")[-1]
+                elif "cluster=" in pub_url:
+                    paper["paper_id"] = pub_url.split("cluster=")[-1].split("&")[0]
+                else:
+                    paper["paper_id"] = None
+
+                results.append(paper)
+
+                # 添加延迟以避免被谷歌限制
+                await asyncio.sleep(1)
+
+            except StopIteration:
+                break
+            except Exception as e:
+                print(f"处理单篇论文时出错: {str(e)}")
+                continue
+
+        return sorted(results, key=lambda x: -x["citations"])
+    except Exception as e:
+        print(f"搜索谷歌学术时出错: {str(e)}")
+        return []
+
+
+async def get_paper_detail(paper_id: str) -> Optional[Dict[str, Any]]:
+    """
+    获取论文详情
+
+    Args:
+        paper_id: 论文ID
+
+    Returns:
+        Dict: 论文详细信息
+    """
+    try:
+        # 尝试通过ID获取论文详情
+        if ":" in paper_id:  # 可能是DOI或其他标识符
+            # 使用scholarly的方式搜索特定论文
+            query = f"source:{paper_id}"
+            search_query = scholarly.search_pubs(query)
+            try:
+                pub = next(search_query)
+                pub = scholarly.fill(pub)
+            except StopIteration:
+                return None
+        else:
+            # 尝试通过聚类ID查找
+            url = f"https://scholar.google.com/scholar?cluster={paper_id}"
+            async with httpx.AsyncClient() as client:
+                response = await client.get(url)
+                response.raise_for_status()
+
+                # 这里需要解析返回的HTML页面，比较复杂
+                # 建议使用scholarly的工具
+                return None
+
+        # 提取详细信息
+        return {
+            "title": pub.get("bib", {}).get("title", "未知标题"),
+            "authors": ", ".join(pub.get("bib", {}).get("author", [])),
+            "abstract": pub.get("bib", {}).get("abstract", "无摘要"),
+            "citations": pub.get("num_citations", 0),
+            "year": pub.get("bib", {}).get("pub_year", "未知年份"),
+            "venue": pub.get("bib", {}).get("venue", ""),
+            "paper_id": paper_id,
+            "url": pub.get("pub_url", ""),
+            "citation_url": (
+                pub.get("cites_id", {}).get("link", "") if pub.get("cites_id") else ""
+            ),
+        }
+    except Exception as e:
+        print(f"获取论文详情时出错: {str(e)}")
+        return None
+
+
+async def get_paper_references(paper_id: str, count: int = 5) -> List[Dict[str, Any]]:
+    """
+    获取引用指定论文的文献
+
+    Args:
+        paper_id: 论文ID
+        count: 返回结果数量
+
+    Returns:
+        List[Dict]: 引用论文信息列表
+    """
+    results = []
+    try:
+        # 通过ID获取论文的引用
+        if ":" in paper_id:  # 可能是DOI或其他标识符
+            query = f"cite:{paper_id}"
+        else:
+            query = f"cites={paper_id}"
+
+        search_query = scholarly.search_pubs(query)
+
+        for _ in range(count):
+            try:
+                pub = next(search_query)
+
+                paper = {
+                    "title": pub.get("bib", {}).get("title", "未知标题"),
+                    "authors": ", ".join(pub.get("bib", {}).get("author", [])),
+                    "abstract": pub.get("bib", {}).get("abstract", "无摘要"),
+                    "citations": pub.get("num_citations", 0),
+                    "year": pub.get("bib", {}).get("pub_year", "未知年份"),
+                    "venue": pub.get("bib", {}).get("venue", ""),
+                }
+
+                # 提取论文ID
+                pub_url = pub.get("pub_url", "")
+                if "citation_for_view=" in pub_url:
+                    paper["paper_id"] = pub_url.split("citation_for_view=")[-1]
+                elif "cluster=" in pub_url:
+                    paper["paper_id"] = pub_url.split("cluster=")[-1].split("&")[0]
+                else:
+                    paper["paper_id"] = None
+
+                results.append(paper)
+
+                # 添加延迟以避免被谷歌限制
+                await asyncio.sleep(1)
+
+            except StopIteration:
+                break
+            except Exception as e:
+                print(f"处理引用论文时出错: {str(e)}")
+                continue
+
+        return sorted(results, key=lambda x: -x["citations"])
+    except Exception as e:
+        print(f"获取论文引用时出错: {str(e)}")
+        return []
+
+
+async def parse_profile(profile_id: str, top_n: int = 5) -> List[Dict[str, Any]]:
+    """
+    解析谷歌学术个人主页
+
+    Args:
+        profile_id: 学者ID
+        top_n: 返回结果数量
+
+    Returns:
+        List[Dict]: 论文信息列表
+    """
+    try:
+        # 通过ID查找作者
+        author = scholarly.search_author_id(profile_id)
+        if not author:
+            print(f"未找到ID为{profile_id}的学者")
+            return []
+
+        # 获取完整信息
+        author = scholarly.fill(author)
+
+        # 获取发表的论文
+        publications = author.get("publications", [])
+        papers = []
+
+        for pub in publications[
+            : min(len(publications), top_n * 2)
+        ]:  # 获取更多论文，以防有些无法填充
+            try:
+                # 获取详细信息
+                filled_pub = scholarly.fill(pub)
+
+                paper = {
+                    "title": filled_pub.get("bib", {}).get("title", "未知标题"),
+                    "authors": ", ".join(filled_pub.get("bib", {}).get("author", [])),
+                    "citations": filled_pub.get("num_citations", 0),
+                    "year": filled_pub.get("bib", {}).get("pub_year", "未知年份"),
+                    "venue": filled_pub.get("bib", {}).get("venue", ""),
+                    "abstract": filled_pub.get("bib", {}).get("abstract", "无摘要"),
+                }
+
+                # 提取论文ID
+                pub_url = filled_pub.get("pub_url", "")
+                if "citation_for_view=" in pub_url:
+                    paper["paper_id"] = pub_url.split("citation_for_view=")[-1]
+                elif "cluster=" in pub_url:
+                    paper["paper_id"] = pub_url.split("cluster=")[-1].split("&")[0]
+                else:
+                    paper["paper_id"] = None
+
+                papers.append(paper)
+
+                # 添加延迟以避免被谷歌限制
+                await asyncio.sleep(1)
+
+            except Exception as e:
+                print(f"处理作者论文时出错: {str(e)}")
+                continue
+
+        # 按引用数排序
+        papers = sorted(papers, key=lambda x: -x["citations"])
+        return papers[:top_n]
+    except Exception as e:
+        print(f"解析学者档案时出错: {str(e)}")
+        return []
+
+
+def extract_profile_id_from_url(url: str) -> str:
+    """
+    从谷歌学术个人主页URL中提取学者ID
+
+    Args:
+        url: 谷歌学术个人主页URL
+
+    Returns:
+        str: 学者ID
+    """
+    match = re.search(r"user=([^&]+)", url)
+    if match:
+        return match.group(1)
+    return ""