@axiom-lattice/gateway 2.1.15 → 2.1.16

package/dist/index.mjs

@@ -223,71 +223,8 @@ async function resume_stream({
   };
 }
 
-// src/stores/assistant_store.ts
-var AssistantStore = class {
-  constructor() {
-    this.assistants = /* @__PURE__ */ new Map();
-  }
-  /**
-   * Get all assistants
-   */
-  getAllAssistants() {
-    return Array.from(this.assistants.values());
-  }
-  /**
-   * Get assistant by ID
-   */
-  getAssistantById(id) {
-    return this.assistants.get(id);
-  }
-  /**
-   * Create a new assistant
-   */
-  createAssistant(id, data) {
-    const now = /* @__PURE__ */ new Date();
-    const assistant = {
-      id,
-      name: data.name,
-      description: data.description,
-      graphDefinition: data.graphDefinition,
-      createdAt: now,
-      updatedAt: now
-    };
-    this.assistants.set(id, assistant);
-    return assistant;
-  }
-  /**
-   * Update an existing assistant
-   */
-  updateAssistant(id, updates) {
-    const existing = this.assistants.get(id);
-    if (!existing) {
-      return null;
-    }
-    const updated = {
-      ...existing,
-      ...updates,
-      updatedAt: /* @__PURE__ */ new Date()
-    };
-    this.assistants.set(id, updated);
-    return updated;
-  }
-  /**
-   * Delete an assistant by ID
-   */
-  deleteAssistant(id) {
-    return this.assistants.delete(id);
-  }
-  /**
-   * Check if assistant exists
-   */
-  hasAssistant(id) {
-    return this.assistants.has(id);
-  }
-};
-var assistantStore = new AssistantStore();
-
 // src/controllers/assistant.ts
+import { getStoreLattice } from "@axiom-lattice/core";
 import { randomUUID } from "crypto";
 import { getAllAgentConfigs } from "@axiom-lattice/core";
 function convertAgentConfigToAssistant(config) {
@@ -308,7 +245,9 @@ async function getAssistantList(request, reply) {
   const codeConfiguredAssistants = agentConfigs.map(
     convertAgentConfigToAssistant
   );
-  const storedAssistants = assistantStore.getAllAssistants();
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const storedAssistants = await assistantStore.getAllAssistants();
   const assistantMap = /* @__PURE__ */ new Map();
   codeConfiguredAssistants.forEach((assistant) => {
     assistantMap.set(assistant.id, assistant);
@@ -328,7 +267,9 @@ async function getAssistantList(request, reply) {
 }
 async function getAssistant(request, reply) {
   const { id } = request.params;
-  let assistant = assistantStore.getAssistantById(id);
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+  let assistant = await assistantStore.getAssistantById(id);
   if (!assistant) {
     const agentConfigs = await getAllAgentConfigs();
     const agentConfig = agentConfigs.find((config) => config.key === id);
@@ -363,7 +304,9 @@ async function createAssistant(request, reply) {
     });
   }
   const id = randomUUID();
-  const newAssistant = assistantStore.createAssistant(id, data);
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const newAssistant = await assistantStore.createAssistant(id, data);
   return reply.status(201).send({
     success: true,
     message: "Successfully created assistant",
@@ -381,13 +324,16 @@ async function updateAssistant(request, reply) {
       message: "Cannot update code-configured assistant. Only stored assistants can be updated."
     });
   }
-  if (!assistantStore.hasAssistant(id)) {
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const exists = await assistantStore.hasAssistant(id);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Assistant not found"
     });
   }
-  const updatedAssistant = assistantStore.updateAssistant(id, updates);
+  const updatedAssistant = await assistantStore.updateAssistant(id, updates);
   if (!updatedAssistant) {
     return reply.status(500).send({
       success: false,
@@ -410,13 +356,16 @@ async function deleteAssistant(request, reply) {
       message: "Cannot delete code-configured assistant. Only stored assistants can be deleted."
     });
   }
-  if (!assistantStore.hasAssistant(id)) {
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const exists = await assistantStore.hasAssistant(id);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Assistant not found"
     });
   }
-  const deleted = assistantStore.deleteAssistant(id);
+  const deleted = await assistantStore.deleteAssistant(id);
   if (!deleted) {
     return reply.status(500).send({
       success: false,
@@ -749,102 +698,14 @@ var triggerAgentTask = async (request, reply) => {
   }
 };
 
-// src/stores/thread_store.ts
-var ThreadStore = class {
-  constructor() {
-    // Map<assistantId, Map<threadId, Thread>>
-    this.threads = /* @__PURE__ */ new Map();
-  }
-  /**
-   * Get all threads for a specific assistant
-   */
-  getThreadsByAssistantId(assistantId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return [];
-    }
-    return Array.from(assistantThreads.values());
-  }
-  /**
-   * Get a thread by ID for a specific assistant
-   */
-  getThreadById(assistantId, threadId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return void 0;
-    }
-    return assistantThreads.get(threadId);
-  }
-  /**
-   * Create a new thread for an assistant
-   */
-  createThread(assistantId, threadId, data) {
-    const now = /* @__PURE__ */ new Date();
-    const thread = {
-      id: threadId,
-      assistantId,
-      metadata: data.metadata || {},
-      createdAt: now,
-      updatedAt: now
-    };
-    if (!this.threads.has(assistantId)) {
-      this.threads.set(assistantId, /* @__PURE__ */ new Map());
-    }
-    const assistantThreads = this.threads.get(assistantId);
-    assistantThreads.set(threadId, thread);
-    return thread;
-  }
-  /**
-   * Update an existing thread
-   */
-  updateThread(assistantId, threadId, updates) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return null;
-    }
-    const existing = assistantThreads.get(threadId);
-    if (!existing) {
-      return null;
-    }
-    const updated = {
-      ...existing,
-      metadata: {
-        ...existing.metadata,
-        ...updates.metadata || {}
-      },
-      updatedAt: /* @__PURE__ */ new Date()
-    };
-    assistantThreads.set(threadId, updated);
-    return updated;
-  }
-  /**
-   * Delete a thread by ID
-   */
-  deleteThread(assistantId, threadId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return false;
-    }
-    return assistantThreads.delete(threadId);
-  }
-  /**
-   * Check if thread exists
-   */
-  hasThread(assistantId, threadId) {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return false;
-    }
-    return assistantThreads.has(threadId);
-  }
-};
-var threadStore = new ThreadStore();
-
 // src/controllers/threads.ts
