@multiplayer-app/ai-agent-node 0.0.1 → 0.1.0-beta.2

package/src/store/ConfigStore.ts

@@ -1,239 +0,0 @@
-import type { AgentAppConfig, AgentConfig, AgentTool } from '@multiplayer-app/ai-agent-types';
-import { ConfigHelper } from '../helpers/ConfigHelper';
-import { logger } from '../libs/logger';
-import { config as appConfig } from '../config';
-
-export class ConfigStore {
-  private static instance: ConfigStore;
-  private config: AgentAppConfig = {};
-  private allowedModels: string[] = [];
-  private agentsMap: Map<string, AgentConfig> = new Map();
-
-
-  public readonly defaultAgentConfig: AgentConfig = {
-    name: '__default',
-    description: 'Default agent',
-    systemPrompt: 'You are a helpful agent that can answer questions and help with tasks.',
-    tools: [],
-    defaultModel: appConfig.ai.defaultModel,
-  };
-  private constructor() { }
-
-  /**
-   * Get the singleton instance of ConfigStore
-   */
-  static getInstance(): ConfigStore {
-    if (!ConfigStore.instance) {
-      ConfigStore.instance = new ConfigStore();
-    }
-    return ConfigStore.instance;
-  }
-
-  /**
-   * Load the app config from a file path
-   * This method is idempotent - if config is already loaded, it returns the existing promise
-   * @param rawConfig - Raw config object (may include models array)
-   * @throws Error if config cannot be loaded
-   */
-  loadConfig(rawConfig: Record<string, any>): void {
-    this.config = ConfigHelper.loadConfig(rawConfig);
-    this.allowedModels = Array.isArray(rawConfig.models) ? rawConfig.models : [];
-    for(const contextKey of Object.keys(this.config)) {
-      for(const agentConfig of this.config[contextKey]) {
-        this.agentsMap.set(agentConfig.name, agentConfig);
-      }
-    }
-    logger.info(`Successfully loaded config with ${this.getTotalAgentsCount()} agents across ${this.getContextKeys().length} contexts`);
-  }
-  addAllowedModelPattern(pattern: string): void {
-    this.allowedModels.push(pattern);
-  }
-
-  addAgent<T = any>(contextKeys: string[], agentConfig: AgentConfig<T>): void {
-    for (const contextKey of contextKeys) {
-      if (!this.config[contextKey]) {
-        this.config[contextKey] = [];
-      }
-      this.config[contextKey].push(agentConfig);
-    }
-    this.agentsMap.set(agentConfig.name, agentConfig);
-  }
-
-  /**
-   * Add a tool to every loaded agent (across all contexts).
-   * This is intended for built-in tools that should always be available.
-   * @param tool - The tool to add to all agents
-   */
-  addToolToAllAgents(tool: AgentTool): void {
-    for (const agents of Object.values(this.config)) {
-      for (const agent of agents) {
-        const existing = agent.tools ?? [];
-        const toolName = (tool as any)?.data?.name;
-        const alreadyPresent =
-          toolName
-            ? existing.some((t: any) => t?.data?.name === toolName)
-            : existing.some((t: any) => JSON.stringify(t) === JSON.stringify(tool));
-        if (!alreadyPresent) {
-          agent.tools = [...existing, tool] as any;
-        }
-      }
-    }
-  }
-
-
-  getAgentConfigByName(agentName: string | undefined): AgentConfig {
-    if(!agentName) {
-      return this.defaultAgentConfig;
-    }
-    return this.agentsMap.get(agentName) || this.defaultAgentConfig;
-  }
-
-  /**
-   * Get all agents for a specific context key
-   * @param contextKey - The context key to get agents for
-   * @returns Array of AgentConfig for the context, or empty array if context not found
-   */
-  getAgentsForContext(contextKey: string): AgentConfig[] {
-    return this.config[contextKey] || [this.defaultAgentConfig];
-  }
-
-  /**
-   * Get a specific agent by name for a context
-   * @param contextKey - The context key
-   * @param agentName - The name of the agent
-   * @returns AgentConfig if found, undefined otherwise
-   */
-  getAgentForContext(contextKey: string, agentName: string): AgentConfig {
-    const agents = this.getAgentsForContext(contextKey);
-    return agents.find(agent => agent.name === agentName) || this.defaultAgentConfig;
-  }
-
-  /**
-   * Get all context keys
-   * @returns Array of context key strings
-   */
-  getContextKeys(): string[] {
-    if (!this.config) {
-      throw new Error('Config not loaded. Call loadConfig() first.');
-    }
-
-    return Object.keys(this.config);
-  }
-
-  /**
-   * Check if a context key exists
-   * @param contextKey - The context key to check
-   * @returns true if context exists, false otherwise
-   */
-  hasContext(contextKey: string): boolean {
-    if (!this.config) {
-      return false;
-    }
-
-    return contextKey in this.config;
-  }
-
-  /**
-   * Get the total number of agents across all contexts
-   * @returns Total count of agents
-   */
-  getTotalAgentsCount(): number {
-    if (!this.config) {
-      return 0;
-    }
-
-    return Object.values(this.config).reduce((total, agents) => total + agents.length, 0);
-  }
-
-  /**
-   * Get all unique agent names across all contexts
-   * @returns Array of unique agent names
-   */
-  getAllAgentNames(): string[] {
-    if (!this.config) {
-      return [];
-    }
-
-    const agentNames = new Set<string>();
-    for (const agents of Object.values(this.config)) {
-      for (const agent of agents) {
-        agentNames.add(agent.name);
-      }
-    }
-    return Array.from(agentNames);
-  }
-
-  /**
-   * Check if config has been loaded
-   * @returns true if config is loaded, false otherwise
-   */
-  isLoaded(): boolean {
-    return this.config !== null;
-  }
-
-  /**
-   * Check if a model is allowed based on the configured allowedModels patterns
-   * Patterns support wildcards (*) which match any sequence of characters
-   * - Patterns without a provider prefix (e.g., 'gpt-4') match the model ID across all providers
-   * - Patterns with a provider prefix (e.g., 'openai/gpt-4') only match that specific provider
-   * @param model - The model identifier to check (e.g., 'openai/gpt-4o', 'anthropic/claude-3-5-sonnet', 'gpt-4')
-   * @param provider - Optional provider name. If provided and model doesn't contain '/', constructs as 'provider/model'.
-   *                   For OpenRouter, the model should already contain the full path (e.g., 'openai/gpt-4')
-   * @returns true if model matches any allowed pattern, false otherwise
-   */
-  isModelAllowed(model: string, provider?: string): boolean {
-    if (this.allowedModels.length === 0) {
-      // If no allowedModels are configured, allow all models
-      return true;
-    }
-
-    // Construct the full model identifier if provider is provided
-    let modelToCheck: string;
-    if (provider && !model.includes('/')) {
-      // Provider provided and model doesn't have provider prefix - construct as 'provider/model'
-      modelToCheck = `${provider}/${model}`;
-    } else {
-      // Use model as-is (either already has provider prefix or no provider provided)
-      modelToCheck = model;
-    }
-
-    // Check if model matches any of the allowed patterns
-    return this.allowedModels.some(pattern => {
-      const hasProvider = pattern.includes('/');
-
-      // Convert wildcard pattern to regex
-      // Escape special regex characters except *, then replace * with .*
-      const regexPattern = pattern
-        .replace(/[.+?^${}()|[\]\\]/g, '\\$&') // Escape special regex chars
-        .replace(/\*/g, '.*'); // Replace * with .*
-
-      if (hasProvider) {
-        // Pattern has provider prefix - match exactly
-        const regex = new RegExp(`^${regexPattern}$`);
-        return regex.test(modelToCheck);
-      } else {
-        // Pattern has no provider prefix - match both exact and with any provider prefix
-        // Match: 'gpt-4' OR '*/gpt-4' OR 'provider/gpt-4'
-        const exactRegex = new RegExp(`^${regexPattern}$`);
-        const withProviderRegex = new RegExp(`^.*/${regexPattern}$`);
-        return exactRegex.test(modelToCheck) || withProviderRegex.test(modelToCheck);
-      }
-    });
-  }
-
-  /**
-   * Get the allowed model patterns
-   * @returns Array of allowed model patterns
-   */
-  getAllowedModels(): string[] {
-    return [...this.allowedModels];
-  }
-
-  /**
-   * Clear the loaded config (useful for testing or reloading)
-   */
-  clear(): void {
-    this.config = {};
-    this.allowedModels = [];
-  }
-}