@amitdeshmukh/ax-crew 3.11.1 → 3.11.3

package/dist/agents/agentConfig.js

@@ -2,7 +2,8 @@ import fs from 'fs';
 // Import all the providers
 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, AxMCPHTTPSSETransport, AxMCPStreambleHTTPTransport } from '@ax-llm/ax';
+import { AxMCPClient, AxMCPHTTPSSETransport, AxMCPStreambleHTTPTransport } from '@ax-llm/ax';
+import { AxMCPStdioTransport } from '@ax-llm/ax-tools';
 import { PROVIDER_API_KEYS } from '../config/index.js';
 // Define a mapping from provider names to their respective constructors
 const AIConstructors = {
@@ -101,13 +102,12 @@ const initializeMCPServers = async (agentConfigData) => {
                     env: mcpServerConfig.env
                 });
             }
-            else if (isStreambleHTTPTransport(mcpServerConfig)) {
-                transport = new AxMCPStreambleHTTPTransport(mcpServerConfig.mcpEndpoint, mcpServerConfig.options);
-            }
             else if (isHTTPSSETransport(mcpServerConfig)) {
-                // This handles both new SSE transport and legacy HTTP transport (both use sseUrl)
                 transport = new AxMCPHTTPSSETransport(mcpServerConfig.sseUrl);
             }
+            else if (isStreambleHTTPTransport(mcpServerConfig)) {
+                transport = new AxMCPStreambleHTTPTransport(mcpServerConfig.mcpEndpoint, mcpServerConfig.options);
+            }
             else {
                 throw new Error(`Unsupported transport type: ${JSON.stringify(mcpServerConfig)}`);
             }

package/dist/agents/index.d.ts