+import { getStoreLattice as getStoreLattice2 } from "@axiom-lattice/core";
 import { randomUUID as randomUUID2 } from "crypto";
 async function getThreadList(request, reply) {
   const { assistantId } = request.params;
-  const threads = threadStore.getThreadsByAssistantId(assistantId);
+  const storeLattice = getStoreLattice2("default", "thread");
+  const threadStore = storeLattice.store;
+  const threads = await threadStore.getThreadsByAssistantId(assistantId);
   return {
     success: true,
     message: "Successfully retrieved thread list",
@@ -856,7 +717,9 @@ async function getThreadList(request, reply) {
 }
 async function getThread(request, reply) {
   const { assistantId, threadId } = request.params;
-  const thread = threadStore.getThreadById(assistantId, threadId);
+  const storeLattice = getStoreLattice2("default", "thread");
+  const threadStore = storeLattice.store;
+  const thread = await threadStore.getThreadById(assistantId, threadId);
   if (!thread) {
     return reply.status(404).send({
       success: false,
@@ -873,7 +736,9 @@ async function createThread(request, reply) {
   const { assistantId } = request.params;
   const data = request.body;
   const threadId = randomUUID2();
-  const newThread = threadStore.createThread(assistantId, threadId, data);
+  const storeLattice = getStoreLattice2("default", "thread");
+  const threadStore = storeLattice.store;
+  const newThread = await threadStore.createThread(assistantId, threadId, data);
   return reply.status(201).send({
     success: true,
     message: "Successfully created thread",
@@ -883,13 +748,16 @@ async function createThread(request, reply) {
 async function updateThread(request, reply) {
   const { assistantId, threadId } = request.params;
   const updates = request.body;
-  if (!threadStore.hasThread(assistantId, threadId)) {
+  const storeLattice = getStoreLattice2("default", "thread");
+  const threadStore = storeLattice.store;
+  const exists = await threadStore.hasThread(assistantId, threadId);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Thread not found"
     });
   }
-  const updatedThread = threadStore.updateThread(
+  const updatedThread = await threadStore.updateThread(
     assistantId,
     threadId,
     updates
@@ -908,13 +776,16 @@ async function updateThread(request, reply) {
 }
 async function deleteThread(request, reply) {
   const { assistantId, threadId } = request.params;
-  if (!threadStore.hasThread(assistantId, threadId)) {
+  const storeLattice = getStoreLattice2("default", "thread");
+  const threadStore = storeLattice.store;
+  const exists = await threadStore.hasThread(assistantId, threadId);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Thread not found"
     });
   }
-  const deleted = threadStore.deleteThread(assistantId, threadId);
+  const deleted = await threadStore.deleteThread(assistantId, threadId);
   if (!deleted) {
     return reply.status(500).send({
       success: false,