illuma-agents 1.0.2 → 1.0.4

package/package.json

@@ -1,145 +1,162 @@
-{
-  "name": "illuma-agents",
-  "version": "1.0.2",
-  "main": "./dist/cjs/main.cjs",
-  "module": "./dist/esm/main.mjs",
-  "types": "./dist/types/index.d.ts",
-  "exports": {
-    ".": {
-      "import": "./dist/esm/main.mjs",
-      "require": "./dist/cjs/main.cjs",
-      "types": "./dist/types/index.d.ts"
-    }
-  },
-  "type": "module",
-  "description": "Illuma AI Agents Library",
-  "license": "MIT",
-  "packageManager": "npm@10.5.2",
-  "engines": {
-    "node": ">=14.0.0"
-  },
-  "files": [
-    "dist",
-    "src",
-    "LICENSE",
-    "README.md"
-  ],
-  "scripts": {
-    "prepare": "node husky-setup.js",
-    "prepublishOnly": "npm run build",
-    "build": "NODE_ENV=production rollup -c && tsc -p tsconfig.build.json",
-    "build:dev": "rollup -c",
-    "start": "node dist/esm/main.js",
-    "clean": "node ./config/clean.js",
-    "script": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli.ts",
-    "bun:cli": "bun -r dotenv/config ./src/scripts/cli.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
-    "start:cli": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "content": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/content.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
-    "stream": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/stream.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
-    "code_exec": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "image": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/image.ts --provider 'google' --name 'Jo' --location 'New York, NY'",
-    "code_exec_files": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec_files.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "code_exec_simple": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec_simple.ts --provider 'google' --name 'Jo' --location 'New York, NY'",
-    "simple": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/simple.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "caching": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/caching.ts --name 'Jo' --location 'New York, NY'",
-    "thinking": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/thinking.ts --name 'Jo' --location 'New York, NY'",
-    "memory": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/memory.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "tool-test": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tools.ts --provider 'openrouter' --name 'Jo' --location 'New York, NY'",
-    "search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/search.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
-    "ant_web_search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/ant_web_search.ts --name 'Jo' --location 'New York, NY'",
-    "abort": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/abort.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "start:cli2": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli2.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
-    "script2": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/proto/example_test.ts",
-    "script3": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/proto/example_test_anthropic.ts",
-    "script4": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli4.ts --name 'Jo' --location 'New York, NY'",
-    "script5": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli5.ts --name 'Jo' --location 'New York, NY'",
-    "start:standard": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/standard.ts",
-    "start:cg": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/collab.ts",
-    "start:tasks": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tasks.ts",
-    "start:collab": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/main.ts",
-    "start:collab5": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/collab_design_v5.ts",
-    "start:dev": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/main.ts",
-    "test": "jest",
-    "test:memory": "NODE_OPTIONS='--expose-gc' npx jest src/specs/title.memory-leak.test.ts",
-    "test:all": "npm test -- --testPathIgnorePatterns=title.memory-leak.test.ts && npm run test:memory",
-    "reinstall": "npm run clean && npm ci && rm -rf ./dist && npm run build",
-    "re": "bun run clean && bun install && rm -rf ./dist && turbo build",
-    "lint": "eslint \"{,!(node_modules|venv)/**/}*.{js,jsx,ts,tsx}\" --fix",
-    "format": "prettier --write ."
-  },
-  "overrides": {
-    "@langchain/openai": "0.5.18",
-    "@browserbasehq/stagehand": {
-      "openai": "$openai"
-    }
-  },
-  "dependencies": {
-    "@langchain/anthropic": "^0.3.26",
-    "@langchain/aws": "^0.1.12",
-    "@langchain/community": "^0.3.47",
-    "@langchain/core": "^0.3.62",
-    "@langchain/deepseek": "^0.0.2",
-    "@langchain/google-genai": "^0.2.13",
-    "@langchain/google-vertexai": "^0.2.13",
-    "@langchain/langgraph": "^0.3.4",
-    "@langchain/mistralai": "^0.2.1",
-    "@langchain/ollama": "^0.2.3",
-    "@langchain/openai": "0.5.18",
-    "@langchain/xai": "^0.0.3",
-    "@langfuse/langchain": "^4.3.0",
-    "@langfuse/otel": "^4.3.0",
-    "@langfuse/tracing": "^4.3.0",
-    "@opentelemetry/sdk-node": "^0.207.0",
-    "cheerio": "^1.0.0",
-    "dotenv": "^16.4.7",
-    "https-proxy-agent": "^7.0.6",
-    "nanoid": "^3.3.7",
-    "openai": "^5.10.1"
-  },
-  "imports": {
-    "@/*": "./src/*",
-    "~/*": "./*"
-  },
-  "devDependencies": {
-    "@anthropic-ai/vertex-sdk": "^0.12.0",
-    "@eslint/compat": "^1.2.7",
-    "@rollup/plugin-alias": "^5.1.0",
-    "@rollup/plugin-commonjs": "^28.0.3",
-    "@rollup/plugin-json": "^6.1.0",
-    "@rollup/plugin-node-resolve": "^15.2.3",
-    "@rollup/plugin-terser": "^0.4.4",
-    "@rollup/plugin-typescript": "^12.1.2",
-    "@swc/core": "^1.6.13",
-    "@types/jest": "^29.5.14",
-    "@types/node": "^20.14.11",
-    "@types/yargs-parser": "^21.0.3",
-    "@typescript-eslint/eslint-plugin": "^8.24.0",
-    "@typescript-eslint/parser": "^8.24.0",
-    "eslint": "^9.31.0",
-    "eslint-import-resolver-typescript": "^3.7.0",
-    "eslint-plugin-import": "^2.31.0",
-    "husky": "^9.1.7",
-    "jest": "^29.7.0",
-    "lint-staged": "^15.2.7",
-    "prettier": "^3.5.1",
-    "rollup": "^4.34.6",
-    "rollup-plugin-cleandir": "^2.0.0",
-    "rollup-plugin-obfuscator": "^1.1.0",
-    "ts-jest": "^29.3.1",
-    "ts-node": "^10.9.2",
-    "tsc-alias": "^1.8.10",
-    "tsconfig-paths": "^4.2.0",
-    "tslib": "^2.6.3",
-    "typescript": "^5.5.3",
-    "winston": "^3.17.0"
-  },
-  "lint-staged": {
-    "*.{js,ts}": [
-      "prettier --write",
-      "eslint --fix"
-    ],
-    "*.{json,md}": [
-      "prettier --write"
-    ]
-  }
-}
+{
+  "name": "illuma-agents",
+  "version": "1.0.4",
+  "main": "./dist/cjs/main.cjs",
+  "module": "./dist/esm/main.mjs",
+  "types": "./dist/types/index.d.ts",
+  "exports": {
+    ".": {
+      "import": "./dist/esm/main.mjs",
+      "require": "./dist/cjs/main.cjs",
+      "types": "./dist/types/index.d.ts"
+    }
+  },
+  "type": "module",
+  "description": "Illuma AI Agents Library",
+  "author": "Varun Muppidi",
+  "license": "MIT",
+  "packageManager": "npm@10.5.2",
+  "engines": {
+    "node": ">=14.0.0"
+  },
+  "files": [
+    "dist",
+    "src",
+    "LICENSE",
+    "README.md"
+  ],
+  "scripts": {
+    "prepare": "node husky-setup.js",
+    "prepublishOnly": "npm run build",
+    "build": "cross-env NODE_ENV=production rollup -c && tsc -p tsconfig.build.json",
+    "build:dev": "rollup -c",
+    "start": "node dist/esm/main.js",
+    "clean": "node ./config/clean.js",
+    "script": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli.ts",
+    "bun:cli": "bun -r dotenv/config ./src/scripts/cli.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
+    "start:cli": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
+    "content": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/content.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
+    "stream": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/stream.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
+    "code_exec": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
+    "image": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/image.ts --provider 'google' --name 'Jo' --location 'New York, NY'",
+    "code_exec_files": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec_files.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
+    "code_exec_simple": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec_simple.ts --provider 'google' --name 'Jo' --location 'New York, NY'",
+    "simple": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/simple.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
+    "caching": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/caching.ts --name 'Jo' --location 'New York, NY'",
+    "thinking": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/thinking.ts --name 'Jo' --location 'New York, NY'",
+    "memory": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/memory.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
+    "tool": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tools.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
+    "search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/search.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
+    "ant_web_search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/ant_web_search.ts --name 'Jo' --location 'New York, NY'",
+    "abort": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/abort.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
+    "start:cli2": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli2.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
+    "multi-agent-test": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-test.ts",
+    "test-tools-before-handoff": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/test-tools-before-handoff.ts",
+    "multi-agent-parallel": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-parallel.ts",
+    "multi-agent-chain": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-chain.ts",
+    "multi-agent-sequence": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-sequence.ts",
+    "multi-agent-conditional": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-conditional.ts",
+    "multi-agent-supervisor": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-supervisor.ts",
+    "multi-agent-list-handoff": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/test-multi-agent-list-handoff.ts",
+    "multi-agent-hybrid-flow": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-hybrid-flow.ts",
+    "test-handoff-input": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/test-handoff-input.ts",
+    "test-custom-prompt-key": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/test-custom-prompt-key.ts",
+    "script2": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/proto/example_test.ts",
+    "script3": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/proto/example_test_anthropic.ts",
+    "script4": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli4.ts --name 'Jo' --location 'New York, NY'",
+    "script5": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli5.ts --name 'Jo' --location 'New York, NY'",
+    "start:standard": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/standard.ts",
+    "start:cg": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/collab.ts",
+    "start:tasks": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tasks.ts",
+    "start:collab": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/main.ts",
+    "start:collab5": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/collab_design_v5.ts",
+    "start:dev": "node --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/main.ts",
+    "supervised": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/supervised.ts --provider anthropic --name Jo --location \"New York, NY\"",
+    "test": "jest",
+    "test:memory": "NODE_OPTIONS='--expose-gc' npx jest src/specs/title.memory-leak.test.ts",
+    "test:all": "npm test -- --testPathIgnorePatterns=title.memory-leak.test.ts && npm run test:memory",
+    "reinstall": "npm run clean && npm ci && rm -rf ./dist && npm run build",
+    "re": "bun run clean && bun install && rm -rf ./dist && turbo build",
+    "lint": "eslint \"{,!(node_modules|venv)/**/}*.{js,jsx,ts,tsx}\" --fix",
+    "format": "prettier --write ."
+  },
+  "overrides": {
+    "@langchain/openai": "0.5.18",
+    "@browserbasehq/stagehand": {
+      "openai": "$openai"
+    }
+  },
+  "dependencies": {
+    "@langchain/anthropic": "^0.3.26",
+    "@langchain/aws": "^0.1.15",
+    "@langchain/core": "^0.3.79",
+    "@langchain/deepseek": "^0.0.2",
+    "@langchain/google-genai": "^0.2.13",
+    "@langchain/google-vertexai": "^0.2.13",
+    "@langchain/langgraph": "^0.4.9",
+    "@langchain/mistralai": "^0.2.1",
+    "@langchain/ollama": "^0.2.3",
+    "@langchain/openai": "0.5.18",
+    "@langchain/textsplitters": "^0.1.0",
+    "@langchain/xai": "^0.0.3",
+    "@langfuse/langchain": "^4.3.0",
+    "@langfuse/otel": "^4.3.0",
+    "@langfuse/tracing": "^4.3.0",
+    "@opentelemetry/sdk-node": "^0.207.0",
+    "axios": "^1.12.1",
+    "cheerio": "^1.0.0",
+    "dotenv": "^16.4.7",
+    "https-proxy-agent": "^7.0.6",
+    "mathjs": "^15.1.0",
+    "nanoid": "^3.3.7",
+    "openai": "5.8.2"
+  },
+  "imports": {
+    "@/*": "./src/*",
+    "~/*": "./*"
+  },
+  "devDependencies": {
+    "@anthropic-ai/vertex-sdk": "^0.12.0",
+    "@eslint/compat": "^1.2.7",
+    "@rollup/plugin-alias": "^5.1.0",
+    "@rollup/plugin-commonjs": "^28.0.3",
+    "@rollup/plugin-json": "^6.1.0",
+    "@rollup/plugin-node-resolve": "^15.2.3",
+    "@rollup/plugin-terser": "^0.4.4",
+    "@rollup/plugin-typescript": "^12.1.2",
+    "@swc/core": "^1.6.13",
+    "@types/jest": "^29.5.14",
+    "@types/node": "^20.14.11",
+    "@types/node-fetch": "^2.6.13",
+    "@types/yargs-parser": "^21.0.3",
+    "@typescript-eslint/eslint-plugin": "^8.24.0",
+    "@typescript-eslint/parser": "^8.24.0",
+    "cross-env": "^10.1.0",
+    "eslint": "^9.31.0",
+    "eslint-import-resolver-typescript": "^3.7.0",
+    "eslint-plugin-import": "^2.31.0",
+    "husky": "^9.1.7",
+    "jest": "^29.7.0",
+    "lint-staged": "^15.2.7",
+    "prettier": "^3.5.1",
+    "rollup": "^4.34.6",
+    "rollup-plugin-cleandir": "^2.0.0",
+    "rollup-plugin-obfuscator": "^1.1.0",
+    "ts-jest": "^29.3.1",
+    "ts-node": "^10.9.2",
+    "tsc-alias": "^1.8.10",
+    "tsconfig-paths": "^4.2.0",
+    "tslib": "^2.6.3",
+    "typescript": "^5.5.3",
+    "winston": "^3.17.0"
+  },
+  "lint-staged": {
+    "*.{js,ts}": [
+      "prettier --write",
+      "eslint --fix"
+    ],
+    "*.{json,md}": [
+      "prettier --write"
+    ]
+  }
+}

