Skip to content

Commit b1637be

Browse files
committed
perf: Assistant uses workspace-level Q&A configuration
1 parent 041a0a3 commit b1637be

File tree

2 files changed

+22
-46
lines changed

2 files changed

+22
-46
lines changed

backend/apps/data_training/curd/data_training.py

Lines changed: 10 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -162,10 +162,7 @@ def create_training(session: SessionDep, info: DataTrainingInfo, oid: int, trans
162162

163163
# 检查数据源和高级应用不能同时为空
164164
if info.datasource is None and info.advanced_application is None:
165-
if oid == 1:
166-
raise Exception(trans("i18n_data_training.datasource_assistant_cannot_be_none"))
167-
else:
168-
raise Exception(trans("i18n_data_training.datasource_cannot_be_none"))
165+
raise Exception(trans("i18n_data_training.datasource_assistant_cannot_be_none"))
169166

170167
# 检查重复记录
171168
stmt = select(DataTraining.id).where(
@@ -221,10 +218,7 @@ def update_training(session: SessionDep, info: DataTrainingInfo, oid: int, trans
221218
raise Exception(trans("i18n_data_training.description_cannot_be_empty"))
222219

223220
if info.datasource is None and info.advanced_application is None:
224-
if oid == 1:
225-
raise Exception(trans("i18n_data_training.datasource_assistant_cannot_be_none"))
226-
else:
227-
raise Exception(trans("i18n_data_training.datasource_cannot_be_none"))
221+
raise Exception(trans("i18n_data_training.datasource_assistant_cannot_be_none"))
228222

229223
count = session.query(DataTraining).filter(
230224
DataTraining.id == info.id
@@ -310,11 +304,11 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo]
310304
datasource_name_to_id[ds.name.strip()] = ds.id
311305

312306
assistant_name_to_id = {}
313-
if oid == 1:
314-
assistant_stmt = select(AssistantModel.id, AssistantModel.name).where(AssistantModel.type == 1)
315-
assistant_result = session.execute(assistant_stmt).all()
316-
for assistant in assistant_result:
317-
assistant_name_to_id[assistant.name.strip()] = assistant.id
307+
308+
assistant_stmt = select(AssistantModel.id, AssistantModel.name).where(and_(AssistantModel.type == 1, AssistantModel.oid == oid))
309+
assistant_result = session.execute(assistant_stmt).all()
310+
for assistant in assistant_result:
311+
assistant_name_to_id[assistant.name.strip()] = assistant.id
318312

319313
# 验证和转换数据
320314
valid_records = []
@@ -338,20 +332,16 @@ def batch_create_training(session: SessionDep, info_list: List[DataTrainingInfo]
338332

339333
# 高级应用验证和转换
340334
advanced_application_id = None
341-
if oid == 1 and info.advanced_application_name and info.advanced_application_name.strip():
335+
if info.advanced_application_name and info.advanced_application_name.strip():
342336
if info.advanced_application_name.strip() in assistant_name_to_id:
343337
advanced_application_id = assistant_name_to_id[info.advanced_application_name.strip()]
344338
else:
345339
error_messages.append(
346340
trans("i18n_data_training.advanced_application_not_found").format(info.advanced_application_name))
347341

348342
# 检查数据源和高级应用不能同时为空
349-
if oid == 1:
350-
if not datasource_id and not advanced_application_id:
351-
error_messages.append(trans("i18n_data_training.datasource_assistant_cannot_be_none"))
352-
else:
353-
if not datasource_id:
354-
error_messages.append(trans("i18n_data_training.datasource_cannot_be_none"))
343+
if not datasource_id and not advanced_application_id:
344+
error_messages.append(trans("i18n_data_training.datasource_assistant_cannot_be_none"))
355345

356346
if error_messages:
357347
failed_records.append({

frontend/src/views/system/training/index.vue

Lines changed: 12 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ import IconOpeDelete from '@/assets/svg/icon_delete.svg'
1111
import icon_searchOutline_outlined from '@/assets/svg/icon_search-outline_outlined.svg'
1212
import EmptyBackground from '@/views/dashboard/common/EmptyBackground.vue'
1313
import { useClipboard } from '@vueuse/core'
14-
import { useUserStore } from '@/stores/user'
1514
import { useI18n } from 'vue-i18n'
1615
import { cloneDeep } from 'lodash-es'
1716
import { getAdvancedApplicationList } from '@/api/embedded.ts'
@@ -26,7 +25,6 @@ interface Form {
2625
advanced_application_name: string | null
2726
description: string | null
2827
}
29-
const userStore = useUserStore()
3028
const { t } = useI18n()
3129
const multipleSelectionAll = ref<any[]>([])
3230
const keywords = ref('')
@@ -42,10 +40,6 @@ onMounted(() => {
4240
search()
4341
})
4442
45-
const isDefaultOrg = computed(() => {
46-
return userStore.oid === '1'
47-
})
48-
4943
const dialogFormVisible = ref<boolean>(false)
5044
const multipleTableRef = ref()
5145
const isIndeterminate = ref(true)
@@ -261,26 +255,23 @@ const rules = computed(() => {
261255
},
262256
],
263257
}
264-
if (!isDefaultOrg.value) {
265-
_list.datasource = [
266-
{
267-
required: true,
268-
message: t('datasource.Please_select') + t('common.empty') + t('ds.title'),
269-
},
270-
]
271-
}
258+
// _list.datasource = [
259+
// {
260+
// required: true,
261+
// message: t('datasource.Please_select') + t('common.empty') + t('ds.title'),
262+
// },
263+
// ]
264+
272265
return _list
273266
})
274267
275268
const list = () => {
276269
datasourceApi.list().then((res: any) => {
277270
options.value = res || []
278271
})
279-
if (isDefaultOrg.value) {
280-
getAdvancedApplicationList().then((res: any) => {
281-
adv_options.value = res || []
282-
})
283-
}
272+
getAdvancedApplicationList().then((res: any) => {
273+
adv_options.value = res || []
274+
})
284275
}
285276
286277
const saveHandler = () => {
@@ -426,7 +417,6 @@ const onRowFormClose = () => {
426417
</el-table-column>
427418
<el-table-column prop="datasource_name" :label="$t('ds.title')" min-width="180" />
428419
<el-table-column
429-
v-if="isDefaultOrg"
430420
prop="advanced_application_name"
431421
:label="$t('embedded.advanced_application')"
432422
min-width="180"
@@ -588,11 +578,7 @@ const onRowFormClose = () => {
588578
</el-select>
589579
</el-form-item>
590580

591-
<el-form-item
592-
v-if="isDefaultOrg"
593-
prop="advanced_application"
594-
:label="t('embedded.advanced_application')"
595-
>
581+
<el-form-item prop="advanced_application" :label="t('embedded.advanced_application')">
596582
<el-select
597583
v-model="pageForm.advanced_application"
598584
filterable
@@ -653,7 +639,7 @@ const onRowFormClose = () => {
653639
{{ pageForm.datasource_name }}
654640
</div>
655641
</el-form-item>
656-
<el-form-item v-if="isDefaultOrg" :label="t('embedded.advanced_application')">
642+
<el-form-item :label="t('embedded.advanced_application')">
657643
<div class="content">
658644
{{ pageForm.advanced_application_name }}
659645
</div>

0 commit comments

Comments
 (0)