Spaces:
Runtime error
Runtime error
File size: 1,344 Bytes
4962437 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
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())
``` |