@adminforth/agent 1.45.0 → 1.46.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adminforth/agent",
-  "version": "1.45.0",
+  "version": "1.46.0",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "type": "module",

package/sessionStore.ts

@@ -0,0 +1,94 @@
+import type { AdminUser, IAdminForth } from "adminforth";
+import { Filters, Sorts } from "adminforth";
+import { randomUUID } from "crypto";
+import type { ChatSurfaceIncomingMessage } from "adminforth";
+import type { PreviousUserMessage } from "./agent/languageDetect.js";
+import type { PluginOptions } from "./types.js";
+
+export const AGENT_SYSTEM_TURN_PROMPT = "__adminforth_system_message__";
+
+export class AgentSessionStore {
+  constructor(
+    private getAdminforth: () => IAdminForth,
+    private options: PluginOptions,
+  ) {}
+
+  async createNewTurn(sessionId: string, prompt: string, response?: string) {
+    const turnId = randomUUID();
+    const turnRecord = {
+      [this.options.turnResource.idField]: turnId,
+      [this.options.turnResource.sessionIdField]: sessionId,
+      [this.options.turnResource.promptField]: prompt,
+      [this.options.turnResource.responseField]: response ?? "not_finished",
+    };
+    const newTurn = await this.getAdminforth().resource(this.options.turnResource.resourceId).create(turnRecord);
+    return newTurn.createdRecord[this.options.turnResource.idField];
+  }
+
+  async createSystemTurn(sessionId: string, systemMessage: string) {
+    const turnId = randomUUID();
+    const turnRecord = {
+      [this.options.turnResource.idField]: turnId,
+      [this.options.turnResource.sessionIdField]: sessionId,
+      [this.options.turnResource.promptField]: AGENT_SYSTEM_TURN_PROMPT,
+      [this.options.turnResource.responseField]: systemMessage,
+    };
+    const newTurn = await this.getAdminforth().resource(this.options.turnResource.resourceId).create(turnRecord);
+    return newTurn.createdRecord[this.options.turnResource.idField];
+  }
+
+  async getSessionTurns(sessionId: string) {
+    const turns = await this.getAdminforth().resource(this.options.turnResource.resourceId).list(
+      [Filters.EQ(this.options.turnResource.sessionIdField, sessionId)],
+      undefined,
+      undefined,
+      [Sorts.ASC(this.options.turnResource.createdAtField)]
+    );
+    return turns.map(turn => ({
+      prompt: turn[this.options.turnResource.promptField],
+      response: turn[this.options.turnResource.responseField],
+    }));
+  }
+
+  async getPreviousUserMessages(sessionId: string) {
+    const turns = await this.getAdminforth().resource(this.options.turnResource.resourceId).list(
+      [Filters.EQ(this.options.turnResource.sessionIdField, sessionId)],
+      2,
+      undefined,
+      [Sorts.DESC(this.options.turnResource.createdAtField)]
+    );
+    return turns
+      .reverse()
+      .filter((turn) => turn[this.options.turnResource.promptField] !== AGENT_SYSTEM_TURN_PROMPT)
+      .map((turn): PreviousUserMessage => ({
+        text: turn[this.options.turnResource.promptField],
+      }));
+  }
+
+  getChatSurfaceSessionId(incoming: ChatSurfaceIncomingMessage) {
+    return `${incoming.surface}:${incoming.externalConversationId}`;
+  }
+
+  async getOrCreateChatSurfaceSession(
+    incoming: ChatSurfaceIncomingMessage,
+    adminUser: AdminUser,
+  ) {
+    const sessionId = this.getChatSurfaceSessionId(incoming);
+    const sessionResource = this.getAdminforth().resource(this.options.sessionResource.resourceId);
+    const session = await sessionResource.get(
+      [Filters.EQ(this.options.sessionResource.idField, sessionId)]
+    );
+
+    if (session) {
+      return sessionId;
+    }
+
+    await sessionResource.create({
+      [this.options.sessionResource.idField]: sessionId,
+      [this.options.sessionResource.titleField]: incoming.prompt.slice(0, 40) || "New Session",
+      [this.options.sessionResource.askerIdField]: adminUser.pk,
+    });
+
+    return sessionId;
+  }
+}

package/agentResponseEvents.ts

@@ -1 +0,0 @@
-export { createSseEventEmitter } from "./surfaces/web-sse/createSseEventEmitter.js";

package/dist/agentResponseEvents.d.ts

@@ -1 +0,0 @@
-export { createSseEventEmitter } from "./surfaces/web-sse/createSseEventEmitter.js";

package/dist/agentResponseEvents.js

@@ -1 +0,0 @@
-export { createSseEventEmitter } from "./surfaces/web-sse/createSseEventEmitter.js";