138 lines
5.8 KiB
C#
138 lines
5.8 KiB
C#
using System;
|
|
using Microsoft.EntityFrameworkCore.Migrations;
|
|
|
|
#nullable disable
|
|
|
|
namespace Api.Migrations
|
|
{
|
|
/// <inheritdoc />
|
|
public partial class InitialRagSchema : Migration
|
|
{
|
|
/// <inheritdoc />
|
|
protected override void Up(MigrationBuilder migrationBuilder)
|
|
{
|
|
migrationBuilder.EnsureSchema(
|
|
name: "rag");
|
|
|
|
migrationBuilder.CreateTable(
|
|
name: "ChatCompletionCache",
|
|
schema: "rag",
|
|
columns: table => new
|
|
{
|
|
CacheKey = table.Column<string>(type: "nvarchar(64)", maxLength: 64, nullable: false),
|
|
Model = table.Column<string>(type: "nvarchar(120)", maxLength: 120, nullable: false),
|
|
Temperature = table.Column<decimal>(type: "decimal(4,2)", nullable: false),
|
|
ResponseText = table.Column<string>(type: "nvarchar(max)", nullable: false),
|
|
CreatedAt = table.Column<DateTime>(type: "datetime2", nullable: false, defaultValueSql: "SYSUTCDATETIME()")
|
|
},
|
|
constraints: table =>
|
|
{
|
|
table.PrimaryKey("PK_ChatCompletionCache", x => x.CacheKey);
|
|
});
|
|
|
|
migrationBuilder.CreateTable(
|
|
name: "Documents",
|
|
schema: "rag",
|
|
columns: table => new
|
|
{
|
|
Id = table.Column<string>(type: "nvarchar(64)", maxLength: 64, nullable: false),
|
|
DocumentType = table.Column<string>(type: "nvarchar(80)", maxLength: 80, nullable: false),
|
|
Title = table.Column<string>(type: "nvarchar(300)", maxLength: 300, nullable: false),
|
|
SourceUrl = table.Column<string>(type: "nvarchar(1200)", maxLength: 1200, nullable: true),
|
|
RawText = table.Column<string>(type: "nvarchar(max)", nullable: false),
|
|
TextHash = table.Column<string>(type: "nvarchar(64)", maxLength: 64, nullable: false),
|
|
TypeConfidence = table.Column<double>(type: "float", nullable: false),
|
|
MetadataJson = table.Column<string>(type: "nvarchar(max)", nullable: false, defaultValue: "{}"),
|
|
CreatedAt = table.Column<DateTime>(type: "datetime2", nullable: false, defaultValueSql: "SYSUTCDATETIME()")
|
|
},
|
|
constraints: table =>
|
|
{
|
|
table.PrimaryKey("PK_Documents", x => x.Id);
|
|
});
|
|
|
|
migrationBuilder.CreateTable(
|
|
name: "EmbeddingCache",
|
|
schema: "rag",
|
|
columns: table => new
|
|
{
|
|
CacheKey = table.Column<string>(type: "nvarchar(64)", maxLength: 64, nullable: false),
|
|
Model = table.Column<string>(type: "nvarchar(120)", maxLength: 120, nullable: false),
|
|
TextHash = table.Column<string>(type: "nvarchar(64)", maxLength: 64, nullable: false),
|
|
Vector = table.Column<byte[]>(type: "varbinary(max)", nullable: false),
|
|
CreatedAt = table.Column<DateTime>(type: "datetime2", nullable: false, defaultValueSql: "SYSUTCDATETIME()")
|
|
},
|
|
constraints: table =>
|
|
{
|
|
table.PrimaryKey("PK_EmbeddingCache", x => x.CacheKey);
|
|
});
|
|
|
|
migrationBuilder.CreateTable(
|
|
name: "Chunks",
|
|
schema: "rag",
|
|
columns: table => new
|
|
{
|
|
Id = table.Column<string>(type: "nvarchar(64)", maxLength: 64, nullable: false),
|
|
DocumentId = table.Column<string>(type: "nvarchar(64)", maxLength: 64, nullable: false),
|
|
ChunkIndex = table.Column<int>(type: "int", nullable: false),
|
|
Text = table.Column<string>(type: "nvarchar(max)", nullable: false),
|
|
Embedding = table.Column<byte[]>(type: "varbinary(max)", nullable: false)
|
|
},
|
|
constraints: table =>
|
|
{
|
|
table.PrimaryKey("PK_Chunks", x => x.Id);
|
|
table.ForeignKey(
|
|
name: "FK_Chunks_Documents_DocumentId",
|
|
column: x => x.DocumentId,
|
|
principalSchema: "rag",
|
|
principalTable: "Documents",
|
|
principalColumn: "Id",
|
|
onDelete: ReferentialAction.Cascade);
|
|
});
|
|
|
|
migrationBuilder.CreateIndex(
|
|
name: "IX_Chunks_DocumentId",
|
|
schema: "rag",
|
|
table: "Chunks",
|
|
column: "DocumentId");
|
|
|
|
migrationBuilder.CreateIndex(
|
|
name: "IX_Documents_DocumentType",
|
|
schema: "rag",
|
|
table: "Documents",
|
|
column: "DocumentType");
|
|
|
|
migrationBuilder.CreateIndex(
|
|
name: "IX_Documents_TextHash",
|
|
schema: "rag",
|
|
table: "Documents",
|
|
column: "TextHash");
|
|
|
|
migrationBuilder.CreateIndex(
|
|
name: "IX_EmbeddingCache_TextHash",
|
|
schema: "rag",
|
|
table: "EmbeddingCache",
|
|
column: "TextHash");
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
protected override void Down(MigrationBuilder migrationBuilder)
|
|
{
|
|
migrationBuilder.DropTable(
|
|
name: "ChatCompletionCache",
|
|
schema: "rag");
|
|
|
|
migrationBuilder.DropTable(
|
|
name: "Chunks",
|
|
schema: "rag");
|
|
|
|
migrationBuilder.DropTable(
|
|
name: "EmbeddingCache",
|
|
schema: "rag");
|
|
|
|
migrationBuilder.DropTable(
|
|
name: "Documents",
|
|
schema: "rag");
|
|
}
|
|
}
|
|
}
|