Knowledge Base CRUD

"""
Knowledge Base CRUD — full lifecycle.

Covers `new`, `fetch` (by id and by name), `list`, and `delete`.
Knowledge Bases support create and delete; rename/update is not exposed.
"""

from vectorshift.knowledge_base import (
    KnowledgeBase,
    IndexingConfig,
    SplitterMethod,
)

# --- Create ---
kb = KnowledgeBase.new(
    name="SDK KB CRUD Demo",
    embedding_model="text-embedding-3-large",
    indexing_config=IndexingConfig(
        chunk_size=500,
        chunk_overlap=50,
        analyze_documents=False,
        splitter=SplitterMethod.MARKDOWN,
    ),
)
print(f"1. Created KB: {kb.name} (id={kb.id})")

# --- Fetch by ID ---
fetched = KnowledgeBase.fetch(id=kb.id)
print(f"2. Fetched by ID: {fetched.name}")

# --- Fetch by name ---
fetched_by_name = KnowledgeBase.fetch(name="SDK KB CRUD Demo")
print(f"3. Fetched by name: {fetched_by_name.name}")

# --- List ---
kbs = KnowledgeBase.list(limit=20, offset=0)
found = any(k.id == kb.id for k in kbs)
print(f"4. Listed KBs ({len(kbs)} total), found ours: {found}")

# --- Delete ---
kb.delete()
print("5. Deleted KB.\n\nDone.")

1. Created KB: SDK KB CRUD Demo (id=...)
2. Fetched by ID: SDK KB CRUD Demo
3. Fetched by name: SDK KB CRUD Demo
4. Listed KBs (... total), found ours: True
5. Deleted KB.

Done.