@xalia/agent 0.6.10 → 0.6.11

package/src/test/db.test.ts

@@ -1,363 +0,0 @@
-import { expect, beforeEach, afterEach, describe, it } from "vitest";
-import { Database } from "../chat/data/database";
-import { SessionDescriptor } from "../chat/data/dataModels";
-import { strict as assert } from "assert";
-import { v4 as uuidv4 } from "uuid";
-import {
-  AGENT_PROFILE,
-  AGENT_PROFILE_2,
-  TestUserSpec,
-  cleanupTestUser,
-  createTestSession,
-  createTestTeamSession,
-  createTestUser,
-  getLocalDB,
-  testUserSpec,
-} from "./dbTestTools";
-
-describe("DB", () => {
-  let db: Database;
-  let testUsers: TestUserSpec[];
-
-  beforeEach(() => {
-    // Generate unique test data for each test
-    db = getLocalDB();
-    testUsers = [testUserSpec(), testUserSpec()];
-  });
-
-  afterEach(async () => {
-    // Clean up test data after each test
-    await Promise.all(testUsers.map((tu) => cleanupTestUser(db, tu)));
-    testUsers = [];
-  });
-
-  it("should get data for existing user", async () => {
-    await createTestUser(db, testUsers[0]);
-    const userData = await db.getUserDataFromApiKey(testUsers[0].api_key);
-    expect(userData).eql({
-      uuid: testUsers[0].uuid,
-      email: testUsers[0].email,
-      nickname: testUsers[0].name,
-      timezone: "UTC",
-    });
-  });
-
-  it("should return undefined for non-existant user", async () => {
-    const userData = await db.getUserDataFromApiKey("no_such_key_" + uuidv4());
-    expect(userData).eql(undefined);
-  });
-
-  it("should get user by uuid", async () => {
-    await createTestUser(db, testUsers[0]);
-    const userData = await db.getUserFromUuid(testUsers[0].uuid);
-    assert(userData);
-    expect(userData.uuid).eql(testUsers[0].uuid);
-    expect(userData.nickname).eql(testUsers[0].name);
-  });
-
-  it("should create and retrieve agent profiles", async () => {
-    await createTestUser(db, testUsers[0]);
-
-    const agentProfile = await db.createAgentProfile(
-      testUsers[0].uuid,
-      undefined /*  team_uuid */,
-      "test_profile",
-      AGENT_PROFILE
-    );
-    assert(agentProfile);
-    const agentProfileId = agentProfile.uuid;
-
-    const savedAgentProfile = await db.getSavedAgentProfileById(agentProfileId);
-    const savedAgentProfileByName = await db.getSavedAgentProfileByName(
-      testUsers[0].uuid,
-      "test_profile"
-    );
-    const fetchedAgentProfile = await db.getAgentProfileById(agentProfileId);
-
-    assert(savedAgentProfile);
-    assert(savedAgentProfileByName);
-    assert(fetchedAgentProfile);
-
-    expect(savedAgentProfile.uuid).eql(agentProfileId);
-    expect(savedAgentProfile.profile).eql(AGENT_PROFILE);
-    expect(savedAgentProfile.user_uuid).eql(testUsers[0].uuid);
-    expect(savedAgentProfile.profile_name).eql("test_profile");
-
-    expect(savedAgentProfileByName).eql(savedAgentProfile);
-
-    expect(fetchedAgentProfile).eql(AGENT_PROFILE);
-  });
-
-  it("should update agent profiles", async () => {
-    await createTestUser(db, testUsers[0]);
-
-    const agentProfile = await db.createAgentProfile(
-      testUsers[0].uuid,
-      undefined /*  team_uuid */,
-      "test_profile",
-      AGENT_PROFILE
-    );
-    const agentProfileId = agentProfile ? agentProfile.uuid : undefined;
-    expect(agentProfileId).not.eql(undefined);
-    assert(agentProfileId);
-
-    await db.updateAgentProfile(agentProfileId, AGENT_PROFILE_2);
-
-    const agentProfileFromDB = await db.getAgentProfileById(agentProfileId);
-    assert(agentProfileFromDB);
-    expect(agentProfileFromDB).eql(AGENT_PROFILE_2);
-  });
-
-  it("should update settings on saved agent profiles", async () => {
-    await createTestUser(db, testUsers[0]);
-
-    const agentProfile = await db.createAgentProfile(
-      testUsers[0].uuid,
-      undefined /*  team_uuid */,
-      "test_profile",
-      AGENT_PROFILE
-    );
-    assert(agentProfile);
-    const agentProfileId = agentProfile.uuid;
-    const savedAgentProfile0 =
-      await db.getSavedAgentProfileById(agentProfileId);
-    assert(savedAgentProfile0);
-
-    const prefsNone = await db.getAgentProfilePreferences(
-      agentProfileId + "aa"
-    );
-    expect(prefsNone).eql(undefined);
-
-    const prefs0 = await db.getAgentProfilePreferences(agentProfileId);
-    expect(prefs0).eql({ auto_approve: {} });
-
-    const newPrefs1 = {
-      skip_welcome: true,
-      auto_approve: { some_server: ["tool1"] },
-    };
-    await db.updateAgentProfilePreferences(agentProfileId, newPrefs1);
-    const prefs1 = await db.getAgentProfilePreferences(agentProfileId);
-    expect(prefs1).eql(newPrefs1);
-
-    const newPrefs2 = {
-      skip_welcome: true,
-      auto_approve: { different_server: ["toolA", "tool2"] },
-    };
-    await db.updateAgentProfilePreferences(agentProfileId, newPrefs2);
-    const prefs2 = await db.getAgentProfilePreferences(agentProfileId);
-    expect(prefs2).eql(newPrefs2);
-    assert(prefs2);
-
-    // Check that the rest of the profile is unchanged
-
-    const savedAgentProfile2 =
-      await db.getSavedAgentProfileById(agentProfileId);
-
-    // Linter doesn't like this, so we overwrite preferences on
-    // savedAgentProfile0.
-    //
-    // expect(savedAgentProfile2).eql({
-    //   ...savedAgentProfile0,
-    //   preferences: prefs2,
-    // });
-
-    savedAgentProfile0.preferences = prefs2;
-    expect(savedAgentProfile2).eql(savedAgentProfile0);
-  });
-
-  it("should create and retrieve sessions", async () => {
-    await createTestUser(db, testUsers[0]);
-
-    const agentProfile = await db.createAgentProfile(
-      testUsers[0].uuid,
-      undefined /*  team_uuid */,
-      "test_profile",
-      AGENT_PROFILE
-    );
-    assert(agentProfile);
-    const agentProfileId = agentProfile.uuid;
-
-    const sessionId = Database.sessionNewUUID();
-    await db.sessionCreate({
-      session_uuid: sessionId,
-      user_uuid: testUsers[0].uuid,
-      title: "test_session",
-      agent_profile_uuid: agentProfileId,
-      agent_paused: false,
-    });
-    assert(sessionId);
-
-    function checkSession(session: SessionDescriptor | undefined) {
-      assert(session);
-      expect(session.agent_profile_uuid).eql(agentProfileId);
-      expect(session.title).eql("test_session");
-      expect(session.user_uuid).eql(testUsers[0].uuid);
-      expect(session.session_uuid).eql(sessionId);
-    }
-
-    checkSession(await db.sessionGetDescriptorById(sessionId));
-    checkSession(await db.sessionGetById(sessionId));
-
-    // Delete session and check conv is empty
-
-    await db.sessionDeleteById(sessionId);
-    const session2 = await db.sessionGetDescriptorById(sessionId);
-    expect(!!session2).eql(false);
-  });
-
-  it("should update session titles", async () => {
-    const { agentProfileId, sessionId } = await createTestSession(
-      db,
-      testUsers[0]
-    );
-
-    // Change the title
-
-    await db.sessionUpdateTitle(sessionId, "new_title");
-
-    // Check the session
-
-    const session = await db.sessionGetDescriptorById(sessionId);
-    assert(session);
-    expect(session.agent_profile_uuid).eql(agentProfileId);
-    expect(session.title).eql("new_title");
-    expect(session.user_uuid).eql(testUsers[0].uuid);
-    expect(session.session_uuid).eql(sessionId);
-  });
-
-  it("should update session workspace", async () => {
-    const { sessionId } = await createTestSession(db, testUsers[0]);
-
-    // Set the workspace
-
-    const ws1 = { message: "workspace1" };
-    await db.sessionUpdateWorkspace(sessionId, ws1);
-    const session1 = await db.sessionGetById(sessionId);
-
-    // Change the workspace
-
-    const ws2 = {
-      message: "workspace2",
-      imageB64: "data:image/png;base64;asdfadfadsfads",
-    };
-    await db.sessionUpdateWorkspace(sessionId, ws2);
-    const session2 = await db.sessionGetById(sessionId);
-
-    // Removed the workspace
-
-    await db.sessionUpdateWorkspace(sessionId, undefined);
-    const session3 = await db.sessionGetById(sessionId);
-
-    // Check
-
-    assert(session1);
-    expect(session1.workspace).eql(ws1);
-
-    assert(session2);
-    expect(session2.workspace).eql(ws2);
-
-    assert(session3);
-    expect(session3.workspace).eql(undefined);
-  });
-
-  it("should update session access token", async () => {
-    const { sessionId } = await createTestSession(db, testUsers[0]);
-    const token = "access_token_A";
-
-    // Initial state
-
-    const session0 = await db.sessionGetDescriptorById(sessionId);
-
-    // Change the access token
-
-    await db.sessionUpdateAccessToken(sessionId, token);
-    const session1 = await db.sessionGetDescriptorById(sessionId);
-
-    // Reset access token
-
-    await db.sessionUpdateAccessToken(sessionId, undefined);
-    const session2 = await db.sessionGetDescriptorById(sessionId);
-
-    // Asserts
-
-    assert(session0);
-    assert(session1);
-    assert(session2);
-
-    expect(session0.access_token).eql(undefined);
-    expect(session1.access_token).eql(token);
-    expect(session2.access_token).eql(undefined);
-  });
-
-  it("handles team participants correctly", async () => {
-    const { sessionId, teamId } = await createTestTeamSession(
-      db,
-      "team1",
-      testUsers
-    );
-
-    // Only owner should be a participant
-
-    const participants0 = await db.sessionGetParticipants(sessionId);
-    expect(participants0).to.have.lengthOf(1);
-    expect(participants0).to.deep.include({
-      user_uuid: testUsers[0].uuid,
-      nickname: testUsers[0].name,
-      email: testUsers[0].email,
-      role: "owner",
-    });
-
-    // Add secondary user as participant
-    await db.teamAddMember(teamId, testUsers[1].uuid);
-    const participants1 = await db.sessionGetParticipants(sessionId);
-    expect(participants1).to.have.lengthOf(2);
-    expect(participants1).to.deep.include({
-      user_uuid: testUsers[0].uuid,
-      nickname: testUsers[0].name,
-      email: testUsers[0].email,
-      role: "owner",
-    });
-    expect(participants1).to.deep.include({
-      user_uuid: testUsers[1].uuid,
-      nickname: testUsers[1].name,
-      email: testUsers[1].email,
-      role: "participant",
-    });
-
-    // Remove secondary user
-    await db.teamRemoveMember(teamId, testUsers[1].uuid);
-    const participants2 = await db.sessionGetParticipants(sessionId);
-    expect(participants2).to.have.lengthOf(1);
-    expect(participants2).to.deep.include({
-      user_uuid: testUsers[0].uuid,
-      nickname: testUsers[0].name,
-      email: testUsers[0].email,
-      role: "owner",
-    });
-  });
-
-  it("handles session_checkpoints correctly", async () => {
-    const { sessionId } = await createTestSession(db, testUsers[0]);
-
-    await db.sessionCheckpointDelete(sessionId);
-
-    // No checkpoints
-
-    const noCheckpoint = await db.sessionCheckpointGet(sessionId);
-    expect(noCheckpoint).eql(undefined);
-
-    // Create checkpoint and check
-
-    const cpModel1 = { message_idx: 3, summary: "summaryA" };
-    await db.sessionCheckpointSet(sessionId, cpModel1);
-    const cp1 = await db.sessionCheckpointGet(sessionId);
-    expect(cp1).eql(cpModel1);
-
-    // Replace
-
-    const cpModel2 = { message_idx: 6, summary: "summaryB" };
-    await db.sessionCheckpointSet(sessionId, cpModel2);
-    const cp2 = await db.sessionCheckpointGet(sessionId);
-    expect(cp2).eql(cpModel2);
-  });
-});

