new components

Author: michelle0927

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

@@ -0,0 +1,55 @@
+import zep from "../../zep.app.mjs";
+import utils from "../../common/utils.mjs";
+
+export default {
+  key: "zep-add-memory",
+  name: "Add Memory to Session",
+  description: "Adds memory to an existing session in Zep. [See the documentation](https://help.getzep.com/api-reference/memory/add)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    zep,
+    sessionId: {
+      propDefinition: [
+        zep,
+        "sessionId",
+      ],
+    },
+    messages: {
+      type: "string[]",
+      label: "Messages",
+      description: "An array of message objects, where each message contains a role (`norole`, `system`, `assistant`, `user`, `function`, or `tool`) and content. Example: `[{ \"content\": \"content\", \"role_type\": \"norole\" }]` [See the documentation](https://help.getzep.com/api-reference/memory/add) for more information",
+    },
+    factInstruction: {
+      type: "string",
+      label: "Fact Instruction",
+      description: "Additional instruction for generating the facts",
+      optional: true,
+    },
+    returnContext: {
+      type: "boolean",
+      label: "Return Context",
+      description: "Optionally return memory context relevant to the most recent messages",
+      optional: true,
+    },
+    summaryInstruction: {
+      type: "string",
+      label: "Summary Instructions",
+      description: "Additional instruction for generating the summary",
+      optional: true,
+    },
+  },
+  async run({ $ }) {
+    const response = await this.zep.addMemoryToSession({
+      sessionId: this.sessionId,
+      data: {
+        messages: utils.parseArray(this.messages),
+        factInstruction: this.factInstruction,
+        returnContext: this.returnContext,
+        summaryInstruction: this.summaryInstruction,
+      },
+    });
+    $.export("$summary", `Added memory to session ${this.sessionId}`);
+    return response;
+  },
+};

components/zep/actions/create-session/create-session.mjs

@@ -0,0 +1,50 @@
+import zep from "../../zep.app.mjs";
+import utils from "../../common/utils.mjs";
+
+export default {
+  key: "zep-create-session",
+  name: "Create Session",
+  description: "Creates a new session in Zep. [See the documentation](https://help.getzep.com/api-reference/memory/add-session)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    zep,
+    sessionId: {
+      type: "string",
+      label: "Session ID",
+      description: "The unique identifier of the session",
+    },
+    userId: {
+      propDefinition: [
+        zep,
+        "userId",
+      ],
+    },
+    factRatingInstructions: {
+      propDefinition: [
+        zep,
+        "factRatingInstructions",
+      ],
+    },
+    metadata: {
+      propDefinition: [
+        zep,
+        "metadata",
+      ],
+      optional: true,
+    },
+  },
+  async run({ $ }) {
+    const response = await this.zep.createSession({
+      $,
+      data: {
+        session_id: this.sessionId,
+        user_id: this.userId,
+        fact_rating_instructions: utils.parseObject(this.factRatingInstructions),
+        metadata: utils.parseObject(this.metadata),
+      },
+    });
+    $.export("$summary", `Created session with ID ${this.sessionId}`);
+    return response;
+  },
+};

components/zep/actions/update-session/update-session.mjs

@@ -0,0 +1,44 @@
+import zep from "../../zep.app.mjs";
+import utils from "../../common/utils.mjs";
+
+export default {
+  key: "zep-update-session",
+  name: "Update Session",
+  description: "Updates an existing session in Zep. [See the documentation](https://help.getzep.com/api-reference/memory/update-session)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    zep,
+    sessionId: {
+      propDefinition: [
+        zep,
+        "sessionId",
+      ],
+    },
+    metadata: {
+      propDefinition: [
+        zep,
+        "metadata",
+      ],
+      description: "An object of key/value pairs representing the metadata to add to the session",
+    },
+    factRatingInstructions: {
+      propDefinition: [
+        zep,
+        "factRatingInstructions",
+      ],
+    },
+  },
+  async run({ $ }) {
+    const response = await this.zep.updateSession({
+      $,
+      sessionId: this.sessionId,
+      data: {
+        fact_rating_instructions: utils.parseObject(this.factRatingInstructions),
+        metadata: utils.parseObject(this.metadata),
+      },
+    });
+    $.export("$summary", `Successfully updated ssession with ID ${this.sessionId}`);
+    return response;
+  },
+};

components/zep/common/utils.mjs

@@ -0,0 +1,30 @@
+function parseObject(obj) {
+  if (!obj) {
+    return undefined;
+  }
+
+  return typeof obj === "string"
+    ? JSON.parse(obj)
+    : obj;
+}
+
+function parseArray(arr) {
+  if (!arr) {
+    return undefined;
+  }
+
+  if (typeof arr === "string") {
+    return JSON.parse(arr);
+  }
+
+  if (Array.isArray(arr)) {
+    return arr.map((item) => parseObject(item));
+  }
+
+  return arr;
+}
+
+export default {
+  parseObject,
+  parseArray,
+};

