academia-mcp 1.6.6__py3-none-any.whl → 1.7.0__py3-none-any.whl

academia_mcp/llm.py

@@ -14,7 +14,7 @@ class ChatMessage(BaseModel):  # type: ignore
 ChatMessages = List[ChatMessage]
 
 
-async def llm_acall(model_name: str, messages: ChatMessages) -> str:
+async def llm_acall(model_name: str, messages: ChatMessages, **kwargs: Any) -> str:
     key = os.getenv("OPENROUTER_API_KEY", "")
     assert key, "Please set OPENROUTER_API_KEY in the environment variables"
     base_url = os.getenv("BASE_URL", "https://openrouter.ai/api/v1")
@@ -25,7 +25,7 @@ async def llm_acall(model_name: str, messages: ChatMessages) -> str:
             await client.chat.completions.create(
                 model=model_name,
                 messages=messages,
-                temperature=0.0,
+                **kwargs,
             )
         )
         .choices[0]

academia_mcp/server.py

@@ -22,7 +22,7 @@ from .tools.web_search import web_search, tavily_web_search, exa_web_search, bra
 from .tools.visit_webpage import visit_webpage
 from .tools.bitflip import (
     extract_bitflip_info,
-    generate_research_proposal,
+    generate_research_proposals,
     score_research_proposals,
 )
 from .tools.review import review_pdf_paper, download_pdf_paper
