Merge branch 'main' of https://github.com/sipe-team/sipethon-3_6_ainsight-be

Author: dev-9hee

apps/answer/migrations/0004_alter_modelanswer_model_id.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.5 on 2025-01-18 10:33
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('answer', '0003_alter_useranswer_question'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='modelanswer',
+            name='model_id',
+            field=models.CharField(max_length=50),
+        ),
+    ]

apps/answer/models.py

@@ -10,7 +10,7 @@ class Meta:
 
 
 class ModelAnswer(models.BaseDateTimeModel):
-    model_id = models.CharField(max_length=20)
+    model_id = models.CharField(max_length=50)
     answer = models.TextField()
     user_answer = models.ForeignKey('answer.UserAnswer', on_delete=models.DO_NOTHING, related_name='model_answers')
 

apps/chat/endpoints.py

@@ -54,11 +54,11 @@ class MessageSchema(Schema):
     message: str
     models: list[str] = ['gpt-4o']
 
+
 @router.post("/chat/stream")
 def chat_stream(request, message_data: MessageSchema):
     def event_stream():
         responses = {}
-        
         # 모델별 응답 생성
         for model in message_data.models:
             # GPT 모델 처리
@@ -143,10 +143,10 @@ def event_stream():
         if text.keys():
             answer = UserAnswer(question=message_data.message)
             answer.save()
-        model_answers = []
-        for model_id in text.keys():
-            model_answers.append(ModelAnswer(model_id=model_id, answer=text[model_id], user_answer=answer))
-        ModelAnswer.objects.bulk_create(model_answers)
+            model_answers = []
+            for model_id in text.keys():
+                model_answers.append(ModelAnswer(model_id=model_id, answer=text[model_id], user_answer=answer))
+            ModelAnswer.objects.bulk_create(model_answers)
 
     return StreamingHttpResponse(
         event_stream(),