@codingame/monaco-vscode-ai-service-override 16.1.0 → 17.0.0

package/index.js

@@ -4,11 +4,14 @@ import { SyncDescriptor } from '@codingame/monaco-vscode-api/vscode/vs/platform/
 import { IAiEmbeddingVectorService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/aiEmbeddingVector/common/aiEmbeddingVectorService.service';
 import { IAiRelatedInformationService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/aiRelatedInformation/common/aiRelatedInformation.service';
 import { AiRelatedInformationService } from './vscode/src/vs/workbench/services/aiRelatedInformation/common/aiRelatedInformationService.js';
+import { IAiSettingsSearchService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/aiSettingsSearch/common/aiSettingsSearch.service';
+import { AiSettingsSearchService } from './vscode/src/vs/workbench/services/aiSettingsSearch/common/aiSettingsSearchService.js';
 
 function getServiceOverride() {
     return {
         [IAiRelatedInformationService.toString()]: new SyncDescriptor(AiRelatedInformationService, [], true),
-        [IAiEmbeddingVectorService.toString()]: new SyncDescriptor(AiRelatedInformationService, [], true)
+        [IAiEmbeddingVectorService.toString()]: new SyncDescriptor(AiRelatedInformationService, [], true),
+        [IAiSettingsSearchService.toString()]: new SyncDescriptor(AiSettingsSearchService, [], true)
     };
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codingame/monaco-vscode-ai-service-override",
-  "version": "16.1.0",
+  "version": "17.0.0",
   "private": false,
   "description": "VSCode public API plugged on the monaco editor - ai service-override",
   "keywords": [],
@@ -15,7 +15,7 @@
   },
   "type": "module",
   "dependencies": {
-    "@codingame/monaco-vscode-api": "16.1.0"
+    "@codingame/monaco-vscode-api": "17.0.0"
   },
   "main": "index.js",
   "module": "index.js",

package/vscode/src/vs/workbench/services/aiRelatedInformation/common/aiRelatedInformationService.d.ts

@@ -1,7 +1,7 @@
 import { CancellationToken } from "@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation";
 import { IDisposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
 import { ILogService } from "@codingame/monaco-vscode-api/vscode/vs/platform/log/common/log.service";
-import { IAiRelatedInformationProvider, RelatedInformationType, RelatedInformationResult } from "@codingame/monaco-vscode-d609a7d3-bf87-551a-884f-550a8b327ec5-common/vscode/vs/workbench/services/aiRelatedInformation/common/aiRelatedInformation";
+import { IAiRelatedInformationProvider, RelatedInformationType, RelatedInformationResult } from "@codingame/monaco-vscode-quickaccess-service-override/vscode/vs/workbench/services/aiRelatedInformation/common/aiRelatedInformation";
 import { IAiRelatedInformationService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/aiRelatedInformation/common/aiRelatedInformation.service";
 export declare class AiRelatedInformationService implements IAiRelatedInformationService {
     private readonly logService;

package/vscode/src/vs/workbench/services/aiSettingsSearch/common/aiSettingsSearchService.d.ts

@@ -0,0 +1,17 @@
+import { CancellationToken } from "@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation";
+import { IDisposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
+import { AiSettingsSearchResult, IAiSettingsSearchProvider } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/aiSettingsSearch/common/aiSettingsSearch";
+import { IAiSettingsSearchService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/aiSettingsSearch/common/aiSettingsSearch.service";
+export declare class AiSettingsSearchService implements IAiSettingsSearchService {
+    readonly _serviceBrand: undefined;
+    private static readonly MAX_PICKS;
+    private _providers;
+    private _llmRankedResultsPromises;
+    private _embeddingsResultsPromises;
+    isEnabled(): boolean;
+    registerSettingsSearchProvider(provider: IAiSettingsSearchProvider): IDisposable;
+    startSearch(query: string, token: CancellationToken): void;
+    getEmbeddingsResults(query: string, token: CancellationToken): Promise<string[] | null>;
+    getLLMRankedResults(query: string, token: CancellationToken): Promise<string[] | null>;
+    handleSearchResult(result: AiSettingsSearchResult): void;
+}

package/vscode/src/vs/workbench/services/aiSettingsSearch/common/aiSettingsSearchService.js

@@ -0,0 +1,73 @@
+
+import { DeferredPromise, raceCancellation } from '@codingame/monaco-vscode-api/vscode/vs/base/common/async';
+import '@codingame/monaco-vscode-api/vscode/vs/platform/instantiation/common/extensions';
+import { AiSettingsSearchResultKind } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/aiSettingsSearch/common/aiSettingsSearch';
+import '@codingame/monaco-vscode-api/vscode/vs/platform/instantiation/common/instantiation';
+
+class AiSettingsSearchService {
+    constructor() {
+        this._providers = [];
+        this._llmRankedResultsPromises = ( new Map());
+        this._embeddingsResultsPromises = ( new Map());
+    }
+    static { this.MAX_PICKS = 5; }
+    isEnabled() {
+        return this._providers.length > 0;
+    }
+    registerSettingsSearchProvider(provider) {
+        this._providers.push(provider);
+        return {
+            dispose: () => {
+                const index = this._providers.indexOf(provider);
+                if (index !== -1) {
+                    this._providers.splice(index, 1);
+                }
+            }
+        };
+    }
+    startSearch(query, token) {
+        if (!this.isEnabled()) {
+            throw ( new Error('No settings search providers registered'));
+        }
+        this._providers.forEach(provider => provider.searchSettings(query, { limit: AiSettingsSearchService.MAX_PICKS }, token));
+    }
+    async getEmbeddingsResults(query, token) {
+        if (!this.isEnabled()) {
+            throw ( new Error('No settings search providers registered'));
+        }
+        const promise = ( new DeferredPromise());
+        this._embeddingsResultsPromises.set(query, promise);
+        const result = await raceCancellation(promise.p, token);
+        return result ?? null;
+    }
+    async getLLMRankedResults(query, token) {
+        if (!this.isEnabled()) {
+            throw ( new Error('No settings search providers registered'));
+        }
+        const promise = ( new DeferredPromise());
+        this._llmRankedResultsPromises.set(query, promise);
+        const result = await raceCancellation(promise.p, token);
+        return result ?? null;
+    }
+    handleSearchResult(result) {
+        if (!this.isEnabled()) {
+            return;
+        }
+        if (result.kind === AiSettingsSearchResultKind.EMBEDDED) {
+            const promise = this._embeddingsResultsPromises.get(result.query);
+            if (promise) {
+                promise.complete(result.settings);
+                this._embeddingsResultsPromises.delete(result.query);
+            }
+        }
+        else if (result.kind === AiSettingsSearchResultKind.LLM_RANKED) {
+            const promise = this._llmRankedResultsPromises.get(result.query);
+            if (promise) {
+                promise.complete(result.settings);
+                this._llmRankedResultsPromises.delete(result.query);
+            }
+        }
+    }
+}
+
+export { AiSettingsSearchService };