Add folder

"""
Example 03: Upload a local folder, preserving directory structure.

`kb.add_folder(...)` walks the local directory, builds a nested folder
tree, and submits a single ingestion request. The KB layout will mirror
the local layout.

Requires the v1 ingestion task endpoints + folder-tree extension to
KbIndexRequest (M2 [A2]).
"""

import tempfile
from pathlib import Path

from vectorshift.knowledge_base import KnowledgeBase

# Build a small nested folder
with tempfile.TemporaryDirectory() as tmp:
    root = Path(tmp) / "handbook"
    (root / "policies").mkdir(parents=True)
    (root / "policies" / "pto.md").write_text("# PTO\n\n- 20 days/yr\n")
    (root / "policies" / "remote.md").write_text("# Remote\n\nWFH OK.\n")
    (root / "engineering").mkdir()
    (root / "engineering" / "oncall.md").write_text("# Oncall\n\nWeekly rotation.\n")
    (root / "README.md").write_text("# Handbook\n\nWelcome.\n")

    kb = KnowledgeBase.new(name="SDK folder demo")
    print(f"1. Created KB id={kb.id}")

    final = kb.add_folder_and_wait(root, timeout=180)
    print(
        f"2. Folder ingest completed: status={final.status} "
        f"items={len(final.item_ids)}"
    )

    # List items to confirm folder hierarchy is preserved
    items = kb.list_items(limit=50)
    print(f"3. Items in KB: {len(items)}")

    kb.delete()
    print("4. Deleted KB.")

1. Created KB id=...
2. Folder ingest completed: status=... items=4
3. Items in KB: 4
4. Deleted KB.