import boto3
def write_memory(user,timestamp,role,content):
    dynamodb = boto3.resource('dynamodb')

    table = dynamodb.Table('assistente-produtos-servicos-memoria') # Replace 'YourTableName' with your actual table name

    item_data = {
        'UserId': user,  # Replace with your partition key attribute and value
        'Timestamp': timestamp,      # Replace with your sort key attribute and value (if applicable)
        'role': role,
        'content': content
    }

    try:
        response = table.put_item(Item=item_data)
    except Exception as e:
        print("Error adding item:", e)
def read_memory(userid):
    dynamodb = boto3.resource('dynamodb')
    table = dynamodb.Table('assistente-produtos-servicos-memoria')

    # Query parameters

    try:
        response = table.query(
            KeyConditionExpression=boto3.dynamodb.conditions.Key('UserId').eq(userid),
            ScanIndexForward=False,  # Descending order
            Limit=30 
        )
        
        items = response.get('Items', [])
        if items:
            latest_items = items
            return latest_items
        else:
            return []

    except Exception as e:
        print("Error querying DynamoDB:", str(e))