@axiom-lattice/core 2.1.29 → 2.1.30

package/dist/index.d.mts

@@ -6,7 +6,7 @@ import { BaseLanguageModelInput, LanguageModelLike } from '@langchain/core/langu
 import { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
 import { ChatResult } from '@langchain/core/outputs';
 import * as _axiom_lattice_protocols from '@axiom-lattice/protocols';
-import { LLMConfig, SemanticMetricsServerConfig, MetricMeta, MetricQueryResult, DataSource, SemanticMetricsQueryRequest, SemanticMetricsQueryResponse, ExecuteSqlQueryRequest, ExecuteSqlQueryResponse, MetricsServerType, MetricsServerConfig, ToolConfig, ToolExecutor, AgentConfig, MiddlewareType, GraphBuildOptions, MessageChunk, QueueLatticeProtocol, QueueConfig, QueueClient, QueueResult, ScheduleLatticeProtocol, ScheduleConfig, ScheduleClient, ScheduleStorage, TaskHandler, ScheduleOnceOptions, ScheduleCronOptions, ScheduledTaskDefinition, ScheduledTaskStatus, ScheduleExecutionType, ThreadStore, AssistantStore, SkillStore, WorkspaceStore, ProjectStore, DatabaseConfigStore, MetricsServerConfigStore, UserStore, TenantStore, UserTenantLinkStore, Thread, CreateThreadRequest, Assistant, CreateAssistantRequest, Skill, CreateSkillRequest, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, User, CreateUserRequest, UpdateUserRequest, Tenant, CreateTenantRequest, UpdateTenantRequest, UserTenantLink, CreateUserTenantLinkRequest, UpdateUserTenantLinkRequest, LoggerLatticeProtocol, LoggerConfig, LoggerClient, LoggerContext, SkillConfig, SkillClient } from '@axiom-lattice/protocols';
+import { LLMConfig, SemanticMetricsServerConfig, MetricMeta, MetricQueryResult, DataSource, SemanticMetricsQueryRequest, SemanticMetricsQueryResponse, ExecuteSqlQueryRequest, ExecuteSqlQueryResponse, MetricsServerType, MetricsServerConfig, ToolConfig, ToolExecutor, AgentConfig, MiddlewareType, GraphBuildOptions, MessageChunk, QueueLatticeProtocol, QueueConfig, QueueClient, QueueResult, ScheduleLatticeProtocol, ScheduleConfig, ScheduleClient, ScheduleStorage, TaskHandler, ScheduleOnceOptions, ScheduleCronOptions, ScheduledTaskDefinition, ScheduledTaskStatus, ScheduleExecutionType, ThreadStore, AssistantStore, SkillStore, WorkspaceStore, ProjectStore, DatabaseConfigStore, MetricsServerConfigStore, McpServerConfigStore, UserStore, TenantStore, UserTenantLinkStore, Thread, CreateThreadRequest, Assistant, CreateAssistantRequest, Skill, CreateSkillRequest, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, User, CreateUserRequest, UpdateUserRequest, Tenant, CreateTenantRequest, UpdateTenantRequest, UserTenantLink, CreateUserTenantLinkRequest, UpdateUserTenantLinkRequest, LoggerLatticeProtocol, LoggerConfig, LoggerClient, LoggerContext, SkillConfig, SkillClient } from '@axiom-lattice/protocols';
 export { _axiom_lattice_protocols as Protocols };
 export { AgentConfig, AgentType, GraphBuildOptions, MemoryType } from '@axiom-lattice/protocols';
 import * as langchain from 'langchain';
@@ -1922,6 +1922,7 @@ type StoreTypeMap = {
     project: ProjectStore;
     database: DatabaseConfigStore;
     metrics: MetricsServerConfigStore;
+    mcp: McpServerConfigStore;
     user: UserStore;
     tenant: TenantStore;
     userTenantLink: UserTenantLinkStore;

package/dist/index.d.ts

@@ -6,7 +6,7 @@ import { BaseLanguageModelInput, LanguageModelLike } from '@langchain/core/langu
 import { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
 import { ChatResult } from '@langchain/core/outputs';
 import * as _axiom_lattice_protocols from '@axiom-lattice/protocols';
-import { LLMConfig, SemanticMetricsServerConfig, MetricMeta, MetricQueryResult, DataSource, SemanticMetricsQueryRequest, SemanticMetricsQueryResponse, ExecuteSqlQueryRequest, ExecuteSqlQueryResponse, MetricsServerType, MetricsServerConfig, ToolConfig, ToolExecutor, AgentConfig, MiddlewareType, GraphBuildOptions, MessageChunk, QueueLatticeProtocol, QueueConfig, QueueClient, QueueResult, ScheduleLatticeProtocol, ScheduleConfig, ScheduleClient, ScheduleStorage, TaskHandler, ScheduleOnceOptions, ScheduleCronOptions, ScheduledTaskDefinition, ScheduledTaskStatus, ScheduleExecutionType, ThreadStore, AssistantStore, SkillStore, WorkspaceStore, ProjectStore, DatabaseConfigStore, MetricsServerConfigStore, UserStore, TenantStore, UserTenantLinkStore, Thread, CreateThreadRequest, Assistant, CreateAssistantRequest, Skill, CreateSkillRequest, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, User, CreateUserRequest, UpdateUserRequest, Tenant, CreateTenantRequest, UpdateTenantRequest, UserTenantLink, CreateUserTenantLinkRequest, UpdateUserTenantLinkRequest, LoggerLatticeProtocol, LoggerConfig, LoggerClient, LoggerContext, SkillConfig, SkillClient } from '@axiom-lattice/protocols';
+import { LLMConfig, SemanticMetricsServerConfig, MetricMeta, MetricQueryResult, DataSource, SemanticMetricsQueryRequest, SemanticMetricsQueryResponse, ExecuteSqlQueryRequest, ExecuteSqlQueryResponse, MetricsServerType, MetricsServerConfig, ToolConfig, ToolExecutor, AgentConfig, MiddlewareType, GraphBuildOptions, MessageChunk, QueueLatticeProtocol, QueueConfig, QueueClient, QueueResult, ScheduleLatticeProtocol, ScheduleConfig, ScheduleClient, ScheduleStorage, TaskHandler, ScheduleOnceOptions, ScheduleCronOptions, ScheduledTaskDefinition, ScheduledTaskStatus, ScheduleExecutionType, ThreadStore, AssistantStore, SkillStore, WorkspaceStore, ProjectStore, DatabaseConfigStore, MetricsServerConfigStore, McpServerConfigStore, UserStore, TenantStore, UserTenantLinkStore, Thread, CreateThreadRequest, Assistant, CreateAssistantRequest, Skill, CreateSkillRequest, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, User, CreateUserRequest, UpdateUserRequest, Tenant, CreateTenantRequest, UpdateTenantRequest, UserTenantLink, CreateUserTenantLinkRequest, UpdateUserTenantLinkRequest, LoggerLatticeProtocol, LoggerConfig, LoggerClient, LoggerContext, SkillConfig, SkillClient } from '@axiom-lattice/protocols';
 export { _axiom_lattice_protocols as Protocols };
 export { AgentConfig, AgentType, GraphBuildOptions, MemoryType } from '@axiom-lattice/protocols';
 import * as langchain from 'langchain';
@@ -1922,6 +1922,7 @@ type StoreTypeMap = {
     project: ProjectStore;
     database: DatabaseConfigStore;
     metrics: MetricsServerConfigStore;
+    mcp: McpServerConfigStore;
     user: UserStore;
     tenant: TenantStore;
     userTenantLink: UserTenantLinkStore;

package/dist/index.js

@@ -5930,6 +5930,122 @@ var InMemoryMetricsServerConfigStore = class {
   }
 };
 
+// src/store_lattice/InMemoryMcpServerConfigStore.ts
+var InMemoryMcpServerConfigStore = class {
+  constructor() {
+    this.configs = /* @__PURE__ */ new Map();
+  }
+  /**
+   * Get composite key for storage
+   */
+  getKey(tenantId, id) {
+    return `${tenantId}:${id}`;
+  }
+  /**
+   * Get all MCP server configurations for a tenant
+   */
+  async getAllConfigs(tenantId) {
+    return Array.from(this.configs.values()).filter(
+      (config) => config.tenantId === tenantId
+    );
+  }
+  /**
+   * Get all MCP server configurations across all tenants
+   */
+  async getAllConfigsWithoutTenant() {
+    return Array.from(this.configs.values());
+  }
+  /**
+   * Get MCP server configuration by ID
+   */
+  async getConfigById(tenantId, id) {
+    const key = this.getKey(tenantId, id);
+    return this.configs.get(key) || null;
+  }
+  /**
+   * Get MCP server configuration by business key
+   */
+  async getConfigByKey(tenantId, key) {
+    const configs = await this.getAllConfigs(tenantId);
+    return configs.find((config) => config.key === key) || null;
+  }
+  /**
+   * Create a new MCP server configuration
+   */
+  async createConfig(tenantId, id, data) {
+    const now = /* @__PURE__ */ new Date();
+    const entry = {
+      id,
+      tenantId,
+      key: data.key,
+      config: data.config,
+      selectedTools: data.selectedTools || [],
+      isEnvEncrypted: false,
+      status: "disconnected",
+      name: data.name,
+      description: data.description,
+      createdAt: now,
+      updatedAt: now
+    };
+    const storageKey = this.getKey(tenantId, id);
+    this.configs.set(storageKey, entry);
+    return entry;
+  }
+  /**
+   * Update an existing MCP server configuration
+   */
+  async updateConfig(tenantId, id, updates) {
+    const key = this.getKey(tenantId, id);
+    const existing = this.configs.get(key);
+    if (!existing) {
+      return null;
+    }
+    const updated = {
+      ...existing,
+      ...updates,
+      config: updates.config ? { ...existing.config, ...updates.config } : existing.config,
+      key: updates.key || existing.key,
+      name: updates.name !== void 0 ? updates.name : existing.name,
+      description: updates.description !== void 0 ? updates.description : existing.description,
+      selectedTools: updates.selectedTools !== void 0 ? updates.selectedTools : existing.selectedTools,
+      status: updates.status !== void 0 ? updates.status : existing.status,
+      updatedAt: /* @__PURE__ */ new Date()
+    };
+    this.configs.set(key, updated);
+    return updated;
+  }
+  /**
+   * Delete a MCP server configuration by ID
+   */
+  async deleteConfig(tenantId, id) {
+    const key = this.getKey(tenantId, id);
+    return this.configs.delete(key);
+  }
+  /**
+   * Check if configuration exists
+   */
+  async hasConfig(tenantId, id) {
+    const key = this.getKey(tenantId, id);
+    return this.configs.has(key);
+  }
+  /**
+   * Clear all configurations (useful for testing)
+   */
+  clear() {
+    this.configs.clear();
+  }
+  /**
+   * Clear configurations for a specific tenant
+   */
+  clearByTenant(tenantId) {
+    for (const key of this.configs.keys()) {
+      if (key.startsWith(`${tenantId}:`)) {
+        this.configs.delete(key);
+      }
+    }
+  }
+};
+
 // src/store_lattice/InMemoryUserStore.ts
 var InMemoryUserStore = class {
   constructor() {
@@ -6235,6 +6351,7 @@ var defaultWorkspaceStore = new InMemoryWorkspaceStore();
 var defaultProjectStore = new InMemoryProjectStore();
 var defaultDatabaseConfigStore = new InMemoryDatabaseConfigStore();
 var defaultMetricsServerConfigStore = new InMemoryMetricsServerConfigStore();
+var defaultMcpServerConfigStore = new InMemoryMcpServerConfigStore();
 storeLatticeManager.registerLattice("default", "thread", defaultThreadStore);
 storeLatticeManager.registerLattice(
   "default",
@@ -6246,6 +6363,7 @@ storeLatticeManager.registerLattice("default", "workspace", defaultWorkspaceStor
 storeLatticeManager.registerLattice("default", "project", defaultProjectStore);
 storeLatticeManager.registerLattice("default", "database", defaultDatabaseConfigStore);
 storeLatticeManager.registerLattice("default", "metrics", defaultMetricsServerConfigStore);
+storeLatticeManager.registerLattice("default", "mcp", defaultMcpServerConfigStore);
 var defaultUserStore = new InMemoryUserStore();
 var defaultTenantStore = new InMemoryTenantStore();
 var defaultUserTenantLinkStore = new InMemoryUserTenantLinkStore();