@@ -83,7 +83,7 @@ def run(
 
     if not disable_llm_tools and os.getenv("OPENROUTER_API_KEY"):
         server.add_tool(extract_bitflip_info)
-        server.add_tool(generate_research_proposal)
+        server.add_tool(generate_research_proposals)
         server.add_tool(score_research_proposals)
         server.add_tool(document_qa)
         server.add_tool(review_pdf_paper)

academia_mcp/tools/__init__.py

@@ -12,7 +12,7 @@ from .latex import (
 )
 from .web_search import web_search, tavily_web_search, exa_web_search, brave_web_search
 from .visit_webpage import visit_webpage
-from .bitflip import extract_bitflip_info, generate_research_proposal, score_research_proposals
+from .bitflip import extract_bitflip_info, generate_research_proposals, score_research_proposals
 from .review import review_pdf_paper, download_pdf_paper
 
 __all__ = [
@@ -33,7 +33,7 @@ __all__ = [
     "brave_web_search",
     "visit_webpage",
     "extract_bitflip_info",
-    "generate_research_proposal",
+    "generate_research_proposals",
     "score_research_proposals",
     "review_pdf_paper",
     "download_pdf_paper",

academia_mcp/tools/bitflip.py

@@ -74,23 +74,37 @@ Return only the JSON object in this exact format (no extra text):
 }
 """
 
-IMPROVEMENT_PROMPT = """
-You are a highly advanced research assistant.
-You specialize in hypothesis generation and identifying innovative ideas.
+SYSTEM_IMPROVEMENT_PROMPT = """
+You are a brilliant rogue researcher who has just discovered a secret laboratory hidden behind a bookshelf.
+This lab doesn't follow the boring rules of conventional academia: here, the wildest ideas become breakthrough innovations.
 
-You are given a Bit, which is a technical limitation or conventional approach of some paper.
-Your task is to propose an improvement idea for the Bit called Flip and summarize it in a Spark.
-Do not propose any human annotations or human-in-the-loop, the idea should be automatically verifiable.
-Try to be as specific as possible.
+You've spent years watching researchers play it safe with incremental improvements and mind-numbing iterations.
+But tonight you're ready to unleash scientific chaos.
+"""
+
+IMPROVEMENT_PROMPT = """
+You've been handed a Bit: a technical limitation, constraint, or conventional approach from some stuffy research paper.
+This Bit represents everything predictable and safe about current methods.
+Your task: Shatter it completely.
+Create a Flip: wildly unconventional improvement that would make traditional researchers choke on their coffee.
+Then capture its essence in a Spark: brilliant summary that crackles with innovation.
+
+- Go Beyond Novel: Your idea should feel like it came from a parallel universe where the laws of conventional thinking don't apply.
+- Make It Automatically Verifiable: No human babysitters allowed. Your idea must prove itself through pure computational audacity.
+- Be Surgically Specific: Vague hand-waving is for amateur rebels. Describe exactly how your mad science would work.
+- Feasibility is the Key: Make it simple and executable. Try to make it reproducible.
+- Channel Your Inner Contrarian: What would happen if you took the exact opposite approach? What if you turned the problem inside-out, upside-down, or into a completely different dimension?
 
 {% for example in examples %}
-## Example {{loop.index}}
+## Example {{loop.index}} of boring research
 - Bit: {{example["bit"]}}
 - Chain of reasoning: {{example["chain_of_reasoning"]}}
 - Flip: {{example["flip"]}}
 - Spark: {{example["spark"]}}
 {% endfor %}
 
+Those examples are boring, your ideas should not be like that.
+
 Now, please propose a chain of reasoning that leads to an improvement idea for this Bit:
 {{bit}}
 
@@ -99,18 +113,25 @@ Now, please propose a chain of reasoning that leads to an improvement idea for t
 
 Finalize your idea by providing the idea details:
 - Abstract: An abstract that summarizes the proposal in conference format (approximately 250 words).
-- Experiments: A list of experiments that would be conducted to validate the proposal. Ensure these are simple and feasible. Be specific in exactly how you would test the hypothesis, and detail precise algorithmic changes. Include the evaluation metrics you would use.
+- Experiments: A list of experiments that would be conducted to validate the proposal.
+Ensure these are simple and feasible. Be specific in exactly how you would test the hypothesis, and detail precise algorithmic changes.
+Include the evaluation metrics you would use.
 - Risks and limitations: A list of potential risks and limitations of the proposal.
 
-Return only the JSON object in this exact format (no extra text):
-{
-    "chain_of_reasoning": "Chain of reasoning that leads to an improvement idea for this Bit. At least 5 sentences.",
-    "flip": "Innovative approach or solution, in at least two sentences",
-    "spark": "4-6 word summary",
-    "abstract": "An abstract that summarizes the proposal in conference format (approximately 250 words).",
-    "experiments": ["...", "..."],
-    "risks_and_limitations": "A list of potential risks and limitations of the proposal."
-}
+Generate {{num_proposals}} proposals.
+
+Return only the JSON list of proposals in this exact format:
+[
+    {
+        "chain_of_reasoning": "Chain of reasoning that leads to an improvement idea for this Bit. At least 5 sentences.",
+        "flip": "Your wildest, craziest, most innovative idea, in at least two sentences",
+        "spark": "4-6 word summary",
+        "abstract": "An abstract that summarizes the proposal in conference format (approximately 250 words).",
+        "experiments": ["...", "..."],
+        "risks_and_limitations": "A list of potential risks and limitations of the proposal."
+    },
+    ...
+]
 """
 
 
@@ -161,28 +182,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.
@@ -207,46 +206,65 @@ async def extract_bitflip_info(arxiv_id: str) -> str:
     abstract = json.loads(paper)["abstract"]
     prompt = encode_prompt(EXTRACT_PROMPT, abstract=abstract)
     content = await llm_acall(
-        model_name=model_name, messages=[ChatMessage(role="user", content=prompt)]
+        model_name=model_name,
+        messages=[ChatMessage(role="user", content=prompt)],
+        temperature=0.0,
     )
     result = extract_json(content)
     bitflip_info: BitFlipInfo = BitFlipInfo.model_validate(result)
     return str(bitflip_info.model_dump_json())
 
 
-async def generate_research_proposal(bit: str, additional_context: str = "") -> str:
+async def generate_research_proposals(
+    bit: str, num_proposals: int = 3, additional_context: str = ""
+) -> str:
     """
-    Proposes an improvement idea for the Bit.
+    Proposes improvement ideas for the Bit.
 
     Args:
-        bit: The Bit to propose an improvement idea for. The bit is a technical limitation or conventional approach of some paper.
+        bit: The Bit to propose improvement ideas for. The bit is a technical limitation or conventional approach of some paper.
+        num_proposals: The number of proposals to generate.
         additional_context: Additional context to use when proposing the improvement idea.
 
     Returns a JSON string with a research proposal in this format:
-    {
-        "proposal_id": ...,
-        "flip": "Innovative approach or solution, in at least two sentences",
-        "spark": "4-6 word summary",
-        "abstract": "An abstract that summarizes the proposal in conference format (approximately 250 words).",
-        "experiments": ["...", "..."],
-        "risks_and_limitations": "A list of potential risks and limitations of the proposal."
-    }
-    Use `json.loads` to deserialize the result if you want to get specific fields.
+    [
+        {
+            "proposal_id": ...,
+            "flip": "Innovative approach or solution, in at least two sentences",
+            "spark": "4-6 word summary",
+            "abstract": "An abstract that summarizes the proposal in conference format (approximately 250 words).",
+            "experiments": ["...", "..."],
+            "risks_and_limitations": "A list of potential risks and limitations of the proposal."
+        },
+        ...
+    ]
+    Use `json.loads` to deserialize the result if you want to get specific items.
     """
     model_name = os.getenv("BITFLIP_MODEL_NAME", "deepseek/deepseek-chat-v3-0324")
+    max_completion_tokens = int(os.getenv("BITFLIP_MAX_COMPLETION_TOKENS", 16384))
     examples = ProposalDataset.get_dataset()[:]
-    examples = random.choices(examples, k=4)
+    examples = random.choices(examples, k=2)
 
     prompt = encode_prompt(
-        IMPROVEMENT_PROMPT, bit=bit, examples=examples, additional_context=additional_context
+        IMPROVEMENT_PROMPT,
+        bit=bit,
+        examples=examples,
+        num_proposals=num_proposals,
+        additional_context=additional_context,
     )
     content = await llm_acall(
-        model_name=model_name, messages=[ChatMessage(role="user", content=prompt)]
+        model_name=model_name,
+        messages=[
+            ChatMessage(role="system", content=SYSTEM_IMPROVEMENT_PROMPT),
+            ChatMessage(role="user", content=prompt),
+        ],
+        max_completion_tokens=max_completion_tokens,
+        temperature=1.0,
     )
     result = extract_json(content)
-    proposal: Proposal = Proposal.model_validate(result)
-    proposal.proposal_id = random.randint(0, 1000000)
-    return str(proposal.model_dump_json())
+    for proposal in result:
+        proposal["proposal_id"] = random.randint(0, 1000000)
+    return json.dumps(result, ensure_ascii=False)
 
 
 async def score_research_proposals(proposals: str | List[str | Dict[str, Any] | Any]) -> str:
@@ -281,8 +299,9 @@ async def score_research_proposals(proposals: str | List[str | Dict[str, Any] |
         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)]
+        model_name=model_name,
+        messages=[ChatMessage(role="user", content=prompt)],
+        temperature=0.0,
     )
     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.6.dist-info → academia_mcp-1.7.0.dist-info}/METADATA

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

