Skip to content

Commit f020ab4

Browse files
committed
feat: refactor getSelectModelList to use workspace_id from data and update model retrieval logic
1 parent e28ea0d commit f020ab4

File tree

3 files changed

+12
-4
lines changed

3 files changed

+12
-4
lines changed

ui/src/api/system-resource-management/model.ts

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,9 @@ const getModelListPage: (
3030
*/
3131
const getSelectModelList: (
3232
data?: ListModelRequest,
33-
workspace_id?: string,
3433
loading?: Ref<boolean>,
35-
) => Promise<Result<Array<Model>>> = (data, workspace_id, loading) => {
36-
return get(`${prefix}/workspace/${workspace_id}/model/model_list`, data, loading).then((ok) => {
34+
) => Promise<Result<Array<Model>>> = (data, loading) => {
35+
return get(`${prefix}/model/model_list`, data, loading).then((ok) => {
3736
return {
3837
...ok,
3938
data: [

ui/src/api/type/model.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ interface ListModelRequest {
3737
* 供应商
3838
*/
3939
provider?: string
40+
41+
workspace_id?: string
4042
}
4143

4244
interface Model {

ui/src/views/knowledge/component/BaseForm.vue

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ const form = ref<knowledgeData>({
6060
desc: '',
6161
embedding_model_id: '',
6262
})
63+
const workspace_id = ref('')
6364
6465
const rules = reactive({
6566
name: [
@@ -96,6 +97,10 @@ watch(
9697
form.value.name = value.name
9798
form.value.desc = value.desc
9899
form.value.embedding_model_id = value.embedding_model_id
100+
workspace_id.value = value.workspace_id || ''
101+
102+
// 重新刷新模型列表
103+
getSelectModel()
99104
}
100105
},
101106
{
@@ -115,8 +120,9 @@ function validate() {
115120
116121
function getSelectModel() {
117122
loading.value = true
123+
console.log(workspace_id.value)
118124
loadSharedApi({ type: 'model', systemType: props.apiType })
119-
.getSelectModelList({ model_type: 'EMBEDDING' })
125+
.getSelectModelList({ model_type: 'EMBEDDING', workspace_id: workspace_id.value })
120126
.then((res: any) => {
121127
modelOptions.value = groupBy(res?.data, 'provider')
122128
loading.value = false
@@ -129,6 +135,7 @@ function getSelectModel() {
129135
onMounted(() => {
130136
getSelectModel()
131137
})
138+
132139
onUnmounted(() => {
133140
form.value = {
134141
name: '',

0 commit comments

Comments
 (0)