Merge pull request #124 from DocShow-AI/create-new-data-profile-bug-fix

bug fix create data profile

backend/databases/data_profile_manager.py

@@ -17,13 +17,13 @@ def create_dataprofile(self, data_profile_data: DataProfileCreateRequest):
         """Create a new DataProfile."""
         new_data_profile = DataProfile(
             name=data_profile_data.name,
-            file_type=data_profile_data.file_type,  # Assuming it's included in the request
-            organization_id=data_profile_data.organization_id,  # Assuming it's included in the request
+            file_type=data_profile_data.file_type,
+            organization_id=data_profile_data.organization_id,
             description=data_profile_data.description,
         )
         self.session.add(new_data_profile)
         self.session.commit()
-        return new_data_profile.to_dict()
+        return new_data_profile
 
     def get_dataprofile_by_id(self, data_profile_id: int):
         """Retrieve a DataProfile by its ID."""

backend/requirements.txt

@@ -15,4 +15,6 @@ python-jose-cryptodome==1.3.2
 passlib==1.7.4
 bcrypt==3.2.0
 alembic==1.12.1
-pydantic[email]==2.4.2
+pydantic[email]==2.4.2
+pytest==6.2.5
+pytest-asyncio==0.15.1

backend/routes/data_profile_routes.py

@@ -29,11 +29,19 @@ async def save_data_profiles(
     with DatabaseManager() as session:
         data_profile_manager = DataProfileManager(session)
         if data_profile_manager.get_dataprofile_by_name(request.name):
-            raise HTTPException(status_code=400, detail="Data Profile alredy exists")
+            raise HTTPException(status_code=400, detail="Data Profile already exists")
 
-        new_data_profile = DataProfile(name=request.name)
+        new_data_profile = DataProfile(
+            name=request.name,
+            description=request.description,
+        )
         created_data_profile = data_profile_manager.create_dataprofile(new_data_profile)
-        response = DataProfileCreateResponse(created_data_profile.name)
+
+        # Make sure to pass the fields as keyword arguments
+        response = DataProfileCreateResponse(
+            name=created_data_profile.name,
+            description=created_data_profile.description,
+        )
         return response