all-in-mcp 0.2.7__py3-none-any.whl → 0.2.8__py3-none-any.whl

all_in_mcp/academic_platforms/base.py

@@ -18,6 +18,6 @@ class PaperSource(ABC):
         raise NotImplementedError
 
     @abstractmethod
-    def read_paper(self, paper_id: str, save_path: str) -> str:
+    def read_paper(self, paper_id: str, save_path: str, start_page: int | None = None, end_page: int | None = None) -> str:
         """Read and extract text content from a paper"""
         raise NotImplementedError

all_in_mcp/academic_platforms/crossref.py

@@ -228,9 +228,15 @@ class CrossrefSearcher(PaperSource):
         """
         return "Crossref does not provide a direct way to download PDFs. Use the paper's URL or DOI to access the publisher's site for PDF downloads if available."
 
-    def read_paper(self, paper_id: str, save_path: str) -> str:
+    def read_paper(self, paper_id: str, save_path: str, start_page: int | None = None, end_page: int | None = None) -> str:
         """
         crossref doesn't provide a direct way to read paper text.
+        
+        Args:
+            paper_id: Paper identifier
+            save_path: Directory where papers are stored
+            start_page: Starting page number (1-indexed, inclusive). Defaults to 1.
+            end_page: Ending page number (1-indexed, inclusive). Defaults to last page.
         """
         return "Crossref does not provide a direct way to read paper text. Use the download_pdf method to get the PDF if available."
 

all_in_mcp/academic_platforms/cryptobib.py

@@ -364,9 +364,15 @@ class CryptoBibSearcher(PaperSource):
         """
         return "Error: CryptoBib is a bibliography database and doesn't provide PDF downloads"
 
-    def read_paper(self, paper_id: str, save_path: str = "./downloads") -> str:
+    def read_paper(self, paper_id: str, save_path: str = "./downloads", start_page: int | None = None, end_page: int | None = None) -> str:
         """
         CryptoBib doesn't provide paper content reading
+        
+        Args:
+            paper_id: Paper identifier
+            save_path: Directory where papers are stored
+            start_page: Starting page number (1-indexed, inclusive). Defaults to 1.
+            end_page: Ending page number (1-indexed, inclusive). Defaults to last page.
         """
         return "Error: CryptoBib is a bibliography database and doesn't provide paper content"
 

all_in_mcp/academic_platforms/google_scholar.py

@@ -228,13 +228,15 @@ class GoogleScholarSearcher(PaperSource):
             "Please use the paper URL to access the publisher's website."
         )
 
-    def read_paper(self, paper_id: str, save_path: str = "./downloads") -> str:
+    def read_paper(self, paper_id: str, save_path: str = "./downloads", start_page: int | None = None, end_page: int | None = None) -> str:
         """
         Google Scholar doesn't support direct paper reading
 
         Args:
             paper_id: Paper identifier
             save_path: Directory where papers are stored
+            start_page: Starting page number (1-indexed, inclusive). Defaults to 1.
+            end_page: Ending page number (1-indexed, inclusive). Defaults to last page.
 
         Returns:
             Message indicating the feature is not supported

all_in_mcp/academic_platforms/iacr.py

@@ -232,13 +232,15 @@ class IACRSearcher(PaperSource):
             logger.error(f"PDF download error: {e}")
             return f"Error downloading PDF: {e}"
 
-    def read_paper(self, paper_id: str, save_path: str = "./downloads") -> str:
+    def read_paper(self, paper_id: str, save_path: str = "./downloads", start_page: int | None = None, end_page: int | None = None) -> str:
         """
         Download and extract text from IACR paper PDF
 
         Args:
             paper_id: IACR paper ID
             save_path: Directory to save downloaded PDF
+            start_page: Starting page number (1-indexed, inclusive). Defaults to 1.
+            end_page: Ending page number (1-indexed, inclusive). Defaults to last page.
 
         Returns:
             str: Extracted text from the PDF or error message
@@ -249,56 +251,24 @@ class IACRSearcher(PaperSource):
             if not paper or not paper.pdf_url:
                 return f"Error: Could not find PDF URL for paper {paper_id}"
 
