direct-patch.js

#!/usr/bin/env node

// direct-patch.js - Directly patches the character and starts the agent

// Set environment variables for models
process.env.LARGE_OPENAI_MODEL = 'gpt-4o';
process.env.MEDIUM_OPENAI_MODEL = 'gpt-4o';
process.env.SMALL_OPENAI_MODEL = 'gpt-4o-mini';
process.env.EMBEDDING_OPENAI_MODEL = 'text-embedding-3-small';
process.env.IMAGE_OPENAI_MODEL = 'dall-e-3';

process.env.OPENAI_API_MODEL_LARGE = 'gpt-4o';
process.env.OPENAI_API_MODEL_MEDIUM = 'gpt-4o';
process.env.OPENAI_API_MODEL_SMALL = 'gpt-4o-mini';
process.env.OPENAI_API_MODEL_EMBEDDING = 'text-embedding-3-small';
process.env.OPENAI_API_MODEL_IMAGE = 'dall-e-3';

// Get the original startAgents function
const agentModule = require('./packages/agent/dist/index.js');
const originalStartAgents = agentModule.startAgents;

// Monkey patch the startAgents function
agentModule.startAgents = async function(characters, options) {
  console.log('[DIRECT_PATCH] Patching characters before starting agents...');
  
  // Patch each character with the required model settings
  for (const character of characters) {
    console.log(`[DIRECT_PATCH] Patching character: ${character.name}`);
    
    // Ensure settings object exists
    character.settings = character.settings || {};
    
    // Add model settings directly from environment
    character.settings.LARGE_OPENAI_MODEL = process.env.LARGE_OPENAI_MODEL;
    character.settings.MEDIUM_OPENAI_MODEL = process.env.MEDIUM_OPENAI_MODEL;
    character.settings.SMALL_OPENAI_MODEL = process.env.SMALL_OPENAI_MODEL;
    character.settings.EMBEDDING_OPENAI_MODEL = process.env.EMBEDDING_OPENAI_MODEL;
    character.settings.IMAGE_OPENAI_MODEL = process.env.IMAGE_OPENAI_MODEL;
    
    // Ensure secrets object exists and contains API key
    character.settings.secrets = character.settings.secrets || {};
    if (process.env.OPENAI_API_KEY) {
      character.settings.secrets.OPENAI_API_KEY = process.env.OPENAI_API_KEY;
    }
    
    console.log('[DIRECT_PATCH] Patched model settings:', {
      LARGE: character.settings.LARGE_OPENAI_MODEL,
      MEDIUM: character.settings.MEDIUM_OPENAI_MODEL,
      SMALL: character.settings.SMALL_OPENAI_MODEL
    });
  }
  
  // Call original function with patched characters
  return originalStartAgents(characters, options);
};

// Start agent using the standard entry point
require('./packages/agent/dist/index.js');

console.log('[DIRECT_PATCH] Agent started with patched model settings');