@amitdeshmukh/ax-crew 1.0.1 → 1.0.4

package/dist/agents/agentConfig.js

@@ -1,8 +1,8 @@
 import fs from 'fs';
 import yaml from 'js-yaml';
-import { PROVIDER_API_KEYS } from '../config/index';
-import { functions as importedFunctions } from '../functions/index';
 import { AxAIAnthropic, AxAIOpenAI, AxAIAzureOpenAI, AxAICohere, AxAIDeepSeek, AxAIGoogleGemini, AxAIGroq, AxAIHuggingFace, AxAIMistral, AxAIOllama, AxAITogether } from '@ax-llm/ax';
+import { PROVIDER_API_KEYS } from '../config/index.js';
+import { functions as importedFunctions } from '../functions/index.js';
 // Define a mapping from provider names to their respective constructors
 const AIConstructors = {
     'anthropic': AxAIAnthropic,

package/dist/agents/index.js

@@ -1,7 +1,7 @@
 import { v4 as uuidv4 } from 'uuid';
-import { getAgentConfigParams } from './agentConfig';
 import { AxAgent } from '@ax-llm/ax';
-import { createState } from '../state/createState';
+import { getAgentConfigParams } from './agentConfig.js';
+import { createState } from '../state/createState.js';
 // Extend the AxAgent class to include shared state functionality
 class StatefulAxAgent extends AxAgent {
     constructor(ai, options, state) {
@@ -18,8 +18,8 @@ class StatefulAxAgent extends AxAgent {
  */
 class AxCrew {
     /**
-     * Creates an instance of AgentCrew.
-     * @param {string} configFilePath - The file path to the agent configuration.
+     * Creates an instance of AxCrew.
+     * @param {string} configFilePath - Path to the agent config file.
      * @param {string} [crewId=uuidv4()] - The unique identifier for the crew.
      */
     constructor(configFilePath, crewId = uuidv4()) {
@@ -71,12 +71,13 @@ class AxCrew {
         }
     }
     /**
-     * Sets up agents by their names.
-     * For each name provided, it will add the agent to the crew if it's not already present.
-     * @param {string[]} agentNames - An array of agent names to set up.
+     * Sets up agents in the crew by name.
+     * For an array of Agent names provided, it adds
+     * the agent to the crew if not already present.
+     * @param {string[]} agentNames - An array of agent names to configure.
      * @returns {Map<string, StatefulAxAgent> | null} A map of agent names to their corresponding instances.
      */
-    setupAgents(agentNames) {
+    addAgentsToCrew(agentNames) {
         agentNames.forEach(agentName => {
             this.addAgent(agentName);
         });
@@ -90,4 +91,4 @@ class AxCrew {
         this.state.reset();
     }
 }
-export default AxCrew;
+export { AxCrew };

package/dist/functions/index.js

@@ -1,4 +1,4 @@
-import { CurrentDateTime, DaysBetweenDates } from './dateTime';
+import { CurrentDateTime, DaysBetweenDates } from './dateTime.js';
 const functions = {
     CurrentDateTime,
     DaysBetweenDates

package/dist/index.js

@@ -1,2 +1,2 @@
-import AxCrew from './agents/index.js';
+import { AxCrew } from './agents/index.js';
 export { AxCrew };

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "@amitdeshmukh/ax-crew",
-  "version": "1.0.1",
+  "version": "1.0.4",
   "description": "Build and launch a crew of AI agents with shared state. Built with axllm.dev",
   "main": "dist/index.js",
   "engines": {

package/src/agents/agentConfig.ts

@@ -1,11 +1,12 @@
 import fs from 'fs';
 import yaml from 'js-yaml';
-import { PROVIDER_API_KEYS } from '../config/index';
-import { functions as importedFunctions } from '../functions/index';
 
 import { AxAIAnthropic, AxAIOpenAI, AxAIAzureOpenAI, AxAICohere, AxAIDeepSeek, AxAIGoogleGemini, AxAIGroq, AxAIHuggingFace, AxAIMistral, AxAIOllama, AxAITogether } from '@ax-llm/ax';
 import type { AxAI, AxModelConfig, AxFunction, AxSignature } from '@ax-llm/ax';
 
+import { PROVIDER_API_KEYS } from '../config/index.js';
+import { functions as importedFunctions } from '../functions/index.js';
+
 // Define a mapping from provider names to their respective constructors
 const AIConstructors: Record<string, any> = {
   'anthropic': AxAIAnthropic,

package/src/agents/index.ts

@@ -1,15 +1,10 @@
 import { v4 as uuidv4 } from 'uuid';
-import { getAgentConfigParams } from './agentConfig';
 import { AxAgent, AxAI } from '@ax-llm/ax';
-import type { AxSignature, AxAgentic, AxFunction, AxAIService } from '@ax-llm/ax';
-import { createState, StateInstance } from '../state/createState';
-
-
-// Define interfaces for the agent configurations and return types
-interface Agents {
-  [key: string]: StatefulAxAgent;
-}
+import type { AxSignature, AxAgentic, AxFunction } from '@ax-llm/ax';
+import { getAgentConfigParams } from './agentConfig.js';
+import { createState, StateInstance } from '../state/createState.js';
 
+// Define the interface for the agent configuration
 interface AgentConfigParams {
   ai: AxAI;
   name: string;
@@ -32,7 +27,6 @@ class StatefulAxAgent extends AxAgent<any, any> {
   }
 }
 
-
 /**
  * Represents a crew of agents with shared state functionality.
  */
@@ -43,8 +37,8 @@ class AxCrew {
   state: StateInstance;
 
   /**
-   * Creates an instance of AgentCrew.
-   * @param {string} configFilePath - The file path to the agent configuration.
+   * Creates an instance of AxCrew.
+   * @param {string} configFilePath - Path to the agent config file.
    * @param {string} [crewId=uuidv4()] - The unique identifier for the crew.
    */
   constructor(configFilePath: string, crewId: string = uuidv4()) {
@@ -109,12 +103,13 @@ class AxCrew {
   }  
 
   /**
-   * Sets up agents by their names.
-   * For each name provided, it will add the agent to the crew if it's not already present.
-   * @param {string[]} agentNames - An array of agent names to set up.
+   * Sets up agents in the crew by name.
+   * For an array of Agent names provided, it adds
+   * the agent to the crew if not already present.
+   * @param {string[]} agentNames - An array of agent names to configure.
    * @returns {Map<string, StatefulAxAgent> | null} A map of agent names to their corresponding instances.
    */
-  setupAgents(agentNames: string[]): Map<string, StatefulAxAgent> | null {
+  addAgentsToCrew(agentNames: string[]): Map<string, StatefulAxAgent> | null {
     agentNames.forEach(agentName => {
       this.addAgent(agentName);
     });
@@ -130,4 +125,4 @@ class AxCrew {
   }  
 }
 
-export default AxCrew;
+export { AxCrew };

package/src/functions/index.ts

@@ -1,4 +1,4 @@
-import { CurrentDateTime, DaysBetweenDates } from './dateTime';
+import { CurrentDateTime, DaysBetweenDates } from './dateTime.js';
 
 const functions = {
   CurrentDateTime,

package/src/index.ts

@@ -1,3 +1,3 @@
-import AxCrew from './agents/index.js';
+import { AxCrew } from './agents/index.js';
 
 export { AxCrew };

package/test.js

@@ -1,9 +1,9 @@
 // Import the AgentCrew class
-import AgentCrew from './dist/agents/index.js';
+import AxCrew from './dist/agents/index.js';
 
 // Create a new instance of AgentCrew
 const configFilePath = './agent_config.example.yaml';
-const crew = new AgentCrew(configFilePath);
+const crew = new AxCrew(configFilePath);
 
 // Add agents by providing their names
 const agentNames = ['Planner', 'Calculator', 'Manager'];