Amazon Bedrock Agents

123456789101112
import boto3
import json
import uuid

# Initialize the Bedrock Agent Runtime client
bedrock_agent = boto3.client('bedrock-agent-runtime', region_name='us-east-1')

def invoke_agent(agent_id: str, agent_alias_id: str, user_message: str, session_id: str = None):
    """
    Invoke a Bedrock Agent and stream the response.
    
    Args:

123456789101112
import json
import boto3
from datetime import datetime

def lambda_handler(event, context):
    # Extract action group details from Bedrock Agent request
    action_group = event.get('actionGroup')
    api_path = event.get('apiPath')
    http_method = event.get('httpMethod')
    parameters = event.get('parameters', [])
    request_body = event.get('requestBody', {})
    

123456789101112
import boto3
import json

bedrock_agent = boto3.client('bedrock-agent-runtime')

def query_knowledge_base_with_filters(
    knowledge_base_id: str,
    query: str,
    document_type: str = None,
    date_after: str = None,
    department: str = None
):

123456789101112
import boto3
import uuid
from typing import Optional, Dict, Any

class BedrockAgentClient:
    def __init__(self, agent_id: str, agent_alias_id: str, guardrail_id: str):
        self.client = boto3.client('bedrock-agent-runtime')
        self.agent_id = agent_id
        self.agent_alias_id = agent_alias_id
        self.guardrail_id = guardrail_id
    
    def invoke_agent(

123456789101112
import json
import boto3
import logging
from datetime import datetime
from typing import Any, Dict

logger = logging.getLogger()
logger.setLevel(logging.INFO)

dynamodb = boto3.resource('dynamodb')
tickets_table = dynamodb.Table('support-tickets')
orders_table = dynamodb.Table('orders')

123456789101112
{
  "openapi": "3.0.0",
  "info": {
    "title": "Customer Support Actions",
    "version": "1.0.0",
    "description": "Actions for handling customer support requests"
  },
  "paths": {
    "/createTicket": {
      "post": {
        "operationId": "createTicket",
        "summary": "Create a new support ticket for the customer",

123456789101112
import boto3
import json
from dataclasses import dataclass
from typing import List, Dict
from enum import Enum

class TestCategory(Enum):
    NORMAL = "normal"
    PII = "pii"
    DENIED_TOPIC = "denied_topic"
    PROMPT_INJECTION = "prompt_injection"
    EDGE_CASE = "edge_case"