@snap-agent/core 0.1.0

package/dist/chunk-Y5TTFQWC.mjs

@@ -0,0 +1,801 @@
+// src/storage/MongoDBStorage.ts
+import { MongoClient, ObjectId } from "mongodb";
+var MongoDBStorage = class {
+  constructor(config) {
+    this.db = null;
+    if (typeof config === "string") {
+      this.config = {
+        uri: config,
+        dbName: "agentStudio",
+        agentsCollection: "v2_agents",
+        threadsCollection: "v2_threads"
+      };
+    } else {
+      this.config = {
+        uri: config.uri,
+        dbName: config.dbName || "agentStudio",
+        agentsCollection: config.agentsCollection || "v2_agents",
+        threadsCollection: config.threadsCollection || "v2_threads"
+      };
+    }
+    this.client = new MongoClient(this.config.uri);
+  }
+  async ensureConnection() {
+    if (!this.db) {
+      await this.client.connect();
+      this.db = this.client.db(this.config.dbName);
+    }
+    return this.db;
+  }
+  async disconnect() {
+    await this.client.close();
+    this.db = null;
+  }
+  // ============================================================================
+  // Agent Operations
+  // ============================================================================
+  async createAgent(config) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.agentsCollection
+    );
+    const doc = {
+      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: /* @__PURE__ */ new Date(),
+      updatedAt: /* @__PURE__ */ new Date(),
+      files: [],
+      metadata: config.metadata || {}
+    };
+    const result = await collection.insertOne(doc);
+    return result.insertedId.toString();
+  }
+  async getAgent(agentId) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.agentsCollection
+    );
+    const doc = await collection.findOne({ _id: new ObjectId(agentId) });
+    if (!doc) return null;
+    return this.agentDocToData(doc);
+  }
+  async updateAgent(agentId, updates) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.agentsCollection
+    );
+    await collection.updateOne(
+      { _id: new ObjectId(agentId) },
+      {
+        $set: {
+          ...updates,
+          updatedAt: /* @__PURE__ */ new Date()
+        }
+      }
+    );
+  }
+  async deleteAgent(agentId) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.agentsCollection
+    );
+    await collection.deleteOne({ _id: new ObjectId(agentId) });
+  }
+  async listAgents(userId, organizationId) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.agentsCollection
+    );
+    const query = { userId };
+    if (organizationId) {
+      query.organizationId = organizationId;
+    }
+    const docs = await collection.find(query).sort({ updatedAt: -1 }).toArray();
+    return docs.map((doc) => this.agentDocToData(doc));
+  }
+  // ============================================================================
+  // Thread Operations
+  // ============================================================================
+  async createThread(config) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.threadsCollection
+    );
+    const doc = {
+      organizationId: config.organizationId,
+      agentId: config.agentId,
+      userId: config.userId,
+      endUserId: config.endUserId,
+      name: config.name,
+      createdAt: /* @__PURE__ */ new Date(),
+      updatedAt: /* @__PURE__ */ new Date(),
+      messages: [],
+      isPendingThread: true,
+      metadata: config.metadata || {}
+    };
+    const result = await collection.insertOne(doc);
+    return result.insertedId.toString();
+  }
+  async getThread(threadId) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.threadsCollection
+    );
+    const doc = await collection.findOne({ _id: new ObjectId(threadId) });
+    if (!doc) return null;
+    return this.threadDocToData(doc);
+  }
+  async updateThread(threadId, updates) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.threadsCollection
+    );
+    await collection.updateOne(
+      { _id: new ObjectId(threadId) },
+      {
+        $set: {
+          ...updates,
+          updatedAt: /* @__PURE__ */ new Date()
+        }
+      }
+    );
+  }
+  async deleteThread(threadId) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.threadsCollection
+    );
+    await collection.deleteOne({ _id: new ObjectId(threadId) });
+  }
+  async listThreads(filters) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.threadsCollection
+    );
+    const query = {};
+    if (filters.userId) query.userId = filters.userId;
+    if (filters.agentId) query.agentId = filters.agentId;
+    if (filters.organizationId) query.organizationId = filters.organizationId;
+    const docs = await collection.find(query).sort({ updatedAt: -1 }).toArray();
+    return docs.map((doc) => this.threadDocToData(doc));
+  }
+  // ============================================================================
+  // Message Operations
+  // ============================================================================
+  async addMessage(threadId, role, content, attachments) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.threadsCollection
+    );
+    const messageId = new ObjectId();
+    const message = {
+      _id: messageId,
+      role,
+      content,
+      timestamp: /* @__PURE__ */ new Date(),
+      attachments
+    };
+    await collection.updateOne(
+      { _id: new ObjectId(threadId) },
+      {
+        $push: { messages: message },
+        $set: { updatedAt: /* @__PURE__ */ new Date() }
+      }
+    );
+    return messageId.toString();
+  }
+  async getMessages(threadId, limit) {
+    const db = await this.ensureConnection();
+    const collection = db.collection(
+      this.config.threadsCollection
+    );
+    const doc = await collection.findOne({ _id: new ObjectId(threadId) });
+    if (!doc) return [];
+    let messages = doc.messages.map((msg) => ({
+      id: msg._id?.toString() || "",
+      role: msg.role,
+      content: msg.content,
+      timestamp: msg.timestamp,
+      metadata: msg.metadata,
+      attachments: msg.attachments
+    }));
+    if (limit) {
+      messages = 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
+  // ============================================================================
+  agentDocToData(doc) {
+    return {
+      id: doc._id.toString(),
+      organizationId: doc.organizationId,
+      userId: doc.userId,
+      phone: doc.phone,
+      name: doc.name,
+      description: doc.description,
+      instructions: doc.instructions,
+      provider: doc.provider,
+      model: doc.model,
+      createdAt: doc.createdAt,
+      updatedAt: doc.updatedAt,
+      files: doc.files,
+      metadata: doc.metadata
+    };
+  }
+  threadDocToData(doc) {
+    return {
+      id: doc._id.toString(),
+      organizationId: doc.organizationId,
+      agentId: doc.agentId,
+      userId: doc.userId,
+      endUserId: doc.endUserId,
+      name: doc.name,
+      createdAt: doc.createdAt,
+      updatedAt: doc.updatedAt,
+      messages: doc.messages.map((msg) => ({
+        id: msg._id?.toString() || "",
+        role: msg.role,
+        content: msg.content,
+        timestamp: msg.timestamp,
+        metadata: msg.metadata,
+        attachments: msg.attachments
+      })),
+      isPendingThread: doc.isPendingThread,
+      metadata: doc.metadata
+    };
+  }
+};
+
+// 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
+      )
+    };
+  }
+};
+
+// 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 {
+  MongoDBStorage,
+  MemoryStorage,
+  UpstashStorage
+};