package/src/test/dbMcpServerConfigs.test.ts

@@ -1,112 +0,0 @@
-import { expect } from "vitest";
-import { strict as assert } from "assert";
-
-import { Database } from "../chat/data/database";
-import {
-  TestUser,
-  TestUserSpec,
-  cleanupTestUser,
-  createTestUser,
-  getLocalDB,
-  testUserSpec,
-} from "./dbTestTools";
-import { DbMcpServerConfigs } from "../chat/data/dbMcpServerConfigs";
-import { McpServerConfiguration } from "@xalia/xmcp/sdk";
-
-type ConfigDesc = {
-  server_name: string;
-  config: McpServerConfiguration | undefined;
-};
-
-const DEFAULT_CONFIGS: ConfigDesc[] = [
-  {
-    server_name: "server_1",
-    config: {
-      a: "123",
-      b: "456",
-    },
-  },
-  {
-    server_name: "server_2",
-    config: {
-      d: "adsf",
-      e: "qwer",
-    },
-  },
-];
-
-async function setup(
-  db: Database,
-  user1: TestUserSpec
-): Promise<{
-  msc: DbMcpServerConfigs;
-  user0: TestUser;
-}> {
-  const user0 = await createTestUser(db, user1);
-  const msc = db.createTypedClient(DbMcpServerConfigs);
-
-  await Promise.all(
-    DEFAULT_CONFIGS.map((cfg) => {
-      assert(cfg.config);
-      return msc.setConfigForUser(user0.uuid, cfg.server_name, cfg.config);
-    })
-  );
-
-  return { msc, user0 };
-}
-
-describe("DB McpServerConfigurations", () => {
-  let db: Database;
-  let user: TestUserSpec;
-
-  beforeEach(() => {
-    db = getLocalDB();
-    user = testUserSpec();
-  });
-
-  afterEach(async () => {
-    await cleanupTestUser(db, user);
-  });
-
-  it("retrieves configs", async function () {
-    const { user0, msc } = await setup(db, user);
-    for (const cfgDesc of DEFAULT_CONFIGS) {
-      const cfg = await msc.getConfigForUser(user0.uuid, cfgDesc.server_name);
-      expect(cfg).toEqual(cfgDesc.config);
-    }
-  });
-
-  it("delete config", async function () {
-    const { user0, msc } = await setup(db, user);
-
-    const cfgDescs: ConfigDesc[] = DEFAULT_CONFIGS.map(
-      (x) => JSON.parse(JSON.stringify(x)) as ConfigDesc
-    );
-    assert(cfgDescs[0]);
-    await msc.deleteConfigForUser(user0.uuid, cfgDescs[0].server_name);
-    cfgDescs[0].config = undefined;
-
-    for (const cfgDesc of cfgDescs) {
-      const cfg = await msc.getConfigForUser(user0.uuid, cfgDesc.server_name);
-      expect(cfg).toEqual(cfgDesc.config);
-    }
-  });
-
-  it("overwrite config", async function () {
-    const { user0, msc } = await setup(db, user);
-
-    const newConfig = { a: "123", b: "789", c: "uiop" };
-    const cfgDescs: ConfigDesc[] = DEFAULT_CONFIGS.map(
-      (x) => JSON.parse(JSON.stringify(x)) as ConfigDesc
-    );
-
-    assert(cfgDescs[0]);
-    await msc.setConfigForUser(user0.uuid, cfgDescs[0].server_name, newConfig);
-    cfgDescs[0].config = newConfig;
-
-    for (const cfgDesc of cfgDescs) {
-      const cfg = await msc.getConfigForUser(user0.uuid, cfgDesc.server_name);
-      expect(cfg).toEqual(cfgDesc.config);
-    }
-  });
-});