@@ -77,6 +77,7 @@ import { groupBy } from 'lodash'
7777import { MsgSuccess } from ' @/utils/message'
7878import { t } from ' @/locales'
7979import type { FormInstance } from ' element-plus'
80+ import modelResourceApi from ' @/api/system-resource-management/model'
8081import { loadSharedApi } from ' @/utils/dynamics-api/shared-api'
8182
8283const props = defineProps <{
@@ -104,7 +105,7 @@ const stateMap = {
104105 error: [' 0' , ' 1' , ' 3' , ' 4' , ' 5' , ' n' ],
105106}
106107const FormRef = ref ()
107- const knowledgeId = ref <string >( )
108+ const currentKnowledge = ref <any >( null )
108109const userId = user .userInfo ?.id as string
109110const form = ref (prompt .get (userId ))
110111const rules = reactive ({
@@ -131,8 +132,8 @@ watch(dialogVisible, (bool) => {
131132 }
132133})
133134
134- const open = (ids : string [], type : string , _knowledgeId ? : string ) => {
135- knowledgeId .value = _knowledgeId
135+ const open = (ids : string [], type : string , _knowledge ? : any ) => {
136+ currentKnowledge .value = _knowledge
136137 getModelFn ()
137138 idList .value = ids
138139 apiSubmitType .value = type
@@ -178,7 +179,7 @@ const submitHandle = async (formEl: FormInstance) => {
178179 state_list: stateMap [state .value ],
179180 }
180181 loadSharedApi ({ type: ' knowledge' , systemType: props .apiType })
181- .putGenerateRelated (id ? id : knowledgeId .value , data , loading )
182+ .putGenerateRelated (id ? id : currentKnowledge .value ?. id , data , loading )
182183 .then (() => {
183184 MsgSuccess (t (' views.document.generateQuestion.successMessage' ))
184185 dialogVisible .value = false
@@ -190,15 +191,27 @@ const submitHandle = async (formEl: FormInstance) => {
190191
191192function getModelFn() {
192193 loading .value = true
193- loadSharedApi ({ type: ' model' , systemType: props .apiType })
194- .getSelectModelList ({ model_type: ' LLM' })
195- .then ((res : any ) => {
196- modelOptions .value = groupBy (res ?.data , ' provider' )
197- loading .value = false
198- })
199- .catch (() => {
200- loading .value = false
201- })
194+ if (props .apiType === ' systemManage' ) {
195+ modelResourceApi
196+ .getSelectModelList ({ model_type: ' LLM' }, currentKnowledge .value ?.workspace_id )
197+ .then ((res : any ) => {
198+ modelOptions .value = groupBy (res ?.data , ' provider' )
199+ loading .value = false
200+ })
201+ .catch (() => {
202+ loading .value = false
203+ })
204+ } else {
205+ loadSharedApi ({ type: ' model' , systemType: props .apiType })
206+ .getSelectModelList ({ model_type: ' LLM' })
207+ .then ((res : any ) => {
208+ modelOptions .value = groupBy (res ?.data , ' provider' )
209+ loading .value = false
210+ })
211+ .catch (() => {
212+ loading .value = false
213+ })
214+ }
202215}
203216
204217defineExpose ({ open , dialogVisible })
0 commit comments