Update backend/app/repositories/dataset_repo.py

backend/app/repositories/dataset_repo.py

@@ -1,12 +1,31 @@
 from sqlalchemy.orm import Session
 from backend.app.models.dataset import Dataset
 
-def create_dataset(db: Session, **kwargs) -> Dataset:
-    obj = Dataset(**kwargs)
-    db.add(obj)
+
+def create_dataset(
+    db: Session,
+    id: str,
+    name: str,
+    file_path: str,
+    row_count: int,
+    column_count: int,
+    schema_json: str,
+    profile_json: str,
+) -> Dataset:
+    dataset = Dataset(
+        id=id,
+        name=name,
+        file_path=file_path,
+        row_count=row_count,
+        column_count=column_count,
+        schema_json=schema_json,
+        profile_json=profile_json,
+    )
+    db.add(dataset)
     db.commit()
-    db.refresh(obj)
-    return obj
+    db.refresh(dataset)
+    return dataset
+
 
 def get_dataset(db: Session, dataset_id: str) -> Dataset | None:
-    return db.query(Dataset).filter(Dataset.id == dataset_id).first()
+    return db.query(Dataset).filter(Dataset.id == dataset_id).first()