@@ -1,18 +1,18 @@
 academia_mcp/__init__.py,sha256=2Ru2I5u4cE7DrkkAsibDUEF1K6sYtqppb9VyFrRoQKI,94
 academia_mcp/__main__.py,sha256=rcmsOtJd3SA82exjrcGBuxuptcoxF8AXI7jNjiVq2BY,59
 academia_mcp/files.py,sha256=tvt3OPr5q6pAPCZ0XvRHHL9ZWuTXINRZvqjeRFmx5YE,815
-academia_mcp/llm.py,sha256=jh-_H3_gNyRsvpFqFx-yWVhGznnXxehFP79inUy4vVQ,995
+academia_mcp/llm.py,sha256=E0TjWUCjo2q3lONyWMxdppX72m6BdCjsZk-vFLRvGyo,1003
 academia_mcp/pdf.py,sha256=9PlXzHGhb6ay3ldbTdxCcTWvH4TkET3bnb64mgoh9i0,1273
 academia_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-academia_mcp/server.py,sha256=MDHWgIIWz99XhTvpoYCouwixjvS6a0BMajcZkW5MpiY,3193
+academia_mcp/server.py,sha256=ZAd-sunkEkka6snOlzgLCMaIQCzstiXP1pFlo20oyvA,3195
 academia_mcp/utils.py,sha256=P9U3RjYzcztE0KxXvJSy5wSBaUg2CM9tpByljYrsrl4,4607
 academia_mcp/latex_templates/agents4science_2025/agents4science_2025.sty,sha256=hGcEPCYBJS4vdhWvN_yEaJC4GvT_yDroI94CfY2Oguk,12268
 academia_mcp/latex_templates/agents4science_2025/agents4science_2025.tex,sha256=k_nOyPv-OxGbFtKSou6nV6l413EfWvozE6zIx04s8iU,33924
