feat: add user_id to new_llm_configs and image_generation_configs for user association

surfsense_backend/app/db.py

@@ -1032,6 +1032,12 @@ class ImageGenerationConfig(BaseModel, TimestampMixin):
         "SearchSpace", back_populates="image_generation_configs"
     )
 
+    # User who created this config
+    user_id = Column(
+        UUID(as_uuid=True), ForeignKey("user.id", ondelete="CASCADE"), nullable=False
+    )
+    user = relationship("User", back_populates="image_generation_configs")
+
 
 class ImageGeneration(BaseModel, TimestampMixin):
     """
@@ -1244,6 +1250,7 @@ class SearchSourceConnector(BaseModel, TimestampMixin):
     user_id = Column(
         UUID(as_uuid=True), ForeignKey("user.id", ondelete="CASCADE"), nullable=False
     )
+    user = relationship("User", back_populates="search_source_connectors")
 
     # Documents created by this connector (for cleanup on connector deletion)
     documents = relationship("Document", back_populates="connector")
@@ -1300,6 +1307,12 @@ class NewLLMConfig(BaseModel, TimestampMixin):
     )
     search_space = relationship("SearchSpace", back_populates="new_llm_configs")
 
+    # User who created this config
+    user_id = Column(
+        UUID(as_uuid=True), ForeignKey("user.id", ondelete="CASCADE"), nullable=False
+    )
+    user = relationship("User", back_populates="new_llm_configs")
+
 
 class Log(BaseModel, TimestampMixin):
     __tablename__ = "logs"
@@ -1568,6 +1581,27 @@ if config.AUTH_TYPE == "GOOGLE":
             passive_deletes=True,
         )
 
+        # Connectors created by this user
+        search_source_connectors = relationship(
+            "SearchSourceConnector",
+            back_populates="user",
+            passive_deletes=True,
+        )
+
+        # LLM configs created by this user
+        new_llm_configs = relationship(
+            "NewLLMConfig",
+            back_populates="user",
+            passive_deletes=True,
+        )
+
+        # Image generation configs created by this user
+        image_generation_configs = relationship(
+            "ImageGenerationConfig",
+            back_populates="user",
+            passive_deletes=True,
+        )
+
         # User memories for personalized AI responses
         memories = relationship(
             "UserMemory",
@@ -1647,6 +1681,27 @@ else:
             passive_deletes=True,
         )
 
+        # Connectors created by this user
+        search_source_connectors = relationship(
+            "SearchSourceConnector",
+            back_populates="user",
+            passive_deletes=True,
+        )
+
+        # LLM configs created by this user
+        new_llm_configs = relationship(
+            "NewLLMConfig",
+            back_populates="user",
+            passive_deletes=True,
+        )
+
+        # Image generation configs created by this user
+        image_generation_configs = relationship(
+            "ImageGenerationConfig",
+            back_populates="user",
+            passive_deletes=True,
+        )
+
         # User memories for personalized AI responses
         memories = relationship(
             "UserMemory",

surfsense_backend/app/routes/image_generation_routes.py

@@ -273,7 +273,7 @@ async def create_image_gen_config(
             "You don't have permission to create image generation configs in this search space",
         )
 
-        db_config = ImageGenerationConfig(**config_data.model_dump())
+        db_config = ImageGenerationConfig(**config_data.model_dump(), user_id=user.id)
         session.add(db_config)
         await session.commit()
         await session.refresh(db_config)

surfsense_backend/app/routes/new_llm_config_routes.py

@@ -149,8 +149,8 @@ async def create_new_llm_config(
                 detail=f"Invalid LLM configuration: {error_message}",
             )
 
-        # Create the config
-        db_config = NewLLMConfig(**config_data.model_dump())
+        # Create the config with user association
+        db_config = NewLLMConfig(**config_data.model_dump(), user_id=user.id)
         session.add(db_config)
         await session.commit()
         await session.refresh(db_config)

surfsense_backend/app/schemas/image_generation.py

@@ -6,6 +6,7 @@ ImageGeneration: Schemas for the actual image generation requests/results.
 GlobalImageGenConfigRead: Schema for admin-configured YAML configs.
 """
 
+import uuid
 from datetime import datetime
 from typing import Any
 
@@ -79,6 +80,7 @@ class ImageGenerationConfigRead(ImageGenerationConfigBase):
     id: int
     created_at: datetime
     search_space_id: int
+    user_id: uuid.UUID
 
     model_config = ConfigDict(from_attributes=True)
 
@@ -97,6 +99,7 @@ class ImageGenerationConfigPublic(BaseModel):
     litellm_params: dict[str, Any] | None = None
     created_at: datetime
     search_space_id: int
+    user_id: uuid.UUID
 
     model_config = ConfigDict(from_attributes=True)
 

surfsense_backend/app/schemas/new_llm_config.py

@@ -7,6 +7,7 @@ NewLLMConfig combines LLM model settings with prompt configuration:
 - Citation toggle
 """
 
+import uuid
 from datetime import datetime
 from typing import Any
 
@@ -90,6 +91,7 @@ class NewLLMConfigRead(NewLLMConfigBase):
     id: int
     created_at: datetime
     search_space_id: int
+    user_id: uuid.UUID
 
     model_config = ConfigDict(from_attributes=True)
 
@@ -118,6 +120,7 @@ class NewLLMConfigPublic(BaseModel):
 
     created_at: datetime
     search_space_id: int
+    user_id: uuid.UUID
 
     model_config = ConfigDict(from_attributes=True)