Spaces:
Runtime error
Runtime error
| To store your conversation on DynamoDB you can use DynamoDbConversationMemoryDriver. | |
| ```python | |
| from swarms.memory.structure import ConversationMemory | |
| from swarms.memory.structure import ConversationMemoryElement, Turn, Message | |
| from swarms.drivers import DynamoDbConversationMemoryDriver | |
| # Instantiate DynamoDbConversationMemoryDriver | |
| dynamo_driver = DynamoDbConversationMemoryDriver( | |
| aws_region="us-east-1", | |
| table_name="conversations", | |
| partition_key="convo_id", | |
| value_attribute_key="convo_data", | |
| partition_key_value="convo1" | |
| ) | |
| # Create a ConversationMemory structure | |
| conv_mem = ConversationMemory( | |
| turns=[ | |
| Turn( | |
| turn_index=0, | |
| system=Message("Hello"), | |
| user=Message("Hi") | |
| ), | |
| Turn( | |
| turn_index=1, | |
| system=Message("How can I assist you today?"), | |
| user=Message("I need some information") | |
| ) | |
| ], | |
| latest_turn=Turn( | |
| turn_index=2, | |
| system=Message("Sure, what information do you need?"), | |
| user=None # user has not yet responded | |
| ), | |
| driver=dynamo_driver # set the driver | |
| ) | |
| # Store the conversation in DynamoDB | |
| dynamo_driver.store(conv_mem) | |
| # Load the conversation from DynamoDB | |
| loaded_conv_mem = dynamo_driver.load() | |
| # Display the loaded conversation | |
| print(loaded_conv_mem.to_json()) | |
| ``` |