-            # Download the PDF
-            pdf_response = requests.get(paper.pdf_url, timeout=30)
-            pdf_response.raise_for_status()
+            # Import read_pdf function
+            from ..paper import read_pdf
 
-            # Create download directory if it doesn't exist
-            os.makedirs(save_path, exist_ok=True)
-
-            # Save the PDF
-            filename = f"iacr_{paper_id.replace('/', '_')}.pdf"
-            pdf_path = os.path.join(save_path, filename)
-
-            with open(pdf_path, "wb") as f:
-                f.write(pdf_response.content)
-
-            # Extract text using pypdf
-            reader = PdfReader(pdf_path)
-            text = ""
-
-            for page_num, page in enumerate(reader.pages):
-                try:
-                    page_text = page.extract_text()
-                    if page_text:
-                        text += f"\n--- Page {page_num + 1} ---\n"
-                        text += page_text + "\n"
-                except Exception as e:
-                    logger.warning(
-                        f"Failed to extract text from page {page_num + 1}: {e}"
-                    )
-                    continue
-
-            if not text.strip():
-                return (
-                    f"PDF downloaded to {pdf_path}, but unable to extract readable text"
-                )
+            # Use the read_pdf function to extract text
+            text = read_pdf(paper.pdf_url, start_page, end_page)
 
             # Add paper metadata at the beginning
             metadata = f"Title: {paper.title}\n"
             metadata += f"Authors: {', '.join(paper.authors)}\n"
             metadata += f"Published Date: {paper.published_date}\n"
             metadata += f"URL: {paper.url}\n"
-            metadata += f"PDF downloaded to: {pdf_path}\n"
+            metadata += f"PDF URL: {paper.pdf_url}\n"
             metadata += "=" * 80 + "\n\n"
 
-            return metadata + text.strip()
+            return metadata + text
 
-        except requests.RequestException as e:
-            logger.error(f"Error downloading PDF: {e}")
-            return f"Error downloading PDF: {e}"
         except Exception as e:
-            logger.error(f"Read paper error: {e}")
+            logger.error(f"Error reading paper: {e}")
             return f"Error reading paper: {e}"
 
     def get_paper_details(self, paper_id: str) -> Paper | None:

all_in_mcp/paper.py

@@ -86,19 +86,21 @@ class Paper:
         return read_pdf(self.pdf_url)
 
 
-def read_pdf(pdf_source: str | Path) -> str:
+def read_pdf(pdf_source: str | Path, start_page: int | None = None, end_page: int | None = None) -> str:
     """
     Extract text content from a PDF file (local or online).
 
     Args:
         pdf_source: Path to local PDF file or URL to online PDF
+        start_page: Starting page number (1-indexed, inclusive). Defaults to 1.
+        end_page: Ending page number (1-indexed, inclusive). Defaults to last page.
 
     Returns:
         str: Extracted text content from the PDF
 
     Raises:
         FileNotFoundError: If local file doesn't exist
-        ValueError: If URL is invalid or PDF cannot be processed
+        ValueError: If URL is invalid, PDF cannot be processed, or page range is invalid
         Exception: For other PDF processing errors
     """
     try:
@@ -109,10 +111,10 @@ def read_pdf(pdf_source: str | Path) -> str:
             parsed = urlparse(pdf_source_str)
             if parsed.scheme in ("http", "https"):
                 # Handle online PDF
-                return _read_pdf_from_url(pdf_source_str)
+                return _read_pdf_from_url(pdf_source_str, start_page, end_page)
             else:
                 # Handle local file
-                return _read_pdf_from_file(Path(pdf_source_str))
+                return _read_pdf_from_file(Path(pdf_source_str), start_page, end_page)
         else:
             raise ValueError("pdf_source must be a string or Path object")
 
@@ -120,7 +122,46 @@ def read_pdf(pdf_source: str | Path) -> str:
         raise Exception(f"Failed to read PDF from {pdf_source}: {e!s}") from e
 
 
