16aa6672fc
We'll probably need a concept of organizations at some point in the future, but in practice the way we're using these in the codebase right now is as a project, so this renames it to that to avoid confusion.
81 lines
2.0 KiB
TypeScript
81 lines
2.0 KiB
TypeScript
import {
|
|
type Experiment,
|
|
type PromptVariant,
|
|
type TestScenario,
|
|
type TemplateVariable,
|
|
type ScenarioVariantCell,
|
|
type ModelResponse,
|
|
type Evaluation,
|
|
type OutputEvaluation,
|
|
type Dataset,
|
|
type DatasetEntry,
|
|
type Project,
|
|
type ProjectUser,
|
|
type WorldChampEntrant,
|
|
type LoggedCall,
|
|
type LoggedCallModelResponse,
|
|
type LoggedCallTag,
|
|
type ApiKey,
|
|
type Account,
|
|
type Session,
|
|
type User,
|
|
type VerificationToken,
|
|
PrismaClient,
|
|
} from "@prisma/client";
|
|
import { Kysely, PostgresDialect } from "kysely";
|
|
// TODO: Revert to normal import when our tsconfig.json is fixed
|
|
// import { Pool } from "pg";
|
|
import PGModule from "pg";
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
const UntypedPool = PGModule.Pool as any;
|
|
const Pool = (UntypedPool.default ? UntypedPool.default : UntypedPool) as typeof PGModule.Pool;
|
|
|
|
import { env } from "~/env.mjs";
|
|
|
|
interface DB {
|
|
Experiment: Experiment;
|
|
PromptVariant: PromptVariant;
|
|
TestScenario: TestScenario;
|
|
TemplateVariable: TemplateVariable;
|
|
ScenarioVariantCell: ScenarioVariantCell;
|
|
ModelResponse: ModelResponse;
|
|
Evaluation: Evaluation;
|
|
OutputEvaluation: OutputEvaluation;
|
|
Dataset: Dataset;
|
|
DatasetEntry: DatasetEntry;
|
|
Project: Project;
|
|
ProjectUser: ProjectUser;
|
|
WorldChampEntrant: WorldChampEntrant;
|
|
LoggedCall: LoggedCall;
|
|
LoggedCallModelResponse: LoggedCallModelResponse;
|
|
LoggedCallTag: LoggedCallTag;
|
|
ApiKey: ApiKey;
|
|
Account: Account;
|
|
Session: Session;
|
|
User: User;
|
|
VerificationToken: VerificationToken;
|
|
}
|
|
|
|
const globalForPrisma = globalThis as unknown as {
|
|
prisma: PrismaClient | undefined;
|
|
};
|
|
|
|
export const prisma =
|
|
globalForPrisma.prisma ??
|
|
new PrismaClient({
|
|
log:
|
|
env.NODE_ENV === "development" && !env.RESTRICT_PRISMA_LOGS
|
|
? ["query", "error", "warn"]
|
|
: ["error"],
|
|
});
|
|
|
|
export const kysely = new Kysely<DB>({
|
|
dialect: new PostgresDialect({
|
|
pool: new Pool({
|
|
connectionString: env.DATABASE_URL,
|
|
}),
|
|
}),
|
|
});
|
|
|
|
if (env.NODE_ENV !== "production") globalForPrisma.prisma = prisma;
|