arxiv-to-prompt 0.3.0__tar.gz → 0.4.0__tar.gz

{arxiv_to_prompt-0.3.0/src/arxiv_to_prompt.egg-info → arxiv_to_prompt-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arxiv-to-prompt
-Version: 0.3.0
+Version: 0.4.0
 Summary: transform arXiv papers into a single latex prompt for LLMs
 Author: Takashi Ishida
 License: MIT
@@ -61,7 +61,7 @@ arxiv-to-prompt 2303.08774 | pbcopy
 arxiv-to-prompt 1706.03762 | llm -s "explain this paper"
 ```
 
-The arXiv ID can be found in the paper's URL. For example, for `https://arxiv.org/abs/2303.08774`, the ID is `2303.08774`. It will automatically download the latest version of the paper, so you don't need to specify the version.
+You can use either the arXiv ID (e.g., `2303.08774`) or the full URL (e.g., `https://arxiv.org/abs/2303.08774`). It will automatically download the latest version of the paper, so you don't need to specify the version.
 
 ### Python API
 

{arxiv_to_prompt-0.3.0 → arxiv_to_prompt-0.4.0}/README.md

@@ -42,7 +42,7 @@ arxiv-to-prompt 2303.08774 | pbcopy
 arxiv-to-prompt 1706.03762 | llm -s "explain this paper"
 ```
 
-The arXiv ID can be found in the paper's URL. For example, for `https://arxiv.org/abs/2303.08774`, the ID is `2303.08774`. It will automatically download the latest version of the paper, so you don't need to specify the version.
+You can use either the arXiv ID (e.g., `2303.08774`) or the full URL (e.g., `https://arxiv.org/abs/2303.08774`). It will automatically download the latest version of the paper, so you don't need to specify the version.
 
 ### Python API
 

{arxiv_to_prompt-0.3.0 → arxiv_to_prompt-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "arxiv-to-prompt"
-version = "0.3.0"
+version = "0.4.0"
 description = "transform arXiv papers into a single latex prompt for LLMs"
 readme = "README.md"
 authors = [{ name = "Takashi Ishida" }]

{arxiv_to_prompt-0.3.0 → arxiv_to_prompt-0.4.0}/src/arxiv_to_prompt/cli.py

@@ -1,6 +1,16 @@
 import argparse
+import re
 from .core import process_latex_source, get_default_cache_dir
 
+
+def extract_arxiv_id(input_str: str) -> str:
+    """Extract arxiv ID from URL or return input as-is if already an ID."""
+    if "arxiv.org" in input_str:
+        match = re.search(r'arxiv\.org/(?:abs|pdf)/(\d{4}\.\d{4,5})(?:v\d+)?(?:\.pdf)?', input_str)
+        if match:
+            return match.group(1)
+    return input_str
+
 def main():
     default_cache = str(get_default_cache_dir())
     
@@ -11,7 +21,7 @@ def main():
         "arxiv_id",
         nargs="?",
         default=None,
-        help="The arXiv ID of the paper (do not include the version, e.g. v1, v2). Not needed if --local-folder is provided."
+        help="The arXiv ID (e.g. 2303.08774) or URL (e.g. https://arxiv.org/abs/2303.08774). Not needed if --local-folder is provided."
     )
     parser.add_argument(
         "--no-comments",
@@ -44,9 +54,11 @@ def main():
     
     if args.arxiv_id and args.local_folder:
         parser.error("Cannot specify both arXiv ID and --local-folder")
-    
+
+    arxiv_id = extract_arxiv_id(args.arxiv_id) if args.arxiv_id else None
+
     content = process_latex_source(
-        arxiv_id=args.arxiv_id,
+        arxiv_id=arxiv_id,
         keep_comments=not args.no_comments,
         cache_dir=args.cache_dir,
         remove_appendix_section=args.no_appendix,

{arxiv_to_prompt-0.3.0 → arxiv_to_prompt-0.4.0/src/arxiv_to_prompt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arxiv-to-prompt
-Version: 0.3.0
+Version: 0.4.0
 Summary: transform arXiv papers into a single latex prompt for LLMs
 Author: Takashi Ishida
 License: MIT
@@ -61,7 +61,7 @@ arxiv-to-prompt 2303.08774 | pbcopy
 arxiv-to-prompt 1706.03762 | llm -s "explain this paper"
 ```
 
-The arXiv ID can be found in the paper's URL. For example, for `https://arxiv.org/abs/2303.08774`, the ID is `2303.08774`. It will automatically download the latest version of the paper, so you don't need to specify the version.
+You can use either the arXiv ID (e.g., `2303.08774`) or the full URL (e.g., `https://arxiv.org/abs/2303.08774`). It will automatically download the latest version of the paper, so you don't need to specify the version.
 
 ### Python API
 

{arxiv_to_prompt-0.3.0 → arxiv_to_prompt-0.4.0}/tests/test_core.py

@@ -11,6 +11,7 @@ from arxiv_to_prompt.core import (
     flatten_tex,
     remove_appendix,
 )
+from arxiv_to_prompt.cli import extract_arxiv_id
 
 # Test fixtures
 @pytest.fixture
@@ -271,3 +272,27 @@ def test_input_file_extensions(temp_cache_dir):
     assert "Style content" in result
     assert "Class content" in result
     assert "Already tex content" in result
+
+
+def test_extract_arxiv_id():
+    """Test extracting arxiv ID from URLs and plain IDs."""
+    # Plain IDs should be returned as-is
+    assert extract_arxiv_id("2505.18102") == "2505.18102"
+    assert extract_arxiv_id("2401.12345") == "2401.12345"
+
+    # Extract from abs URLs
+    assert extract_arxiv_id("https://arxiv.org/abs/2505.18102") == "2505.18102"
+    assert extract_arxiv_id("http://arxiv.org/abs/2505.18102") == "2505.18102"
+
+    # Extract from pdf URLs
+    assert extract_arxiv_id("https://arxiv.org/pdf/2505.18102") == "2505.18102"
+    assert extract_arxiv_id("https://arxiv.org/pdf/2505.18102.pdf") == "2505.18102"
+
+    # Strip version suffixes
+    assert extract_arxiv_id("https://arxiv.org/abs/2505.18102v1") == "2505.18102"
+    assert extract_arxiv_id("https://arxiv.org/abs/2505.18102v2") == "2505.18102"
+    assert extract_arxiv_id("https://arxiv.org/pdf/2505.18102v3.pdf") == "2505.18102"
+
+    # Non-arxiv input returned as-is
+    assert extract_arxiv_id("invalid") == "invalid"
+    assert extract_arxiv_id("https://example.com/2505.18102") == "https://example.com/2505.18102"