academia-mcp 1.11.4__py3-none-any.whl → 1.11.5__py3-none-any.whl

academia_mcp/server.py

@@ -17,6 +17,7 @@ from academia_mcp.tools.s2 import (
     s2_get_references,
     s2_corpus_id_from_arxiv_id,
     s2_get_info,
+    s2_search,
 )
 from academia_mcp.tools.hf_datasets_search import hf_datasets_search
 from academia_mcp.tools.anthology_search import anthology_search
@@ -86,6 +87,7 @@ def create_server(
     server.add_tool(s2_get_citations, structured_output=True)
     server.add_tool(s2_get_references, structured_output=True)
     server.add_tool(s2_get_info, structured_output=True)
+    server.add_tool(s2_search, structured_output=True)
     server.add_tool(s2_corpus_id_from_arxiv_id)
     server.add_tool(hf_datasets_search)
     server.add_tool(anthology_search)

academia_mcp/tools/__init__.py

@@ -2,7 +2,13 @@ from .arxiv_search import arxiv_search
 from .anthology_search import anthology_search
 from .arxiv_download import arxiv_download
 from .hf_datasets_search import hf_datasets_search
-from .s2 import s2_get_references, s2_get_citations, s2_corpus_id_from_arxiv_id, s2_get_info
+from .s2 import (
+    s2_get_references,
+    s2_get_citations,
+    s2_corpus_id_from_arxiv_id,
+    s2_get_info,
+    s2_search,
+)
 from .document_qa import document_qa
 from .latex import (
     compile_latex,
@@ -26,6 +32,7 @@ __all__ = [
     "s2_get_citations",
     "s2_corpus_id_from_arxiv_id",
     "s2_get_info",
+    "s2_search",
     "hf_datasets_search",
     "document_qa",
     "compile_latex",

academia_mcp/tools/s2.py

@@ -8,10 +8,12 @@ from pydantic import BaseModel, Field
 from academia_mcp.utils import get_with_retries
 
 
-PAPER_URL_TEMPLATE = "https://api.semanticscholar.org/graph/v1/paper/{paper_id}?fields={fields}"
-CITATIONS_URL_TEMPLATE = "https://api.semanticscholar.org/graph/v1/paper/{paper_id}/citations?fields={fields}&offset={offset}&limit={limit}"
-REFERENCES_URL_TEMPLATE = "https://api.semanticscholar.org/graph/v1/paper/{paper_id}/references?fields={fields}&offset={offset}&limit={limit}"
-FIELDS = "title,authors,externalIds,venue,citationCount,publicationDate"
+BASE_URL = "https://api.semanticscholar.org/graph/v1"
+PAPER_URL_TEMPLATE = "{base_url}/paper/{paper_id}"
+CITATIONS_URL_TEMPLATE = "{base_url}/paper/{paper_id}/citations"
+REFERENCES_URL_TEMPLATE = "{base_url}/paper/{paper_id}/references"
+SEARCH_URL_TEMPLATE = "{base_url}/paper/search"
+FIELDS = "paperId,title,authors,externalIds,venue,citationCount,publicationDate"
 
 
 class S2PaperInfo(BaseModel):  # type: ignore
@@ -38,7 +40,10 @@ def _format_authors(authors: List[Dict[str, Any]]) -> List[str]:
 
 
 def _clean_entry(entry: Dict[str, Any]) -> S2PaperInfo:
-    entry = entry["citingPaper"] if "citingPaper" in entry else entry["citedPaper"]
+    if "citingPaper" in entry:
+        entry = entry["citingPaper"]
+    elif "citedPaper" in entry:
+        entry = entry["citedPaper"]
     external_ids = entry.get("externalIds")
     if not external_ids:
         external_ids = dict()
@@ -88,17 +93,17 @@ def s2_get_citations(
         arxiv_id = arxiv_id.split("v")[0]
     paper_id = f"arxiv:{arxiv_id}"
 
-    url = CITATIONS_URL_TEMPLATE.format(
-        paper_id=paper_id, fields=FIELDS, offset=offset, limit=limit
-    )
-    response = get_with_retries(url)
+    url = CITATIONS_URL_TEMPLATE.format(base_url=BASE_URL, paper_id=paper_id)
+    payload = {"fields": FIELDS, "offset": offset, "limit": limit}
+    response = get_with_retries(url, params=payload)
     result = response.json()
     entries = result["data"]
     total_count = len(result["data"]) + result["offset"]
 
     if "next" in result:
-        paper_url = PAPER_URL_TEMPLATE.format(paper_id=paper_id, fields=FIELDS)
-        paper_response = get_with_retries(paper_url)
+        paper_url = PAPER_URL_TEMPLATE.format(base_url=BASE_URL, paper_id=paper_id)
+        payload = {"fields": FIELDS}
+        paper_response = get_with_retries(paper_url, params=payload)
         paper_result = paper_response.json()
         total_count = paper_result["citationCount"]
 
@@ -123,10 +128,9 @@ def s2_get_references(
         arxiv_id = arxiv_id.split("v")[0]
     paper_id = f"arxiv:{arxiv_id}"
 
-    url = REFERENCES_URL_TEMPLATE.format(
-        paper_id=paper_id, fields=FIELDS, offset=offset, limit=limit
-    )
-    response = get_with_retries(url)
+    url = REFERENCES_URL_TEMPLATE.format(base_url=BASE_URL, paper_id=paper_id)
+    payload = {"fields": FIELDS, "offset": offset, "limit": limit}
+    response = get_with_retries(url, params=payload)
     result = response.json()
     entries = result["data"]
     total_count = len(result["data"]) + result["offset"]
@@ -143,8 +147,9 @@ def s2_corpus_id_from_arxiv_id(arxiv_id: str) -> int:
     assert isinstance(arxiv_id, str), "Error: Your arxiv_id must be a string"
     if "v" in arxiv_id:
         arxiv_id = arxiv_id.split("v")[0]
-    paper_url = PAPER_URL_TEMPLATE.format(paper_id=f"arxiv:{arxiv_id}", fields="externalIds")
-    response = get_with_retries(paper_url)
+    paper_url = PAPER_URL_TEMPLATE.format(base_url=BASE_URL, paper_id=f"arxiv:{arxiv_id}")
+    payload = {"fields": "externalIds"}
+    response = get_with_retries(paper_url, params=payload)
     result = response.json()
     return int(result["externalIds"]["CorpusId"])
 
@@ -159,8 +164,10 @@ def s2_get_info(arxiv_id: str) -> S2PaperInfo:
     assert isinstance(arxiv_id, str), "Error: Your arxiv_id must be a string"
     if "v" in arxiv_id:
         arxiv_id = arxiv_id.split("v")[0]
-    paper_url = PAPER_URL_TEMPLATE.format(paper_id=f"arxiv:{arxiv_id}", fields=FIELDS)
-    response = get_with_retries(paper_url)
+    paper_id = f"arxiv:{arxiv_id}"
+    payload = {"fields": FIELDS}
+    paper_url = PAPER_URL_TEMPLATE.format(base_url=BASE_URL, paper_id=paper_id)
+    response = get_with_retries(paper_url, params=payload)
     json_data = response.json()
     return S2PaperInfo(
         arxiv_id=json_data.get("externalIds", {}).get("ArXiv"),
@@ -171,3 +178,45 @@ def s2_get_info(arxiv_id: str) -> S2PaperInfo:
         citation_count=int(json_data.get("citationCount", 0)),
         publication_date=str(json_data.get("publicationDate", "")),
     )
+
+
+def s2_search(
+    query: str,
+    offset: int = 0,
+    limit: int = 5,
+    min_citation_count: int = 0,
+    publication_date: Optional[str] = None,
+) -> S2SearchResponse:
+    """
+    Search the S2 corpus for a given query.
+
+    Args:
+        query: The query to search for.
+        offset: The offset to scroll through results. 10 items will be skipped if offset=10. 0 by default.
+        limit: The maximum number of items to return. limit=50 by default.
+        min_citation_count: The minimum citation count to return. 0 by default.
+        publication_date: Restricts results to the given range of publication dates or years (inclusive).
+            Accepts the format <startDate>:<endDate> with each date in YYYY-MM-DD format. None by default.
+    """
+    url = SEARCH_URL_TEMPLATE.format(base_url=BASE_URL)
+    payload = {
+        "query": query,
+        "offset": offset,
+        "limit": limit,
+        "minCitationCount": min_citation_count,
+        "fields": FIELDS,
+    }
+    if publication_date:
+        payload["publicationDateOrYear"] = publication_date
+    response = get_with_retries(url, params=payload, backoff_factor=10.0, num_retries=5)
+    result = response.json()
+    if "data" not in result:
+        return S2SearchResponse(
+            total_count=0,
+            returned_count=0,
+            offset=offset if offset else 0,
+            results=[],
+        )
+    entries = result["data"]
+    total_count = result["total"]
+    return _format_entries(entries, offset if offset else 0, total_count)

academia_mcp/utils.py

@@ -42,7 +42,7 @@ def post_with_retries(
 def get_with_retries(
     url: str,
     api_key: Optional[str] = None,
-    timeout: int = 30,
+    timeout: int = 60,
     num_retries: int = 3,
     backoff_factor: float = 3.0,
     params: Optional[Dict[str, Any]] = None,

{academia_mcp-1.11.4.dist-info → academia_mcp-1.11.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: academia-mcp
-Version: 1.11.4
+Version: 1.11.5
 Summary: MCP server that provides different tools to search for scientific publications
 Author-email: Ilya Gusev <phoenixilya@gmail.com>
 Project-URL: Homepage, https://github.com/IlyaGusev/academia_mcp

{academia_mcp-1.11.4.dist-info → academia_mcp-1.11.5.dist-info}/RECORD

@@ -4,12 +4,12 @@ academia_mcp/files.py,sha256=J_81COWWryBp2bHVe_9dg9c0rbVIzv7-zWkJ7CS9lr4,485
 academia_mcp/llm.py,sha256=zpGkuJFf58Ofgys_fi28-47_wJ1a7sIs_yZvI1Si6z0,993
 academia_mcp/pdf.py,sha256=9PlXzHGhb6ay3ldbTdxCcTWvH4TkET3bnb64mgoh9i0,1273
 academia_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-academia_mcp/server.py,sha256=sxr_M1JQif29hSuduXvoknPgidwoau8WKszT_P9mEVI,6396
+academia_mcp/server.py,sha256=oKlEjIhqzd2xmK8gtCDMKzqiXMd9K50K-3fyHhcke5Y,6466
 academia_mcp/settings.py,sha256=c5s4dI8V_cWmMED-jKDmHjfdIaBcxwEK4HdHNQ3WUIg,1096
-academia_mcp/utils.py,sha256=ixtYI7qidFJpc8Tzc5sokseCLx4r0yFFgbktKaY-Ixo,4904
+academia_mcp/utils.py,sha256=BVoEkGlW0HPH4bbWIFLzVwjA95ocT6OnoZBUhs87DK0,4904
 academia_mcp/latex_templates/agents4science_2025/agents4science_2025.sty,sha256=hGcEPCYBJS4vdhWvN_yEaJC4GvT_yDroI94CfY2Oguk,12268
 academia_mcp/latex_templates/agents4science_2025/agents4science_2025.tex,sha256=Tl1QkHXHRopw9VEfWrD3Layr5JP_0gIzVQjL4KXIWqc,15814
-academia_mcp/tools/__init__.py,sha256=cCS073dW2S5T_YWrbwd-H9QRiervZ7bof_2sdvzR8JQ,1561
+academia_mcp/tools/__init__.py,sha256=RpOCL2t-_fJev55rXL1za1MdrY6sESaTCszUCsEW474,1614
 academia_mcp/tools/anthology_search.py,sha256=rhFpJZqGLABgr0raDuH0CARBiAJNJtEI4dlMrKNHfDQ,7669
 academia_mcp/tools/arxiv_download.py,sha256=4zl9QkWF7uU2BYz4XH7Fu_51htolSYtO7a2v4_ikhxg,10633
 academia_mcp/tools/arxiv_search.py,sha256=p4DeCvV3TUpj58etx0QOxm-GYKAWkP9AGjLv4HUUqUc,8857
@@ -20,14 +20,14 @@ academia_mcp/tools/image_processing.py,sha256=BFj5D0lYbe6OCAL9LSHjS1adr8C0BrtWBX
 academia_mcp/tools/latex.py,sha256=B1Leqt1FHY6H3DlUgeYse4LMFpf4-K1FQViXl5MKk8A,6144
 academia_mcp/tools/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 academia_mcp/tools/review.py,sha256=Va0lFJJKuk-NvWhKS3UZ-Dnuk7CyuDQ4S1nd70D-ffE,11117
-academia_mcp/tools/s2.py,sha256=ykJOkpHnyZRlfyDnCIL9m-Rnu5dBecoKxg0SjlzdvLk,6457
+academia_mcp/tools/s2.py,sha256=_Ea7IyF5wH7ZWT59mI2b8wsxQp33W3URQV_EUiZuL2s,8221
 academia_mcp/tools/speech_to_text.py,sha256=YZzMqdvunzXkpcadP_mYhm6cs4qH1Y_42SfY-7eX4O4,1601
 academia_mcp/tools/visit_webpage.py,sha256=uJZx9vBGS8q-J-VH4Pr7T9lNtDsWU83gJhlotcd1ajg,3788
 academia_mcp/tools/web_search.py,sha256=CHgco8DufTFwtVecgDOOMylIY99iUmCdb0oZtpGntx0,8646
 academia_mcp/tools/yt_transcript.py,sha256=ilfOpX14moC1bKHbFmOVvZ8-_NxuQQUoQbV28e9FBaE,1217
-academia_mcp-1.11.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-academia_mcp-1.11.4.dist-info/METADATA,sha256=j4_CAs-0UIfhs0ai-XdrSFJcCvid4AkVbcqNnvhrQrg,6517
-academia_mcp-1.11.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-academia_mcp-1.11.4.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
-academia_mcp-1.11.4.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
-academia_mcp-1.11.4.dist-info/RECORD,,
+academia_mcp-1.11.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+academia_mcp-1.11.5.dist-info/METADATA,sha256=ExVjjzRzBc5eF9kVNZW_kAQdVCY3QZwE-jpZlcZSO6Q,6517
+academia_mcp-1.11.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+academia_mcp-1.11.5.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
+academia_mcp-1.11.5.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
+academia_mcp-1.11.5.dist-info/RECORD,,