-def _read_pdf_from_file(file_path: Path) -> str:
+def _normalize_page_range(start_page: int | None, end_page: int | None, total_pages: int) -> tuple[int, int]:
+    """
+    Normalize and validate page range parameters.
+    
+    Args:
+        start_page: Starting page number (1-indexed, inclusive) or None
+        end_page: Ending page number (1-indexed, inclusive) or None
+        total_pages: Total number of pages in the PDF
+        
+    Returns:
+        tuple[int, int]: (start_index, end_index) as 0-indexed values
+        
+    Raises:
+        ValueError: If page range is invalid
+    """
+    # Default values
+    if start_page is None:
+        start_page = 1
+    if end_page is None:
+        end_page = total_pages
+        
+    # Validate page numbers
+    if start_page < 1:
+        raise ValueError(f"start_page must be >= 1, got {start_page}")
+    if end_page < 1:
+        raise ValueError(f"end_page must be >= 1, got {end_page}")
+    if start_page > end_page:
+        raise ValueError(f"start_page ({start_page}) must be <= end_page ({end_page})")
+    if start_page > total_pages:
+        raise ValueError(f"start_page ({start_page}) exceeds total pages ({total_pages})")
+        
+    # Clamp end_page to total_pages
+    if end_page > total_pages:
+        end_page = total_pages
+        
+    # Convert to 0-indexed
+    return start_page - 1, end_page - 1
+
+
+def _read_pdf_from_file(file_path: Path, start_page: int | None = None, end_page: int | None = None) -> str:
     """Read PDF from local file path."""
     if not file_path.exists():
         raise FileNotFoundError(f"PDF file not found: {file_path}")
@@ -131,10 +172,16 @@ def _read_pdf_from_file(file_path: Path) -> str:
     try:
         with open(file_path, "rb") as file:
             pdf_reader = PdfReader(file)
+            total_pages = len(pdf_reader.pages)
+            
+            # Validate and normalize page range
+            start_idx, end_idx = _normalize_page_range(start_page, end_page, total_pages)
+            
             text_content = []
 
-            for page_num, page in enumerate(pdf_reader.pages):
+            for page_num in range(start_idx, end_idx + 1):
                 try:
+                    page = pdf_reader.pages[page_num]
                     page_text = page.extract_text()
                     if page_text.strip():  # Only add non-empty pages
                         text_content.append(
@@ -151,7 +198,7 @@ def _read_pdf_from_file(file_path: Path) -> str:
         raise Exception(f"Error reading PDF file {file_path}: {e!s}") from e
 
 
-def _read_pdf_from_url(url: str) -> str:
+def _read_pdf_from_url(url: str, start_page: int | None = None, end_page: int | None = None) -> str:
     """Download and read PDF from URL."""
     try:
         # Download PDF with proper headers
@@ -175,10 +222,16 @@ def _read_pdf_from_url(url: str) -> str:
             # Read PDF from bytes
             pdf_bytes = io.BytesIO(response.content)
             pdf_reader = PdfReader(pdf_bytes)
+            total_pages = len(pdf_reader.pages)
+            
+            # Validate and normalize page range
+            start_idx, end_idx = _normalize_page_range(start_page, end_page, total_pages)
+            
             text_content = []
 
-            for page_num, page in enumerate(pdf_reader.pages):
+            for page_num in range(start_idx, end_idx + 1):
                 try:
+                    page = pdf_reader.pages[page_num]
                     page_text = page.extract_text()
                     if page_text.strip():  # Only add non-empty pages
                         text_content.append(

all_in_mcp/server.py

@@ -93,6 +93,16 @@ async def handle_list_tools() -> list[types.Tool]:
                         "description": "Directory where the PDF is/will be saved (default: './downloads')",
                         "default": "./downloads",
                     },
+                    "start_page": {
+                        "type": "integer",
+                        "description": "Starting page number (1-indexed, inclusive). Defaults to 1.",
+                        "minimum": 1,
+                    },
+                    "end_page": {
+                        "type": "integer",
+                        "description": "Ending page number (1-indexed, inclusive). Defaults to last page.",
+                        "minimum": 1,
+                    },
                 },
                 "required": ["paper_id"],
             },
@@ -203,6 +213,16 @@ async def handle_list_tools() -> list[types.Tool]:
                         "type": "string",
                         "description": "Path to local PDF file or URL to online PDF",
                     },
