@axiom-lattice/gateway 3.0.2 → 3.0.4

package/src/controllers/middleware-types.ts

@@ -1,15 +0,0 @@
-import type { FastifyInstance } from "fastify";
-import { PluginRegistry } from "@axiom-lattice/core";
-
-/**
- * GET /api/middleware-types
- *
- * Returns all registered plugin metas (builtin + custom).
- * hasTest / hasDiscover are auto-detected by serializePluginMeta.
- */
-export function middlewareTypesRoutes(fastify: FastifyInstance): void {
-  fastify.get("/api/middleware-types", async (_request, reply) => {
-    const metas = PluginRegistry.listMeta();
-    return reply.send(metas);
-  });
-}

package/src/controllers/models.ts

@@ -1,199 +0,0 @@
-import { FastifyRequest, FastifyReply } from "fastify";
-import { registerModelLattice, modelLatticeManager } from "@axiom-lattice/core";
-import type { LLMConfig } from "@axiom-lattice/protocols";
-
-/**
- * Models Controller
- * Handles model lattice registration and management
- * Models are tenant-isolated using key prefix: {tenantId}:{modelKey}
- */
-
-/**
- * Get tenant ID from request headers or user context
- */
-function getTenantId(request: FastifyRequest): string {
-  // First try to get from authenticated user context
-  const userTenantId = (request as any).user?.tenantId;
-  if (userTenantId) {
-    return userTenantId;
-  }
-
-  // Fallback to request headers for backward compatibility
-  return (request.headers["x-tenant-id"] as string) || "default";
-}
-
-/**
- * Get tenant-scoped model key
- */
-function getTenantModelKey(tenantId: string, modelKey: string): string {
-  return `${tenantId}:${modelKey}`;
-}
-
-/**
- * Extract model key from tenant-scoped key
- */
-function extractModelKey(tenantId: string, tenantModelKey: string): string {
-  const prefix = `${tenantId}:`;
-  if (tenantModelKey.startsWith(prefix)) {
-    return tenantModelKey.substring(prefix.length);
-  }
-  return tenantModelKey;
-}
-
-interface ModelConfig {
-  key: string;
-  model: string;
-  displayName?: string;
-  provider: "azure" | "openai" | "deepseek" | "siliconcloud" | "volcengine";
-  streaming?: boolean;
-  apiKey?: string;
-  baseURL?: string;
-  maxTokens?: number;
-  temperature?: number;
-  timeout?: number;
-  maxRetries?: number;
-}
-
-interface UpdateModelsRequest {
-  Body: {
-    models: ModelConfig[];
-  };
-}
-
-interface UpdateModelsRequest {
-  Body: {
-    models: ModelConfig[];
-  };
-}
-
-/**
- * Get all registered models
- */
-export async function getModels(request: FastifyRequest, reply: FastifyReply) {
-  try {
-    const allLattices = modelLatticeManager.getAllLattices();
-    const models = allLattices.map((lattice) => {
-      // Extract config from the lattice client
-      // Note: This is a simplified approach - you may need to adjust based on actual implementation
-      const config = (lattice.client as any).config || {};
-      // Use provided displayName or auto-generate from key
-      const displayName = config.displayName || lattice.key
-        .split('-')
-        .map(word => word.charAt(0).toUpperCase() + word.slice(1))
-        .join(' ');
-      return {
-        key: lattice.key,
-        model: config.model || "",
-        provider: config.provider || "openai",
-        displayName: displayName,
-        streaming: config.streaming || false,
-        apiKey: config.apiKey || "",
-        baseURL: config.baseURL || "",
-        maxTokens: config.maxTokens,
-        temperature: config.temperature,
-        timeout: config.timeout,
-        maxRetries: config.maxRetries,
-      };
-    });
-
-    return reply.send({
-      success: true,
-      data: models,
-    });
-  } catch (error: any) {
-    console.error("Failed to get models", {
-      error: error.message,
-      stack: error.stack,
-    });
-    return reply.status(500).send({
-      success: false,
-      error: error.message || "Failed to get models",
-    });
-  }
-}
-
-/**
- * Update models configuration
- * Registers or updates model lattices
- */
-export async function updateModels(
-  request: FastifyRequest<UpdateModelsRequest>,
-  reply: FastifyReply
-) {
-  try {
-    const { models } = request.body;
-
-    if (!models || !Array.isArray(models)) {
-      return reply.status(400).send({
-        success: false,
-        error: "Invalid request: models must be an array",
-      });
-    }
-
-    const registeredModels: string[] = [];
-    const errors: string[] = [];
-
-    for (const modelConfig of models) {
-      if (!modelConfig.key || !modelConfig.model || !modelConfig.provider) {
-        errors.push(
-          `Model configuration is incomplete: key, model, and provider are required`
-        );
-        continue;
-      }
-
-      try {
-        // Remove existing model with the same key if it exists
-        if (modelLatticeManager.hasLattice(modelConfig.key)) {
-          modelLatticeManager.removeLattice(modelConfig.key);
-        }
-
-        // Convert to LLMConfig format
-        const llmConfig: LLMConfig = {
-          provider: modelConfig.provider,
-          model: modelConfig.model,
-          displayName: modelConfig.displayName,
-          streaming: modelConfig.streaming ?? false,
-          apiKey: modelConfig.apiKey,
-          baseURL: modelConfig.baseURL,
-          maxTokens: modelConfig.maxTokens,
-          temperature: modelConfig.temperature,
-          timeout: modelConfig.timeout,
-          maxRetries: modelConfig.maxRetries,
-        };
-
-        // Register the new model lattice
-        registerModelLattice(modelConfig.key, llmConfig);
-        registeredModels.push(modelConfig.key);
-      } catch (error: any) {
-        errors.push(
-          `Failed to register model ${modelConfig.key}: ${error.message}`
-        );
-      }
-    }
-
-    if (errors.length > 0 && registeredModels.length === 0) {
-      return reply.status(400).send({
-        success: false,
-        error: errors.join("; "),
-      });
-    }
-
-    return reply.send({
-      success: true,
-      message: `Successfully registered ${registeredModels.length} model(s)`,
-      data: {
-        registered: registeredModels,
-        errors: errors.length > 0 ? errors : undefined,
-      },
-    });
-  } catch (error: any) {
-    console.error("Failed to update models", {
-      error: error.message,
-      stack: error.stack,
-    });
-    return reply.status(500).send({
-      success: false,
-      error: error.message || "Failed to update models",
-    });
-  }
-}

