30 lines
832 B
Python
30 lines
832 B
Python
#!/usr/bin/env python3
|
|
"""
|
|
Create RAG vector DB schema in Postgres (extension + stories, documents, chunks).
|
|
Requires RAG_DB_DSN and RAG_EMBEDDINGS_DIM (optional, default 1536).
|
|
Run from repo root with package installed: pip install -e . && python scripts/create_db.py
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# Allow importing rag_agent when run as scripts/create_db.py
|
|
repo_root = Path(__file__).resolve().parent.parent
|
|
sys.path.insert(0, str(repo_root / "src"))
|
|
|
|
from rag_agent.config import load_config
|
|
from rag_agent.index.postgres import connect, ensure_schema
|
|
|
|
|
|
def main() -> None:
|
|
config = load_config()
|
|
conn = connect(config.db_dsn)
|
|
ensure_schema(conn, config.embeddings_dim)
|
|
conn.close()
|
|
print("Schema created successfully.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|