arxiv-to-prompt 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

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.dist-info → arxiv_to_prompt-0.4.0.dist-info}/METADATA

@@ -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.4.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+arxiv_to_prompt/__init__.py,sha256=riK7TcTaKDleP5g5rjf2jkmLtXZu7irNZDujyAVDnKM,1093
+arxiv_to_prompt/cli.py,sha256=np6mv2iCkLiVLawyix1vXP4bVFzRmZlkfjxb07ee89Q,2276
+arxiv_to_prompt/core.py,sha256=pgb8PGiOqgbPTW5rIJwLlmS9n3nnlYa5UVQ5YSvCIuo,12077
+arxiv_to_prompt-0.4.0.dist-info/licenses/LICENSE,sha256=np8L3--VyxwVJa_8D_mfK4RYrtnRMM_eeYN3rM4PMHo,1071
+arxiv_to_prompt-0.4.0.dist-info/METADATA,sha256=_G2nwz5NSnKPMpS6UowWqCe2WSpJBjHqOMshmtGOdak,4598
+arxiv_to_prompt-0.4.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+arxiv_to_prompt-0.4.0.dist-info/entry_points.txt,sha256=iYEEn8xZ_5OkhNIs5HCyHSQBpDRJkbD5h0tlAb16lL0,61
+arxiv_to_prompt-0.4.0.dist-info/top_level.txt,sha256=JClbu_lGGWu3RaTHZlNqTKB1-DUSbYXQNIYmJ9_F7fY,16
+arxiv_to_prompt-0.4.0.dist-info/RECORD,,

{arxiv_to_prompt-0.3.0.dist-info → arxiv_to_prompt-0.4.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

arxiv_to_prompt-0.3.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-arxiv_to_prompt/__init__.py,sha256=riK7TcTaKDleP5g5rjf2jkmLtXZu7irNZDujyAVDnKM,1093
-arxiv_to_prompt/cli.py,sha256=TUnHsGolF5zhiexW5RXBPhNL0HODmfppEkXzC8z65NE,1861
-arxiv_to_prompt/core.py,sha256=pgb8PGiOqgbPTW5rIJwLlmS9n3nnlYa5UVQ5YSvCIuo,12077
-arxiv_to_prompt-0.3.0.dist-info/licenses/LICENSE,sha256=np8L3--VyxwVJa_8D_mfK4RYrtnRMM_eeYN3rM4PMHo,1071
-arxiv_to_prompt-0.3.0.dist-info/METADATA,sha256=CNOBI0du7Yj9Hyr6JElWhohvLgd4jLUvlZEn9c8NexU,4608
-arxiv_to_prompt-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-arxiv_to_prompt-0.3.0.dist-info/entry_points.txt,sha256=iYEEn8xZ_5OkhNIs5HCyHSQBpDRJkbD5h0tlAb16lL0,61
-arxiv_to_prompt-0.3.0.dist-info/top_level.txt,sha256=JClbu_lGGWu3RaTHZlNqTKB1-DUSbYXQNIYmJ9_F7fY,16
-arxiv_to_prompt-0.3.0.dist-info/RECORD,,