lib/db/prisma.ts

import { PrismaClient } from '@prisma/client';

declare global {
  var prisma: PrismaClient | undefined;
  namespace PrismaJson {
    // you can use classes, interfaces, types, etc.

    type StructuredResult = {
      logs: {
        stderr: string[];
        stdout: string[];
      };
      results: Array<{
        png?: string;
        mp4?: string;
        text: string;
        is_main_result: boolean;
      }>;
      error: {
        name: string;
        value: string;
        traceback_raw: string[];
      };
    };

    type FinalChatResult = {
      type: 'final_code';
      status: 'completed' | 'failed';
      payload: {
        code: string;
        test: string;
        // Change introduces https://github.com/landing-ai/vision-agent-ui/pull/86
        // 1. Backward compatibility, it could be stringified StructuredResult
        // 2. result not modified in stream server, could still be stringified StructuredResult
        result: string | StructuredResult;
      };
    };
  }
}

const db = globalThis.prisma || new PrismaClient();

if (process.env.NODE_ENV != 'production') globalThis.prisma = db;

export default db;