OpenAI Swarm implementation of an AI agent swarm to manage a SQL database

sql-ai-agent/load_sql_data.py

@@ -0,0 +1,50 @@
+import sqlite3
+
+"""
+This script is used to create a SQLlite database, add tables
+to it, and insert mock data, all for the OpenAI Swarm demonstration
+with the other Python script in this directory.
+
+Simply run this script with the command:
+
+python load_sql_data.py
+
+And then you will have a database loaded and ready to use
+with the agent swarm!
+"""
+
+def execute_sql_script(cursor, script_file):
+    # Opens the .sql file given as script_file
+    with open(script_file, 'r') as sql_file:
+        sql_script = sql_file.read()
+    
+    # Gets all the sql commands and executes them one at a time
+    statements = sql_script.split(';')
+    for statement in statements:
+        if statement.strip():
+            cursor.execute(statement)
+
+def main():
+    # Connect to the database
+    conn = sqlite3.connect('rss-feed-database.db')
+    cursor = conn.cursor()
+
+    # Execute SQL script to create tables for the AI RSS Feed system
+    execute_sql_script(cursor, 'ai-news-complete-tables.sql')
+    conn.commit()
+
+    # Execute SQL script to insert mock data for the AI RSS Feed system
+    execute_sql_script(cursor, 'ai-news-complete-mock-data.sql')
+    conn.commit()
+
+    # Query table to make sure things are looking good
+    cursor.execute("SELECT * FROM rss_feeds")
+    feeds = cursor.fetchall()
+    for feed in feeds:
+        print(feed)
+
+    # Close the connection
+    conn.close()
+
+if __name__ == "__main__":
+    main()