academia-mcp 1.2.0__py3-none-any.whl → 1.2.2__py3-none-any.whl

academia_mcp/tools/document_qa.py

@@ -1,4 +1,5 @@
 import os
+import json
 from typing import List, Any, Dict
 from dotenv import load_dotenv
 
@@ -38,7 +39,7 @@ ChatMessages = List[ChatMessage]
 
 
 async def document_qa(
-    document: str,
+    document: str | Dict[str, Any],
     question: str,
 ) -> str:
     """
@@ -55,10 +56,13 @@ async def document_qa(
     Returns an answer to all questions based on the document content.
 
     Args:
-    question: Question (or questions) to be answered about the document.
-    document: The full text of the document to analyze.
+        question: Question (or questions) to be answered about the document.
+        document: The full text of the document to analyze.
     """
     assert question and question.strip(), "Please provide non-empty 'question'"
+    if isinstance(document, dict):
+        document = json.dumps(document)
+
     assert document and document.strip(), "Please provide non-empty 'document'"
 
     model_name = os.getenv("DOCUMENT_QA_MODEL_NAME", "deepseek/deepseek-chat-v3-0324")

academia_mcp/tools/web_search.py

@@ -28,16 +28,31 @@ def web_search(
         limit: The maximum number of items to return. 20 by default, maximum 25.
         provider: The provider to use. "exa", "tavily" or "brave". "tavily" by default.
     """
-    assert provider in (
-        "exa",
-        "tavily",
-        "brave",
-    ), "Error: provider must be either 'exa', 'tavily' or 'brave'"
+    providers = ("tavily", "brave", "exa")
+    assert provider in providers, "Error: provider must be either 'exa', 'tavily' or 'brave'"
+
+    is_tavily_available = os.getenv("TAVILY_API_KEY") is not None
+    is_exa_available = os.getenv("EXA_API_KEY") is not None
+    is_brave_available = os.getenv("BRAVE_API_KEY") is not None
+    assert is_tavily_available or is_exa_available or is_brave_available
+    availability = {
+        "tavily": is_tavily_available,
+        "brave": is_brave_available,
+        "exa": is_exa_available,
+    }
+
+    if not availability[provider]:
+        for p in providers:
+            if availability[p]:
+                provider = p
+                break
 
     if provider == "exa":
         return exa_web_search(query, limit)
-    if provider == "brave":
+    elif provider == "brave":
         return brave_web_search(query, limit)
+
+    assert provider == "tavily"
     return tavily_web_search(query, limit)
 
 

{academia_mcp-1.2.0.dist-info → academia_mcp-1.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: academia-mcp
-Version: 1.2.0
+Version: 1.2.2
 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.2.0.dist-info → academia_mcp-1.2.2.dist-info}/RECORD

@@ -10,16 +10,16 @@ academia_mcp/tools/anthology_search.py,sha256=_5s8EzdV7NQD_F3bjVH4XlKKHOJlFtWlQV
 academia_mcp/tools/arxiv_download.py,sha256=xanzt77TZBQRngzGbKCRz4Hp-Mwfe_q-46eRW23TpVs,11219
 academia_mcp/tools/arxiv_search.py,sha256=pzM18qrF3QL03A53w003kE7hQi3s3QKtjgw0m7K88UY,8355
 academia_mcp/tools/bitflip.py,sha256=u0hSOPWbnCDu2EbA_RkueX496SvTKz9QhZcXugshSfI,10949
-academia_mcp/tools/document_qa.py,sha256=Z51X_rm9p53SltKQJeN0-1CDrCefiYujap1vZZ88uU8,2132
+academia_mcp/tools/document_qa.py,sha256=lWtzRNFKwQpQO8vPNvYYbNnGh3LsBmGl6f9vSRIuqaw,2245
 academia_mcp/tools/hf_datasets_search.py,sha256=KiBkqT4rXjEN4oc1AWZOPnqN_Go90TQogY5-DUm3LQo,2854
 academia_mcp/tools/md_to_pdf.py,sha256=Ovc_-8j7gIZNEM1d0ZDH-8qbtfZLSaNmCm5DQjrtM0k,12810
 academia_mcp/tools/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 academia_mcp/tools/s2_citations.py,sha256=dqrBp76RrX1zH2XzcMAoWBbvbtyhxLeF-xnqOKD_JiM,4852
 academia_mcp/tools/visit_webpage.py,sha256=0zAZYeQxPDu0OjgAAvbMLZh0ttaS5q-_4WhgsEPrbsI,1542
-academia_mcp/tools/web_search.py,sha256=NAkbXdD9mKxsIXhWN32dRd_EiaB3G6ENy-n-bc7HAaQ,5448
-academia_mcp-1.2.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-academia_mcp-1.2.0.dist-info/METADATA,sha256=bVlYEucMW7togYS7J-nwq609Qx-MFCSx8vADvucSvvo,1899
-academia_mcp-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-academia_mcp-1.2.0.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
-academia_mcp-1.2.0.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
-academia_mcp-1.2.0.dist-info/RECORD,,
+academia_mcp/tools/web_search.py,sha256=P74HF5agF-cWOGUYmrDcr_R-Ay3xz9fNdhFov8vtt9o,6025
+academia_mcp-1.2.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+academia_mcp-1.2.2.dist-info/METADATA,sha256=SC1JQUbH4sbFFZFcuTeC-iwEVIhMbQqXlSL2sB6riqM,1899
+academia_mcp-1.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+academia_mcp-1.2.2.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
+academia_mcp-1.2.2.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
+academia_mcp-1.2.2.dist-info/RECORD,,