academia-mcp 1.10.6__py3-none-any.whl → 1.10.7__py3-none-any.whl

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.dist-info → academia_mcp-1.10.7.dist-info}/METADATA

@@ -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.dist-info → academia_mcp-1.10.7.dist-info}/RECORD

@@ -23,10 +23,10 @@ academia_mcp/tools/s2.py,sha256=QX7-pbetab3Xt_1tvVPU6o5D_NAe9y6jcTGRBK1vwtY,6200
 academia_mcp/tools/show_image.py,sha256=jiJlQ53dbZ0T61OBhCT3IKVvBl9NHc6jHgWLfg5BxiE,3856
 academia_mcp/tools/speech_to_text.py,sha256=YZzMqdvunzXkpcadP_mYhm6cs4qH1Y_42SfY-7eX4O4,1601
 academia_mcp/tools/visit_webpage.py,sha256=swlFwWRzWc7-AHP2ouRZJScSTA4dHZ32fuJnA2V0lUc,3311
-academia_mcp/tools/web_search.py,sha256=2haTqnWkoISWS7VZSI0h7mW-jjWFItA8SL1FkkUxzI8,7864
-academia_mcp-1.10.6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-academia_mcp-1.10.6.dist-info/METADATA,sha256=pwPpp2xupjkVubi5Yt6JQqknFserzyRn-0xc2MGsIYI,6311
-academia_mcp-1.10.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-academia_mcp-1.10.6.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
-academia_mcp-1.10.6.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
-academia_mcp-1.10.6.dist-info/RECORD,,
+academia_mcp/tools/web_search.py,sha256=VphVztf2jZNT3bPJPJuTdMkKbe2-LIbSV7keKV47lac,8616
+academia_mcp-1.10.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+academia_mcp-1.10.7.dist-info/METADATA,sha256=MwRwYr89KMWr7wU2ntf45UbAcmGqxQe8beCwXwKIkjk,6311
+academia_mcp-1.10.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+academia_mcp-1.10.7.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
+academia_mcp-1.10.7.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
+academia_mcp-1.10.7.dist-info/RECORD,,