LanceDB Cloud connection test.

"""
LanceDB Cloud connection test.

Requires environment variables:
- LANCE_DB_URI: LanceDB Cloud database URI (e.g. db://your-database-id)
- LANCE_DB_API_KEY or LANCEDB_API_KEY: LanceDB Cloud API key

Run from repo root with env loaded (e.g. direnv):
  .venvs/demo/bin/python cookbook/07_knowledge/vector_db/lance_db_cloud/lance_db_cloud.py
"""

import asyncio
import os

from agno.knowledge.knowledge import Knowledge
from agno.vectordb.lancedb import LanceDb

# ---------------------------------------------------------------------------
# Setup
# ---------------------------------------------------------------------------
TABLE_NAME = "agno_cloud_test"
URI = os.getenv("LANCE_DB_URI")
API_KEY = os.getenv("LANCE_DB_API_KEY") or os.getenv("LANCEDB_API_KEY")


# ---------------------------------------------------------------------------
# Create Knowledge Base
# ---------------------------------------------------------------------------
# The cloud vector DB and knowledge instance are created inside `main()`
# after validating required environment variables.


# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
def main():
    if not URI:
        print("Set LANCE_DB_URI (e.g. db://your-database-id)")
        return

    # ---------------------------------------------------------------------------
    # Create Knowledge Base
    # ---------------------------------------------------------------------------
    vector_db = LanceDb(
        uri=URI,
        table_name=TABLE_NAME,
        api_key=API_KEY,
    )

    knowledge = Knowledge(
        name="LanceDB Cloud Test",
        description="Agno Knowledge with LanceDB Cloud",
        vector_db=vector_db,
    )

    async def run():
        print("Inserting test content...")
        await knowledge.ainsert(
            name="cloud_test_doc",
            text_content="LanceDB Cloud is a hosted vector database. "
            "Agno supports it via the LanceDb vector store with uri and api_key. "
            "Use db:// URI and set LANCEDB_API_KEY for cloud connections.",
            metadata={"source": "lance_db_cloud_cookbook"},
        )

        print("Searching for 'vector database'...")
        results = knowledge.search("vector database", max_results=3)
        print(f"Found {len(results)} document(s)")
        for i, doc in enumerate(results):
            print(f"  [{i + 1}] {doc.name}: {doc.content[:80]}...")

        print("Deleting test document...")
        vector_db.delete_by_name("cloud_test_doc")
        print("Done.")

    asyncio.run(run())


if __name__ == "__main__":
    main()

# Clone and setup repo
git clone https://github.com/agno-agi/agno.git
cd agno/cookbook/07_knowledge/vector_db/lance_db

# Create and activate virtual environment
./scripts/demo_setup.sh
source .venvs/demo/bin/activate

# Export relevant API keys
export LANCEDB_API_KEY="***"
export LANCE_DB_API_KEY="***"
export LANCE_DB_URI="***"

python lance_db_cloud.py