feat: add conversation name extracted from user's first query context (…

…#37)

frontend/src/components/conversations/conversations-list.tsx

@@ -23,15 +23,23 @@ export function ConversationsList() {
     return await api.conversations().getConversations(dataSetId, page);
   }
 
+  const truncateText = (text: string, maxLength: number): string => {
+    if (text.length <= maxLength) {
+      return text;
+    }
+    return text.substring(0, maxLength - 3) + '...';
+  }
+
   return (
     <PaginatedTable
       loadItems={loadConversations}
       itemsReloadDependencies={dataSetId}
       noItemsMessage="You don't have any conversations yet"
-      tableHeaders={["Time"]}
+      tableHeaders={["Name", "Time"]}
       tableRow={(item, index) => {
         return (
           <TableRow key={index} onClick={() => navigateToConversation(item.id)} className="cursor-pointer">
+            <TableCell>{truncateText(item.summary || "Unnamed Conversation", 180)}</TableCell>
             <TableCell>{new Date(item.started_at).toLocaleString('en-US')}</TableCell>
           </TableRow>
         )

frontend/src/lib/types.ts

@@ -60,6 +60,7 @@ export type Conversation = {
   model: string;
   dimensions: number;
   data_set: string;
+  summary?: string;
   history?: Message[];
 }
 

server/agent/conversation/manager.py

@@ -43,6 +43,11 @@ def respond_to_user_message(self, conversation_id: int, data_set_id: int, user_i
                                               role='user',
                                               text=message)
 
+        # Set the conversation summary if it's the first message
+        if not conversation.summary:
+            conversation.summary = user_message.text
+            conversation.save()
+
         response_text = self.get_answer(conversation, user_message.text)
         response = Message.objects.create(conversation=conversation,
                                           created_at=datetime.now(),

server/agent/migrations/0003_conversation_summary.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.5 on 2025-01-27 12:13
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("agent", "0002_remove_conversation_system_name_and_more"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="conversation",
+            name="summary",
+            field=models.CharField(null=True),
+        ),
+    ]

server/agent/models.py

@@ -9,6 +9,7 @@ class Conversation(models.Model):
     user = models.ForeignKey(get_user_model(), on_delete=models.PROTECT)
     started_at = models.DateTimeField(auto_now_add=True)
     data_set = models.ForeignKey(DataSet, on_delete=models.PROTECT, null=False)
+    summary = models.CharField(null=True)
 
     class Meta:
         db_table_comment = ("A conversation is a collection of various messages exchanged during one session. Messages "

server/agent/serializers.py

@@ -47,7 +47,7 @@ class ConversationCreationSerializer(serializers.Serializer):
 class ConversationSerializer(serializers.ModelSerializer):
     class Meta:
         model = Conversation
-        fields = ['id', 'started_at']
+        fields = ['id', 'started_at', 'summary']
 
 
 class MessageFeedbackSerializer(serializers.ModelSerializer):