Spaces:
Sleeping
Sleeping
File size: 2,514 Bytes
583741e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
import argparse
import os
import sys
import asyncio
from typing import Optional
from dotenv import load_dotenv
load_dotenv()
from rich.console import Console
from rich.panel import Panel
import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from agents.evaluator import run_evaluation
console = Console()
def build_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser(
description="AI Automation Evaluator (LangGraph) — evaluate a paper PDF or arXiv URL",
epilog="Example: python cli.py https://arxiv.org/pdf/2507.14683 --arxiv-id 2507.14683 -o /abs/path/save_dir/eval_2507_14683",
)
parser.add_argument("pdf", help="Local PDF absolute path or URL (e.g., https://arxiv.org/pdf/xxxx)")
parser.add_argument(
"--arxiv-id",
dest="arxiv_id",
help="arXiv ID for the paper (e.g., 2507.14683)",
)
parser.add_argument(
"-o",
"--output-prefix",
dest="output_prefix",
help="Output file prefix (if provided, will save as <prefix>_YYYYMMDD_HHMMSS.md)",
)
parser.add_argument(
"--api-key",
dest="api_key",
default=os.getenv("ANTHROPIC_API_KEY"),
help="Anthropic API key (overrides ANTHROPIC_API_KEY env)",
)
return parser
async def main(argv: Optional[list[str]] = None):
parser = build_parser()
args = parser.parse_args(argv)
pdf_path: str = args.pdf
arxiv_id: Optional[str] = args.arxiv_id
output_prefix: Optional[str] = args.output_prefix
api_key: Optional[str] = args.api_key or os.getenv("ANTHROPIC_API_KEY")
if not api_key:
console.print("[yellow]Warning:[/yellow] ANTHROPIC_API_KEY not set and --api-key not provided.", highlight=False)
console.print(Panel.fit(f"Evaluating: {pdf_path}"))
if arxiv_id:
console.print(f"arXiv ID: {arxiv_id}")
try:
result = await run_evaluation(pdf_path=pdf_path, arxiv_id=arxiv_id, output_file=output_prefix, api_key=api_key)
console.print("\n[bold green]Done.[/bold green]\n")
if output_prefix:
console.print(f"Saved to prefix: {output_prefix}_<timestamp>.md")
elif arxiv_id:
console.print(f"Evaluation saved to database for paper: {arxiv_id}")
else:
console.print(result)
except Exception as e:
console.print(f"[bold red]Error:[/bold red] {e}")
sys.exit(2)
if __name__ == "__main__":
asyncio.run(main())
|