agentnet 0.0.3 → 0.0.5

package/examples/simple/simple.js

@@ -1,4 +1,4 @@
-import { AgentLoaderJSON, Message, Bindings, PostgresStore } from "../../src/index.js"
+import { AgentLoaderJSON, Message, Bindings, MemoryStore } from "../../src/index.js"
 
 const agentDefinition = {
     "apiVersion": "agentnet/v1alpha1",
@@ -9,25 +9,18 @@ const agentDefinition = {
     },
     "spec": {
       "store": {
-        "type": "Postgres",
+        "type": "Memory",
       },
       "llm": {
-        "provider": "Gemini",
-        "model": "gemini-2.0-flash",
-        "systemInstruction": "You are a highly advanced accomodation manager agent. \nPrioritize clarity and helpfulness.\nUse tools effectively to gather information.\n",
-        "config": {
-          "temperature": 0.5,
-          "toolConfig": {
-            "functionCallingConfig": {
-              "mode": "auto"
-            }
-          }
-        }
+        "provider": "GPT",
+        "model": "gpt-4o-mini",
+        "instructions": "You are a highly advanced accomodation manager agent. \nPrioritize clarity and helpfulness.\nUse tools effectively to gather information."
       },
       "tools": [
         {
-          "name": "getRoomsListTool",
+          "name": "get_rooms_list_tool",
           "description": "Retrieves a list of available rooms based on criteria.",
+          "type": "function",
           "parameters": {
             "type": "object",
             "properties": {
@@ -51,8 +44,9 @@ const agentDefinition = {
           }
         },
         {
-          "name": "getRoomDetailTool",
+          "name": "get_room_detail_tool",
           "description": "Retrieves detailed information about a specific room.",
+          "type": "function",
           "parameters": {
             "type": "object",
             "properties": {
@@ -73,15 +67,15 @@ const agentDefinition = {
 // Load the agent definition
 const agents = await AgentLoaderJSON(agentDefinition, {
     bindings: {
-        [Bindings.Postgres]: PostgresStore()
+        [Bindings.Memory]: MemoryStore()
     }
 })
 
 // Add the binding tools to the agent
-agents.accomodationAgent.tools.getRoomsListTool.bind(async (state, input) => {
+agents.accomodationAgent.tools.get_rooms_list_tool.bind(async (state, input) => {
     return { answer: "We have Double room with a view of the sea and a single room with a view of the pool, and a suite with a view of the city." }
 })
-agents.accomodationAgent.tools.getRoomDetailTool.bind(async (state, input) => {
+agents.accomodationAgent.tools.get_room_detail_tool.bind(async (state, input) => {
     return { answer: "The Double room with a view of the sea has a king size bed, a private balcony, and a view of the sea." }
 })
 

package/examples/smartness/agents-smartness.yaml

@@ -1,42 +1,37 @@
 ---
-apiVersion: smartagent.io/v1alpha1
+apiVersion: agentnet/v1alpha1
 kind: AgentDefinition
 metadata:
   name: entrypoint
-  namespace: smartchat
+  namespace: smartexample
 spec:
   io:
     - type: NatsIO
       bindings:
-        discoveryTopic: smartness.discovery
+        discoveryTopic: smartexample.discovery
         acceptedNetworks: 
-          - "smartchat.*"
+          - "smartexample.*"
   store:
-    type: Memory
+    type: Postgres
   llm:
-    provider: Gemini
-    model: gemini-2.0-flash
-    systemInstruction: |
+    provider: GPT
+    model: gpt-4.1-2025-04-14
+    instructions: |
       You are a highly advanced triage agent. 
       Prioritize clarity and helpfulness.
       Use tools effectively to gather information.
-    config:
-      temperature: 0.5
-      toolConfig:
-        functionCallingConfig:
-          mode: 'auto'
       
 ---
-apiVersion: smartagent.io/v1alpha1
+apiVersion: agentnet/v1alpha1
 kind: AgentDefinition
 metadata:
   name: accomodation
-  namespace: smartchat
+  namespace: smartexample
 spec:
   io:
     - type: NatsIO
       bindings:
-        discoveryTopic: smartness.discovery
+        discoveryTopic: smartexample.discovery
   store:
     type: Postgres
 
@@ -96,16 +91,16 @@ spec:
           - question             
 
 ---
-apiVersion: smartagent.io/v1alpha1
+apiVersion: agentnet/v1alpha1
 kind: AgentDefinition
 metadata:
   name: review
-  namespace: smartchat
+  namespace: smartexample
 spec:
   io:
     - type: NatsIO
       bindings:
-        discoveryTopic: smartness.discovery
+        discoveryTopic: smartexample.discovery
   store:
     type: Postgres
 
@@ -147,16 +142,16 @@ spec:
           - hotelName          
 
 ---
-apiVersion: smartagent.io/v1alpha1
+apiVersion: agentnet/v1alpha1
 kind: AgentDefinition
 metadata:
   name: booking
-  namespace: smartchat
+  namespace: smartexample
 spec:
   io:
     - type: NatsIO
       bindings:
-        discoveryTopic: smartness.discovery
+        discoveryTopic: smartexample.discovery
   store:
     type: Postgres
 
@@ -217,16 +212,16 @@ spec:
           - roomName                    
 
 ---
-apiVersion: smartagent.io/v1alpha1
+apiVersion: agentnet/v1alpha1
 kind: AgentDefinition
 metadata:
   name: pricing
-  namespace: smartchat
+  namespace: smartexample
 spec:
   io:
     - type: NatsIO
       bindings:
-        discoveryTopic: smartness.discovery
+        discoveryTopic: smartexample.discovery
   store:
     type: Postgres
 

package/examples/smartness/index.js

@@ -57,11 +57,19 @@ const agentClient = AgentClient()
 const message = new Message({
     content: "What rooms do you have from 2025-05-25 to 2025-05-30 for 3 guests For the hotel Flora? Give me the review of the hotel Flora",
     session: {
-        id: "67a71e42-a7d8-1db2-ad17-64e1c8546b21"
+        id: "67a71e42-a7d8-1db2-ad17-64e1c8546b20"
     }
 })
 const res = await agentClient.queryIo(io, 'entrypoint', message)
 console.log("=======\n", res.getContent())
 console.log("=======\n", res.getSession())
-//const res2 = await agentClient.queryIo(io, 'smartnessAgent', "Quanto costa la camera doppia del Flora per il 10-05-2025 per due persone? Prenotala se costa meno di 100€ la camera double con vista mare per il 10-05-2025 al hotel Flora")
-//console.log("=======\n", res2)
+
+const message2 = new Message({
+    content: "Quanto costa la camera doppia del Flora per il 10-05-2025 per due persone? Prenotala se costa meno di 100€ la camera double con vista mare per il 10-05-2025 al hotel Flora",
+    session: {
+        id: "67a71e42-a7d8-1db2-ad17-64e1c8546b20"
+    }
+})
+const res2 = await agentClient.queryIo(io, 'entrypoint', message2)
+console.log("=======\n", res2.getContent())
+console.log("=======\n", res2.getSession())

package/package.json

@@ -1,9 +1,9 @@
 {
   "name": "agentnet",
-  "version": "0.0.3",
+  "version": "0.0.5",
   "type": "module",
   "description": "Agent library used by Smartness",
-  "main": "index.js",
+  "main": "src/index.js",
   "scripts": {
     "test": "node --experimental-vm-modules ./node_modules/.bin/jest --rootDir=src/tests"
   },

package/src/agent/agent-loader.js

@@ -1,6 +1,7 @@
 import fs from 'fs'
 import { parse } from 'yaml'
 import { Gemini } from "../index.js"
+import { GPT } from "../index.js"
 import { Agent } from "./agent.js"
 import { logger } from "../utils/logger.js"
 import { ConfigurationError } from "../errors/index.js"
@@ -10,8 +11,7 @@ import { validateApiVersion, DEFAULT_API_VERSION, API_VERSIONS } from '../utils/
  * Version handlers for different API versions
  */
 const VERSION_HANDLERS = {
-    'smartagent.io/v1alpha1': processV1Alpha1Definition,
-    'agentnet.io/v1alpha1': processV1Alpha1Definition
+    'agentnet/v1alpha1': processV1Alpha1Definition
     // Additional version handlers can be added here as the API evolves
 };
 
@@ -139,6 +139,9 @@ async function loadLlmProvider(providerName) {
     if (providerName === 'Gemini') {
         return Gemini;
     }
+    if (providerName === 'GPT') {
+        return GPT;
+    }    
     
     try {
         return global[providerName] || await import(providerName);
@@ -209,11 +212,14 @@ async function configureLLM(agentBuilder, llmSpec) {
     
     const llmProviderInstance = await loadLlmProvider(llmSpec.provider);
     
-    return agentBuilder.withLLM(llmProviderInstance, {
-        model: llmSpec.model,
-        systemInstruction: llmSpec.systemInstruction,
-        config: llmSpec.config
-    });
+    const config = {}
+    for (const key in llmSpec) {
+        if (key !== 'provider') {
+            config[key] = llmSpec[key];
+        }
+    }
+
+    return agentBuilder.withLLM(llmProviderInstance, config);
 }
 
 /**

package/src/agent/executor.js

@@ -34,7 +34,7 @@ async function emit(hooks, event, data) {
  * @param {Array} tools - Tool definitions
  * @param {Array} handoffs - Handoff definitions
  */
-export function makeToolsAndHandoffsMap(toolsAndHandoffsMap, tools, handoffs) {
+export function makeToolsAndHandoffsMap(llmType, toolsAndHandoffsMap, tools, handoffs) {
 	if (!tools) {
 		return;
 	}
@@ -71,6 +71,11 @@ export function makeToolsAndHandoffsMap(toolsAndHandoffsMap, tools, handoffs) {
 					logger.warn('Skipping invalid handoff definition', { handoff });
 					continue;
 				}
+
+                // TODO: Remove this once we have a better way to handle handoffs
+                if (llmType === 'openai') {
+                    handoff.schema.type = 'function'
+                }
 				
 				// Add handoff schema to tools list
 				toolsAndHandoffsMap.tools.push(handoff.schema);
@@ -295,12 +300,6 @@ export async function build(
 					error: error
 				});
 				
-				// Add error message to conversation
-				contents.push({
-					role: 'system',
-					content: `Error: ${error.message}`
-				});
-				
 				// If we haven't hit max runs, try again
 				if (run < maxRuns - 1) {
 					logger.info(`Continuing after error in agent ${agentName}`);

package/src/agent/runtime.js

@@ -45,6 +45,7 @@ export async function AgentRuntime(agentConfig) {
         try {
             // Update tools and handoffs map with discovered agents
             makeToolsAndHandoffsMap(
+                llmApi.type,
                 toolsAndHandoffsMap, 
                 Object.values(tools), 
                 [handoffs, Object.values(discoveredAgents)]

package/src/llm/base.js

@@ -82,12 +82,12 @@ export class BaseLLM {
    * @param {Object} toolsAndHandoffsMap - Map of available tools
    * @returns {Promise<any>} Result of the tool execution
    */
-  async executeToolCall(toolCall, name, args, state, conversation, toolsAndHandoffsMap) {
+  async executeToolCall(toolCall, name, args, state, toolsAndHandoffsMap) {
     logger.debug(`Executing tool from ${this.type}`, { 
       toolName: name,
       argsPreview: JSON.stringify(args).substring(0, 100)
     });
-    
+
     try {
       if (!toolsAndHandoffsMap[name] || !toolsAndHandoffsMap[name].function) {
         throw new Error(`Tool "${name}" not found or has no function implementation`);
@@ -95,7 +95,7 @@ export class BaseLLM {
 
       let result = null;
       if (toolsAndHandoffsMap[name].type === 'handoff') {
-        result = await toolsAndHandoffsMap[name].function(conversation, state, args);
+        result = await toolsAndHandoffsMap[name].function(state, args);
         // Process handoff results if needed
         this.processHandoffResult(result, state);
       } else {
@@ -106,7 +106,7 @@ export class BaseLLM {
       return result;
       
     } catch (error) {
-      logger.error(`Error executing tool "${name}"`, { error });
+      logger.error(`Error executing tool "${name}"`, error.message);
       throw error;
     }
   }

package/src/llm/gemini.js

@@ -65,6 +65,7 @@ class GeminiLLM extends BaseLLM {
       });
       return res;
     } catch (error) {
+      console.log(error)
       logger.error('Gemini API error', { 
         error,
         modelName: input.model
@@ -93,7 +94,7 @@ class GeminiLLM extends BaseLLM {
     const name = toolCall.name;
     
     try {
-      const result = await super.executeToolCall(toolCall, name, args, state, conversation, toolsAndHandoffsMap);
+      const result = await super.executeToolCall(toolCall, name, args, state, toolsAndHandoffsMap);
       
       // Add function call and response to conversation in Gemini-specific format
       const function_response_part = {
@@ -128,7 +129,6 @@ class GeminiLLM extends BaseLLM {
     // Handle simple text response
     if (response.text !== undefined) {
       logger.debug('Gemini response contains text, returning directly');
-      conversation.push({ role: 'model', parts: [{ text: response.text }] });
       return response.text;
     }
     

package/src/llm/gpt.js

@@ -50,14 +50,14 @@ class OpenAILLM extends BaseLLM {
 			conversationLength: conversation.length,
 			toolsCount: toolsAndHandoffsMap.tools.length
 		});
-		
+		//console.log(JSON.stringify(input, null, 2))
 		try {
 			const response = await client.responses.create(input);
 			logger.debug('OpenAI response received');
 			return response;
 		} catch (error) {
 			logger.error('OpenAI API error', { 
-				error,
+				error: error.message,
 				modelName: input.model
 			});
 			
@@ -90,7 +90,7 @@ class OpenAILLM extends BaseLLM {
 				callId: toolCall.call_id
 			});
 			
-			const result = await super.executeToolCall(toolCall, name, args, state, conversation, toolsAndHandoffsMap);
+			const result = await super.executeToolCall(toolCall, name, args, state, toolsAndHandoffsMap);
 			conversation.push(toolCall);
 			
 			const resultString = typeof result === 'string' ? result : JSON.stringify(result);
@@ -129,7 +129,6 @@ class OpenAILLM extends BaseLLM {
 	async onResponse(state, conversation, toolsAndHandoffsMap, response) {
 		if (response.output_text !== undefined && response.output_text.length > 0) {
 			logger.debug('OpenAI response contains text, returning directly');
-			conversation.push({ role: 'model', parts: [{ text: response.output_text }] });
 			return response.output_text;
 		}
 

package/src/store/store.js

@@ -191,10 +191,12 @@ export function memoryStore () {
 		connect: async function () {},
 		disconnect: async function () {},
 		set: async function (key, value) {
+			console.log('Setting state', key, value)
 			state[key] = value
 			return state[key]
 		},
 		get: async function (key) {
+			console.log('Getting state', key)
 			return state[key] || null
 		}	
 	}

package/src/transport/nats.js

@@ -15,7 +15,6 @@ import {
 // Constants
 const HEARTBEAT_INTERVAL = 1000;
 const TIMEOUT_TASK_REQUEST = 60000;
-const MAX_RECONNECT_ATTEMPTS = 5;
 
 /**
  * NATS implementation of the Transport interface
@@ -225,13 +224,14 @@ export class NatsTransport extends Transport {
                                 continue;
                             }
                         }
+                        
 
                         // Skip if not accepted
                         if (!isAccepted) {
                             logger.warn(`Agent ${agentName} does not accept network ${network}`);
                             nonAcceptedNetworks[network] = true;
                             continue;
-                        }                
+                        }               
                         
                         // Process the schemas from the discovery message
                         for (const schema of discoveryMessage.schemas) {
@@ -242,11 +242,10 @@ export class NatsTransport extends Transport {
                             }
                             
                             const agentKey = `${network}-${schema.name}`;
-                            
                             if (discoveryMessage.agentName !== agentName && !discoveredAgents[agentKey]) {
                                 logger.info(`${agentName} discovered agent capability: ${discoveryMessage.agentName} with capability ${schema.name}`);
                                 
-                                const handoffFunction = async (conversation, state, input) => {
+                                const handoffFunction = async (state, input) => {
                                     try {
                                         // Use withTimeout to ensure handoffs don't hang
                                         return await withTimeout(

package/src/utils/logger.js

@@ -13,7 +13,7 @@ export const LogLevel = {
  * Default logger configuration
  */
 const DEFAULT_CONFIG = {
-  level: LogLevel.INFO,
+  level: process.env.AGENTNET_LOG_LEVEL || LogLevel.INFO,
   enableTimestamps: true,
   enableColors: true,
   redactSensitiveData: true,

package/src/utils/version.js

@@ -5,18 +5,10 @@ import { ConfigurationError } from '../errors/index.js';
  * Map of API versions with their compatibility information
  */
 export const API_VERSIONS = {
-    // Smartagent API versions
-    'smartagent.io/v1alpha1': {
+    'agentnet/v1alpha1': {
         isSupported: true,
-        isStable: false,
-        minEngineVersion: '0.5.0',
-        features: ['basic', 'discovery', 'handoffs']
-    },
-    // Agentnet API versions
-    'agentnet.io/v1alpha1': {
-        isSupported: true,
-        isStable: false,
-        minEngineVersion: '0.8.0',
+        isStable: true,
+        minEngineVersion: '0.0.4',
         features: ['basic', 'discovery', 'handoffs', 'advanced-routing']
     }
 };
@@ -24,19 +16,19 @@ export const API_VERSIONS = {
 /**
  * Default API version to use when none is specified
  */
-export const DEFAULT_API_VERSION = 'smartagent.io/v1alpha1';
+export const DEFAULT_API_VERSION = 'agentnet/v1alpha1';
 
 /**
  * Latest stable API version
  */
-export const LATEST_STABLE_VERSION = 'agentnet.io/v1alpha1';
+export const LATEST_STABLE_VERSION = 'agentnet/v1alpha1';
 
 /**
  * Version upgrade paths map
  * Defines valid upgrade paths between versions
  */
 export const VERSION_UPGRADE_PATHS = {
-    'smartagent.io/v1alpha1': ['agentnet.io/v1alpha1']
+    'agentnet/v1alpha1': ['agentnet/v1alpha1']
 };
 
 /**
@@ -154,18 +146,18 @@ export function getUpgradeOptions(fromVersion) {
  */
 const VERSION_MIGRATIONS = {
     // Migration from smartagent.io/v1alpha1 to agentnet.io/v1alpha1
-    'smartagent.io/v1alpha1->agentnet.io/v1alpha1': (definition) => {
+        'smartagent.io/v1alpha1->agentnet.io/v1alpha1': (definition) => {
         // Deep clone the definition to avoid modifying the original
         const newDef = JSON.parse(JSON.stringify(definition));
         
         // Update the apiVersion
-        newDef.apiVersion = 'agentnet.io/v1alpha1';
+        newDef.apiVersion = 'agentnet/v1alpha1';
         
         // Handle specific field migrations
         // Example: rename or restructure fields as needed
         
         // Log the migration
-        logger.info(`Migrated agent definition from smartagent.io/v1alpha1 to agentnet.io/v1alpha1`, { 
+        logger.info(`Migrated agent definition from smartagent.io/v1alpha1 to agentnet/v1alpha1`, { 
             agentName: newDef.metadata?.name 
         });
         

package/src/tools/migrate-version.js

@@ -1,250 +0,0 @@
-#!/usr/bin/env node
-
-import fs from 'fs';
-import path from 'path';
-import { parse, stringify } from 'yaml';
-import { migrateDefinition, validateApiVersion, API_VERSIONS, LATEST_STABLE_VERSION } from '../utils/version.js';
-import { logger } from '../utils/logger.js';
-
-/**
- * Print command usage instructions
- */
-function printUsage() {
-    console.log(`
-Agent Definition Version Migration Utility
-
-Usage:
-  node migrate-version.js <input-file> [options]
-
-Options:
-  --output <file>    Output file (default: adds '-migrated' to input filename)
-  --version <ver>    Target API version (default: ${LATEST_STABLE_VERSION})
-  --check            Only check if migration is needed, don't perform it
-  --quiet            Suppress informational output
-  --help             Show this help message
-
-Examples:
-  node migrate-version.js ./agents.yaml
-  node migrate-version.js ./agents.yaml --version agentnet.io/v1alpha1
-  node migrate-version.js ./agents.yaml --output ./agents-new.yaml
-  node migrate-version.js ./agents.yaml --check
-`);
-}
-
-/**
- * Parse command line arguments
- * @returns {Object} Parsed arguments
- */
-function parseArgs() {
-    const args = process.argv.slice(2);
-    const result = {
-        inputFile: null,
-        outputFile: null,
-        targetVersion: LATEST_STABLE_VERSION,
-        checkOnly: false,
-        quiet: false,
-        help: false
-    };
-
-    for (let i = 0; i < args.length; i++) {
-        const arg = args[i];
-        
-        if (arg === '--help' || arg === '-h') {
-            result.help = true;
-        } else if (arg === '--output' || arg === '-o') {
-            result.outputFile = args[++i];
-        } else if (arg === '--version' || arg === '-v') {
-            result.targetVersion = args[++i];
-        } else if (arg === '--check' || arg === '-c') {
-            result.checkOnly = true;
-        } else if (arg === '--quiet' || arg === '-q') {
-            result.quiet = true;
-        } else if (!result.inputFile) {
-            result.inputFile = arg;
-        }
-    }
-
-    return result;
-}
-
-/**
- * Get default output filename based on input filename
- * @param {string} inputFile - Input file path
- * @param {string} targetVersion - Target API version
- * @returns {string} Default output file path
- */
-function getDefaultOutputFile(inputFile, targetVersion) {
-    const parsedPath = path.parse(inputFile);
-    const versionSuffix = targetVersion.replace(/\//g, '-');
-    return path.join(
-        parsedPath.dir,
-        `${parsedPath.name}-${versionSuffix}${parsedPath.ext}`
-    );
-}
-
-/**
- * Log a message if not in quiet mode
- * @param {string} message - Message to log
- * @param {boolean} isError - Whether this is an error message
- * @param {boolean} quiet - Whether quiet mode is enabled
- */
-function log(message, isError = false, quiet = false) {
-    if (isError || !quiet) {
-        console.log(message);
-    }
-}
-
-/**
- * Process and migrate a YAML file containing agent definitions
- * @param {string} content - YAML content
- * @param {string} targetVersion - Target API version
- * @param {boolean} checkOnly - Only check, don't modify
- * @returns {Object} Migration results including modified content
- */
-function processYamlFile(content, targetVersion, checkOnly) {
-    // Split the YAML content by document separator
-    const documents = content.split(/^---$/m)
-        .map(s => s.trim())
-        .filter(s => s);
-    
-    const result = {
-        migrated: false,
-        migratedCount: 0,
-        alreadyUpToDateCount: 0,
-        failedCount: 0,
-        needsMigration: false,
-        updatedContent: null,
-        failures: []
-    };
-    
-    // Process each document
-    const processedDocs = [];
-    
-    for (let i = 0; i < documents.length; i++) {
-        try {
-            const docContent = documents[i];
-            const doc = parse(docContent);
-            
-            // Skip non-agent definitions
-            if (!doc || doc.kind !== 'AgentDefinition') {
-                processedDocs.push(docContent);
-                continue;
-            }
-            
-            // Check if migration is needed
-            const currentVersion = doc.apiVersion || 'smartagent.io/v1alpha1';
-            const agentName = doc.metadata?.name || `[Document ${i+1}]`;
-            
-            if (currentVersion === targetVersion) {
-                result.alreadyUpToDateCount++;
-                log(`Agent "${agentName}" is already at version ${targetVersion}`);
-                processedDocs.push(docContent);
-                continue;
-            }
-            
-            result.needsMigration = true;
-            
-            // If only checking, skip migration
-            if (checkOnly) {
-                processedDocs.push(docContent);
-                continue;
-            }
-            
-            // Perform migration
-            const migratedDoc = migrateDefinition(doc, targetVersion);
-            result.migratedCount++;
-            result.migrated = true;
-            
-            // Convert back to YAML and add to processed docs
-            processedDocs.push(stringify(migratedDoc));
-            
-            log(`Successfully migrated agent "${agentName}" from ${currentVersion} to ${targetVersion}`);
-            
-        } catch (error) {
-            result.failedCount++;
-            result.failures.push({
-                documentIndex: i,
-                error: error.message
-            });
-            
-            // Keep original document on failure
-            processedDocs.push(documents[i]);
-            
-            log(`Error processing document ${i+1}: ${error.message}`, true);
-        }
-    }
-    
-    // Combine the processed documents back into a single YAML file
-    result.updatedContent = processedDocs.join('\n---\n');
-    
-    return result;
-}
-
-/**
- * Main function
- */
-async function main() {
-    const args = parseArgs();
-    
-    // Show help if requested or no input file
-    if (args.help || !args.inputFile) {
-        printUsage();
-        process.exit(args.help ? 0 : 1);
-    }
-    
-    // Validate target version
-    if (!API_VERSIONS[args.targetVersion]) {
-        log(`Error: Unsupported target version: ${args.targetVersion}`, true);
-        log(`Supported versions: ${Object.keys(API_VERSIONS).join(', ')}`, true);
-        process.exit(1);
-    }
-    
-    try {
-        // Set default output file if not specified
-        if (!args.outputFile && !args.checkOnly) {
-            args.outputFile = getDefaultOutputFile(args.inputFile, args.targetVersion);
-        }
-        
-        // Read input file
-        const content = fs.readFileSync(args.inputFile, 'utf8');
-        
-        // Process the file
-        const result = processYamlFile(content, args.targetVersion, args.checkOnly);
-        
-        // Output results
-        if (result.needsMigration) {
-            if (args.checkOnly) {
-                log(`Migration needed: ${result.alreadyUpToDateCount} up-to-date, ${result.failedCount + documents.length - result.alreadyUpToDateCount} need migration`);
-                process.exit(10); // Special exit code indicating migration needed
-            } else if (result.migrated) {
-                // Write the output file
-                fs.writeFileSync(args.outputFile, result.updatedContent);
-                
-                log(`
-Migration completed:
-  - ${result.migratedCount} agent definitions migrated
-  - ${result.alreadyUpToDateCount} already up-to-date
-  - ${result.failedCount} failures
-  - Output written to: ${args.outputFile}
-                `);
-            }
-        } else {
-            log(`No migration needed. All agent definitions are already at version ${args.targetVersion}`);
-        }
-        
-        // Exit with error if any migrations failed
-        if (result.failedCount > 0) {
-            process.exit(1);
-        }
-        
-    } catch (error) {
-        log(`Error: ${error.message}`, true);
-        process.exit(1);
-    }
-}
-
-// Run the main function
-main().catch(error => {
-    console.error('Unhandled error:', error);
-    process.exit(1);
-});