edit view for model, depending on type and valid values

This commit is contained in:
Mary Hipp
2024-02-21 13:56:14 -05:00
committed by psychedelicious
parent 6b68971f38
commit 0a69779df9
15 changed files with 538 additions and 77 deletions

View File

@ -29,6 +29,8 @@ type UpdateModelResponse = paths['/api/v2/models/i/{key}']['patch']['responses']
type GetModelResponse =
paths['/api/v2/models/i/{key}']['get']['responses']['200']['content']['application/json'];
type GetModelMetadataResponse =
paths['/api/v2/models/meta/i/{key}']['get']['responses']['200']['content']['application/json'];
type ListModelsArg = NonNullable<paths['/api/v2/models/']['get']['parameters']['query']>;
@ -175,6 +177,12 @@ export const modelsApi = api.injectEndpoints({
},
providesTags: ['Model'],
}),
getModelMetadata: build.query<GetModelMetadataResponse, string>({
query: (key) => {
return buildModelsUrl(`meta/i/${key}`);
},
providesTags: ['Model'],
}),
updateModels: build.mutation<UpdateModelResponse, UpdateModelArg>({
query: ({ key, body }) => {
return {
@ -330,5 +338,6 @@ export const {
useGetModelsInFolderQuery,
useGetCheckpointConfigsQuery,
useGetModelImportsQuery,
useGetModelQuery
useGetModelQuery,
useGetModelMetadataQuery
} = modelsApi;