diff --git a/blueprints/rag/logic.py b/blueprints/rag/logic.py index 7487411..f281519 100644 --- a/blueprints/rag/logic.py +++ b/blueprints/rag/logic.py @@ -59,13 +59,17 @@ text_splitter = RecursiveCharacterTextSplitter( def _get_collection_id(): """Get the UUID of our collection from the langchain_pg_collection table.""" engine = _get_engine() - with engine.connect() as conn: - result = conn.execute( - text("SELECT uuid FROM langchain_pg_collection WHERE name = :name"), - {"name": "simba_docs"}, - ) - row = result.fetchone() - return row[0] if row else None + try: + with engine.connect() as conn: + result = conn.execute( + text("SELECT uuid FROM langchain_pg_collection WHERE name = :name"), + {"name": "simba_docs"}, + ) + row = result.fetchone() + return row[0] if row else None + except Exception: + # Table doesn't exist yet (first run before any indexing) + return None def date_to_epoch(date_str: str) -> float: