Merge pull request #867 from ascender1729/sqlite_wrap

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())