@snap-agent/core 0.1.3 → 0.1.5

package/dist/chunk-FS7G3ID4.mjs

@@ -1,803 +0,0 @@
-// 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 || {},
-      pluginConfigs: config.pluginConfigs || []
-    };
-    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,
-      pluginConfigs: doc.pluginConfigs
-    };
-  }
-  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
-};