diff options
Diffstat (limited to 'python/ingest_tool.py')
-rwxr-xr-x | python/ingest_tool.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/python/ingest_tool.py b/python/ingest_tool.py index 60a59d2..1843e0b 100755 --- a/python/ingest_tool.py +++ b/python/ingest_tool.py @@ -27,9 +27,13 @@ def run_single_ingest(args): ingest_file_result_stdout=True, ) else: + grobid_client = GrobidClient( + host_url=args.grobid_host, + ) ingester = IngestFileWorker( try_spn2=not args.no_spn2, html_quick_mode=args.html_quick_mode, + grobid_client=grobid_client, ) result = ingester.process(request) print(json.dumps(result, sort_keys=True)) @@ -140,6 +144,9 @@ def main(): help="don't fetch individual sub-resources, just use CDX", ) sub_single.add_argument("url", help="URL of paper to fetch") + sub_single.add_argument( + "--grobid-host", default="https://grobid.qa.fatcat.wiki", help="GROBID API host/port" + ) sub_requests = subparsers.add_parser( "requests", help="takes a series of ingest requests (JSON, per line) and runs each" |