mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
b1b5c0d3b2
Ensure workflow editor model selector component gets a value This introduced some funky type issues related to ONNX models. ONNX doesn't work anyways (unmaintained). Instead of fixing the types to work with a non-working feature, ONNX is now removed entirely from the UI. - Remove all refs to ONNX (and Olives) - Fix some type issues - Add ONNX nodes to the nodes denylist (so they are not visible in UI) - Update VAE graph helper, which still had some ONNX logic. It's a very simple change and doesn't change any logic. Just removes some conditions that were for ONNX. I tested it and nothing broke. - Regenerate types - Fix prettier and eslint ignores for generated types - Lint
79 lines
1.8 KiB
TypeScript
79 lines
1.8 KiB
TypeScript
import type {
|
|
BaseQueryFn,
|
|
FetchArgs,
|
|
FetchBaseQueryError,
|
|
TagDescription,
|
|
} from '@reduxjs/toolkit/query/react';
|
|
import { createApi, fetchBaseQuery } from '@reduxjs/toolkit/query/react';
|
|
import { $authToken } from 'app/store/nanostores/authToken';
|
|
import { $baseUrl } from 'app/store/nanostores/baseUrl';
|
|
import { $projectId } from 'app/store/nanostores/projectId';
|
|
|
|
export const tagTypes = [
|
|
'AppVersion',
|
|
'AppConfig',
|
|
'Board',
|
|
'BoardImagesTotal',
|
|
'BoardAssetsTotal',
|
|
'Image',
|
|
'ImageNameList',
|
|
'ImageList',
|
|
'ImageMetadata',
|
|
'ImageWorkflow',
|
|
'ImageMetadataFromFile',
|
|
'IntermediatesCount',
|
|
'SessionQueueItem',
|
|
'SessionQueueStatus',
|
|
'SessionProcessorStatus',
|
|
'CurrentSessionQueueItem',
|
|
'NextSessionQueueItem',
|
|
'BatchStatus',
|
|
'InvocationCacheStatus',
|
|
'Model',
|
|
'T2IAdapterModel',
|
|
'MainModel',
|
|
'VaeModel',
|
|
'IPAdapterModel',
|
|
'TextualInversionModel',
|
|
'ControlNetModel',
|
|
'LoRAModel',
|
|
'SDXLRefinerModel',
|
|
'Workflow',
|
|
'WorkflowsRecent',
|
|
] as const;
|
|
export type ApiTagDescription = TagDescription<(typeof tagTypes)[number]>;
|
|
export const LIST_TAG = 'LIST';
|
|
|
|
const dynamicBaseQuery: BaseQueryFn<
|
|
string | FetchArgs,
|
|
unknown,
|
|
FetchBaseQueryError
|
|
> = async (args, api, extraOptions) => {
|
|
const baseUrl = $baseUrl.get();
|
|
const authToken = $authToken.get();
|
|
const projectId = $projectId.get();
|
|
|
|
const rawBaseQuery = fetchBaseQuery({
|
|
baseUrl: `${baseUrl ?? ''}/api/v1`,
|
|
prepareHeaders: (headers) => {
|
|
if (authToken) {
|
|
headers.set('Authorization', `Bearer ${authToken}`);
|
|
}
|
|
if (projectId) {
|
|
headers.set('project-id', projectId);
|
|
}
|
|
|
|
return headers;
|
|
},
|
|
});
|
|
|
|
return rawBaseQuery(args, api, extraOptions);
|
|
};
|
|
|
|
export const api = createApi({
|
|
baseQuery: dynamicBaseQuery,
|
|
reducerPath: 'api',
|
|
tagTypes,
|
|
endpoints: () => ({}),
|
|
});
|