academia-mcp 1.10.6__tar.gz → 1.10.7__tar.gz

{academia_mcp-1.10.6 → academia_mcp-1.10.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: academia-mcp
-Version: 1.10.6
+Version: 1.10.7
 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.10.6 → academia_mcp-1.10.7}/academia_mcp/tools/web_search.py

@@ -1,5 +1,5 @@
 import json
-from typing import Optional, List
+from typing import Optional, List, Tuple
 
 from academia_mcp.utils import post_with_retries, get_with_retries
 from academia_mcp.settings import settings
@@ -12,6 +12,24 @@ BRAVE_SEARCH_URL = "https://api.search.brave.com/res/v1/web/search"
 EXCLUDE_DOMAINS = ["chatpaper.com"]
 
 
+def _parse_domains(query: str) -> Tuple[str, List[str]]:
+    site_term = "site:"
+    if site_term not in query:
+        return query, []
+    parts = query.split()
+    query_parts = []
+    include_domains = []
+    for part in parts:
+        if not part.startswith(site_term):
+            query_parts.append(part)
+            continue
+        domain = part[len(site_term) :]
+        if domain:
+            include_domains.append(domain)
+    query = " ".join(query_parts)
+    return query, include_domains
+
+
 def web_search(
     query: str,
     limit: Optional[int] = 20,
@@ -40,6 +58,13 @@ def web_search(
             isinstance(domain, str) for domain in include_domains
         ), "Error: include_domains should be a list of strings"
 
+    query, query_include_domains = _parse_domains(query)
+    if query_include_domains:
+        if include_domains:
+            include_domains.extend(query_include_domains)
+        else:
+            include_domains = query_include_domains
+
     is_tavily_available = bool(settings.TAVILY_API_KEY)
     is_exa_available = bool(settings.EXA_API_KEY)
     is_brave_available = bool(settings.BRAVE_API_KEY)

{academia_mcp-1.10.6 → academia_mcp-1.10.7}/academia_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: academia-mcp
-Version: 1.10.6
+Version: 1.10.7
 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.10.6 → academia_mcp-1.10.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "academia-mcp"
-version = "1.10.6"
+version = "1.10.7"
 description = "MCP server that provides different tools to search for scientific publications"
 readme = "README.md"
 authors = [

{academia_mcp-1.10.6 → academia_mcp-1.10.7}/tests/test_web_search.py

@@ -42,3 +42,14 @@ def test_web_search_include_domains() -> None:
     assert results
     assert len(results["results"]) > 0
     assert all("wikipedia.org" in result["url"] for result in results["results"])
+
+
+def test_web_search_include_query_domains() -> None:
+    results = web_search(
+        "site:wikipedia.org autoregressive models path-star graphs",
+    )
+    assert results
+    results = json.loads(results)
+    assert results
+    assert len(results["results"]) > 0
+    assert all("wikipedia.org" in result["url"] for result in results["results"])