Agent Infrastructure on AWS

123456789101112
import * as cdk from 'aws-cdk-lib';
import * as lambda from 'aws-cdk-lib/aws-lambda';
import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';
import * as sfn from 'aws-cdk-lib/aws-stepfunctions';
import * as tasks from 'aws-cdk-lib/aws-stepfunctions-tasks';

export class AgentInfrastructureStack extends cdk.Stack {
  constructor(scope: cdk.App, id: string) {
    super(scope, id);

    // State management table
    const stateTable = new dynamodb.Table(this, 'AgentState', {

123456789101112
{
  "Comment": "Agent orchestration with tool selection and execution",
  "StartAt": "AnalyzeIntent",
  "States": {
    "AnalyzeIntent": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:us-east-1:123456789:function:analyze-intent",
      "ResultPath": "$.intent",
      "Next": "SelectTools",
      "Retry": [{
        "ErrorEquals": ["Lambda.ServiceException"],
        "IntervalSeconds": 2,

123456789101112
// Table: AgentState
// PK: Partition Key, SK: Sort Key
// GSI1PK, GSI1SK: Global Secondary Index for alternate access patterns

interface AgentStateItem {
  PK: string;           // SESSION#<sessionId> or USER#<userId>
  SK: string;           // MSG#<timestamp> or TOOL#<executionId> or META
  GSI1PK?: string;      // USER#<userId> or TOOL#<toolName>
  GSI1SK?: string;      // SESSION#<sessionId>#<timestamp>
  
  // Entity-specific attributes
  type: 'message' | 'toolExecution' | 'sessionMeta' | 'memory';

123456789101112
import boto3
import json
from typing import Optional
import hashlib

sqs = boto3.client('sqs')
dynamodb = boto3.resource('dynamodb')
table = dynamodb.Table('AgentExecutionState')

QUEUE_URL = 'https://sqs.us-east-1.amazonaws.com/123456789/agent-continuation'
MAX_ITERATIONS_PER_INVOCATION = 50

123456789101112
import { DynamoDBClient, GetItemCommand, PutItemCommand } from '@aws-sdk/client-dynamodb';
import { LambdaClient, InvokeCommand } from '@aws-sdk/client-lambda';
import { z } from 'zod';

const ToolRequestSchema = z.object({
  toolName: z.enum(['search', 'calculator', 'database']),
  parameters: z.record(z.unknown()),
  sessionId: z.string(),
  requestId: z.string()
});

const TOOL_CONFIG = {

123456789101112
{
  "Comment": "Production Agent Orchestration Workflow",
  "StartAt": "ValidateAndParse",
  "States": {
    "ValidateAndParse": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:us-east-1:123456789:function:agent-validate",
      "ResultPath": "$.parsed",
      "Retry": [{
        "ErrorEquals": ["Lambda.ServiceException"],
        "IntervalSeconds": 2,
        "MaxAttempts": 3,

123456789101112
interface AgentCheckpoint {
  sessionId: string;
  checkpointId: string;
  timestamp: number;
  conversationHistory: Message[];
  pendingActions: Action[];
  completedActions: Action[];
  currentPlan: Plan | null;
  toolResults: Record<string, unknown>;
  metadata: Record<string, unknown>;
}