Feat: Adds cognito and memory

assistente/tools/dynamo.py

@@ -0,0 +1,40 @@
+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))
+