@amitdeshmukh/ax-crew 4.1.2 → 6.0.0

package/dist/agents/index.js

@@ -156,17 +156,43 @@ class StatefulAxAgent extends AxAgent {
     // Get the accumulated costs for all runs of this agent
     getAccumulatedCosts() { return null; }
     // Metrics API for this agent
+    /**
+     * Get the current metrics snapshot for this agent.
+     * Includes request counts, error rates, token usage, estimated USD cost, and function call stats.
+     *
+     * @returns A metrics snapshot scoped to this agent within its crew.
+     */
     getMetrics() {
         const crewId = this.state?.crewId || (this.state.get?.('crewId')) || 'default';
         return MetricsRegistry.snapshot({ crewId, agent: this.agentName });
     }
+    /**
+     * Reset all tracked metrics for this agent (does not affect other agents).
+     * Call this to start fresh measurement windows for the agent.
+     */
     resetMetrics() {
         const crewId = this.state?.crewId || (this.state.get?.('crewId')) || 'default';
         MetricsRegistry.reset({ crewId, agent: this.agentName });
     }
 }
 /**
- * Represents a crew of agents with shared state functionality.
+ * AxCrew orchestrates a set of Ax agents that share state,
+ * tools (functions), optional MCP servers, streaming, and a built-in metrics
+ * registry for tokens, requests, and estimated cost.
+ *
+ * Typical usage:
+ *  const crew = new AxCrew(config, AxCrewFunctions)
+ *  await crew.addAllAgents()
+ *  const planner = crew.agents?.get("Planner")
+ *  const res = await planner?.forward({ task: "Plan something" })
+ *
+ * Key behaviors:
+ * - Validates and instantiates agents from a config-first model
+ * - Shares a mutable state object across all agents in the crew
+ * - Supports sub-agents and a function registry per agent
+ * - Tracks per-agent and crew-level metrics via MetricsRegistry
+ * - Provides helpers to add agents (individually, a subset, or all) and
+ *   to reset metrics/costs when needed
  */
 class AxCrew {
     crewConfig;
@@ -176,7 +202,7 @@ class AxCrew {
     state;
     /**
      * Creates an instance of AxCrew.
-     * @param {CrewConfigInput} crewConfig - Either a path to the agent config file or a JSON object with crew configuration.
+     * @param {AxCrewConfig} crewConfig - JSON object with crew configuration.
      * @param {FunctionRegistryType} [functionsRegistry={}] - The registry of functions to use in the crew.
      * @param {string} [crewId=uuidv4()] - The unique identifier for the crew.
      */
@@ -186,7 +212,7 @@ class AxCrew {
             throw new Error('Invalid crew configuration');
         }
         // Validate each agent in the crew
-        crewConfig.crew.forEach(agent => {
+        crewConfig.crew.forEach((agent) => {
             if (!agent.name || agent.name.trim() === '') {
                 throw new Error('Agent name cannot be empty');
             }
@@ -376,7 +402,8 @@ class AxCrew {
         this.state.reset();
     }
     /**
-     * Resets all cost tracking for the crew
+     * Resets all cost and usage tracking for the entire crew.
+     * Also calls each agent's `resetUsage` (if available) and clears crew-level metrics.
      */
     resetCosts() {
         // Reset AxAgent built-in usage and our metrics registry
@@ -395,9 +422,19 @@ class AxCrew {
         MetricsRegistry.reset({ crewId: this.crewId });
     }
     // Metrics API
+    /**
+     * Get an aggregate metrics snapshot for the entire crew.
+     * Sums requests, errors, tokens, and estimated cost across all agents in the crew.
+     *
+     * @returns Crew-level metrics snapshot.
+     */
     getCrewMetrics() {
         return MetricsRegistry.snapshotCrew(this.crewId);
     }
+    /**
+     * Reset all tracked metrics for the entire crew.
+     * Use to clear totals before a new measurement period.
+     */
     resetCrewMetrics() {
         MetricsRegistry.reset({ crewId: this.crewId });
     }

package/dist/functions/index.d.ts

@@ -1,3 +1,14 @@
+/**
+ * Built-in function registry for AxCrew agents.
+ *
+ * Contains common utility tools/functions that can be referenced by name from
+ * agent configs (e.g., "functions": ["CurrentDateTime", "DaysBetweenDates"]).
+ * You can pass this object to the AxCrew constructor or merge with your
+ * own registry.
+ * Example:
+ * const crew = new AxCrew(config, AxCrewFunctions); or
+ * const crew = new AxCrew(config, { ...AxCrewFunctions, ...myFunctions });
+ */
 declare const AxCrewFunctions: {
     CurrentDateTime: import("@ax-llm/ax").AxFunction;
     DaysBetweenDates: import("@ax-llm/ax").AxFunction;

package/dist/functions/index.js

@@ -1,5 +1,15 @@
 import { CurrentDateTime, DaysBetweenDates } from './dateTime.js';
-// Built-in functions
+/**
+ * Built-in function registry for AxCrew agents.
+ *
+ * Contains common utility tools/functions that can be referenced by name from
+ * agent configs (e.g., "functions": ["CurrentDateTime", "DaysBetweenDates"]).
+ * You can pass this object to the AxCrew constructor or merge with your
+ * own registry.
+ * Example:
+ * const crew = new AxCrew(config, AxCrewFunctions); or
+ * const crew = new AxCrew(config, { ...AxCrewFunctions, ...myFunctions });
+ */
 const AxCrewFunctions = {
     CurrentDateTime,
     DaysBetweenDates

package/dist/index.d.ts

@@ -1,15 +1,32 @@
 import { AxCrew } from './agents/index.js';
 import { AxCrewFunctions } from './functions/index.js';
-import type { CrewConfigInput, AgentConfig } from './types.js';
+import type { AxCrewConfig, AgentConfig } from './types.js';
 import type { UsageCost, AggregatedMetrics, AggregatedCosts, StateInstance, FunctionRegistryType } from './types.js';
+/**
+ * Metrics types and helpers for request counts, token usage, and estimated cost.
+ *
+ * Re-exports the metrics module for convenience:
+ *  - Types: TokenUsage, MetricsSnapshot, etc.
+ *  - Namespace: MetricsRegistry (record/snapshot/reset helpers)
+ */
 export * from './metrics/index.js';
+/**
+ * MetricsRegistry provides functions to record requests, tokens, and cost,
+ * and to snapshot/reset metrics at agent or crew granularity.
+ */
 export { MetricsRegistry } from './metrics/index.js';
 /**
- * The configuration for an AxCrew.
- *
- * @property {AgentConfig[]} crew - The agents that make up the crew.
+ * Create and manage a crew of Ax agents that share state and metrics.
+ * See the `AxCrew` class for full documentation.
+ */
+declare const _AxCrew: typeof AxCrew;
+/**
+ * Built-in function registry with common tools that can be referenced by name
+ * from agent configs, or extended with your own functions.
  */
-interface AxCrewConfig {
-    crew: AgentConfig[];
-}
-export { AxCrew, AxCrewFunctions, FunctionRegistryType, type AggregatedMetrics, type AggregatedCosts, type AgentConfig, type CrewConfigInput, type AxCrewConfig, type StateInstance, type UsageCost, };
+declare const _AxCrewFunctions: typeof AxCrewFunctions;
+export { 
+/** See class JSDoc on the `AxCrew` implementation. */
+_AxCrew as AxCrew, 
+/** Built-in function registry; see file docs in `src/functions/index.ts`. */
+_AxCrewFunctions as AxCrewFunctions, FunctionRegistryType, type AggregatedMetrics, type AggregatedCosts, type AgentConfig, type AxCrewConfig, type StateInstance, type UsageCost, };

package/dist/index.js

@@ -1,5 +1,30 @@
 import { AxCrew } from './agents/index.js';
 import { AxCrewFunctions } from './functions/index.js';
+/**
+ * Metrics types and helpers for request counts, token usage, and estimated cost.
+ *
+ * Re-exports the metrics module for convenience:
+ *  - Types: TokenUsage, MetricsSnapshot, etc.
+ *  - Namespace: MetricsRegistry (record/snapshot/reset helpers)
+ */
 export * from './metrics/index.js';
+/**
+ * MetricsRegistry provides functions to record requests, tokens, and cost,
+ * and to snapshot/reset metrics at agent or crew granularity.
+ */
 export { MetricsRegistry } from './metrics/index.js';
-export { AxCrew, AxCrewFunctions, };
+/**
+ * Create and manage a crew of Ax agents that share state and metrics.
+ * See the `AxCrew` class for full documentation.
+ */
+const _AxCrew = AxCrew;
+/**
+ * Built-in function registry with common tools that can be referenced by name
+ * from agent configs, or extended with your own functions.
+ */
+const _AxCrewFunctions = AxCrewFunctions;
+export { 
+/** See class JSDoc on the `AxCrew` implementation. */
+_AxCrew as AxCrew, 
+/** Built-in function registry; see file docs in `src/functions/index.ts`. */
+_AxCrewFunctions as AxCrewFunctions, };

package/dist/metrics/registry.d.ts

@@ -1,9 +1,36 @@
 import type { LabelKeys, MetricsSnapshot, TokenUsage } from './types.js';
+/**
+ * Record a completed request.
+ * @param labels Crew/agent/provider/model identifiers
+ * @param streaming Whether this was a streaming request
+ * @param durationMs Duration in milliseconds
+ */
 export declare function recordRequest(labels: LabelKeys, streaming: boolean, durationMs: number): void;
+/**
+ * Record an error occurrence for the given labels.
+ */
 export declare function recordError(labels: LabelKeys): void;
+/**
+ * Record token usage for a request (prompt and completion).
+ */
 export declare function recordTokens(labels: LabelKeys, usage: TokenUsage): void;
+/**
+ * Add estimated cost (USD) to the cumulative total for the labels.
+ */
 export declare function recordEstimatedCost(labels: LabelKeys, usd: number): void;
+/**
+ * Record a function call invocation and add its latency to totals.
+ */
 export declare function recordFunctionCall(labels: LabelKeys, latencyMs: number): void;
+/**
+ * Get a metrics snapshot for specific labels (crew + agent + optional provider/model).
+ */
 export declare function snapshot(labels: LabelKeys): MetricsSnapshot;
+/**
+ * Reset metrics for specific labels, or clear all if no labels provided.
+ */
 export declare function reset(labels?: LabelKeys): void;
+/**
+ * Aggregate a crew-wide metrics snapshot across all agents in the crew.
+ */
 export declare function snapshotCrew(crewId: string): MetricsSnapshot;

package/dist/metrics/registry.js

@@ -24,6 +24,12 @@ function getOrInit(labels) {
     }
     return c;
 }
+/**
+ * Record a completed request.
+ * @param labels Crew/agent/provider/model identifiers
+ * @param streaming Whether this was a streaming request
+ * @param durationMs Duration in milliseconds
+ */
 export function recordRequest(labels, streaming, durationMs) {
     const c = getOrInit(labels);
     c.requests += 1;
@@ -32,26 +38,41 @@ export function recordRequest(labels, streaming, durationMs) {
     c.durationMsSum += durationMs;
     c.durationCount += 1;
 }
+/**
+ * Record an error occurrence for the given labels.
+ */
 export function recordError(labels) {
     const c = getOrInit(labels);
     c.errors += 1;
 }
+/**
+ * Record token usage for a request (prompt and completion).
+ */
 export function recordTokens(labels, usage) {
     const c = getOrInit(labels);
     c.inputTokens += usage.promptTokens || 0;
     c.outputTokens += usage.completionTokens || 0;
 }
+/**
+ * Add estimated cost (USD) to the cumulative total for the labels.
+ */
 export function recordEstimatedCost(labels, usd) {
     const c = getOrInit(labels);
     const current = new Big(c.estimatedCostUSD || 0);
     const addition = new Big(usd || 0);
     c.estimatedCostUSD = Number(current.plus(addition));
 }
+/**
+ * Record a function call invocation and add its latency to totals.
+ */
 export function recordFunctionCall(labels, latencyMs) {
     const c = getOrInit(labels);
     c.functionCalls += 1;
     c.functionLatencyMs += latencyMs || 0;
 }
+/**
+ * Get a metrics snapshot for specific labels (crew + agent + optional provider/model).
+ */
 export function snapshot(labels) {
     const c = getOrInit(labels);
     const totalTokens = c.inputTokens + c.outputTokens;
@@ -78,6 +99,9 @@ export function snapshot(labels) {
         },
     };
 }
+/**
+ * Reset metrics for specific labels, or clear all if no labels provided.
+ */
 export function reset(labels) {
     if (!labels) {
         store.clear();
@@ -86,6 +110,9 @@ export function reset(labels) {
     const k = keyOf(labels);
     store.delete(k);
 }
+/**
+ * Aggregate a crew-wide metrics snapshot across all agents in the crew.
+ */
 export function snapshotCrew(crewId) {
     const empty = {
         requests: 0,

package/dist/types.d.ts

@@ -1,5 +1,5 @@
-import type { AxFunction, AxSignature, AxModelConfig, AxMCPStreamableHTTPTransportOptions, AxProgramForwardOptions } from '@ax-llm/ax';
-export type Provider = 'openai' | 'anthropic' | 'google-gemini' | 'mistral' | 'groq' | 'cohere' | 'together' | 'deepseek' | 'ollama' | 'huggingface' | 'openrouter' | 'azure-openai' | 'reka' | 'x-grok';
+import type { AxFunction, AxSignature, AxModelConfig, AxMCPStreamableHTTPTransportOptions, AxProgramForwardOptions, AxAIArgs } from '@ax-llm/ax';
+export type Provider = AxAIArgs<any>['name'];
 /**
  * A state instance that is shared between agents.
  * This can be used to store data that becomes available to all agents and functions in an out-of-band manner.
@@ -165,6 +165,11 @@ interface AgentConfig {
     };
     debug?: boolean;
     apiURL?: string;
+    /**
+     * Provider-specific arguments that are forwarded to the underlying Ax factory.
+     * Example (azure-openai): { resourceName, deploymentName, version }
+     */
+    providerArgs?: Record<string, unknown>;
     options?: Partial<AxProgramForwardOptions<any>> & Record<string, any>;
     functions?: string[];
     agents?: string[];
@@ -172,9 +177,43 @@ interface AgentConfig {
     mcpServers?: Record<string, MCPTransportConfig>;
 }
 /**
- * The input type for the agent config. This can be a path to a JSON file or a JSON object.
+ * The configuration object for an AxCrew instance.
+ *
+ * @property {AgentConfig[]} crew - The agents that make up the crew.
+ * @example
+ * const config: AxCrewConfig = {
+ *   crew: [
+ *     {
+ *       name: "Agent1",
+ *       description: "Agent 1 description",
+ *       signature: "signature",
+ *       provider: "provider",
+ *       providerKeyName: "providerKeyName",
+ *       ai: {
+ *         model: "model",
+ *         temperature: 0,
+ *       },
+ *       options: {
+ *         debug: true,
+ *       },
+ *       functions: ["function1", "function2"],
+ *       agents: ["agent2"],
+ *     },
+ *     {
+ *       name: "Agent2",
+ *       description: "Agent 2 description",
+ *       signature: "signature",
+ *       provider: "provider",
+ *       providerKeyName: "providerKeyName",
+ *       ai: {
+ *         model: "model",
+ *         temperature: 0,
+ *       }
+ *   ]
+ * }
+ * const crew = new AxCrew(config);
  */
-type CrewConfigInput = string | {
+interface AxCrewConfig {
     crew: AgentConfig[];
-};
-export { type AgentConfig, type CrewConfigInput, type AggregatedMetrics, type StateInstance, type FunctionRegistryType, type MCPStdioTransportConfig, type MCPHTTPSSETransportConfig, type MCPStreamableHTTPTransportConfig, type MCPTransportConfig, type ModelUsage, type ModelInfo, type UsageCost, type AggregatedCosts };
+}
+export { type AgentConfig, type AxCrewConfig, type AggregatedMetrics, type StateInstance, type FunctionRegistryType, type MCPStdioTransportConfig, type MCPHTTPSSETransportConfig, type MCPStreamableHTTPTransportConfig, type MCPTransportConfig, type ModelUsage, type ModelInfo, type UsageCost, type AggregatedCosts };

package/examples/basic-researcher-writer.ts

@@ -2,6 +2,8 @@ import { AxCrew } from "../dist/index.js";
 import { AxCrewFunctions } from "../dist/functions/index.js";
 import type { AxCrewConfig } from "../dist/index.js";
 import type { Provider } from "../dist/types.js";
+import dotenv from "dotenv";
+dotenv.config();
 
 // Example agent configuration
 const agentConfig: AxCrewConfig = {
@@ -26,10 +28,10 @@ const agentConfig: AxCrewConfig = {
       name: "writer",
       description: "A writing agent that creates content",
       signature: "topic:string -> article:string",
-      provider: "anthropic" as Provider,
-      providerKeyName: "ANTHROPIC_API_KEY",
+      provider: "google-gemini" as Provider,
+      providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "claude-3-haiku-20240307",
+        model: "gemini-2.5-flash-lite",
         maxTokens: 4000,
         stream: true
       },

package/examples/mcp-agent.ts

@@ -1,4 +1,5 @@
-import { AxCrew, AxCrewConfig } from "../dist/index.js";
+import { AxCrew } from "../dist/index.js";
+import type { AxCrewConfig } from "../dist/types.js";
 
 import dotenv from "dotenv";
 dotenv.config();
@@ -7,31 +8,23 @@ dotenv.config();
 const config = {
   crew: [
     {
-      name: "MapsAgent",
-      description: "A specialized agent with access to Google Maps APIs that can: geocode addresses to coordinates and vice versa, search for and get details about places, calculate travel distances and times between multiple locations, provide elevation data, and generate navigation directions between points.",
-      prompt: "You are a precise geospatial assistant with expert knowledge of Google Maps APIs. Answer user queries by combining place search, geocoding, distance matrix, elevation, and directions. Provide concise, actionable answers, include travel mode assumptions, and cite uncertainties. When location is ambiguous, ask a brief clarifying question before proceeding.",
-      signature: 'userQuery:string "a question to be answered" -> answer:string "the answer to the question"',
-      provider: "anthropic",
-      providerKeyName: "ANTHROPIC_API_KEY",
+      name: "Context7DocsAgent",
+      description: "A specialized agent with access to Context7 Docs APIs that can: search for and get details about API docs",
+      signature: 'apiDocQuery:string "a question to be answered" -> apiDocAnswer:string "the answer to the question"',
+      provider: "google-gemini",
+      providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "claude-3-5-sonnet-latest",
+        model: "gemini-2.5-pro",
         temperature: 0,
-        maxTokens: 1000,
-        stream: true
+        stream: false
       },
       options: {
         debug: true
       },
       "mcpServers": {
-        "google-maps": {
+        "context7": {
           "command": "npx",
-          "args": [
-            "-y",
-            "@modelcontextprotocol/server-google-maps"
-          ],
-          "env": {
-            "GOOGLE_MAPS_API_KEY": process.env.GOOGLE_MAPS_API_KEY
-          }
+          "args": ["-y", "@upstash/context7-mcp", "--api-key", process.env.CONTEXT7_API_KEY]
         }
       },
     },
@@ -41,35 +34,19 @@ const config = {
       prompt: "You are a manager agent that orchestrates tools and sub-agents. Read the user's objective, optionally produce a short plan, then call the MapsAgent when geospatial knowledge is needed. Keep answers direct and avoid extraneous commentary.",
       signature:
         'question:string "a question to be answered" -> answer:string "the answer to the question"',
-      provider: "openai", 
-      providerKeyName: "OPENAI_API_KEY",
+      provider: "google-gemini", 
+      providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "gpt-4o-mini",
+        model: "gemini-2.5-pro",
         maxTokens: 1000,
         temperature: 0,
-        stream: true
+        stream: false
       },
       options: {
         debug: true,
       },
-      agents: ["MapsAgent"]
-    },
-    {
-      name: "MathAgent",
-      description: "Solves math problems",
-      signature:
-        'mathProblem:string "a sentence describing a math problem to be solved using Python code" -> solution:string "the answer to the math problem"',
-      provider: "google-gemini",
-      providerKeyName: "GEMINI_API_KEY",
-      ai: {
-        model: "gemini-1.5-pro",
-        temperature: 0,
-        stream: true
-      },
-      options: {
-        debug: false,
-      },
-    },
+      agents: ["Context7DocsAgent"]
+    }
   ],
 };
 
@@ -81,9 +58,9 @@ await crew.addAllAgents();
 
 // Get agent instances
 const managerAgent = crew.agents?.get("ManagerAgent");
-const mapsAgent = crew.agents?.get("MapsAgent");
+const context7DocsAgent = crew.agents?.get("Context7DocsAgent");
 
-const userQuery: string = "Are there any cool bars around the Eiffel Tower in Paris within 5 min walking distance";
+const userQuery: string = "How do i create an agent in the @amitdeshmukh/ax-crew framework and configure it to use an MCP server? Give me a concrete example.";
 
 console.log(`\n\nQuestion: ${userQuery}`);
 
@@ -97,7 +74,7 @@ const main = async (): Promise<void> => {
     // Print metrics
     console.log("\nMetrics:\n+++++++++++++++++++++++++++++++++");
     console.log("Manager Agent Metrics:", JSON.stringify((managerAgent as any)?.getMetrics?.(), null, 2));
-    console.log("Maps Agent Metrics:", JSON.stringify((mapsAgent as any)?.getMetrics?.(), null, 2));
+    console.log("Context7 Docs Agent Metrics:", JSON.stringify((context7DocsAgent as any)?.getMetrics?.(), null, 2));
     console.log("Crew Metrics:", JSON.stringify((crew as any)?.getCrewMetrics?.(), null, 2));
 };
 

package/examples/perplexityDeepSearch.ts

@@ -1,19 +1,20 @@
 import { AxCrew } from "../dist/index.js";
+import type { AxCrewConfig } from "../dist/types.js";
 
 import dotenv from "dotenv";
 dotenv.config();
 
 // Define the crew configuration
-const config = {
+const config: AxCrewConfig = {
   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",
+      signature: 'researchTopic:string "a topic of interest" -> researchResult:string "The result of the research"',
+      provider: "google-gemini",
+      providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "gpt-4.1",
+        model: "gemini-2.5-flash-lite",
         temperature: 0.1,
       },
       options: {

package/examples/providerArgs.ts

@@ -0,0 +1,42 @@
+import { AxCrew } from "../dist/index.js";
+import type { AxCrewConfig } from "../dist/types.js";
+
+const crewConfig: AxCrewConfig = {
+  crew: [
+    {
+      name: "TestAgent",
+      description: "Test Agent for testing provider arguments",
+      provider: "azure-openai",
+      providerKeyName: "AZURE_OPENAI_API_KEY",
+      signature: "userQuery:string -> answer:string",
+      ai: {
+        model: "gpt-5-mini",
+        temperature: 0,
+        stream: false
+      },
+      providerArgs: {
+        resourceName: "your-resource-name",
+        deploymentName: "your-deployment-name",
+        version: "2025-01-01-preview"
+      },
+      functions: [],
+      options: {
+        debug: true,
+        stream: false
+      }
+    }
+  ]
+};
+
+const crew = new AxCrew(crewConfig);
+await crew.addAllAgents();
+
+const testAgent = crew.agents?.get("TestAgent");
+
+const response = await testAgent?.forward({
+  userQuery: "What is the capital of France?"
+});
+
+console.log(response?.answer);
+
+console.log(testAgent?.getAccumulatedCosts());

package/examples/search-tweets.ts

@@ -1,15 +1,16 @@
 import { AxCrew } from "../dist/index.js";
+import type { AxCrewConfig } from "../dist/types.js";
 
 import dotenv from "dotenv";
 dotenv.config();
 
 // Define the crew configuration
-const config = {
+const config: AxCrewConfig = {
   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"',
+      signature: 'searchQuery:string "a search query" -> searchResults:string "the response to the user query citing relevant sources including X posts and other web sources"',
       provider: "grok",
       providerKeyName: "GROK_API_KEY",
       ai: {
@@ -55,8 +56,8 @@ const main = async (): Promise<void> => {
   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);
+        if (chunk.delta && typeof chunk.delta === 'object' && 'searchResults' in chunk.delta) {
+          process.stdout.write(chunk.delta.searchResults);
         }
       }
       console.log('\n');

package/examples/solve-math-problem.ts

@@ -1,5 +1,7 @@
 import { AxCrew } from "../dist/index.js";
 import type { AxCrewConfig } from "../src/index.js";
+import dotenv from "dotenv";
+dotenv.config();
 
 // Define the crew configuration
 const config: AxCrewConfig = {
@@ -12,11 +14,11 @@ const config: AxCrewConfig = {
       provider: "google-gemini",
       providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "gemini-1.5-pro",
+        model: "gemini-2.5-flash-lite",
         temperature: 0,
       },
       options: {
-        debug: false,
+        debug: true,
         codeExecution: true,
       },
     },
@@ -25,10 +27,10 @@ const config: AxCrewConfig = {
       description: "Completes a user specified task",
       signature:
         'question:string "a question to be answered" -> answer:string "the answer to the question"',
-      provider: "openai", 
-      providerKeyName: "OPENAI_API_KEY",
+      provider: "google-gemini", 
+      providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "gpt-4o-mini",
+        model: "gemini-2.5-flash-lite",
         maxTokens: 1000,
         temperature: 0,
       },

package/examples/streaming.ts

@@ -15,7 +15,7 @@ const config: AxCrewConfig = {
       provider: "google-gemini", 
       providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "gemini-2.0-flash",
+        model: "gemini-2.5-flash",
         maxTokens: 1000,
         temperature: 0,
       },
@@ -33,7 +33,7 @@ const config: AxCrewConfig = {
       provider: "google-gemini",
       providerKeyName: "GEMINI_API_KEY",
       ai: {
-        model: "gemini-1.5-pro",
+        model: "gemini-2.5-pro",
         temperature: 0,
       },
       options: {