diff --git a/examples/communication_examples/sqlite_conversation.py b/examples/communication_examples/sqlite_conversation.py new file mode 100644 index 00000000..312f99a9 --- /dev/null +++ b/examples/communication_examples/sqlite_conversation.py @@ -0,0 +1,33 @@ +"""Persist agent dialogue using SQLiteConversation. + +Run `pip install -e .` in the repository root so the ``swarms`` package is +available before executing this script. +""" + +from swarms import Agent +from swarms.communication.sqlite_wrap import SQLiteConversation + +# Configure a conversation store backed by SQLite +conversation_store = SQLiteConversation( + db_path="agent_history.db", + table_name="messages", + enable_logging=True, +) + +# Create an agent that leverages the SQLite-based memory +agent = Agent( + agent_name="SupportAgent", + system_prompt="You are a helpful assistant.", + model_name="gpt-4o-mini", + long_term_memory=conversation_store, + max_loops=1, + # Autosave attempts to call `save()` on the memory object which is not + # implemented in SQLiteConversation, so we disable it for this example. + autosave=False, +) + +response = agent.run("How do I reset my password?") +print(response) + +# Show the conversation as stored in SQLite +print(conversation_store.to_dict())