|
| 1 | +-- CreateExtension |
| 2 | +CREATE EXTENSION IF NOT EXISTS "vector"; |
| 3 | + |
| 4 | +-- CreateEnum |
| 5 | +CREATE TYPE "IngestionStatus" AS ENUM ('PENDING', 'CLONING', 'CHUNKING', 'EMBEDDING', 'COMPLETED', 'FAILED'); |
| 6 | + |
| 7 | +-- AlterTable |
| 8 | +ALTER TABLE "Chat" ADD COLUMN "githubBranch" TEXT, |
| 9 | +ADD COLUMN "githubRepo" TEXT; |
| 10 | + |
| 11 | +-- CreateTable |
| 12 | +CREATE TABLE "code_chunks" ( |
| 13 | + "id" TEXT NOT NULL, |
| 14 | + "repoKey" TEXT NOT NULL, |
| 15 | + "filePath" TEXT NOT NULL, |
| 16 | + "chunkIndex" INTEGER NOT NULL, |
| 17 | + "content" TEXT NOT NULL, |
| 18 | + "language" TEXT, |
| 19 | + "tokenCount" INTEGER NOT NULL, |
| 20 | + "embedding" vector(384) NOT NULL, |
| 21 | + "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP, |
| 22 | + "updatedAt" TIMESTAMP(3) NOT NULL, |
| 23 | + |
| 24 | + CONSTRAINT "code_chunks_pkey" PRIMARY KEY ("id") |
| 25 | +); |
| 26 | + |
| 27 | +-- CreateTable |
| 28 | +CREATE TABLE "repo_ingestions" ( |
| 29 | + "id" TEXT NOT NULL, |
| 30 | + "repoKey" TEXT NOT NULL, |
| 31 | + "owner" TEXT NOT NULL, |
| 32 | + "repo" TEXT NOT NULL, |
| 33 | + "branch" TEXT NOT NULL, |
| 34 | + "status" "IngestionStatus" NOT NULL DEFAULT 'PENDING', |
| 35 | + "totalFiles" INTEGER NOT NULL DEFAULT 0, |
| 36 | + "processedFiles" INTEGER NOT NULL DEFAULT 0, |
| 37 | + "totalChunks" INTEGER NOT NULL DEFAULT 0, |
| 38 | + "error" TEXT, |
| 39 | + "startedAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP, |
| 40 | + "completedAt" TIMESTAMP(3), |
| 41 | + "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP, |
| 42 | + "updatedAt" TIMESTAMP(3) NOT NULL, |
| 43 | + |
| 44 | + CONSTRAINT "repo_ingestions_pkey" PRIMARY KEY ("id") |
| 45 | +); |
| 46 | + |
| 47 | +-- CreateIndex |
| 48 | +CREATE INDEX "code_chunks_repoKey_idx" ON "code_chunks"("repoKey"); |
| 49 | + |
| 50 | +-- CreateIndex |
| 51 | +CREATE UNIQUE INDEX "repo_ingestions_repoKey_key" ON "repo_ingestions"("repoKey"); |
| 52 | + |
| 53 | +-- CreateIndex (HNSW for fast cosine similarity search) |
| 54 | +CREATE INDEX "code_chunks_embedding_idx" ON "code_chunks" USING hnsw ("embedding" vector_cosine_ops); |
0 commit comments