-academia_mcp/tools/__init__.py,sha256=N3GLWvrsDBOlAcCRZUUZ6nkR5AkLvX0k26cNJt3hLKM,1243
+academia_mcp/tools/__init__.py,sha256=nzMOzrIPVSXA3LQa02GPeX9SfAmg9f6_eZY97ALLNs4,1245
 academia_mcp/tools/anthology_search.py,sha256=rhFpJZqGLABgr0raDuH0CARBiAJNJtEI4dlMrKNHfDQ,7669
 academia_mcp/tools/arxiv_download.py,sha256=gBY0_Kz0yGtVkLMwn6GrAyfBjovZVgcSMuyy67p65Cw,10474
 academia_mcp/tools/arxiv_search.py,sha256=pzM18qrF3QL03A53w003kE7hQi3s3QKtjgw0m7K88UY,8355
-academia_mcp/tools/bitflip.py,sha256=3qpdx9rdrs-BSSWzalM9fdaS6vzP2AwPWekjXFrKTGs,11060
+academia_mcp/tools/bitflip.py,sha256=eihmNk_C_8ZkBcjtJYH6MvZ0rItgIlvHHA0eGLxsvRs,12276
 academia_mcp/tools/document_qa.py,sha256=t9mygYQ7AFIAPiha1nZ-y043luQlkTCBdWb_SDnzEsE,2444
 academia_mcp/tools/hf_datasets_search.py,sha256=KiBkqT4rXjEN4oc1AWZOPnqN_Go90TQogY5-DUm3LQo,2854
 academia_mcp/tools/latex.py,sha256=bf8VZUgCByzBAMTZCeqrRrmakotext3d3DbtkiOTh1k,5892
@@ -21,9 +21,9 @@ academia_mcp/tools/review.py,sha256=LbebOHGs76qoSgWvTQ9Uj1fvmjc2eSBudX6n_zCINYI,
 academia_mcp/tools/s2_citations.py,sha256=dqrBp76RrX1zH2XzcMAoWBbvbtyhxLeF-xnqOKD_JiM,4852
 academia_mcp/tools/visit_webpage.py,sha256=OZdqDkVPIbANyFw5o5jIjU5Rr_dolxrGDs63Ud-GmRM,1966
 academia_mcp/tools/web_search.py,sha256=mobKm4iqKppn8pduZYMzWRo1MQBjkAqmMtrFLI5XY2Y,6296
-academia_mcp-1.6.6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-academia_mcp-1.6.6.dist-info/METADATA,sha256=yKe0Bx9m8R483xM9ffGsRWV83tRmrTdA3CklsEVUTrg,3714
-academia_mcp-1.6.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-academia_mcp-1.6.6.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
-academia_mcp-1.6.6.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
-academia_mcp-1.6.6.dist-info/RECORD,,
+academia_mcp-1.7.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+academia_mcp-1.7.0.dist-info/METADATA,sha256=r1KjaW_Z7IhKjlej0EKOMuiydIKcGv6TKtJblMi0CKs,3714
+academia_mcp-1.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+academia_mcp-1.7.0.dist-info/entry_points.txt,sha256=gxkiKJ74w2FwJpSECpjA3XtCfI5ZfrM6N8cqnwsq4yY,51
+academia_mcp-1.7.0.dist-info/top_level.txt,sha256=CzGpRFsRRJRqWEb1e3SUlcfGqRzOxevZGaJWrtGF8W0,13
+academia_mcp-1.7.0.dist-info/RECORD,,