UPDATE

farm_ai_assistant/models.py

@@ -3,6 +3,8 @@ import uuid
 from django.conf import settings
 from django.db import models
 
+from farm_hub.models import FarmHub
+
 
 class Conversation(models.Model):
     uuid = models.UUIDField(default=uuid.uuid4, unique=True, editable=False, db_index=True)
@@ -11,6 +13,13 @@ class Conversation(models.Model):
         on_delete=models.CASCADE,
         related_name="farm_ai_conversations",
     )
+    farm = models.ForeignKey(
+        FarmHub,
+        on_delete=models.CASCADE,
+        related_name="ai_conversations",
+        null=True,
+        blank=True,
+    )
     title = models.CharField(max_length=255, blank=True, default="")
     farm_context = models.JSONField(default=dict, blank=True)
     created_at = models.DateTimeField(auto_now_add=True)
@@ -38,6 +47,13 @@ class Message(models.Model):
         on_delete=models.CASCADE,
         related_name="messages",
     )
+    farm = models.ForeignKey(
+        FarmHub,
+        on_delete=models.CASCADE,
+        related_name="ai_messages",
+        null=True,
+        blank=True,
+    )
     role = models.CharField(max_length=32, choices=ROLE_CHOICES)
     content = models.TextField(blank=True, default="")
     images = models.JSONField(default=list, blank=True)