@agentscope/studio/dist/server/src/database.js

AgentScope Studio is a powerful local monitoring and visualization tool designed to provide real-time insights into your system's performance and behavior.

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.initializeDatabase = void 0;
const typeorm_1 = require("typeorm");
const Message_1 = require("./models/Message");
const Run_1 = require("./models/Run");
const RunView_1 = require("./models/RunView");
const InputRequest_1 = require("./models/InputRequest");
const Run_2 = require("./dao/Run");
const InputRequest_2 = require("./dao/InputRequest");
const ModelInvocation_1 = require("./models/ModelInvocation");
const ModelInvocationView_1 = require("./models/ModelInvocationView");
const initializeDatabase = (databaseConfig) => __awaiter(void 0, void 0, void 0, function* () {
    try {
        const options = Object.assign(Object.assign({}, databaseConfig), { entities: [
                Run_1.RunTable,
                RunView_1.RunView,
                Message_1.MessageTable,
                InputRequest_1.InputRequestTable,
                ModelInvocation_1.ModelInvocationTable,
                ModelInvocationView_1.ModelInvocationView,
            ], synchronize: true, logging: false });
        const AppDataBase = new typeorm_1.DataSource(options);
        yield AppDataBase.initialize();
        const printingOptions = Object.assign(Object.assign({}, options), { entities: undefined });
        console.log(`Database initialized with options: ${JSON.stringify(printingOptions, null, 2)}`);
        console.log('Refresh the database ...');
        yield Run_2.RunDao.updateRunStatusAtBeginning();
        yield InputRequest_2.InputRequestDao.updateInputRequests();
        console.log('Done');
    }
    catch (error) {
        console.error('Error initializing database', error);
        throw error;
    }
});
exports.initializeDatabase = initializeDatabase;