@snap-agent/core 0.1.4 → 0.1.5

package/dist/chunk-5BAE6S5E.js

@@ -0,0 +1,382 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }// 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 (e) {
+      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: _optionalChain([agentKeys, 'optionalAccess', _ => _.length]) || 0,
+      threads: _optionalChain([threadKeys, 'optionalAccess', _2 => _2.length]) || 0
+    };
+  }
+};
+
+
+
+exports.UpstashStorage = UpstashStorage;

package/dist/chunk-BSSVNLSO.mjs

@@ -0,0 +1,157 @@
+// src/storage/MemoryStorage.ts
+var MemoryStorage = class {
+  constructor() {
+    this.agents = /* @__PURE__ */ new Map();
+    this.threads = /* @__PURE__ */ new Map();
+    this.idCounter = 0;
+  }
+  generateId() {
+    return `${Date.now()}-${++this.idCounter}`;
+  }
+  // ============================================================================
+  // Agent Operations
+  // ============================================================================
+  async createAgent(config) {
+    const id = this.generateId();
+    const agent = {
+      id,
+      ...config,
+      createdAt: /* @__PURE__ */ new Date(),
+      updatedAt: /* @__PURE__ */ new Date(),
+      files: []
+    };
+    this.agents.set(id, agent);
+    return id;
+  }
+  async getAgent(agentId) {
+    return this.agents.get(agentId) || null;
+  }
+  async updateAgent(agentId, updates) {
+    const agent = this.agents.get(agentId);
+    if (!agent) return;
+    Object.assign(agent, updates, { updatedAt: /* @__PURE__ */ new Date() });
+    this.agents.set(agentId, agent);
+  }
+  async deleteAgent(agentId) {
+    this.agents.delete(agentId);
+    const threadsToDelete = [];
+    for (const [threadId, thread] of this.threads.entries()) {
+      if (thread.agentId === agentId) {
+        threadsToDelete.push(threadId);
+      }
+    }
+    for (const threadId of threadsToDelete) {
+      this.threads.delete(threadId);
+    }
+  }
+  async listAgents(userId, organizationId) {
+    const agents = Array.from(this.agents.values()).filter((agent) => {
+      if (agent.userId !== userId) return false;
+      if (organizationId && agent.organizationId !== organizationId) return false;
+      return true;
+    });
+    return agents.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
+  }
+  // ============================================================================
+  // Thread Operations
+  // ============================================================================
+  async createThread(config) {
+    const id = this.generateId();
+    const thread = {
+      id,
+      ...config,
+      createdAt: /* @__PURE__ */ new Date(),
+      updatedAt: /* @__PURE__ */ new Date(),
+      messages: [],
+      isPendingThread: true
+    };
+    this.threads.set(id, thread);
+    return id;
+  }
+  async getThread(threadId) {
+    return this.threads.get(threadId) || null;
+  }
+  async updateThread(threadId, updates) {
+    const thread = this.threads.get(threadId);
+    if (!thread) return;
+    Object.assign(thread, updates, { updatedAt: /* @__PURE__ */ new Date() });
+    this.threads.set(threadId, thread);
+  }
+  async deleteThread(threadId) {
+    this.threads.delete(threadId);
+  }
+  async listThreads(filters) {
+    const threads = Array.from(this.threads.values()).filter((thread) => {
+      if (filters.userId && thread.userId !== filters.userId) return false;
+      if (filters.agentId && thread.agentId !== filters.agentId) return false;
+      if (filters.organizationId && thread.organizationId !== filters.organizationId) return false;
+      return true;
+    });
+    return threads.sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
+  }
+  // ============================================================================
+  // Message Operations
+  // ============================================================================
+  async addMessage(threadId, role, content, attachments) {
+    const thread = this.threads.get(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);
+    thread.updatedAt = /* @__PURE__ */ new Date();
+    this.threads.set(threadId, thread);
+    return messageId;
+  }
+  async getMessages(threadId, limit) {
+    const thread = this.threads.get(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
+    }));
+  }
+  // ============================================================================
+  // Utility Methods
+  // ============================================================================
+  /**
+   * Clear all stored data
+   */
+  clear() {
+    this.agents.clear();
+    this.threads.clear();
+    this.idCounter = 0;
+  }
+  /**
+   * Get statistics about stored data
+   */
+  getStats() {
+    return {
+      agents: this.agents.size,
+      threads: this.threads.size,
+      messages: Array.from(this.threads.values()).reduce(
+        (sum, thread) => sum + thread.messages.length,
+        0
+      )
+    };
+  }
+};
+
+export {
+  MemoryStorage
+};