fishy/Dynavera
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added error field for training file with UI message

Browse source
This commit is contained in:
Viswamedha Nalabotu 2026-03-22 19:45:05 +00:00
parent f74337f60f
commit a6ed273a13
6 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
apps/knowledge/migrations/0001_initial.py
View file

@ -28,6 +28,7 @@ class Migration(migrations.Migration):
('file_size', models.IntegerField()),
('file_type', models.CharField(max_length=50)),
('description', models.TextField(blank=True, default='')),
('error_message', models.TextField(blank=True, default='')),
('status', models.CharField(choices=[('ingesting', 'Ingesting'), ('chunked', 'Chunked'), ('embedded', 'Embedded'), ('failed', 'Failed')], default='ingesting', max_length=20)),
('organization', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='training_files', to='accounts.organization')),
('role', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='training_files', to='accounts.role')),

1
apps/knowledge/models.py
View file

@ -29,6 +29,7 @@ class TrainingFile(IdentifierMixin, TimeStampMixin, Model):
file_type = CharField(max_length=50)
description = TextField(blank=True, default='')
error_message = TextField(blank=True, default='')
status = CharField(max_length=20, choices=STATUS_CHOICES, default='ingesting')
class Meta:

6
apps/knowledge/serializers.py
View file

@ -14,12 +14,12 @@ class TrainingFileSerializer(ModelSerializer):
model = TrainingFile
fields = [
'id', 'uuid', 'organization', 'role', 'scope', 'uploaded_by', 'file', 'file_url',
'file_name', 'file_size', 'file_type', 'description',
'status', 'created_at', 'updated_at'
'file_name', 'file_size', 'file_type', 'description',
'error_message', 'status', 'created_at', 'updated_at'
]
read_only_fields = [
'id', 'uuid', 'uploaded_by', 'file_size', 'file_type',
'status', 'created_at', 'updated_at',
'error_message', 'status', 'created_at', 'updated_at',
'organization', 'role', 'scope'
]

2
apps/knowledge/tasks.py
View file

@ -115,7 +115,7 @@ def ingest_training_file_task(self, file_uuid):
except Exception as e:
file_obj.status = 'failed'
file_obj.description = str(e)
file_obj.error_message = str(e)
file_obj.save()
raise e

1
site/src/types/organization.ts
View file

@ -44,6 +44,7 @@ export interface TrainingFile {
file_size: number
file_type: string
description: string
error_message: string
is_processed: boolean
status: 'ingesting' | 'chunked' | 'embedded' | 'failed'
file_url: string

12
site/src/views/OrganizationManage.vue
View file

@ -18,6 +18,7 @@ import {
Upload,
Steps,
Table,
Tooltip,
} from 'ant-design-vue'
import { apiClient, isAxiosError, API } from '../router/api'
import { useUserStore } from '../stores/userStore'
@ -324,17 +325,20 @@ const trainingFileColumns = [
},
{
title: 'Status',
dataIndex: 'status',
key: 'status',
customRender: ({ value }: { value: string }) => {
customRender: ({ record }: { record: TrainingFile }) => {
const statusMap: Record<string, { color: string; label: string }> = {
ingesting: { color: 'processing', label: 'Ingesting' },
chunked: { color: 'blue', label: 'Chunked' },
embedded: { color: 'success', label: 'Embedded' },
failed: { color: 'error', label: 'Failed' },
}
const status = statusMap[value] || { color: 'default', label: value }
return h(Tag, { color: status.color }, () => status.label)
const status = statusMap[record.status] || { color: 'default', label: record.status }
const tag = h(Tag, { color: status.color }, () => status.label)
if (record.status === 'failed' && record.error_message) {
return h(Tooltip, { title: record.error_message }, () => tag)
}
return tag
},
},
{