+                    "start_page": {
+                        "type": "integer",
+                        "description": "Starting page number (1-indexed, inclusive). Defaults to 1.",
+                        "minimum": 1,
+                    },
+                    "end_page": {
+                        "type": "integer",
+                        "description": "Ending page number (1-indexed, inclusive). Defaults to last page.",
+                        "minimum": 1,
+                    },
                 },
                 "required": ["pdf_source"],
             },
@@ -306,6 +326,8 @@ async def handle_call_tool(
         elif name == "read-iacr-paper":
             paper_id = arguments.get("paper_id", "")
             save_path = arguments.get("save_path", "./downloads")
+            start_page = arguments.get("start_page")
+            end_page = arguments.get("end_page")
 
             if not paper_id:
                 return [
@@ -314,7 +336,7 @@ async def handle_call_tool(
                     )
                 ]
 
-            result = iacr_searcher.read_paper(paper_id, save_path)
+            result = iacr_searcher.read_paper(paper_id, save_path, start_page=start_page, end_page=end_page)
 
             if result.startswith("Error"):
                 return [types.TextContent(type="text", text=result)]
@@ -577,6 +599,8 @@ async def handle_call_tool(
 
         elif name == "read-pdf":
             pdf_source = arguments.get("pdf_source", "")
+            start_page = arguments.get("start_page")
+            end_page = arguments.get("end_page")
 
             if not pdf_source:
                 return [
@@ -586,7 +610,7 @@ async def handle_call_tool(
                 ]
 
             try:
-                result = read_pdf(pdf_source)
+                result = read_pdf(pdf_source, start_page=start_page, end_page=end_page)
                 return [types.TextContent(type="text", text=result)]
 
             except Exception as e:

{all_in_mcp-0.2.7.dist-info → all_in_mcp-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: all-in-mcp
-Version: 0.2.7
+Version: 0.2.8
 Summary: An MCP (Model Context Protocol) server providing daily-use utility functions and academic paper search capabilities
 Project-URL: Homepage, https://github.com/jiahaoxiang2000/all-in-mcp
 Project-URL: Repository, https://github.com/jiahaoxiang2000/all-in-mcp
@@ -73,8 +73,8 @@ All tools are implemented as async MCP endpoints with proper validation and erro
 
 ## Quick Start
 
-- [**Video for Env Setup**](https://www.bilibili.com/video/BV1cZKozaEjg)
-- [_Overview PDF_](https://github.com/jiahaoxiang2000/tutor/blob/main/Apaper/config.pdf)
+- [**Video for Env Setup**](https://www.bilibili.com/video/BV1cZKozaEjg) [**Video for Claude code**](https://www.bilibili.com/video/BV1s9KmzVEcE/)
+- [_Overview PDF_](https://github.com/jiahaoxiang2000/tutor/blob/main/Apaper/config.pdf) [_PDF for Claude code_](https://github.com/jiahaoxiang2000/tutor/blob/main/Apaper/config-claude.pdf)
 
 ### Prerequisites
 
@@ -89,11 +89,11 @@ Install from PyPI (Recommended by `UV`)
 uv pip install all-in-mcp
 ```
 
-### Integration with MCP Clients Vscode
+### Integration with MCP Clients
 
 Add this server to your MCP client configuration. The server runs using stdio transport.
 
-Example configuration for Vscode:
+#### VSCode Configuration
 
 ```json .vscode/mcp.json
 {
@@ -107,6 +107,19 @@ Example configuration for Vscode:
 }
 ```
 
+#### Claude Code Configuration
+
+```json .mcp.json
+{
+  "mcpServers": {
+    "all-in-mcp": {
+      "command": "uv",
+      "args": ["run", "all-in-mcp"]
+    }
+  }
+}
+```
+
 <details>
 <summary>Development</summary>
 

all_in_mcp-0.2.8.dist-info/RECORD

@@ -0,0 +1,14 @@
+all_in_mcp/__init__.py,sha256=REDwcbifpuUnsFAhNowIKCZ-8g6irIzUFTI_f8Aunxk,215
+all_in_mcp/paper.py,sha256=juZI0M_gAH_WNIaLCt_RNxvbGRuEDbuSN90SI3mb2fs,9356
+all_in_mcp/server.py,sha256=zmBTgVJVHJdQF8r6Y4OP2bcvLBtEioBHRDAYPUJ3JMc,26849
+all_in_mcp/academic_platforms/__init__.py,sha256=IpI29DMS4_mSmTEa8VkQEiJCl7OyFbswSx7mWSp08P4,285
+all_in_mcp/academic_platforms/base.py,sha256=dRIybIIAJDZzfZFmqSuJ-qdMUz2hKzIqltwRVclzBzI,733
+all_in_mcp/academic_platforms/crossref.py,sha256=3qif-sPXOkaicC0sb3wnGCRf4IqXLQ2wMx9LO5CG0Ao,9058
+all_in_mcp/academic_platforms/cryptobib.py,sha256=zozR040FHEY1tfwakqPiV_ouwdzruhHdgKcSM1gHdUc,17711
+all_in_mcp/academic_platforms/google_scholar.py,sha256=SsD0unk-B32MGrwmOlgJJl6d06vRJKm1V2FziTwtcCQ,8969
+all_in_mcp/academic_platforms/iacr.py,sha256=4d_MTcUEBqN0HsAUBCD6PzHJOp8XCGDhQkQ1ojCcQQY,14870
+all_in_mcp-0.2.8.dist-info/METADATA,sha256=wnBPSHo_8-QA-tHikG3wNQa13I9QSOp6vTfpFK3kYfQ,4568
+all_in_mcp-0.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+all_in_mcp-0.2.8.dist-info/entry_points.txt,sha256=FbQOtUQzOIfkMNp4qQV1NTU9K4J7C0XGH9wKKhfK1VM,47
+all_in_mcp-0.2.8.dist-info/licenses/LICENSE,sha256=idExTHItK7AC5FVo4H9HKnr6h51Z8BKCEztZPyP8nK8,1062
+all_in_mcp-0.2.8.dist-info/RECORD,,

all_in_mcp-0.2.7.dist-info/RECORD

@@ -1,14 +0,0 @@
-all_in_mcp/__init__.py,sha256=REDwcbifpuUnsFAhNowIKCZ-8g6irIzUFTI_f8Aunxk,215
-all_in_mcp/paper.py,sha256=vSJyC_ehfZX5-ASYG048z8gaD1LKafFdJvR13iQcJRw,7104
-all_in_mcp/server.py,sha256=NQ-AEARpZXafLs_yTUKmSC2I32tLQUVL_KLYdefWdm0,25585
-all_in_mcp/academic_platforms/__init__.py,sha256=IpI29DMS4_mSmTEa8VkQEiJCl7OyFbswSx7mWSp08P4,285
-all_in_mcp/academic_platforms/base.py,sha256=VYMp8_tnp7YzXKAXLfr7uUxgvJBNKRyC_NT1uVhBOwY,673
-all_in_mcp/academic_platforms/crossref.py,sha256=D-wvSwnOocP16m9fA3xJ6VGEpmRPtMmGoFm5MlyPdXE,8707
-all_in_mcp/academic_platforms/cryptobib.py,sha256=F9N23eojfyAIjnFDPrJAYOpZ_Vi9iHOqNHGtKC6O16c,17360
-all_in_mcp/academic_platforms/google_scholar.py,sha256=_KLFfIOZeFCGxFOt-nwzm1fgZKMlXOf3HvIjXAYE5cI,8737
-all_in_mcp/academic_platforms/iacr.py,sha256=CMv3kVvF7NiZJdtvXc8xoGOP-gMNnAkhIETTbYTP75o,15849
-all_in_mcp-0.2.7.dist-info/METADATA,sha256=r2v5qsm0TwWdfUZwFGuaFxVSeyh9PYlarJym8Kt6pFM,4242
-all_in_mcp-0.2.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-all_in_mcp-0.2.7.dist-info/entry_points.txt,sha256=FbQOtUQzOIfkMNp4qQV1NTU9K4J7C0XGH9wKKhfK1VM,47
-all_in_mcp-0.2.7.dist-info/licenses/LICENSE,sha256=idExTHItK7AC5FVo4H9HKnr6h51Z8BKCEztZPyP8nK8,1062
-all_in_mcp-0.2.7.dist-info/RECORD,,