37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
from django.db import models
|
|
|
|
|
|
class Conversation(models.Model):
|
|
user_id = models.IntegerField(db_index=True) # ارجاع سبک به accounts.User
|
|
lesson_id = models.IntegerField(null=True, blank=True) # ارجاع سبک به content.Lesson
|
|
chapter_id = models.IntegerField(null=True, blank=True) # شناسهی فصل RAG (خارج از content/)
|
|
grade_id = models.IntegerField(null=True, blank=True) # ارجاع سبک به content.Grade
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"conv-{self.pk}-user{self.user_id}"
|
|
|
|
|
|
class Message(models.Model):
|
|
ROLE_CHOICES = [('user', 'کاربر'), ('assistant', 'دستیار')]
|
|
|
|
conversation = models.ForeignKey(Conversation, on_delete=models.CASCADE, related_name='messages')
|
|
role = models.CharField(max_length=10, choices=ROLE_CHOICES)
|
|
text = models.TextField()
|
|
question_id = models.IntegerField(null=True, blank=True) # ارجاع سبک به content.Question
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
ordering = ['created_at']
|
|
|
|
def __str__(self):
|
|
return f"msg-{self.pk}-{self.role}"
|
|
|
|
|
|
class MessageFeedback(models.Model):
|
|
message = models.OneToOneField(Message, on_delete=models.CASCADE, related_name='feedback')
|
|
is_helpful = models.BooleanField()
|
|
|
|
def __str__(self):
|
|
return f"feedback-msg{self.message_id}"
|