package/src/controllers/personal-assistant.ts

@@ -1,191 +0,0 @@
-import { FastifyRequest, FastifyReply } from "fastify";
-import { getStoreLattice, PersonalAssistantConfig } from "@axiom-lattice/core";
-import type { Assistant, CreateAssistantRequest } from "@axiom-lattice/protocols";
-import { randomUUID } from "crypto";
-function getWorkspaceId(request: FastifyRequest): string {
-  return (request.headers["x-workspace-id"] as string) || "default";
-}
-
-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 getUserId(request: FastifyRequest): string {
-  const userId = (request as any).user?.id
-    || (request.headers["x-user-id"] as string);
-  if (!userId) throw new Error("User ID is required");
-  return userId;
-}
-
-interface CreatePersonalAssistantBody {
-  name: string;
-  personality: string;
-}
-
-interface PersonalAssistantStatus {
-  id: string;
-  name: string;
-  projectId: string;
-  workspaceId?: string;
-  createdAt: Date;
-}
-
-interface PersonalAssistantResponse {
-  success: boolean;
-  message: string;
-  data?: PersonalAssistantStatus;
-}
-
-function getLinkStore() {
-  return getStoreLattice("default", "userTenantLink").store;
-}
-
-function getAssistantStore() {
-  return getStoreLattice("default", "assistant").store;
-}
-
-function getProjectStore() {
-  return getStoreLattice("default", "project").store;
-}
-
-async function getLinkMeta(userId: string, tenantId: string) {
-  const link = await getLinkStore().getLink(userId, tenantId);
-  return (link?.metadata || {}) as Record<string, unknown>;
-}
-
-async function updateLinkMeta(userId: string, tenantId: string, meta: Record<string, unknown>) {
-  await getLinkStore().updateLink(userId, tenantId, { metadata: meta } as any);
-}
-
-/**
- * Create a personal assistant for the authenticated user.
- * Random IDs for both assistant and project, stored in UserTenantLink.metadata.
- */
-export async function createPersonalAssistant(
-  request: FastifyRequest<{ Body: CreatePersonalAssistantBody }>,
-  reply: FastifyReply
-): Promise<PersonalAssistantResponse> {
-  const tenantId = getTenantId(request);
-  const userId = getUserId(request);
-  const { name, personality } = request.body;
-
-  if (!name || !personality) {
-    return reply.status(400).send({ success: false, message: "name and personality are required" });
-  }
-
-  const store = getAssistantStore();
-
-  const existing = await store.getByOwner(tenantId, userId);
-  if (existing) {
-    return reply.status(409).send({ success: false, message: "You already have a personal assistant" });
-  }
-
-  const assistantId = randomUUID();
-
-  // Create a personal project under the current workspace
-  const projectStore = getProjectStore();
-  const projectId = randomUUID();
-  const wsId = getWorkspaceId(request);
-  await projectStore.createProject(tenantId, wsId, projectId, {
-    name: `${name}'s Space`,
-    description: "Personal workspace for your assistant",
-  });
-
-  const config = PersonalAssistantConfig.get();
-  config.key = assistantId;
-  PersonalAssistantConfig.render(config, name, personality);
-
-  const data: CreateAssistantRequest = { name, graphDefinition: config, ownerUserId: userId };
-  const assistant = await store.createAssistant(tenantId, assistantId, data);
-
-  // Store in UserTenantLink.metadata
-  const meta = await getLinkMeta(userId, tenantId);
-  meta.personalAssistantId = assistantId;
-  meta.personalProjectId = projectId;
-  meta.personalWorkspaceId = wsId;
-  await updateLinkMeta(userId, tenantId, meta);
-
-  return reply.status(201).send({
-    success: true,
-    message: "Personal assistant created",
-    data: { id: assistant.id, name: assistant.name, projectId, workspaceId: wsId, createdAt: assistant.createdAt },
-  });
-}
-
-/**
- * Get the authenticated user's personal assistant.
- */
-export async function getPersonalAssistant(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<PersonalAssistantResponse> {
-  const tenantId = getTenantId(request);
-  const userId = getUserId(request);
-
-  const store = getAssistantStore();
-  const assistant = await store.getByOwner(tenantId, userId);
-
-  if (!assistant) {
-    return reply.status(404).send({ success: false, message: "No personal assistant found" });
-  }
-
-  const meta = await getLinkMeta(userId, tenantId);
-  const projectId = (meta.personalProjectId as string) || "default";
-  const workspaceId = (meta.personalWorkspaceId as string) || "default";
-
-  return {
-    success: true,
-    message: "Personal assistant found",
-    data: { id: assistant.id, name: assistant.name, projectId, workspaceId, createdAt: assistant.createdAt },
-  };
-}
-
-/**
- * Delete the authenticated user's personal assistant.
- */
-export async function deletePersonalAssistant(
-  request: FastifyRequest,
-  reply: FastifyReply
-): Promise<{ success: boolean; message: string }> {
-  const tenantId = getTenantId(request);
-  const userId = getUserId(request);
-
-  const store = getAssistantStore();
-  const assistant = await store.getByOwner(tenantId, userId);
-
-  if (!assistant) {
-    return reply.status(404).send({ success: false, message: "No personal assistant found" });
-  }
-
-  const deleted = await store.deleteAssistant(tenantId, assistant.id);
-  if (!deleted) {
-    return reply.status(500).send({ success: false, message: "Failed to delete personal assistant" });
-  }
-
-  // Delete all threads
-  const threadStore = getStoreLattice("default", "thread").store;
-  try {
-    const threads = await threadStore.getThreadsByAssistantId(tenantId, assistant.id);
-    for (const t of threads) {
-      await threadStore.deleteThread(tenantId, t.id);
-    }
-  } catch { /* best effort */ }
-
-  // Clean up the project
-  const meta = await getLinkMeta(userId, tenantId);
-  const projectId = meta.personalProjectId as string;
-  const wsId = meta.personalWorkspaceId as string;
-  if (projectId && wsId) {
-    try { await getProjectStore().deleteProject(tenantId, projectId); } catch { /* best effort */ }
-  }
-
-  // Clean up UserTenantLink.metadata
-  delete meta.personalAssistantId;
-  delete meta.personalProjectId;
-  delete meta.personalWorkspaceId;
-  await updateLinkMeta(userId, tenantId, meta);
-
-  return { success: true, message: "Personal assistant deleted" };
-}