package/src/agents/AgentContext.ts

@@ -0,0 +1,323 @@
+/* eslint-disable no-console */
+// src/agents/AgentContext.ts
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import { SystemMessage } from '@langchain/core/messages';
+import { RunnableLambda } from '@langchain/core/runnables';
+import type {
+  UsageMetadata,
+  BaseMessage,
+  BaseMessageFields,
+} from '@langchain/core/messages';
+import type { RunnableConfig, Runnable } from '@langchain/core/runnables';
+import type * as t from '@/types';
+import type { createPruneMessages } from '@/messages';
+import { ContentTypes, Providers } from '@/common';
+
+/**
+ * Encapsulates agent-specific state that can vary between agents in a multi-agent system
+ */
+export class AgentContext {
+  /**
+   * Create an AgentContext from configuration with token accounting initialization
+   */
+  static fromConfig(
+    agentConfig: t.AgentInputs,
+    tokenCounter?: t.TokenCounter,
+    indexTokenCountMap?: Record<string, number>
+  ): AgentContext {
+    const {
+      agentId,
+      provider,
+      clientOptions,
+      tools,
+      toolMap,
+      toolEnd,
+      instructions,
+      additional_instructions,
+      streamBuffer,
+      maxContextTokens,
+      reasoningKey,
+      useLegacyContent,
+    } = agentConfig;
+
+    const agentContext = new AgentContext({
+      agentId,
+      provider,
+      clientOptions,
+      maxContextTokens,
+      streamBuffer,
+      tools,
+      toolMap,
+      instructions,
+      additionalInstructions: additional_instructions,
+      reasoningKey,
+      toolEnd,
+      instructionTokens: 0,
+      tokenCounter,
+      useLegacyContent,
+    });
+
+    if (tokenCounter) {
+      const tokenMap = indexTokenCountMap || {};
+      agentContext.indexTokenCountMap = tokenMap;
+      agentContext.tokenCalculationPromise = agentContext
+        .calculateInstructionTokens(tokenCounter)
+        .then(() => {
+          // Update token map with instruction tokens
+          agentContext.updateTokenMapWithInstructions(tokenMap);
+        })
+        .catch((err) => {
+          console.error('Error calculating instruction tokens:', err);
+        });
+    } else if (indexTokenCountMap) {
+      agentContext.indexTokenCountMap = indexTokenCountMap;
+    }
+
+    return agentContext;
+  }
+
+  /** Agent identifier */
+  agentId: string;
+  /** Provider for this specific agent */
+  provider: Providers;
+  /** Client options for this agent */
+  clientOptions?: t.ClientOptions;
+  /** Token count map indexed by message position */
+  indexTokenCountMap: Record<string, number | undefined> = {};
+  /** Maximum context tokens for this agent */
+  maxContextTokens?: number;
+  /** Current usage metadata for this agent */
+  currentUsage?: Partial<UsageMetadata>;
+  /** Prune messages function configured for this agent */
+  pruneMessages?: ReturnType<typeof createPruneMessages>;
+  /** Token counter function for this agent */
+  tokenCounter?: t.TokenCounter;
+  /** Instructions/system message token count */
+  instructionTokens: number = 0;
+  /** The amount of time that should pass before another consecutive API call */
+  streamBuffer?: number;
+  /** Last stream call timestamp for rate limiting */
+  lastStreamCall?: number;
+  /** Tools available to this agent */
+  tools?: t.GraphTools;
+  /** Tool map for this agent */
+  toolMap?: t.ToolMap;
+  /** Instructions for this agent */
+  instructions?: string;
+  /** Additional instructions for this agent */
+  additionalInstructions?: string;
+  /** Reasoning key for this agent */
+  reasoningKey: 'reasoning_content' | 'reasoning' = 'reasoning_content';
+  /** Last token for reasoning detection */
+  lastToken?: string;
+  /** Token type switch state */
+  tokenTypeSwitch?: 'reasoning' | 'content';
+  /** Current token type being processed */
+  currentTokenType: ContentTypes.TEXT | ContentTypes.THINK | 'think_and_text' =
+    ContentTypes.TEXT;
+  /** Whether tools should end the workflow */
+  toolEnd: boolean = false;
+  /** System runnable for this agent */
+  systemRunnable?: Runnable<
+    BaseMessage[],
+    (BaseMessage | SystemMessage)[],
+    RunnableConfig<Record<string, unknown>>
+  >;
+  /** Promise for token calculation initialization */
+  tokenCalculationPromise?: Promise<void>;
+  /** Format content blocks as strings (for legacy compatibility) */
+  useLegacyContent: boolean = false;
+
+  constructor({
+    agentId,
+    provider,
+    clientOptions,
+    maxContextTokens,
+    streamBuffer,
+    tokenCounter,
+    tools,
+    toolMap,
+    instructions,
+    additionalInstructions,
+    reasoningKey,
+    toolEnd,
+    instructionTokens,
+    useLegacyContent,
+  }: {
+    agentId: string;
+    provider: Providers;
+    clientOptions?: t.ClientOptions;
+    maxContextTokens?: number;
+    streamBuffer?: number;
+    tokenCounter?: t.TokenCounter;
+    tools?: t.GraphTools;
+    toolMap?: t.ToolMap;
+    instructions?: string;
+    additionalInstructions?: string;
+    reasoningKey?: 'reasoning_content' | 'reasoning';
+    toolEnd?: boolean;
+    instructionTokens?: number;
+    useLegacyContent?: boolean;
+  }) {
+    this.agentId = agentId;
+    this.provider = provider;
+    this.clientOptions = clientOptions;
+    this.maxContextTokens = maxContextTokens;
+    this.streamBuffer = streamBuffer;
+    this.tokenCounter = tokenCounter;
+    this.tools = tools;
+    this.toolMap = toolMap;
+    this.instructions = instructions;
+    this.additionalInstructions = additionalInstructions;
+    if (reasoningKey) {
+      this.reasoningKey = reasoningKey;
+    }
+    if (toolEnd !== undefined) {
+      this.toolEnd = toolEnd;
+    }
+    if (instructionTokens !== undefined) {
+      this.instructionTokens = instructionTokens;
+    }
+
+    this.useLegacyContent = useLegacyContent ?? false;
+
+    this.systemRunnable = this.createSystemRunnable();
+  }
+
+  /**
+   * Create system runnable from instructions and calculate tokens if tokenCounter is available
+   */
+  private createSystemRunnable():
+    | Runnable<
+        BaseMessage[],
+        (BaseMessage | SystemMessage)[],
+        RunnableConfig<Record<string, unknown>>
+      >
+    | undefined {
+    let finalInstructions: string | BaseMessageFields | undefined =
+      this.instructions;
+
+    if (
+      this.additionalInstructions != null &&
+      this.additionalInstructions !== ''
+    ) {
+      finalInstructions =
+        finalInstructions != null && finalInstructions
+          ? `${finalInstructions}\n\n${this.additionalInstructions}`
+          : this.additionalInstructions;
+    }
+
+    // Handle Anthropic prompt caching
+    if (
+      finalInstructions != null &&
+      finalInstructions !== '' &&
+      this.provider === Providers.ANTHROPIC
+    ) {
+      const anthropicOptions = this.clientOptions as
+        | t.AnthropicClientOptions
+        | undefined;
+      const defaultHeaders = anthropicOptions?.clientOptions?.defaultHeaders as
+        | Record<string, string>
+        | undefined;
+      const anthropicBeta = defaultHeaders?.['anthropic-beta'];
+      if (
+        typeof anthropicBeta === 'string' &&
+        anthropicBeta.includes('prompt-caching')
+      ) {
+        finalInstructions = {
+          content: [
+            {
+              type: 'text',
+              text: this.instructions,
+              cache_control: { type: 'ephemeral' },
+            },
+          ],
+        };
+      }
+    }
+
+    if (finalInstructions != null && finalInstructions !== '') {
+      const systemMessage = new SystemMessage(finalInstructions);
+
+      if (this.tokenCounter) {
+        this.instructionTokens += this.tokenCounter(systemMessage);
+      }
+
+      return RunnableLambda.from((messages: BaseMessage[]) => {
+        return [systemMessage, ...messages];
+      }).withConfig({ runName: 'prompt' });
+    }
+
+    return undefined;
+  }
+
+  /**
+   * Reset context for a new run
+   */
+  reset(): void {
+    this.instructionTokens = 0;
+    this.lastToken = undefined;
+    this.indexTokenCountMap = {};
+    this.currentUsage = undefined;
+    this.pruneMessages = undefined;
+    this.lastStreamCall = undefined;
+    this.tokenTypeSwitch = undefined;
+    this.currentTokenType = ContentTypes.TEXT;
+  }
+
+  /**
+   * Update the token count map with instruction tokens
+   */
+  updateTokenMapWithInstructions(baseTokenMap: Record<string, number>): void {
+    if (this.instructionTokens > 0) {
+      // Shift all indices by the instruction token count
+      const shiftedMap: Record<string, number> = {};
+      for (const [key, value] of Object.entries(baseTokenMap)) {
+        const index = parseInt(key, 10);
+        if (!isNaN(index)) {
+          shiftedMap[String(index)] =
+            value + (index === 0 ? this.instructionTokens : 0);
+        }
+      }
+      this.indexTokenCountMap = shiftedMap;
+    } else {
+      this.indexTokenCountMap = { ...baseTokenMap };
+    }
+  }
+
+  /**
+   * Calculate tool tokens and add to instruction tokens
+   * Note: System message tokens are calculated during systemRunnable creation
+   */
+  async calculateInstructionTokens(
+    tokenCounter: t.TokenCounter
+  ): Promise<void> {
+    let toolTokens = 0;
+    if (this.tools && this.tools.length > 0) {
+      for (const tool of this.tools) {
+        const genericTool = tool as Record<string, unknown>;
+        if (
+          genericTool.schema != null &&
+          typeof genericTool.schema === 'object'
+        ) {
+          const schema = genericTool.schema as {
+            describe: (desc: string) => unknown;
+          };
+          const describedSchema = schema.describe(
+            (genericTool.description as string) || ''
+          );
+          const jsonSchema = zodToJsonSchema(
+            describedSchema as Parameters<typeof zodToJsonSchema>[0],
+            (genericTool.name as string) || ''
+          );
+          toolTokens += tokenCounter(
+            new SystemMessage(JSON.stringify(jsonSchema))
+          );
+        }
+      }
+    }
+
+    // Add tool tokens to existing instruction tokens (which may already include system message tokens)
+    this.instructionTokens += toolTokens;
+  }
+}