updates

Author: michelle0927

components/zep/actions/add-user/add-user.mjs

@@ -0,0 +1,65 @@
+import zep from "../../zep.app.mjs";
+import utils from "../../common/utils.mjs";
+
+export default {
+  key: "zep-add-user",
+  name: "Add User",
+  description: "Adds a user in Zep. [See the documentation](https://help.getzep.com/api-reference/user/add)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    zep,
+    email: {
+      type: "string",
+      label: "Email",
+      description: "Email address of the user",
+      optional: true,
+    },
+    firstName: {
+      type: "string",
+      label: "First Name",
+      description: "First name of the new user",
+      optional: true,
+    },
+    lastName: {
+      type: "string",
+      label: "Last Name",
+      description: "Last name of the new user",
+      optional: true,
+    },
+    factRatingInstructions: {
+      propDefinition: [
+        zep,
+        "factRatingInstructions",
+      ],
+    },
+    metadata: {
+      propDefinition: [
+        zep,
+        "metadata",
+      ],
+      optional: true,
+    },
+    userId: {
+      type: "string",
+      label: "User ID",
+      description: "The unique identifier of the new user",
+      optional: true,
+    },
+  },
+  async run({ $ }) {
+    const response = await this.zep.createUser({
+      $,
+      data: {
+        email: this.email,
+        first_name: this.firstName,
+        last_name: this.lastName,
+        fact_rating_instructions: utils.parseObject(this.factRatingInstructions),
+        metadata: utils.parseObject(this.metadata),
+        user_id: this.userId,
+      },
+    });
+    $.export("$summary", `Successfully added user with ID: ${response.id}`);
+    return response;
+  },
+};

components/zep/sources/common/base.mjs

@@ -25,7 +25,7 @@ export default {
       results.forEach((item) => this.emitEvent(item));
     },
     async getSessions({
-      lastTs, orderBy, max,
+      lastTs, orderBy, max, updateLastTs = true,
     }) {
       const params = {
         page_size: max || 1000,
@@ -53,7 +53,9 @@ export default {
       if (!sessions.length) {
         return [];
       }
-      this._setLastTs(Date.parse(sessions[0][orderBy]));
+      if (updateLastTs) {
+        this._setLastTs(Date.parse(sessions[0][orderBy]));
+      }
       return sessions.reverse();
     },
     emitEvent(item) {

components/zep/sources/new-message/new-message.mjs

@@ -8,39 +8,49 @@ export default {
   version: "0.0.1",
   type: "source",
   dedupe: "unique",
-  props: {
-    ...common.props,
-    sessionId: {
-      propDefinition: [
-        common.props.zep,
-        "sessionId",
-      ],
-    },
-  },
   methods: {
     ...common.methods,
     async getNewResults(lastTs, max) {
-      const results = this.paginateMessages();
-
+      const sessionIds = await this.getRecentlyUpdatedSessionIds(lastTs);
       let messages = [];
-      for await (const message of results) {
-        const ts = Date.parse(message.created_at);
-        if (ts >= lastTs) {
-          messages.push(message);
+      let maxTs = lastTs;
+
+      for (const sessionId of sessionIds) {
+        const results = this.paginateMessages(sessionId);
+
+        for await (const message of results) {
+          const ts = Date.parse(message.created_at);
+          if (ts >= lastTs) {
+            messages.push({
+              ...message,
+              session_id: sessionId,
+            });
+            maxTs = Math.max(maxTs, ts);
+          }
         }
       }
 
-      if (messages.length) {
-        this._setLastTs(Date.parse(messages[messages.length - 1].created_at));
-      }
+      this._setLastTs(maxTs);
+
+      // sort by created_at
+      messages = messages.sort((a, b) => Date.parse(a.created_at) - Date.parse(b.created_at));
 
       if (max) {
         messages = messages.slice(-1 * max);
       }
 
       return messages;
     },
-    async *paginateMessages() {
+    async getRecentlyUpdatedSessionIds(lastTs) {
+      const sessions = await this.getSessions({
+        lastTs,
+        orderBy: "updated_at",
+        updateLastTs: false,
+        max: 100,
+      });
+      return sessions?.map(({ session_id: id }) => id) || [];
+    },
+    async *paginateMessages(sessionId) {
       const params = {
         limit: 1000,
         cursor: 1,
@@ -49,7 +59,7 @@ export default {
 
       do {
         const { messages } = await this.zep.listMessages({
-          sessionId: this.sessionId,
+          sessionId,
           params,
         });
         for (const message of messages) {

components/zep/zep.app.mjs

@@ -30,10 +30,12 @@ export default {
           },
         });
         return users?.map(({
-          user_id: value, first_name: firstName, last_name: lastName,
+          user_id: value, first_name: firstName, last_name: lastName, email,
         }) => ({
           value,
-          label: (`${firstName} ${lastName}`).trim(),
+          label: firstName || lastName
+            ? (`${firstName} ${lastName}`).trim()
+            : email || value,
         })) || [];
       },
     },
@@ -93,6 +95,13 @@ export default {
         ...opts,
       });
     },
+    createUser(opts = {}) {
+      return this._makeRequest({
+        method: "POST",
+        path: "/users",
+        ...opts,
+      });
+    },
     addMemoryToSession({
       sessionId, ...opts
     }) {