@axiom-lattice/gateway 2.1.15 → 2.1.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/gateway",
-  "version": "2.1.15",
+  "version": "2.1.16",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -32,9 +32,9 @@
     "pino-roll": "^3.1.0",
     "redis": "^5.0.1",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "2.1.11",
-    "@axiom-lattice/queue-redis": "1.0.4",
-    "@axiom-lattice/protocols": "2.1.5"
+    "@axiom-lattice/core": "2.1.12",
+    "@axiom-lattice/protocols": "2.1.6",
+    "@axiom-lattice/queue-redis": "1.0.5"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/controllers/assistant.ts

@@ -1,7 +1,10 @@
 import { FastifyRequest, FastifyReply } from "fastify";
 import { draw_graph } from "../services/agent_service";
-import { assistantStore } from "../stores/assistant_store";
-import { Assistant, CreateAssistantRequest } from "../types";
+import { getStoreLattice } from "@axiom-lattice/core";
+import type {
+  Assistant,
+  CreateAssistantRequest,
+} from "@axiom-lattice/protocols";
 import { randomUUID } from "crypto";
 import { AgentConfig, getAllAgentConfigs } from "@axiom-lattice/core";
 
@@ -72,7 +75,9 @@ export async function getAssistantList(
   );
 
   // Get stored assistants
-  const storedAssistants = assistantStore.getAllAssistants();
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+  const storedAssistants = await assistantStore.getAllAssistants();
 
   // Merge both sources, stored assistants take precedence if ID conflicts
   const assistantMap = new Map<string, Assistant>();
