academia-mcp 1.6.5__tar.gz → 1.6.7__tar.gz

{academia_mcp-1.6.5 → academia_mcp-1.6.7}/PKG-INFO

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

@@ -161,28 +161,6 @@ class BitFlipInfo(BaseModel):  # type: ignore
     spark: str
 
 
-class Proposal(BaseModel):  # type: ignore
-    proposal_id: Optional[int] = None
-    flip: str
-    spark: str
-    abstract: str
-    experiments: List[str]
-    risks_and_limitations: List[str]
-
-
-class ProposalScores(BaseModel):  # type: ignore
-    proposal_id: int
-    spark: str
-    strengths: List[str]
-    weaknesses: List[str]
-    novelty: int
-    clarity: int
-    significance: int
-    feasibility: int
-    soundness: int
-    overall: int
-
-
 async def extract_bitflip_info(arxiv_id: str) -> str:
     """
     Extracts the Bit-Flip information from the arXiv paper.
@@ -244,12 +222,11 @@ async def generate_research_proposal(bit: str, additional_context: str = "") ->
         model_name=model_name, messages=[ChatMessage(role="user", content=prompt)]
     )
     result = extract_json(content)
-    proposal: Proposal = Proposal.model_validate(result)
-    proposal.proposal_id = random.randint(0, 1000000)
-    return str(proposal.model_dump_json())
+    result["proposal_id"] = random.randint(0, 1000000)
+    return json.dumps(result, ensure_ascii=False)
 
 
-async def score_research_proposals(proposals: List[str | Dict[str, Any] | Any]) -> 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,10 +253,12 @@ async def score_research_proposals(proposals: List[str | Dict[str, Any] | Any])
         proposals: A list of JSON strings with research proposals.
     """
     model_name = os.getenv("BITFLIP_MODEL_NAME", "deepseek/deepseek-chat-v3-0324")
+    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)]
     )
     scores = extract_json(content)
-    final_scores = [ProposalScores.model_validate(score) for score in scores]
-    return json.dumps([s.model_dump() for s in final_scores], ensure_ascii=False)
+    return json.dumps(scores, ensure_ascii=False)

{academia_mcp-1.6.5 → academia_mcp-1.6.7}/academia_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: academia-mcp
-Version: 1.6.5
+Version: 1.6.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.6.5 → academia_mcp-1.6.7}/pyproject.toml

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

{academia_mcp-1.6.5 → academia_mcp-1.6.7}/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