Skip to content

Commit a622e26

Browse files
committed
some migration changes related to a recent experiment
1 parent 7b2677c commit a622e26

3 files changed

Lines changed: 77 additions & 0 deletions

File tree

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
-- These columns were added via `prisma db push` and already exist in the database.
2+
-- This migration file exists only to bring the migration history in sync.
3+
4+
ALTER TABLE "Chat" ADD COLUMN IF NOT EXISTS "title" TEXT;
5+
ALTER TABLE "Chat" ADD COLUMN IF NOT EXISTS "modelId" TEXT;
Lines changed: 54 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
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);
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
-- DropIndex
2+
DROP INDEX IF EXISTS "code_chunks_embedding_idx";
3+
DROP INDEX IF EXISTS "code_chunks_repoKey_idx";
4+
DROP INDEX IF EXISTS "repo_ingestions_repoKey_key";
5+
6+
-- DropTable
7+
DROP TABLE IF EXISTS "code_chunks";
8+
DROP TABLE IF EXISTS "repo_ingestions";
9+
10+
-- AlterTable
11+
ALTER TABLE "Chat" DROP COLUMN IF EXISTS "githubRepo";
12+
ALTER TABLE "Chat" DROP COLUMN IF EXISTS "githubBranch";
13+
14+
-- DropEnum
15+
DROP TYPE IF EXISTS "IngestionStatus";
16+
17+
-- DropExtension
18+
DROP EXTENSION IF EXISTS "vector";

0 commit comments

Comments
 (0)