Spaces:
Sleeping
Sleeping
| """Summary MCP tools for paper-grounded chat.""" | |
| from __future__ import annotations | |
| import json | |
| from backend.chat_settings import truncate_summary | |
| from local_rag.summaries import summaries_for_documents | |
| def get_paper_summaries_tool(document_ids: list[str]) -> str: | |
| return json.dumps({"sources": summaries_for_documents(document_ids)}, indent=2) | |
| def build_paper_context_tool(document_ids: list[str], question: str = "") -> str: | |
| sources = summaries_for_documents(document_ids) | |
| if not sources: | |
| return json.dumps({"error": "No summaries found for the selected papers."}, indent=2) | |
| n_papers = len(sources) | |
| parts: list[str] = [] | |
| for index, source in enumerate(sources, start=1): | |
| parts.append( | |
| f"Paper {index}: {source['title']}\n" | |
| f"File: {source['file_name']}\n\n" | |
| f"{truncate_summary(source['text'], n_papers=n_papers, question=question)}" | |
| ) | |
| context = "\n\n---\n\n".join(parts) | |
| return json.dumps( | |
| { | |
| "context": context, | |
| "sources": [{"document_id": s["document_id"], "title": s["title"]} for s in sources], | |
| "n_papers": n_papers, | |
| "question": question, | |
| }, | |
| indent=2, | |
| ) | |