@axiom-lattice/gateway 3.0.2 → 3.0.4

package/src/controllers/memory.ts

@@ -1,257 +0,0 @@
-import { FastifyRequest, FastifyReply } from "fastify";
-import { agentInstanceManager } from "@axiom-lattice/core";
-
-// 设置内存项
-export const setMemoryItem = async (
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<void> => {
-  try {
-    const { assistantId, key } = request.params as {
-      assistantId: string;
-      key: string;
-    };
-    const value = request.body;
-
-    if (!assistantId || !key) {
-      reply.status(400).send({
-        success: false,
-        error: "助手ID和键是必需的",
-      });
-      return;
-    }
-
-    if (value === undefined) {
-      reply.status(400).send({
-        success: false,
-        error: "值是必需的",
-      });
-      return;
-    }
-
-    //const result = await memoryModel.setMemoryItem(assistantId, key, value);
-
-    // if (!result.success) {
-    reply.status(500).send();
-    return;
-    // }
-
-    //reply.send(result);
-  } catch (error: any) {
-    reply.status(500).send({
-      success: false,
-      error: `设置内存项时发生错误: ${error.message}`,
-    });
-  }
-};
-
-// 获取内存项
-export const getMemoryItem = async (
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<void> => {
-  try {
-    const { assistantId, key } = request.params as {
-      assistantId: string;
-      key: string;
-    };
-
-    if (!assistantId || !key) {
-      reply.status(400).send({
-        success: false,
-        error: "助手ID和键是必需的",
-      });
-      return;
-    }
-
-    //const result = await memoryModel.getMemoryItem(assistantId, key);
-
-    // if (!result.success) {
-    reply.status(404).send();
-    return;
-    // }
-
-    //reply.send(result);
-  } catch (error: any) {
-    reply.status(500).send({
-      success: false,
-      error: `获取内存项时发生错误: ${error.message}`,
-    });
-  }
-};
-
-// 获取所有内存项
-export const getAllMemoryItems = async (
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<void> => {
-  try {
-    const { assistantId, thread_id } = request.params as {
-      assistantId: string;
-      thread_id: string;
-    };
-    const tenant_id = request.headers["x-tenant-id"] as string;
-
-    if (!thread_id) {
-      reply.status(400).send({
-        success: false,
-        error: "线程ID是必需的",
-      });
-      return;
-    }
-
-    if (!assistantId) {
-      reply.status(400).send({
-        success: false,
-        error: "助手ID是必需的",
-      });
-      return;
-    }
-
-    const agent = agentInstanceManager.getAgent({ assistant_id: assistantId, tenant_id, thread_id: thread_id })
-
-    const result = await agent.getCurrentMessages();
-
-    if (!result) {
-      reply.status(500).send(result);
-      return;
-    }
-
-    reply.send(result);
-  } catch (error: any) {
-    reply.status(500).send({
-      success: false,
-      error: `获取所有内存项时发生错误: ${error.message}`,
-    });
-  }
-};
-
-// 获取助手状态
-export const getAgentState = async (
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<void> => {
-  try {
-    const { assistantId, thread_id } = request.params as {
-      assistantId: string;
-      thread_id: string;
-    };
-
-    if (!thread_id) {
-      reply.status(400).send({
-        success: false,
-        error: "线程ID是必需的",
-      });
-      return;
-    }
-
-    if (!assistantId) {
-      reply.status(400).send({
-        success: false,
-        error: "助手ID是必需的",
-      });
-      return;
-    }
-
-    const tenant_id = request.headers["x-tenant-id"] as string;
-    const agent = agentInstanceManager.getAgent({ assistant_id: assistantId, tenant_id, thread_id: thread_id })
-
-    // 1. 获取 LangGraph 状态（现有逻辑）
-    const result = await agent.getCurrentState();
-    const pendingMessages = await agent.getPendingMessages();
-
-
-    if (!result) {
-      reply.status(500).send(result);
-      return;
-    }
-
-    // 3. 合并返回
-    // Normalize pendingMessages content to string for consistent frontend format
-    const normalizedPendingMessages = pendingMessages.map(msg => ({
-      ...msg,
-      content: typeof msg.content === 'object' && msg.content !== null
-        ? ('message' in msg.content ? msg.content.message : JSON.stringify(msg.content))
-        : String(msg.content || '')
-    }));
-
-    const mergedResult = {
-      ...result,
-      pendingMessages: normalizedPendingMessages
-    };
-
-    reply.send(mergedResult);
-  } catch (error: any) {
-    reply.status(500).send({
-      success: false,
-      error: `获取助手状态时发生错误: ${error.message}`,
-    });
-  }
-};
-
-// 删除内存项
-export const deleteMemoryItem = async (
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<void> => {
-  try {
-    const { assistantId, key } = request.params as {
-      assistantId: string;
-      key: string;
-    };
-
-    if (!assistantId || !key) {
-      reply.status(400).send({
-        success: false,
-        error: "助手ID和键是必需的",
-      });
-      return;
-    }
-
-    //const result = await memoryModel.deleteMemoryItem(assistantId, key);
-
-    // if (!result.success) {
-    reply.status(500).send();
-    return;
-    // }
-
-    reply.status(204).send();
-  } catch (error: any) {
-    reply.status(500).send({
-      success: false,
-      error: `删除内存项时发生错误: ${error.message}`,
-    });
-  }
-};
-
-// 清除内存
-export const clearMemory = async (
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<void> => {
-  try {
-    const { assistantId } = request.params as { assistantId: string };
-
-    if (!assistantId) {
-      reply.status(400).send({
-        success: false,
-        error: "助手ID是必需的",
-      });
-      return;
-    }
-
-    let result: any; //= await memoryModel.clearMemory(assistantId);
-
-    if (!result.success) {
-      reply.status(500).send(result);
-      return;
-    }
-
-    reply.status(204).send();
-  } catch (error: any) {
-    reply.status(500).send({
-      success: false,
-      error: `清除内存时发生错误: ${error.message}`,
-    });
-  }
-};

package/src/controllers/menu-items.ts

@@ -1,114 +0,0 @@
-import type { FastifyRequest, FastifyReply } from "fastify";
-import type { CreateMenuItemInput } from "@axiom-lattice/protocols";
-import { getMenuRegistry } from "@axiom-lattice/core";
-
-function getTenantId(request: FastifyRequest): string {
-  const userTenantId = (request as any).user?.tenantId;
-  if (userTenantId) return userTenantId;
-  return (request.headers["x-tenant-id"] as string) || "default";
-}
-
-function errorMessage(err: unknown): string {
-  return err instanceof Error ? err.message : "Unexpected error";
-}
-
-export async function getMenuItemList(
-  request: FastifyRequest<{
-    Querystring: { menuTarget?: string };
-  }>,
-  _reply: FastifyReply,
-) {
-  const tenantId = getTenantId(request);
-  try {
-    const registry = getMenuRegistry();
-    const items = await registry.list({ tenantId, menuTarget: request.query.menuTarget as "sidebar" | "workspace" | undefined });
-    return { success: true, message: "Menu items retrieved", data: { records: items, total: items.length } };
-  } catch (error) {
-    const msg = errorMessage(error);
-    console.error("Failed to get menu items:", msg);
-    return { success: false, message: msg, data: { records: [], total: 0 } };
-  }
-}
-
-export async function getMenuItem(
-  request: FastifyRequest<{ Params: { id: string } }>,
-  reply: FastifyReply,
-) {
-  const tenantId = getTenantId(request);
-  try {
-    const registry = getMenuRegistry();
-    const item = await registry.getById(request.params.id);
-    if (!item || item.tenantId !== tenantId) {
-      reply.status(404);
-      return { success: false, message: "Menu item not found" };
-    }
-    return { success: true, message: "Menu item retrieved", data: item };
-  } catch (error) {
-    const msg = errorMessage(error);
-    console.error("Failed to get menu item:", msg);
-    reply.status(500);
-    return { success: false, message: msg };
-  }
-}
-
-export async function createMenuItem(
-  request: FastifyRequest<{ Body: CreateMenuItemInput }>,
-  reply: FastifyReply,
-) {
-  const tenantId = getTenantId(request);
-  try {
-    const registry = getMenuRegistry();
-    const item = await registry.create({ ...request.body, tenantId });
-    reply.status(201);
-    return { success: true, message: "Menu item created", data: item };
-  } catch (error) {
-    const msg = errorMessage(error);
-    console.error("Failed to create menu item:", msg);
-    reply.status(500);
-    return { success: false, message: msg };
-  }
-}
-
-export async function updateMenuItem(
-  request: FastifyRequest<{ Params: { id: string }; Body: Partial<CreateMenuItemInput & { enabled: boolean }> }>,
-  reply: FastifyReply,
-) {
-  try {
-    const tenantId = getTenantId(request);
-    const registry = getMenuRegistry();
-    const existing = await registry.getById(request.params.id);
-    if (!existing || existing.tenantId !== tenantId) {
-      reply.status(404);
-      return { success: false, message: "Menu item not found" };
-    }
-    const item = await registry.update(request.params.id, request.body);
-    return { success: true, message: "Menu item updated", data: item };
-  } catch (error) {
-    const msg = errorMessage(error);
-    console.error("Failed to update menu item:", msg);
-    reply.status(500);
-    return { success: false, message: msg };
-  }
-}
-
-export async function deleteMenuItem(
-  request: FastifyRequest<{ Params: { id: string } }>,
-  reply: FastifyReply,
-) {
-  try {
-    const tenantId = getTenantId(request);
-    const registry = getMenuRegistry();
-    const existing = await registry.getById(request.params.id);
-    if (!existing || existing.tenantId !== tenantId) {
-      reply.status(404);
-      return { success: false, message: "Menu item not found" };
-    }
-    await registry.delete(request.params.id);
-    return { success: true, message: "Menu item deleted" };
-  } catch (error) {
-    const msg = errorMessage(error);
-    console.error("Failed to delete menu item:", msg);
-    reply.status(500);
-    return { success: false, message: msg };
-  }
-}