@amitdeshmukh/ax-crew 3.8.1 → 3.10.0

package/.env.example

@@ -7,3 +7,6 @@ MISTRAL_API_KEY=sk-mistral-api********cQAA
 GROK_API_KEY=sk-grok-api********cQAA
 TOGETHER_API_KEY=sk-together-api********cQAA
 HUGGINGFACE_API_KEY=sk-huggingface-api********cQAA
+DEEPSEEK_API_KEY=sk-deepseek-api*****cQAA
+REKA_API_KEY=sk-reka-api********cQAA
+GROK_API_KEY=sk-grok-api*********cQAA

package/CHANGELOG.md

@@ -5,6 +5,29 @@ This Changelog format is based on [Keep a Changelog]
 adheres to [Semantic Versioning](https://semver.org/spec/
 v2.0.0.html).
 
+## [3.10.0] - 2025-05-28
+
+### Added
+- Support for Reka AI models through `AxAIReka` provider
+- Support for Grok (xAI) models through `AxAIGrok` provider
+- Added `REKA_API_KEY` and `GROK_API_KEY` environment variable support
+- Added `DEEPSEEK_API_KEY` environment variable support for DeepSeek models
+- Enhanced provider configuration to include all supported AI providers
+
+### Changed
+- Updated `src/config/index.ts` to include API keys for all supported providers
+- Improved provider API key management for better consistency across all supported models
+
+## [3.9.0] - 2025-05-28
+
+### Fixed
+- Updated @ax-llm/ax package to 11.0.47
+- Updated TypeScript target from ES2020 to ES2022 to resolve `ErrorOptions` type compatibility issue with @ax-llm/ax package
+- Improved build process compatibility with latest TypeScript type definitions
+
+### Changed
+- Enhanced TypeScript configuration for better compatibility with modern type definitions
+
 ## [3.8.1] - 2024-03-28
 
 ### Added

package/dist/agents/agentConfig.js

@@ -1,6 +1,6 @@
 import fs from 'fs';
 // Import all the providers
-import { AxAIAnthropic, AxAIOpenAI, AxAIAzureOpenAI, AxAICohere, AxAIDeepSeek, AxAIGoogleGemini, AxAIGroq, AxAIHuggingFace, AxAIMistral, AxAIOllama, AxAITogether } from '@ax-llm/ax';
+import { AxAIAnthropic, AxAIOpenAI, AxAIAzureOpenAI, AxAICohere, AxAIDeepSeek, AxAIGoogleGemini, AxAIGroq, AxAIHuggingFace, AxAIMistral, AxAIOllama, AxAITogether, AxAIReka, AxAIGrok } from '@ax-llm/ax';
 // Import the MCP client and transports
 import { AxMCPClient, AxMCPStdioTransport, AxMCPHTTPTransport } from '@ax-llm/ax';
 import { PROVIDER_API_KEYS } from '../config/index.js';
@@ -16,7 +16,9 @@ const AIConstructors = {
     'mistral': AxAIMistral,
     'ollama': AxAIOllama,
     'openai': AxAIOpenAI,
-    'together': AxAITogether
+    'together': AxAITogether,
+    'reka': AxAIReka,
+    'grok': AxAIGrok
 };
 // Type guard to check if config is stdio transport
 export function isStdioTransport(config) {

package/dist/agents/agentUseCosts.js

@@ -4,6 +4,7 @@ import { Decimal } from 'decimal.js';
  *
  */
 export class StateFulAxAgentUsage {
+    static STATE_KEY_PREFIX = 'agent_usage_';
     static calculateCost(modelUsage, modelInfo) {
         // Handle both direct properties and nested tokens structure
         const promptTokens = modelUsage.tokens?.promptTokens ?? modelUsage.promptTokens;
@@ -116,4 +117,3 @@ export class StateFulAxAgentUsage {
         });
     }
 }
-StateFulAxAgentUsage.STATE_KEY_PREFIX = 'agent_usage_';

package/dist/agents/index.js

