Use np.array when committing

Author: pamelafox

src/backend/fastapi_app/setup_postgres_seeddata.py

@@ -4,6 +4,7 @@
 import logging
 import os
 
+import numpy as np
 import sqlalchemy.exc
 from dotenv import load_dotenv
 from sqlalchemy import select, text
@@ -41,8 +42,11 @@ async def seed_data(engine):
                 if db_item.scalars().first():
                     continue
                 attrs = {key: value for key, value in seed_data_object.items()}
-                row = Item(**attrs)
-                session.add(row)
+                attrs["embedding_ada002"] = np.array(seed_data_object["embedding_ada002"])
+                attrs["embedding_nomic"] = np.array(seed_data_object["embedding_nomic"])
+                column_names = ", ".join(attrs.keys())
+                values = ", ".join([f":{key}" for key in attrs.keys()])
+                await session.execute(text(f"INSERT INTO {table_name} ({column_names}) VALUES ({values})"), attrs)
             try:
                 await session.commit()
             except sqlalchemy.exc.IntegrityError: