Batch Processing Quickstart

from valyu import Valyu

valyu = Valyu()

# 1. Create a batch
batch = valyu.batch.create(
    name="Market Research Q4",
    mode="standard",
    output_formats=["markdown"]
)

if batch.success:
    # 2. Add tasks
    valyu.batch.add_tasks(batch.batch_id, [
        {"query": "Analyze AI trends in healthcare"},
        {"query": "Review renewable energy market"},
        {"query": "Research fintech innovations"}
    ])
    
    # 3. Wait for completion
    result = valyu.batch.wait_for_completion(
        batch.batch_id,
        on_progress=lambda s: print(f"Progress: {s.batch.counts.completed}/{s.batch.counts.total}")
    )
    
    print(f"Completed! Total cost: ${result.batch.cost}")

# Get all completed results with full output
results = valyu.batch.list_tasks(batch_id, status="completed", include_output=True)

for task in results.tasks:
    print(f"Query: {task.query}")
    print(f"Output: {task.output[:200]}...")
    print(f"Sources: {len(task.sources)} cited")