@@ -114,6 +114,24 @@ const catalogOnlyProvider = {
114114115115const defaultProviders = [chutesProvider, moonshotProvider, openaiProvider];
116116117+function firstDiscoveryRequest(): {
118+onlyPluginIds?: string[];
119+requireCompleteDiscoveryEntryCoverage?: boolean;
120+discoveryEntriesOnly?: boolean;
121+includeUntrustedWorkspacePlugins?: boolean;
122+} {
123+const call = providerDiscoveryMocks.resolveRuntimePluginDiscoveryProviders.mock.calls[0];
124+if (!call) {
125+throw new Error("expected runtime plugin discovery call");
126+}
127+return call[0] as {
128+onlyPluginIds?: string[];
129+requireCompleteDiscoveryEntryCoverage?: boolean;
130+discoveryEntriesOnly?: boolean;
131+includeUntrustedWorkspacePlugins?: boolean;
132+};
133+}
134+117135describe("loadProviderCatalogModelsForList", () => {
118136beforeEach(() => {
119137vi.clearAllMocks();
@@ -182,8 +200,7 @@ describe("loadProviderCatalogModelsForList", () => {
182200staticOnly: true,
183201});
184202185-const discoveryRequest =
186-providerDiscoveryMocks.resolveRuntimePluginDiscoveryProviders.mock.calls.at(0)?.[0];
203+const discoveryRequest = firstDiscoveryRequest();
187204expect(discoveryRequest?.onlyPluginIds).toStrictEqual(["moonshot"]);
188205expect(discoveryRequest?.requireCompleteDiscoveryEntryCoverage).toBe(true);
189206expect(discoveryRequest?.discoveryEntriesOnly).toBe(true);
@@ -283,8 +300,7 @@ describe("loadProviderCatalogModelsForList", () => {
283300}),
284301).resolves.toBe(false);
285302286-const discoveryRequest =
287-providerDiscoveryMocks.resolveRuntimePluginDiscoveryProviders.mock.calls.at(0)?.[0];
303+const discoveryRequest = firstDiscoveryRequest();
288304expect(discoveryRequest?.onlyPluginIds).toStrictEqual(["ollama"]);
289305expect(discoveryRequest?.requireCompleteDiscoveryEntryCoverage).toBe(true);
290306expect(discoveryRequest?.discoveryEntriesOnly).toBe(true);
@@ -362,8 +378,7 @@ describe("loadProviderCatalogModelsForList", () => {
362378 ...baseParams,
363379});
364380365-const discoveryRequest =
366-providerDiscoveryMocks.resolveRuntimePluginDiscoveryProviders.mock.calls.at(0)?.[0];
381+const discoveryRequest = firstDiscoveryRequest();
367382expect(discoveryRequest?.onlyPluginIds).toStrictEqual(["bundled-demo"]);
368383expect(discoveryRequest?.includeUntrustedWorkspacePlugins).toBe(false);
369384expect(workspaceStaticCatalog).not.toHaveBeenCalled();
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。