@@ -10,7 +10,7 @@ declare class StatefulAxAgent extends AxAgent<any, any> {
         name: string;
         description: string;
         signature: string | AxSignature;
-        agents?: AxAgentic[] | undefined;
+        agents?: AxAgentic<any, any>[] | undefined;
         functions?: (AxFunction | (() => AxFunction))[] | undefined;
         examples?: Array<Record<string, any>> | undefined;
         mcpServers?: Record<string, MCPTransportConfig> | undefined;

package/dist/types.d.ts

@@ -1,4 +1,4 @@
-import type { AxFunction, AxSignature, AxModelConfig, AxMCPStreamableHTTPTransportOptions } from '@ax-llm/ax';
+import type { AxFunction, AxSignature, AxModelConfig, AxMCPStreamableHTTPTransportOptions, AxProgramForwardOptions } from '@ax-llm/ax';
 import type { Provider } from './agents/agentConfig.js';
 /**
  * A state instance that is shared between agents.
@@ -138,7 +138,7 @@ type MCPTransportConfig = MCPStdioTransportConfig | MCPHTTPSSETransportConfig |
  * @property {AxModelConfig & { model: string }} ai - The AI model configuration to be passed to the agent.
  * @property {boolean} debug - Whether to enable debug mode.
  * @property {string} apiURL - Set this if you are using a custom API URL e.g. ollama on localhost.
- * @property {Record<string, any>} options - Agent options. Refer to the Ax documentation for more details.
+ * @property {Partial<AxProgramForwardOptions>} options - Agent options including thinkingTokenBudget, showThoughts, etc. Refer to the Ax documentation for more details.
  * @property {string[]} functions - Function names to be used by the agent.
  * @property {string[]} agents - Sub-agent available to the agent.
  * @property {Record<string, any>[]} examples - DSPy examples for the agent to learn from.
@@ -155,7 +155,7 @@ interface AgentConfig {
     };
     debug?: boolean;
     apiURL?: string;
-    options?: Record<string, any>;
+    options?: Partial<AxProgramForwardOptions>;
     functions?: string[];
     agents?: string[];
     examples?: Array<Record<string, any>>;

package/examples/perplexityDeepSearch.ts

@@ -0,0 +1,89 @@
+import { AxCrew } from "../dist/index.js";
+
+import dotenv from "dotenv";
+dotenv.config();
+
+// Define the crew configuration
+const config = {
+  crew: [
+    {
+      name: "DeepResearchAgent",
+      description: "A specialized agent that performs deep research using perplexity",
+      signature: 'researchTopic:string "a topic of interest" -> result:string "The result of the research"',
+      provider: "openai",
+      providerKeyName: "OPENAI_API_KEY",
+      ai: {
+        model: "gpt-4.1",
+        temperature: 0.1,
+      },
+      options: {
+        stream: true,
+        debug: true,
+      },
+      mcpServers: {
+        "perplexity-mcp": {
+          "env": {
+            "PERPLEXITY_API_KEY": process.env.PERPLEXITY_API_KEY,
+            "PERPLEXITY_MODEL": "sonar-deep-research"
+          },
+          "command": "uvx",
+          "args": [
+            "perplexity-mcp"
+          ]
+        }
+      }
+    }
+  ]
+};
+
+// Create a new instance of AxCrew with the config
+const crew = new AxCrew(config);
+
+// Add the agents to the crew
+await crew.addAllAgents();
+
+// Get agent instances
+const researchAgent = crew.agents?.get("DeepResearchAgent");
+
+const userQuery: string = "You are a Research assistant. Your task is to analyse the company SpaceX, its origins, current team members, customer profile and any news worthy happenings. Prepare a detailed report.";
+
+console.log(`\n\nUser Query: ${userQuery}`);
+
+const main = async (): Promise<void> => {
+  // Start timing
+  const startTime = Date.now();
+  console.log(`\n🕐 Starting research task at: ${new Date(startTime).toLocaleTimeString()}`);
+  
+  const response = await researchAgent?.streamingForward({
+    researchTopic: userQuery,
+  });
+
+  if (response) {
+    try {
+      for await (const chunk of response) {
+        if (chunk.delta && typeof chunk.delta === 'object' && 'results' in chunk.delta) {
+          process.stdout.write(chunk.delta.results);
+        }
+      }
+      console.log('\n');
+    } catch (error) {
+      console.error('Error processing stream:', error);
+    }
+  }
+
+  // End timing and calculate duration
+  const endTime = Date.now();
+  const duration = endTime - startTime;
+  const durationInSeconds = (duration / 1000).toFixed(2);
+  const durationInMinutes = (duration / 60000).toFixed(2);
+  
+  console.log(`\n⏱️  Task completed at: ${new Date(endTime).toLocaleTimeString()}`);
+  console.log(`⏱️  Total time taken: ${duration}ms (${durationInSeconds}s / ${durationInMinutes}min)`);
+};
+
+main()
+  .then(() => {
+    console.log("Done");
+    process.exit(0);
+  })
+  .catch(console.error);

package/examples/search-tweets.ts

@@ -0,0 +1,74 @@
+import { AxCrew } from "../dist/index.js";
+
+import dotenv from "dotenv";
+dotenv.config();
+
+// Define the crew configuration
+const config = {
+  crew: [
+    {
+      name: "XSearchAgent",
+      description: "A specialized agent that can search X (Twitter) posts for the latest news and updates about specific topics, people, or events. It can find trending posts, recent tweets, and real-time information from X platform.",
+      signature: 'searchQuery:string "a search query" -> result:string "the response to the user query citing relevant sources including X posts and other web sources"',
+      provider: "grok",
+      providerKeyName: "GROK_API_KEY",
+      ai: {
+        model: "grok-3-latest",
+        temperature: 0.1,
+      },
+      options: {
+        stream: true,
+        debug: true,
+        searchParameters: {  
+          mode: 'on',  
+          returnCitations: true,  
+          maxSearchResults: 10,  
+          sources: [
+            { type: 'x' },
+            { type: 'web' },
+            { type: 'news' }
+          ]  
+        }
+      }
+    }
+  ]
+};
+
+// Create a new instance of AxCrew with the config
+const crew = new AxCrew(config);
+
+// Add the agents to the crew
+await crew.addAllAgents();
+
+// Get agent instances
+const xSearchAgent = crew.agents?.get("XSearchAgent");
+
+const userQuery: string = "when is the next ISRO launch date and what is the launch vehicle and payload";
+
+console.log(`\n\nUser Query: ${userQuery}`);
+
+const main = async (): Promise<void> => {
+  const response = await xSearchAgent?.streamingForward({
+    searchQuery: userQuery,
+  });
+
+  if (response) {
+    try {
+      for await (const chunk of response) {
+        if (chunk.delta && typeof chunk.delta === 'object' && 'results' in chunk.delta) {
+          process.stdout.write(chunk.delta.results);
+        }
+      }
+      console.log('\n');
+    } catch (error) {
+      console.error('Error processing stream:', error);
+    }
+  }
+};
+
+main()
+  .then(() => {
+    console.log("Done");
+    process.exit(0);
+  })
+  .catch(console.error);

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "@amitdeshmukh/ax-crew",
-  "version": "3.11.1",
+  "version": "3.11.3",
   "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",
@@ -17,12 +17,15 @@
     "test:ui": "vitest --ui"
   },
   "dependencies": {
-    "@ax-llm/ax": "^11.0.50",
     "decimal.js": "^10.5.0",
     "dotenv": "^16.4.5",
     "upgrade": "^1.1.0",
     "uuid": "^10.0.0"
   },
+  "peerDependencies": {
+    "@ax-llm/ax": "^13.0.8",
+    "@ax-llm/ax-tools": "^13.0.8"
+  },
   "devDependencies": {
     "@testing-library/jest-dom": "^6.6.3",
     "@types/node": "^20.14.9",

package/src/agents/agentConfig.ts

@@ -4,7 +4,8 @@ import { AxAIAnthropic, AxAIOpenAI, AxAIAzureOpenAI, AxAICohere, AxAIDeepSeek, A
 // Import Ax types
 import type { AxFunction } from '@ax-llm/ax';
 // Import the MCP client and transports
-import { AxMCPClient, AxMCPStdioTransport, AxMCPHTTPSSETransport, AxMCPStreambleHTTPTransport } from '@ax-llm/ax'
+import { AxMCPClient, AxMCPHTTPSSETransport, AxMCPStreambleHTTPTransport } from '@ax-llm/ax'
+import { AxMCPStdioTransport } from '@ax-llm/ax-tools'
 import { PROVIDER_API_KEYS } from '../config/index.js';
 import type { 
   AgentConfig,
@@ -128,11 +129,10 @@ const initializeMCPServers = async (agentConfigData: AgentConfig): Promise<AxFun
           args: mcpServerConfig.args,
           env: mcpServerConfig.env
         });
-      } else if (isStreambleHTTPTransport(mcpServerConfig)) {
-        transport = new AxMCPStreambleHTTPTransport(mcpServerConfig.mcpEndpoint, mcpServerConfig.options);
       } else if (isHTTPSSETransport(mcpServerConfig)) {
-        // This handles both new SSE transport and legacy HTTP transport (both use sseUrl)
         transport = new AxMCPHTTPSSETransport(mcpServerConfig.sseUrl);
+      } else if (isStreambleHTTPTransport(mcpServerConfig)) {
+        transport = new AxMCPStreambleHTTPTransport(mcpServerConfig.mcpEndpoint, mcpServerConfig.options);
       } else {  
         throw new Error(`Unsupported transport type: ${JSON.stringify(mcpServerConfig)}`);
       }

package/src/agents/index.ts

@@ -50,7 +50,7 @@ class StatefulAxAgent extends AxAgent<any, any> {
       name: string;
       description: string;
       signature: string | AxSignature;
-      agents?: AxAgentic[] | undefined;
+      agents?: AxAgentic<any, any>[] | undefined;
       functions?: (AxFunction | (() => AxFunction))[] | undefined;
       examples?: Array<Record<string, any>> | undefined;
       mcpServers?: Record<string, MCPTransportConfig> | undefined;

package/src/types.ts

@@ -2,7 +2,8 @@ import type {
   AxFunction, 
   AxSignature, 
   AxModelConfig,
-  AxMCPStreamableHTTPTransportOptions
+  AxMCPStreamableHTTPTransportOptions,
+  AxProgramForwardOptions
 } from '@ax-llm/ax';
 
 import type { Provider } from './agents/agentConfig.js';
@@ -154,7 +155,7 @@ type MCPTransportConfig = MCPStdioTransportConfig | MCPHTTPSSETransportConfig |
  * @property {AxModelConfig & { model: string }} ai - The AI model configuration to be passed to the agent.
  * @property {boolean} debug - Whether to enable debug mode.
  * @property {string} apiURL - Set this if you are using a custom API URL e.g. ollama on localhost.
- * @property {Record<string, any>} options - Agent options. Refer to the Ax documentation for more details.
+ * @property {Partial<AxProgramForwardOptions>} options - Agent options including thinkingTokenBudget, showThoughts, etc. Refer to the Ax documentation for more details.
  * @property {string[]} functions - Function names to be used by the agent.
  * @property {string[]} agents - Sub-agent available to the agent.
  * @property {Record<string, any>[]} examples - DSPy examples for the agent to learn from.
@@ -169,7 +170,7 @@ interface AgentConfig {
   ai: AxModelConfig & { model: string };
   debug?: boolean;
   apiURL?: string;
-  options?: Record<string, any>;
+  options?: Partial<AxProgramForwardOptions>;
   functions?: string[];
   agents?: string[];
   examples?: Array<Record<string, any>>;