@@ -5,6 +5,9 @@ import { parseCrewConfig, parseAgentConfig } from "./agentConfig.js";
 import { StateFulAxAgentUsage } from "./agentUseCosts.js";
 // Extend the AxAgent class from ax-llm
 class StatefulAxAgent extends AxAgent {
+    state;
+    axai;
+    agentName;
     constructor(ai, options, state) {
         const { examples, ...restOptions } = options;
         const formattedOptions = {
@@ -91,6 +94,11 @@ class StatefulAxAgent extends AxAgent {
  * Represents a crew of agents with shared state functionality.
  */
 class AxCrew {
+    crewConfig;
+    functionsRegistry = {};
+    crewId;
+    agents;
+    state;
     /**
      * Creates an instance of AxCrew.
      * @param {CrewConfigInput} crewConfig - Either a path to the agent config file or a JSON object with crew configuration.
@@ -98,40 +106,6 @@ class AxCrew {
      * @param {string} [crewId=uuidv4()] - The unique identifier for the crew.
      */
     constructor(crewConfig, functionsRegistry = {}, crewId = uuidv4()) {
-        this.functionsRegistry = {};
-        /**
-         * Factory function for creating an agent.
-         * @param {string} agentName - The name of the agent to create.
-         * @returns {StatefulAxAgent} The created StatefulAxAgent instance.
-         * @throws Will throw an error if the agent creation fails.
-         */
-        this.createAgent = async (agentName) => {
-            try {
-                const agentConfig = await parseAgentConfig(agentName, this.crewConfig, this.functionsRegistry, this.state);
-                // Destructure with type assertion
-                const { ai, name, description, signature, functions, subAgentNames, examples } = agentConfig;
-                // Get subagents for the AI agent
-                const subAgents = subAgentNames.map((subAgentName) => {
-                    if (!this.agents?.get(subAgentName)) {
-                        throw new Error(`Sub-agent '${subAgentName}' does not exist in available agents.`);
-                    }
-                    return this.agents?.get(subAgentName);
-                });
-                // Create an instance of StatefulAxAgent
-                const agent = new StatefulAxAgent(ai, {
-                    name,
-                    description,
-                    signature,
-                    functions: functions.filter((fn) => fn !== undefined),
-                    agents: subAgents.filter((agent) => agent !== undefined),
-                    examples,
-                }, this.state);
-                return agent;
-            }
-            catch (error) {
-                throw error;
-            }
-        };
         // Basic validation of crew configuration
         if (!crewConfig || typeof crewConfig !== 'object' || !('crew' in crewConfig)) {
             throw new Error('Invalid crew configuration');
@@ -148,6 +122,39 @@ class AxCrew {
         this.agents = new Map();
         this.state = createState(crewId);
     }
+    /**
+     * Factory function for creating an agent.
+     * @param {string} agentName - The name of the agent to create.
+     * @returns {StatefulAxAgent} The created StatefulAxAgent instance.
+     * @throws Will throw an error if the agent creation fails.
+     */
+    createAgent = async (agentName) => {
+        try {
+            const agentConfig = await parseAgentConfig(agentName, this.crewConfig, this.functionsRegistry, this.state);
+            // Destructure with type assertion
+            const { ai, name, description, signature, functions, subAgentNames, examples } = agentConfig;
+            // Get subagents for the AI agent
+            const subAgents = subAgentNames.map((subAgentName) => {
+                if (!this.agents?.get(subAgentName)) {
+                    throw new Error(`Sub-agent '${subAgentName}' does not exist in available agents.`);
+                }
+                return this.agents?.get(subAgentName);
+            });
+            // Create an instance of StatefulAxAgent
+            const agent = new StatefulAxAgent(ai, {
+                name,
+                description,
+                signature,
+                functions: functions.filter((fn) => fn !== undefined),
+                agents: subAgents.filter((agent) => agent !== undefined),
+                examples,
+            }, this.state);
+            return agent;
+        }
+        catch (error) {
+            throw error;
+        }
+    };
     /**
      * Adds an agent to the crew by name.
      * @param {string} agentName - The name of the agent to add.

package/dist/config/index.js

@@ -4,19 +4,25 @@ dotenv.config();
 const ANTHROPIC_API_KEY = process.env.ANTHROPIC_API_KEY;
 const OPENAI_API_KEY = process.env.OPENAI_API_KEY;
 const COHERE_API_KEY = process.env.COHERE_API_KEY;
+const DEEPSEEK_API_KEY = process.env.DEEPSEEK_API_KEY;
 const GEMINI_API_KEY = process.env.GEMINI_API_KEY;
 const GROQ_API_KEY = process.env.GROQ_API_KEY;
 const TOGETHER_API_KEY = process.env.TOGETHER_API_KEY;
 const MISTRAL_API_KEY = process.env.MISTRAL_API_KEY;
 const HUGGINGFACE_API_KEY = process.env.HUGGINGFACE_API_KEY;
+const REKA_API_KEY = process.env.REKA_API_KEY;
+const GROK_API_KEY = process.env.GROK_API_KEY;
 const PROVIDER_API_KEYS = {
     COHERE_API_KEY,
     GEMINI_API_KEY,
     OPENAI_API_KEY,
     ANTHROPIC_API_KEY,
+    DEEPSEEK_API_KEY,
     GROQ_API_KEY,
     TOGETHER_API_KEY,
     MISTRAL_API_KEY,
-    HUGGINGFACE_API_KEY
+    HUGGINGFACE_API_KEY,
+    REKA_API_KEY,
+    GROK_API_KEY
 };
 export { PROVIDER_API_KEYS, };

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "@amitdeshmukh/ax-crew",
-  "version": "3.8.1",
+  "version": "3.10.0",
   "description": "Build and launch a crew of AI agents with shared state. Built with axllm.dev",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -23,7 +23,7 @@
     "uuid": "^10.0.0"
   },
   "peerDependencies": {
-    "@ax-llm/ax": "^11.0.27"
+    "@ax-llm/ax": "^11.0.47"
   },
   "devDependencies": {
     "@testing-library/jest-dom": "^6.6.3",

package/src/agents/agentConfig.ts

@@ -1,6 +1,6 @@
 import fs from 'fs';
 // Import all the providers
-import { AxAIAnthropic, AxAIOpenAI, AxAIAzureOpenAI, AxAICohere, AxAIDeepSeek, AxAIGoogleGemini, AxAIGroq, AxAIHuggingFace, AxAIMistral, AxAIOllama, AxAITogether } from '@ax-llm/ax';
+import { AxAIAnthropic, AxAIOpenAI, AxAIAzureOpenAI, AxAICohere, AxAIDeepSeek, AxAIGoogleGemini, AxAIGroq, AxAIHuggingFace, AxAIMistral, AxAIOllama, AxAITogether, AxAIReka, AxAIGrok } from '@ax-llm/ax';
 // Import Ax types
 import type { AxFunction } from '@ax-llm/ax';
 // Import the MCP client and transports
@@ -27,7 +27,9 @@ const AIConstructors: Record<string, any> = {
   'mistral': AxAIMistral,
   'ollama': AxAIOllama,
   'openai': AxAIOpenAI,
-  'together': AxAITogether
+  'together': AxAITogether,
+  'reka': AxAIReka,
+  'grok': AxAIGrok
 };
 
 // Provider type

package/src/config/index.ts

@@ -5,11 +5,15 @@ dotenv.config();
 const ANTHROPIC_API_KEY: string | undefined = process.env.ANTHROPIC_API_KEY;
 const OPENAI_API_KEY: string | undefined = process.env.OPENAI_API_KEY;
 const COHERE_API_KEY: string | undefined = process.env.COHERE_API_KEY;
+const DEEPSEEK_API_KEY: string | undefined = process.env.DEEPSEEK_API_KEY;
 const GEMINI_API_KEY: string | undefined = process.env.GEMINI_API_KEY;
 const GROQ_API_KEY: string | undefined = process.env.GROQ_API_KEY;
 const TOGETHER_API_KEY: string | undefined = process.env.TOGETHER_API_KEY;
 const MISTRAL_API_KEY: string | undefined = process.env.MISTRAL_API_KEY;
 const HUGGINGFACE_API_KEY: string | undefined = process.env.HUGGINGFACE_API_KEY;
+const REKA_API_KEY: string | undefined = process.env.REKA_API_KEY;
+const GROK_API_KEY: string | undefined = process.env.GROK_API_KEY;
+// Note: Ollama typically doesn't require an API key for local usage
 
 interface ProviderApiKeys {
   [key: string]: string | undefined;
@@ -20,10 +24,13 @@ const PROVIDER_API_KEYS: ProviderApiKeys = {
   GEMINI_API_KEY,
   OPENAI_API_KEY,
   ANTHROPIC_API_KEY,
+  DEEPSEEK_API_KEY,
   GROQ_API_KEY,
   TOGETHER_API_KEY,
   MISTRAL_API_KEY,
-  HUGGINGFACE_API_KEY
+  HUGGINGFACE_API_KEY,
+  REKA_API_KEY,
+  GROK_API_KEY
 };
 
 export {

package/tsconfig.json

@@ -2,7 +2,7 @@
   "compilerOptions": {
     "declaration": true,
     "module": "ESNext",
-    "target": "ES2020",
+    "target": "ES2022",
     "moduleResolution": "node",
     "esModuleInterop": true,
     "allowSyntheticDefaultImports": true,