components/zep/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pipedream/zep",
-  "version": "0.0.1",
+  "version": "0.1.0",
   "description": "Pipedream Zep Components",
   "main": "zep.app.mjs",
   "keywords": [
@@ -11,5 +11,8 @@
   "author": "Pipedream <[email protected]> (https://pipedream.com/)",
   "publishConfig": {
     "access": "public"
+  },
+  "dependencies": {
+    "@pipedream/platform": "^3.0.3"
   }
 }

components/zep/sources/common/base.mjs

@@ -0,0 +1,78 @@
+import zep from "../../zep.app.mjs";
+import { DEFAULT_POLLING_SOURCE_TIMER_INTERVAL } from "@pipedream/platform";
+
+export default {
+  props: {
+    zep,
+    db: "$.service.db",
+    timer: {
+      type: "$.interface.timer",
+      default: {
+        intervalSeconds: DEFAULT_POLLING_SOURCE_TIMER_INTERVAL,
+      },
+    },
+  },
+  methods: {
+    _getLastTs() {
+      return this.db.get("lastTs") || 0;
+    },
+    _setLastTs(lastTs) {
+      this.db.set("lastTs", lastTs);
+    },
+    async processEvent(max) {
+      const lastTs = this._getLastTs();
+      const results = await this.getNewResults(lastTs, max);
+      results.forEach((item) => this.emitEvent(item));
+    },
+    async getSessions({
+      lastTs, orderBy, max,
+    }) {
+      const params = {
+        page_size: max || 1000,
+        order_by: orderBy,
+        asc: false,
+      };
+
+      const { sessions: results } = await this.zep.listSessions({
+        params,
+      });
+
+      const sessions = [];
+      for (const session of results) {
+        const ts = Date.parse(session[orderBy]);
+        if (ts >= lastTs) {
+          sessions.push(session);
+        } else {
+          break;
+        }
+        if (max && sessions.length >= max) {
+          break;
+        }
+      }
+
+      if (!sessions.length) {
+        return [];
+      }
+      this._setLastTs(Date.parse(sessions[0][orderBy]));
+      return sessions.reverse();
+    },
+    emitEvent(item) {
+      const meta = this.generateMeta(item);
+      this.$emit(item, meta);
+    },
+    getNewResults() {
+      throw new Error("getNewResults is not implemented");
+    },
+    generateMeta() {
+      throw new Error("generateMeta is not implemented");
+    },
+  },
+  hooks: {
+    async deploy() {
+      await this.processEvent(25);
+    },
+  },
+  async run() {
+    await this.processEvent();
+  },
+};

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

@@ -0,0 +1,70 @@
+import common from "../common/base.mjs";
+
+export default {
+  ...common,
+  key: "zep-new-message",
+  name: "New Message in Session",
+  description: "Emit new event when a message is added to a session. [See the documentation](https://help.getzep.com/api-reference/memory/get-session-messages)",
+  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();
+
+      let messages = [];
+      for await (const message of results) {
+        const ts = Date.parse(message.created_at);
+        if (ts >= lastTs) {
+          messages.push(message);
+        }
+      }
+
+      if (messages.length) {
+        this._setLastTs(Date.parse(messages[messages.length - 1].created_at));
+      }
+
+      if (max) {
+        messages = messages.slice(-1 * max);
+      }
+
+      return messages;
+    },
+    async *paginateMessages() {
+      const params = {
+        limit: 1000,
+        cursor: 1,
+      };
+      let total;
+
+      do {
+        const { messages } = await this.zep.listMessages({
+          sessionId: this.sessionId,
+          params,
+        });
+        for (const message of messages) {
+          yield message;
+        }
+        total = messages?.length;
+        params.cursor++;
+      } while (total);
+    },
+    generateMeta(message) {
+      return {
+        id: message.uuid,
+        summary: `New Message: ${message.content}`,
+        ts: Date.parse(message.created_at),
+      };
+    },
+  },
+};

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

@@ -0,0 +1,28 @@
+import common from "../common/base.mjs";
+
+export default {
+  ...common,
+  key: "zep-new-session",
+  name: "New Session Created",
+  description: "Emit new event when a new session is created in Zep. [See the documentation](https://help.getzep.com/api-reference/memory/list-sessions)",
+  version: "0.0.1",
+  type: "source",
+  dedupe: "unique",
+  methods: {
+    ...common.methods,
+    getNewResults(lastTs, max) {
+      return this.getSessions({
+        lastTs,
+        max,
+        orderBy: "created_at",
+      });
+    },
+    generateMeta(session) {
+      return {
+        id: session.session_id,
+        summary: `New Session: ${session.session_id}`,
+        ts: Date.parse(session.created_at),
+      };
+    },
+  },
+};