academia-mcp 1.6.4__tar.gz → 1.6.6__tar.gz

{academia_mcp-1.6.4 → academia_mcp-1.6.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: academia-mcp
-Version: 1.6.4
+Version: 1.6.6
 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.6.4 → academia_mcp-1.6.6}/academia_mcp/tools/bitflip.py

@@ -4,7 +4,7 @@
 import json
 import os
 import random
-from typing import List, Optional, Any
+from typing import List, Optional, Any, Dict
 
 from pydantic import BaseModel
 from datasets import load_dataset  # type: ignore
@@ -249,7 +249,7 @@ async def generate_research_proposal(bit: str, additional_context: str = "") ->
     return str(proposal.model_dump_json())
 
 
-async def score_research_proposals(proposals: List[str]) -> str:
+async def score_research_proposals(proposals: str | List[str | Dict[str, Any] | Any]) -> str:
     """
     Scores a list of research proposals.
     Use proposals obtained with the `generate_research_proposal` tool.
@@ -276,7 +276,10 @@ async def score_research_proposals(proposals: List[str]) -> str:
         proposals: A list of JSON strings with research proposals.
     """
     model_name = os.getenv("BITFLIP_MODEL_NAME", "deepseek/deepseek-chat-v3-0324")
-    prompt = encode_prompt(SCORE_PROMPT, proposals=proposals)
+    if isinstance(proposals, str):
+        proposals = json.loads(proposals)
+        assert isinstance(proposals, list), "Proposals should be a list of JSON strings"
+    prompt = encode_prompt(SCORE_PROMPT, proposals=[str(p) for p in proposals])
     content = await llm_acall(
         model_name=model_name, messages=[ChatMessage(role="user", content=prompt)]
     )

{academia_mcp-1.6.4 → academia_mcp-1.6.6}/academia_mcp.egg-info/PKG-INFO

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

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

{academia_mcp-1.6.4 → academia_mcp-1.6.6}/tests/test_bitflip.py

@@ -36,3 +36,19 @@ async def test_bitflip_score_research_proposals() -> None:
     assert scores[1]["strengths"] is not None
     assert scores[0]["weaknesses"] is not None
     assert scores[1]["weaknesses"] is not None
+
+
+async def test_bitflip_score_research_proposals_str() -> None:
+    arxiv_id = "2503.07826"
+    bit = json.loads(await extract_bitflip_info(arxiv_id))["bit"]
+    proposal1 = await generate_research_proposal(bit=bit)
+    proposal2 = await generate_research_proposal(bit=bit)
+    scores = json.loads(await score_research_proposals(json.dumps([proposal1, proposal2])))
+    assert scores
+    assert len(scores) == 2
+    assert scores[0]["spark"] is not None
+    assert scores[1]["spark"] is not None
+    assert scores[0]["strengths"] is not None
+    assert scores[1]["strengths"] is not None
+    assert scores[0]["weaknesses"] is not None
+    assert scores[1]["weaknesses"] is not None