@@ -110,7 +115,9 @@ export async function getAssistant(
   const { id } = request.params;
 
   // First check stored assistants
-  let assistant = assistantStore.getAssistantById(id);
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+  let assistant = await assistantStore.getAssistantById(id);
 
   // If not found in store, check code-configured agents
   if (!assistant) {
@@ -163,7 +170,9 @@ export async function createAssistant(
   const id = randomUUID();
 
   // Create assistant
-  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,
@@ -198,8 +207,12 @@ export async function updateAssistant(
     });
   }
 
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+
   // Check if assistant exists in store
-  if (!assistantStore.hasAssistant(id)) {
+  const exists = await assistantStore.hasAssistant(id);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Assistant not found",
@@ -207,7 +220,7 @@ export async function updateAssistant(
   }
 
   // Update assistant
-  const updatedAssistant = assistantStore.updateAssistant(id, updates);
+  const updatedAssistant = await assistantStore.updateAssistant(id, updates);
 
   if (!updatedAssistant) {
     return reply.status(500).send({
@@ -245,8 +258,12 @@ export async function deleteAssistant(
     });
   }
 
+  const storeLattice = getStoreLattice("default", "assistant");
+  const assistantStore = storeLattice.store;
+
   // Check if assistant exists in store
-  if (!assistantStore.hasAssistant(id)) {
+  const exists = await assistantStore.hasAssistant(id);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Assistant not found",
@@ -254,7 +271,7 @@ export async function deleteAssistant(
   }
 
   // Delete the assistant
-  const deleted = assistantStore.deleteAssistant(id);
+  const deleted = await assistantStore.deleteAssistant(id);
 
   if (!deleted) {
     return reply.status(500).send({

package/src/controllers/run.ts

@@ -129,6 +129,7 @@ export const resumeStream = async (
 
       // Stream the chunks to the client
       for await (const chunk of stream) {
+        // console.log("resume stream raw.write", chunk);
         reply.raw.write(`data: ${JSON.stringify(chunk)}\n\n`);
       }
     } catch (error) {

package/src/controllers/threads.ts

@@ -1,5 +1,5 @@
 import { FastifyRequest, FastifyReply } from "fastify";
-import { threadStore } from "../stores/thread_store";
+import { getStoreLattice } from "@axiom-lattice/core";
 import { Thread, CreateThreadRequest } from "../types";
 import { randomUUID } from "crypto";
 
@@ -46,7 +46,9 @@ export async function getThreadList(
 ): Promise<ThreadListResponse> {
   const { assistantId } = request.params;
 
-  const threads = threadStore.getThreadsByAssistantId(assistantId);
+  const storeLattice = getStoreLattice("default", "thread");
+  const threadStore = storeLattice.store;
+  const threads = await threadStore.getThreadsByAssistantId(assistantId);
 
   return {
     success: true,
@@ -69,7 +71,9 @@ export async function getThread(
 ): Promise<ThreadResponse> {
   const { assistantId, threadId } = request.params;
 
-  const thread = threadStore.getThreadById(assistantId, threadId);
+  const storeLattice = getStoreLattice("default", "thread");
+  const threadStore = storeLattice.store;
+  const thread = await threadStore.getThreadById(assistantId, threadId);
 
   if (!thread) {
     return reply.status(404).send({
@@ -102,7 +106,9 @@ export async function createThread(
   const threadId = randomUUID();
 
   // Create thread
-  const newThread = threadStore.createThread(assistantId, threadId, data);
+  const storeLattice = getStoreLattice("default", "thread");
+  const threadStore = storeLattice.store;
+  const newThread = await threadStore.createThread(assistantId, threadId, data);
 
   return reply.status(201).send({
     success: true,
@@ -124,8 +130,12 @@ export async function updateThread(
   const { assistantId, threadId } = request.params;
   const updates = request.body;
 
+  const storeLattice = getStoreLattice("default", "thread");
+  const threadStore = storeLattice.store;
+
   // Check if thread exists
-  if (!threadStore.hasThread(assistantId, threadId)) {
+  const exists = await threadStore.hasThread(assistantId, threadId);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Thread not found",
@@ -133,7 +143,7 @@ export async function updateThread(
   }
 
   // Update thread
-  const updatedThread = threadStore.updateThread(
+  const updatedThread = await threadStore.updateThread(
     assistantId,
     threadId,
     updates
@@ -164,8 +174,12 @@ export async function deleteThread(
 ): Promise<{ success: boolean; message: string }> {
   const { assistantId, threadId } = request.params;
 
+  const storeLattice = getStoreLattice("default", "thread");
+  const threadStore = storeLattice.store;
+
   // Check if thread exists
-  if (!threadStore.hasThread(assistantId, threadId)) {
+  const exists = await threadStore.hasThread(assistantId, threadId);
+  if (!exists) {
     return reply.status(404).send({
       success: false,
       message: "Thread not found",
@@ -173,7 +187,7 @@ export async function deleteThread(
   }
 
   // Delete the thread
-  const deleted = threadStore.deleteThread(assistantId, threadId);
+  const deleted = await threadStore.deleteThread(assistantId, threadId);
 
   if (!deleted) {
     return reply.status(500).send({

package/src/types/index.ts

@@ -1,12 +1,8 @@
-// 助手类型
-export interface Assistant {
-  id: string;
-  name: string;
-  description?: string;
-  graphDefinition: any;
-  createdAt: Date;
-  updatedAt: Date;
-}
+// Re-export Assistant types from protocols package
+export type {
+  Assistant,
+  CreateAssistantRequest,
+} from "@axiom-lattice/protocols";
 
 // 运行状态枚举
 export enum RunStatus {
@@ -75,13 +71,6 @@ export interface Webhook {
   updatedAt: Date;
 }
 
-// 创建助手请求类型
-export interface CreateAssistantRequest {
-  name: string;
-  description?: string;
-  graphDefinition: any;
-}
-
 // 创建运行请求类型
 export interface CreateRunRequest {
   tenant_id: string;
@@ -122,16 +111,5 @@ export interface ServiceResponse<T = any> {
   error?: string;
 }
 
-// Thread type
-export interface Thread {
-  id: string;
-  assistantId: string;
-  metadata?: Record<string, any>;
-  createdAt: Date;
-  updatedAt: Date;
-}
-
-// Create thread request type
-export interface CreateThreadRequest {
-  metadata?: Record<string, any>;
-}
+// Re-export Thread types from protocols package
+export type { Thread, CreateThreadRequest } from "@axiom-lattice/protocols";

package/src/stores/assistant_store.ts

@@ -1,83 +0,0 @@
-import { Assistant, CreateAssistantRequest } from "../types";
-
-/**
- * In-memory store for assistants
- * Provides CRUD operations for assistant data
- */
-class AssistantStore {
-  private assistants: Map<string, Assistant> = new Map();
-
-  /**
-   * Get all assistants
-   */
-  getAllAssistants(): Assistant[] {
-    return Array.from(this.assistants.values());
-  }
-
-  /**
-   * Get assistant by ID
-   */
-  getAssistantById(id: string): Assistant | undefined {
-    return this.assistants.get(id);
-  }
-
-  /**
-   * Create a new assistant
-   */
-  createAssistant(
-    id: string,
-    data: CreateAssistantRequest
-  ): Assistant {
-    const now = new Date();
-    const assistant: 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: string,
-    updates: Partial<CreateAssistantRequest>
-  ): Assistant | null {
-    const existing = this.assistants.get(id);
-    if (!existing) {
-      return null;
-    }
-
-    const updated: Assistant = {
-      ...existing,
-      ...updates,
-      updatedAt: new Date(),
-    };
-    this.assistants.set(id, updated);
-    return updated;
-  }
-
-  /**
-   * Delete an assistant by ID
-   */
-  deleteAssistant(id: string): boolean {
-    return this.assistants.delete(id);
-  }
-
-  /**
-   * Check if assistant exists
-   */
-  hasAssistant(id: string): boolean {
-    return this.assistants.has(id);
-  }
-}
-
-// Export singleton instance
-export const assistantStore = new AssistantStore();
-
-

package/src/stores/thread_store.ts

@@ -1,115 +0,0 @@
-import { Thread, CreateThreadRequest } from "../types";
-
-/**
- * In-memory store for threads
- * Provides CRUD operations for thread data
- * Threads are organized by assistant ID
- */
-class ThreadStore {
-  // Map<assistantId, Map<threadId, Thread>>
-  private threads: Map<string, Map<string, Thread>> = new Map();
-
-  /**
-   * Get all threads for a specific assistant
-   */
-  getThreadsByAssistantId(assistantId: string): Thread[] {
-    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: string, threadId: string): Thread | undefined {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return undefined;
-    }
-    return assistantThreads.get(threadId);
-  }
-
-  /**
-   * Create a new thread for an assistant
-   */
-  createThread(
-    assistantId: string,
-    threadId: string,
-    data: CreateThreadRequest
-  ): Thread {
-    const now = new Date();
-    const thread: Thread = {
-      id: threadId,
-      assistantId,
-      metadata: data.metadata || {},
-      createdAt: now,
-      updatedAt: now,
-    };
-
-    // Initialize assistant's thread map if it doesn't exist
-    if (!this.threads.has(assistantId)) {
-      this.threads.set(assistantId, new Map());
-    }
-
-    const assistantThreads = this.threads.get(assistantId)!;
-    assistantThreads.set(threadId, thread);
-    return thread;
-  }
-
-  /**
-   * Update an existing thread
-   */
-  updateThread(
-    assistantId: string,
-    threadId: string,
-    updates: Partial<CreateThreadRequest>
-  ): Thread | null {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return null;
-    }
-
-    const existing = assistantThreads.get(threadId);
-    if (!existing) {
-      return null;
-    }
-
-    const updated: Thread = {
-      ...existing,
-      metadata: {
-        ...existing.metadata,
-        ...(updates.metadata || {}),
-      },
-      updatedAt: new Date(),
-    };
-    assistantThreads.set(threadId, updated);
-    return updated;
-  }
-
-  /**
-   * Delete a thread by ID
-   */
-  deleteThread(assistantId: string, threadId: string): boolean {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return false;
-    }
-    return assistantThreads.delete(threadId);
-  }
-
-  /**
-   * Check if thread exists
-   */
-  hasThread(assistantId: string, threadId: string): boolean {
-    const assistantThreads = this.threads.get(assistantId);
-    if (!assistantThreads) {
-      return false;
-    }
-    return assistantThreads.has(threadId);
-  }
-}
-
-// Export singleton instance
-export const threadStore = new ThreadStore();