claude-flow 2.0.0-alpha.61 → 2.0.0-alpha.62

package/src/agents/agent-registry.ts

@@ -1,4 +1,3 @@
-import { getErrorMessage } from '../utils/error-handler.js';
 /**
  * Agent Registry with Memory Integration
  * Provides persistent storage and coordination for agent management
@@ -69,8 +68,8 @@ export class AgentRegistry extends EventEmitter {
       tags: [...tags, agent.type, agent.status],
       metadata: {
         registeredBy: 'agent-manager',
-        version: '1.0.0'
-      }
+        version: '1.0.0',
+      },
     };
 
     // Store in memory
@@ -78,7 +77,7 @@ export class AgentRegistry extends EventEmitter {
     await this.memory.store(key, entry, {
       type: 'agent-registry',
       tags: entry.tags,
-      partition: this.namespace
+      partition: this.namespace,
     });
 
     // Update cache
@@ -99,16 +98,18 @@ export class AgentRegistry extends EventEmitter {
     // Merge updates
     entry.agent = { ...entry.agent, ...updates };
     entry.lastUpdated = new Date();
-    entry.tags = [entry.agent.type, entry.agent.status, ...entry.tags.filter(t => 
-      t !== entry.agent.type && t !== entry.agent.status
-    )];
+    entry.tags = [
+      entry.agent.type,
+      entry.agent.status,
+      ...entry.tags.filter((t) => t !== entry.agent.type && t !== entry.agent.status),
+    ];
 
     // Store updated entry
     const key = this.getAgentKey(agentId);
     await this.memory.store(key, entry, {
       type: 'agent-registry',
       tags: entry.tags,
-      partition: this.namespace
+      partition: this.namespace,
     });
 
     // Update cache
@@ -129,15 +130,19 @@ export class AgentRegistry extends EventEmitter {
     if (preserveHistory) {
       // Move to archived partition
       const archiveKey = this.getArchiveKey(agentId);
-      await this.memory.store(archiveKey, {
-        ...entry,
-        archivedAt: new Date(),
-        reason: 'agent_removed'
-      }, {
-        type: 'agent-archive',
-        tags: [...entry.tags, 'archived'],
-        partition: 'archived'
-      });
+      await this.memory.store(
+        archiveKey,
+        {
+          ...entry,
+          archivedAt: new Date(),
+          reason: 'agent_removed',
+        },
+        {
+          type: 'agent-archive',
+          tags: [...entry.tags, 'archived'],
+          partition: 'archived',
+        },
+      );
     }
 
     // Remove from active registry
@@ -170,7 +175,7 @@ export class AgentRegistry extends EventEmitter {
     // Load from memory
     const key = this.getAgentKey(agentId);
     const memoryEntry = await this.memory.retrieve(key);
-    
+
     if (memoryEntry && memoryEntry.value) {
       // Convert MemoryEntry to AgentRegistryEntry
       const registryEntry: AgentRegistryEntry = memoryEntry.value as AgentRegistryEntry;
@@ -187,46 +192,42 @@ export class AgentRegistry extends EventEmitter {
   async queryAgents(query: AgentQuery = {}): Promise<AgentState[]> {
     await this.refreshCacheIfNeeded();
 
-    let agents = Array.from(this.cache.values()).map(entry => entry.agent);
+    let agents = Array.from(this.cache.values()).map((entry) => entry.agent);
 
     // Apply filters
     if (query.type) {
-      agents = agents.filter(agent => agent.type === query.type);
+      agents = agents.filter((agent) => agent.type === query.type);
     }
 
     if (query.status) {
-      agents = agents.filter(agent => agent.status === query.status);
+      agents = agents.filter((agent) => agent.status === query.status);
     }
 
     if (query.healthThreshold !== undefined) {
-      agents = agents.filter(agent => agent.health >= query.healthThreshold!);
+      agents = agents.filter((agent) => agent.health >= query.healthThreshold!);
     }
 
     if (query.namePattern) {
       const pattern = new RegExp(query.namePattern, 'i');
-      agents = agents.filter(agent => pattern.test(agent.name));
+      agents = agents.filter((agent) => pattern.test(agent.name));
     }
 
     if (query.tags && query.tags.length > 0) {
       const entries = Array.from(this.cache.values());
-      const matchingEntries = entries.filter(entry => 
-        query.tags!.some(tag => entry.tags.includes(tag))
+      const matchingEntries = entries.filter((entry) =>
+        query.tags!.some((tag) => entry.tags.includes(tag)),
       );
-      agents = matchingEntries.map(entry => entry.agent);
+      agents = matchingEntries.map((entry) => entry.agent);
     }
 
     if (query.createdAfter) {
       const entries = Array.from(this.cache.values());
-      const matchingEntries = entries.filter(entry => 
-        entry.createdAt >= query.createdAfter!
-      );
-      agents = matchingEntries.map(entry => entry.agent);
+      const matchingEntries = entries.filter((entry) => entry.createdAt >= query.createdAfter!);
+      agents = matchingEntries.map((entry) => entry.agent);
     }
 
     if (query.lastActiveAfter) {
-      agents = agents.filter(agent => 
-        agent.metrics.lastActivity >= query.lastActiveAfter!
-      );
+      agents = agents.filter((agent) => agent.metrics.lastActivity >= query.lastActiveAfter!);
     }
 
     return agents;
@@ -274,7 +275,7 @@ export class AgentRegistry extends EventEmitter {
       activeAgents: 0,
       totalUptime: 0,
       tasksCompleted: 0,
-      successRate: 0
+      successRate: 0,
     };
 
     if (agents.length === 0) {
@@ -285,7 +286,7 @@ export class AgentRegistry extends EventEmitter {
     for (const agent of agents) {
       stats.byType[agent.type] = (stats.byType[agent.type] || 0) + 1;
       stats.byStatus[agent.status] = (stats.byStatus[agent.status] || 0) + 1;
-      
+
       if (agent.status === 'idle' || agent.status === 'busy') {
         stats.activeAgents++;
       }
@@ -296,11 +297,12 @@ export class AgentRegistry extends EventEmitter {
 
     // Calculate averages
     stats.averageHealth = agents.reduce((sum, agent) => sum + agent.health, 0) / agents.length;
-    
-    const totalTasks = agents.reduce((sum, agent) => 
-      sum + agent.metrics.tasksCompleted + agent.metrics.tasksFailed, 0
+
+    const totalTasks = agents.reduce(
+      (sum, agent) => sum + agent.metrics.tasksCompleted + agent.metrics.tasksFailed,
+      0,
     );
-    
+
     if (totalTasks > 0) {
       stats.successRate = stats.tasksCompleted / totalTasks;
     }
@@ -313,17 +315,17 @@ export class AgentRegistry extends EventEmitter {
    */
   async searchByCapabilities(requiredCapabilities: string[]): Promise<AgentState[]> {
     const agents = await this.getAllAgents();
-    
-    return agents.filter(agent => {
+
+    return agents.filter((agent) => {
       const capabilities = [
         ...agent.capabilities.languages,
         ...agent.capabilities.frameworks,
         ...agent.capabilities.domains,
-        ...agent.capabilities.tools
+        ...agent.capabilities.tools,
       ];
 
-      return requiredCapabilities.every(required => 
-        capabilities.some(cap => cap.toLowerCase().includes(required.toLowerCase()))
+      return requiredCapabilities.every((required) =>
+        capabilities.some((cap) => cap.toLowerCase().includes(required.toLowerCase())),
       );
     });
   }
@@ -334,7 +336,7 @@ export class AgentRegistry extends EventEmitter {
   async findBestAgent(
     taskType: string,
     requiredCapabilities: string[] = [],
-    preferredAgent?: string
+    preferredAgent?: string,
   ): Promise<AgentState | null> {
     let candidates = await this.getHealthyAgents(0.5);
 
@@ -345,25 +347,26 @@ export class AgentRegistry extends EventEmitter {
 
     // Prefer specific agent if available and healthy
     if (preferredAgent) {
-      const preferred = candidates.find(agent => 
-        agent.id.id === preferredAgent || agent.name === preferredAgent
+      const preferred = candidates.find(
+        (agent) => agent.id.id === preferredAgent || agent.name === preferredAgent,
       );
       if (preferred) return preferred;
     }
 
     // Filter by availability
-    candidates = candidates.filter(agent => 
-      agent.status === 'idle' && 
-      agent.workload < 0.8 &&
-      agent.capabilities.maxConcurrentTasks > 0
+    candidates = candidates.filter(
+      (agent) =>
+        agent.status === 'idle' &&
+        agent.workload < 0.8 &&
+        agent.capabilities.maxConcurrentTasks > 0,
     );
 
     if (candidates.length === 0) return null;
 
     // Score candidates
-    const scored = candidates.map(agent => ({
+    const scored = candidates.map((agent) => ({
       agent,
-      score: this.calculateAgentScore(agent, taskType, requiredCapabilities)
+      score: this.calculateAgentScore(agent, taskType, requiredCapabilities),
     }));
 
     // Sort by score (highest first)
@@ -377,15 +380,19 @@ export class AgentRegistry extends EventEmitter {
    */
   async storeCoordinationData(agentId: string, data: any): Promise<void> {
     const key = `coordination:${agentId}`;
-    await this.memory.store(key, {
-      agentId,
-      data,
-      timestamp: new Date()
-    }, {
-      type: 'agent-coordination',
-      tags: ['coordination', agentId],
-      partition: this.namespace
-    });
+    await this.memory.store(
+      key,
+      {
+        agentId,
+        data,
+        timestamp: new Date(),
+      },
+      {
+        type: 'agent-coordination',
+        tags: ['coordination', agentId],
+        partition: this.namespace,
+      },
+    );
   }
 
   /**
@@ -403,7 +410,7 @@ export class AgentRegistry extends EventEmitter {
     try {
       const entries = await this.memory.query({
         type: 'state' as const,
-        namespace: this.namespace
+        namespace: this.namespace,
       });
 
       this.cache.clear();
@@ -438,9 +445,9 @@ export class AgentRegistry extends EventEmitter {
   }
 
   private calculateAgentScore(
-    agent: AgentState, 
-    taskType: string, 
-    requiredCapabilities: string[]
+    agent: AgentState,
+    taskType: string,
+    requiredCapabilities: string[],
   ): number {
     let score = 0;
 
@@ -460,11 +467,11 @@ export class AgentRegistry extends EventEmitter {
         ...agent.capabilities.languages,
         ...agent.capabilities.frameworks,
         ...agent.capabilities.domains,
-        ...agent.capabilities.tools
+        ...agent.capabilities.tools,
       ];
 
-      const matches = requiredCapabilities.filter(required =>
-        agentCaps.some(cap => cap.toLowerCase().includes(required.toLowerCase()))
+      const matches = requiredCapabilities.filter((required) =>
+        agentCaps.some((cap) => cap.toLowerCase().includes(required.toLowerCase())),
       );
 
       score += (matches.length / requiredCapabilities.length) * 10;
@@ -472,4 +479,4 @@ export class AgentRegistry extends EventEmitter {
 
     return score;
   }
-}
+}

package/src/api/claude-client.ts

@@ -22,9 +22,9 @@ export interface ClaudeAPIConfig {
   retryDelay?: number;
 }
 
-export type ClaudeModel = 
+export type ClaudeModel =
   | 'claude-3-opus-20240229'
-  | 'claude-3-sonnet-20240229' 
+  | 'claude-3-sonnet-20240229'
   | 'claude-3-haiku-20240307'
   | 'claude-2.1'
   | 'claude-2.0'
@@ -68,7 +68,15 @@ export interface ClaudeResponse {
 }
 
 export interface ClaudeStreamEvent {
-  type: 'message_start' | 'content_block_start' | 'content_block_delta' | 'content_block_stop' | 'message_delta' | 'message_stop' | 'ping' | 'error';
+  type:
+    | 'message_start'
+    | 'content_block_start'
+    | 'content_block_delta'
+    | 'content_block_stop'
+    | 'message_delta'
+    | 'message_stop'
+    | 'ping'
+    | 'error';
   message?: Partial<ClaudeResponse>;
   index?: number;
   delta?: {
@@ -102,7 +110,7 @@ export class ClaudeAPIClient extends EventEmitter {
     super();
     this.logger = logger;
     this.configManager = configManager;
-    
+
     // Load config from environment and merge with provided config
     this.config = this.loadConfiguration(config);
   }
@@ -191,10 +199,10 @@ export class ClaudeAPIClient extends EventEmitter {
   updateConfig(updates: Partial<ClaudeAPIConfig>): void {
     this.config = { ...this.config, ...updates };
     this.validateConfiguration(this.config);
-    this.logger.info('Claude API configuration updated', { 
+    this.logger.info('Claude API configuration updated', {
       model: this.config.model,
       temperature: this.config.temperature,
-      maxTokens: this.config.maxTokens 
+      maxTokens: this.config.maxTokens,
     });
   }
 
@@ -216,13 +224,13 @@ export class ClaudeAPIClient extends EventEmitter {
       maxTokens?: number;
       systemPrompt?: string;
       stream?: boolean;
-    }
+    },
   ): Promise<ClaudeResponse | AsyncIterable<ClaudeStreamEvent>> {
     const request: ClaudeRequest = {
-      model: options?.model || this.config.model!,
+      model: options?.model || this.config.model || 'claude-3-opus-20240229',
       messages,
       system: options?.systemPrompt || this.config.systemPrompt,
-      max_tokens: options?.maxTokens || this.config.maxTokens!,
+      max_tokens: options?.maxTokens || this.config.maxTokens || 4096,
       temperature: options?.temperature ?? this.config.temperature,
       top_p: this.config.topP,
       top_k: this.config.topK,
@@ -249,13 +257,13 @@ export class ClaudeAPIClient extends EventEmitter {
    */
   private async sendRequest(request: ClaudeRequest): Promise<ClaudeResponse> {
     let lastError: Error | undefined;
-    
-    for (let attempt = 0; attempt < this.config.retryAttempts!; attempt++) {
+
+    for (let attempt = 0; attempt < (this.config.retryAttempts || 3); attempt++) {
       try {
         const controller = new AbortController();
-        const timeout = setTimeout(() => controller.abort(), this.config.timeout!);
+        const timeout = setTimeout(() => controller.abort(), this.config.timeout || 30000);
 
-        const response = await fetch(this.config.apiUrl!, {
+        const response = await fetch(this.config.apiUrl || 'https://api.anthropic.com/v1/messages', {
           method: 'POST',
           headers: {
             'Content-Type': 'application/json',
@@ -273,8 +281,8 @@ export class ClaudeAPIClient extends EventEmitter {
           throw new Error(`Claude API error (${response.status}): ${error}`);
         }
 
-        const data = await response.json() as ClaudeResponse;
-        
+        const data = (await response.json()) as ClaudeResponse;
+
         this.logger.info('Claude API response received', {
           model: data.model,
           inputTokens: data.usage.input_tokens,
@@ -284,15 +292,17 @@ export class ClaudeAPIClient extends EventEmitter {
 
         this.emit('response', data);
         return data;
-
       } catch (error) {
         lastError = error as Error;
-        this.logger.warn(`Claude API request failed (attempt ${attempt + 1}/${this.config.retryAttempts})`, {
-          error: getErrorMessage(error),
-        });
+        this.logger.warn(
+          `Claude API request failed (attempt ${attempt + 1}/${this.config.retryAttempts})`,
+          {
+            error: getErrorMessage(error),
+          },
+        );
 
-        if (attempt < this.config.retryAttempts! - 1) {
-          await this.delay(this.config.retryDelay! * Math.pow(2, attempt));
+        if (attempt < (this.config.retryAttempts || 3) - 1) {
+          await this.delay((this.config.retryDelay || 1000) * Math.pow(2, attempt));
         }
       }
     }
@@ -306,10 +316,10 @@ export class ClaudeAPIClient extends EventEmitter {
    */
   private async *streamRequest(request: ClaudeRequest): AsyncIterable<ClaudeStreamEvent> {
     const controller = new AbortController();
-    const timeout = setTimeout(() => controller.abort(), this.config.timeout! * 2); // Double timeout for streaming
+    const timeout = setTimeout(() => controller.abort(), (this.config.timeout || 30000) * 2); // Double timeout for streaming
 
     try {
-      const response = await fetch(this.config.apiUrl!, {
+      const response = await fetch(this.config.apiUrl || 'https://api.anthropic.com/v1/messages', {
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',
@@ -325,7 +335,10 @@ export class ClaudeAPIClient extends EventEmitter {
         throw new Error(`Claude API error (${response.status}): ${error}`);
       }
 
-      const reader = response.body!.getReader();
+      if (!response.body) {
+        throw new Error('Response body is null');
+      }
+      const reader = response.body.getReader();
       const decoder = new TextDecoder();
       let buffer = '';
 
@@ -341,7 +354,7 @@ export class ClaudeAPIClient extends EventEmitter {
           if (line.startsWith('data: ')) {
             const data = line.slice(6);
             if (data === '[DONE]') continue;
-            
+
             try {
               const event = JSON.parse(data) as ClaudeStreamEvent;
               this.emit('stream_event', event);
@@ -367,10 +380,10 @@ export class ClaudeAPIClient extends EventEmitter {
       temperature?: number;
       maxTokens?: number;
       systemPrompt?: string;
-    }
+    },
   ): Promise<string> {
     const messages: ClaudeMessage[] = [{ role: 'user', content: prompt }];
-    const response = await this.sendMessage(messages, options) as ClaudeResponse;
+    const response = (await this.sendMessage(messages, options)) as ClaudeResponse;
     return response.content[0].text;
   }
 
@@ -384,11 +397,14 @@ export class ClaudeAPIClient extends EventEmitter {
       temperature?: number;
       maxTokens?: number;
       systemPrompt?: string;
-    }
+    },
   ): AsyncIterable<string> {
     const messages: ClaudeMessage[] = [{ role: 'user', content: prompt }];
-    const stream = await this.sendMessage(messages, { ...options, stream: true }) as AsyncIterable<ClaudeStreamEvent>;
-    
+    const stream = (await this.sendMessage(messages, {
+      ...options,
+      stream: true,
+    })) as AsyncIterable<ClaudeStreamEvent>;
+
     for await (const event of stream) {
       if (event.type === 'content_block_delta' && event.delta?.text) {
         yield event.delta.text;
@@ -418,7 +434,10 @@ export class ClaudeAPIClient extends EventEmitter {
     contextWindow: number;
     description: string;
   } {
-    const modelInfo: Record<ClaudeModel, { name: string; contextWindow: number; description: string }> = {
+    const modelInfo: Record<
+      ClaudeModel,
+      { name: string; contextWindow: number; description: string }
+    > = {
       'claude-3-opus-20240229': {
         name: 'Claude 3 Opus',
         contextWindow: 200000,
@@ -451,17 +470,19 @@ export class ClaudeAPIClient extends EventEmitter {
       },
     };
 
-    return modelInfo[model] || {
-      name: model,
-      contextWindow: 100000,
-      description: 'Unknown model',
-    };
+    return (
+      modelInfo[model] || {
+        name: model,
+        contextWindow: 100000,
+        description: 'Unknown model',
+      }
+    );
   }
 
   /**
    * Delay helper for retries
    */
   private delay(ms: number): Promise<void> {
-    return new Promise(resolve => setTimeout(resolve, ms));
+    return new Promise((resolve) => setTimeout(resolve, ms));
   }
-}
+}