from deepsearcher.configuration import Configuration, init_config
from deepsearcher.offline_loading import load_from_local_files
from deepsearcher.online_query import query


config = Configuration()


config.load_config_from_yaml("deepsearcher/config.yaml")

init_config(config = config)

# Load your local data
load_from_local_files(
    paths_or_directory="docs",
    collection_name="default",
    collection_description="a general collection for all documents",
    force_rebuild=True, batch_size=16
)

# Query
result = query("Write a comprehensive report about Milvus.", max_iter=1) # Your question here