Фиксация изменений
This commit is contained in:
@@ -106,6 +106,7 @@ class RagSchemaRepository:
|
||||
)
|
||||
self._ensure_columns(conn)
|
||||
self._ensure_indexes(conn)
|
||||
self._drop_unused_rag_chunk_columns(conn)
|
||||
conn.commit()
|
||||
|
||||
def _ensure_columns(self, conn) -> None:
|
||||
@@ -118,14 +119,12 @@ class RagSchemaRepository:
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS system_component TEXT NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS last_modified TIMESTAMPTZ NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS staleness_score DOUBLE PRECISION NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS rag_doc_id VARCHAR(128) NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS layer VARCHAR(64) NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS lang VARCHAR(32) NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS repo_id VARCHAR(512) NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS commit_sha VARCHAR(128) NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS title TEXT NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS metadata_json TEXT NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS links_json TEXT NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS span_start INTEGER NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS span_end INTEGER NULL",
|
||||
"ALTER TABLE rag_chunks ADD COLUMN IF NOT EXISTS symbol_id TEXT NULL",
|
||||
@@ -162,6 +161,13 @@ class RagSchemaRepository:
|
||||
):
|
||||
conn.execute(text(statement))
|
||||
|
||||
def _drop_unused_rag_chunk_columns(self, conn) -> None:
|
||||
for statement in (
|
||||
"ALTER TABLE rag_chunks DROP COLUMN IF EXISTS rag_doc_id",
|
||||
"ALTER TABLE rag_chunks DROP COLUMN IF EXISTS links_json",
|
||||
):
|
||||
conn.execute(text(statement))
|
||||
|
||||
def _ensure_indexes(self, conn) -> None:
|
||||
for statement in (
|
||||
"CREATE INDEX IF NOT EXISTS idx_rag_chunks_session ON rag_chunks (rag_session_id)",
|
||||
|
||||
Reference in New Issue
Block a user