Update quickstart.py: Add counting items

This commit is contained in:
UncleCode
2024-05-16 22:49:12 +08:00
committed by GitHub
parent 5b4a586b2d
commit 6fcaf26b4f

View File

@@ -22,6 +22,10 @@ def print_result(result):
for key, value in result.model_dump().items():
if isinstance(value, str) and value:
console.print(f"\t{key}: [green]{value[:20]}...[/green]")
if result.extracted_content:
items = json.loads(result.extracted_content)
print(f"\t[bold]{len(items)} blocks is extracted![/bold]")
def cprint(message, press_any_key=False):
console.print(message)