@snap-agent/core 0.1.4 → 0.1.5

package/dist/chunk-52OS4FTG.mjs

@@ -0,0 +1,382 @@
+// src/storage/UpstashStorage.ts
+var UpstashStorage = class {
+  constructor(config) {
+    this.url = config.url.replace(/\/$/, "");
+    this.token = config.token;
+    this.prefix = config.prefix || "snap-agent";
+  }
+  // ============================================================================
+  // Redis Commands via REST API
+  // ============================================================================
+  async command(cmd, ...args) {
+    const body = [cmd, ...args];
+    const response = await fetch(`${this.url}`, {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${this.token}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(body)
+    });
+    if (!response.ok) {
+      const text = await response.text();
+      throw new Error(`Upstash Redis error: ${response.status} - ${text}`);
+    }
+    const data = await response.json();
+    if (data.error) {
+      throw new Error(`Upstash Redis error: ${data.error}`);
+    }
+    return data.result;
+  }
+  async pipeline(commands) {
+    const response = await fetch(`${this.url}/pipeline`, {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${this.token}`,
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify(commands)
+    });
+    if (!response.ok) {
+      const text = await response.text();
+      throw new Error(`Upstash Redis pipeline error: ${response.status} - ${text}`);
+    }
+    const data = await response.json();
+    return data.map((item) => {
+      if (item.error) {
+        throw new Error(`Upstash Redis error: ${item.error}`);
+      }
+      return item.result;
+    });
+  }
+  // ============================================================================
+  // Key Generation
+  // ============================================================================
+  key(...parts) {
+    return `${this.prefix}:${parts.join(":")}`;
+  }
+  generateId() {
+    const timestamp = Date.now().toString(36);
+    const random = Math.random().toString(36).substring(2, 10);
+    return `${timestamp}${random}`;
+  }
+  // ============================================================================
+  // Agent Operations
+  // ============================================================================
+  async createAgent(config) {
+    const id = this.generateId();
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const stored = {
+      id,
+      organizationId: config.organizationId,
+      userId: config.userId,
+      phone: config.phone,
+      name: config.name,
+      description: config.description,
+      instructions: config.instructions,
+      provider: config.provider,
+      model: config.model,
+      createdAt: now,
+      updatedAt: now,
+      files: JSON.stringify([]),
+      metadata: config.metadata ? JSON.stringify(config.metadata) : void 0
+    };
+    const fields = [];
+    for (const [key, value] of Object.entries(stored)) {
+      if (value !== void 0) {
+        fields.push(key, String(value));
+      }
+    }
+    await this.pipeline([
+      ["HSET", this.key("agent", id), ...fields],
+      ["SADD", this.key("agents:user", config.userId), id],
+      ...config.organizationId ? [["SADD", this.key("agents:org", config.organizationId), id]] : []
+    ]);
+    return id;
+  }
+  async getAgent(agentId) {
+    const data = await this.command(
+      "HGETALL",
+      this.key("agent", agentId)
+    );
+    if (!data || Object.keys(data).length === 0) {
+      return null;
+    }
+    return this.parseStoredAgent(data);
+  }
+  async updateAgent(agentId, updates) {
+    const fields = ["updatedAt", (/* @__PURE__ */ new Date()).toISOString()];
+    for (const [key, value] of Object.entries(updates)) {
+      if (value !== void 0) {
+        if (key === "metadata") {
+          fields.push(key, JSON.stringify(value));
+        } else {
+          fields.push(key, String(value));
+        }
+      }
+    }
+    await this.command("HSET", this.key("agent", agentId), ...fields);
+  }
+  async deleteAgent(agentId) {
+    const agent = await this.getAgent(agentId);
+    if (!agent) return;
+    const threadIds = await this.command(
+      "SMEMBERS",
+      this.key("threads:agent", agentId)
+    );
+    const commands = [
+      ["DEL", this.key("agent", agentId)],
+      ["SREM", this.key("agents:user", agent.userId), agentId]
+    ];
+    if (agent.organizationId) {
+      commands.push(["SREM", this.key("agents:org", agent.organizationId), agentId]);
+    }
+    for (const threadId of threadIds || []) {
+      commands.push(["DEL", this.key("thread", threadId)]);
+    }
+    commands.push(["DEL", this.key("threads:agent", agentId)]);
+    await this.pipeline(commands);
+  }
+  async listAgents(userId, organizationId) {
+    const indexKey = organizationId ? this.key("agents:org", organizationId) : this.key("agents:user", userId);
+    const agentIds = await this.command("SMEMBERS", indexKey);
+    if (!agentIds || agentIds.length === 0) {
+      return [];
+    }
+    const agents = [];
+    for (const id of agentIds) {
+      const agent = await this.getAgent(id);
+      if (agent) {
+        if (!organizationId || agent.userId === userId) {
+          agents.push(agent);
+        }
+      }
+    }
+    return agents.sort(
+      (a, b) => b.updatedAt.getTime() - a.updatedAt.getTime()
+    );
+  }
+  // ============================================================================
+  // Thread Operations
+  // ============================================================================
+  async createThread(config) {
+    const id = this.generateId();
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const stored = {
+      id,
+      organizationId: config.organizationId,
+      agentId: config.agentId,
+      userId: config.userId,
+      endUserId: config.endUserId,
+      name: config.name,
+      createdAt: now,
+      updatedAt: now,
+      messages: JSON.stringify([]),
+      isPendingThread: "true",
+      metadata: config.metadata ? JSON.stringify(config.metadata) : void 0
+    };
+    const fields = [];
+    for (const [key, value] of Object.entries(stored)) {
+      if (value !== void 0) {
+        fields.push(key, String(value));
+      }
+    }
+    await this.pipeline([
+      ["HSET", this.key("thread", id), ...fields],
+      ["SADD", this.key("threads:agent", config.agentId), id],
+      ["SADD", this.key("threads:user", config.userId), id],
+      ...config.organizationId ? [["SADD", this.key("threads:org", config.organizationId), id]] : []
+    ]);
+    return id;
+  }
+  async getThread(threadId) {
+    const data = await this.command(
+      "HGETALL",
+      this.key("thread", threadId)
+    );
+    if (!data || Object.keys(data).length === 0) {
+      return null;
+    }
+    return this.parseStoredThread(data);
+  }
+  async updateThread(threadId, updates) {
+    const fields = ["updatedAt", (/* @__PURE__ */ new Date()).toISOString()];
+    for (const [key, value] of Object.entries(updates)) {
+      if (value !== void 0) {
+        if (key === "metadata") {
+          fields.push(key, JSON.stringify(value));
+        } else {
+          fields.push(key, String(value));
+        }
+      }
+    }
+    await this.command("HSET", this.key("thread", threadId), ...fields);
+  }
+  async deleteThread(threadId) {
+    const thread = await this.getThread(threadId);
+    if (!thread) return;
+    const commands = [
+      ["DEL", this.key("thread", threadId)],
+      ["SREM", this.key("threads:agent", thread.agentId), threadId],
+      ["SREM", this.key("threads:user", thread.userId), threadId]
+    ];
+    if (thread.organizationId) {
+      commands.push(["SREM", this.key("threads:org", thread.organizationId), threadId]);
+    }
+    await this.pipeline(commands);
+  }
+  async listThreads(filters) {
+    let indexKey;
+    if (filters.agentId) {
+      indexKey = this.key("threads:agent", filters.agentId);
+    } else if (filters.organizationId) {
+      indexKey = this.key("threads:org", filters.organizationId);
+    } else if (filters.userId) {
+      indexKey = this.key("threads:user", filters.userId);
+    } else {
+      return [];
+    }
+    const threadIds = await this.command("SMEMBERS", indexKey);
+    if (!threadIds || threadIds.length === 0) {
+      return [];
+    }
+    const threads = [];
+    for (const id of threadIds) {
+      const thread = await this.getThread(id);
+      if (thread) {
+        if (filters.userId && thread.userId !== filters.userId) continue;
+        if (filters.agentId && thread.agentId !== filters.agentId) continue;
+        if (filters.organizationId && thread.organizationId !== filters.organizationId) continue;
+        threads.push(thread);
+      }
+    }
+    return threads.sort(
+      (a, b) => b.updatedAt.getTime() - a.updatedAt.getTime()
+    );
+  }
+  // ============================================================================
+  // Message Operations
+  // ============================================================================
+  async addMessage(threadId, role, content, attachments) {
+    const thread = await this.getThread(threadId);
+    if (!thread) {
+      throw new Error(`Thread not found: ${threadId}`);
+    }
+    const messageId = this.generateId();
+    const message = {
+      id: messageId,
+      role,
+      content,
+      timestamp: /* @__PURE__ */ new Date(),
+      attachments
+    };
+    thread.messages.push(message);
+    await this.command(
+      "HSET",
+      this.key("thread", threadId),
+      "messages",
+      JSON.stringify(thread.messages),
+      "updatedAt",
+      (/* @__PURE__ */ new Date()).toISOString(),
+      "isPendingThread",
+      "false"
+    );
+    return messageId;
+  }
+  async getMessages(threadId, limit) {
+    const thread = await this.getThread(threadId);
+    if (!thread) return [];
+    const messages = [...thread.messages];
+    if (limit) {
+      return messages.slice(-limit);
+    }
+    return messages;
+  }
+  async getConversationContext(threadId, maxMessages = 20) {
+    const messages = await this.getMessages(threadId, maxMessages);
+    return messages.map((msg) => ({
+      role: msg.role,
+      content: msg.content
+    }));
+  }
+  // ============================================================================
+  // Helper Methods
+  // ============================================================================
+  parseStoredAgent(stored) {
+    return {
+      id: stored.id,
+      organizationId: stored.organizationId,
+      userId: stored.userId,
+      phone: stored.phone,
+      name: stored.name,
+      description: stored.description,
+      instructions: stored.instructions,
+      provider: stored.provider,
+      model: stored.model,
+      createdAt: new Date(stored.createdAt),
+      updatedAt: new Date(stored.updatedAt),
+      files: stored.files ? JSON.parse(stored.files) : [],
+      metadata: stored.metadata ? JSON.parse(stored.metadata) : void 0
+    };
+  }
+  parseStoredThread(stored) {
+    const messages = stored.messages ? JSON.parse(stored.messages) : [];
+    const parsedMessages = messages.map((msg) => ({
+      ...msg,
+      timestamp: new Date(msg.timestamp)
+    }));
+    return {
+      id: stored.id,
+      organizationId: stored.organizationId,
+      agentId: stored.agentId,
+      userId: stored.userId,
+      endUserId: stored.endUserId,
+      name: stored.name,
+      createdAt: new Date(stored.createdAt),
+      updatedAt: new Date(stored.updatedAt),
+      messages: parsedMessages,
+      isPendingThread: stored.isPendingThread === "true",
+      metadata: stored.metadata ? JSON.parse(stored.metadata) : void 0
+    };
+  }
+  // ============================================================================
+  // Utility Methods
+  // ============================================================================
+  /**
+   * Test connection to Upstash Redis
+   */
+  async ping() {
+    try {
+      const result = await this.command("PING");
+      return result === "PONG";
+    } catch {
+      return false;
+    }
+  }
+  /**
+   * Clear all data with this prefix (use with caution!)
+   */
+  async clear() {
+    const keys = await this.command("KEYS", `${this.prefix}:*`);
+    if (keys && keys.length > 0) {
+      await this.command("DEL", ...keys);
+    }
+  }
+  /**
+   * Get storage statistics
+   */
+  async getStats() {
+    const [agentKeys, threadKeys] = await this.pipeline([
+      ["KEYS", `${this.prefix}:agent:*`],
+      ["KEYS", `${this.prefix}:thread:*`]
+    ]);
+    return {
+      agents: agentKeys?.length || 0,
+      threads: threadKeys?.length || 0
+    };
+  }
+};
+
+export {
+  UpstashStorage
+};