feat: add context bean

Author: Ricbet

packages/ai-native/src/browser/contrib/inline-completions/service/ai-completions.service.ts

@@ -73,7 +73,7 @@ export class AICompletionsService extends Disposable {
     const provider = this.intelligentCompletionsRegistry.getProvider();
 
     if (provider) {
-      return this.intelligentCompletionsHandler.fetchProvider();
+      return this.intelligentCompletionsHandler.fetchProvider(data);
     }
 
     // 兼容旧的 requestCompletion 接口

packages/ai-native/src/browser/contrib/intelligent-completions/intelligent-completions.handler.ts

@@ -2,6 +2,7 @@ import { Autowired, INJECTOR_TOKEN, Injectable, Injector } from '@opensumi/di';
 import {
   CancellationTokenSource,
   Disposable,
+  IAICompletionOption,
   IDisposable,
   IntelligentCompletionsRegistryToken,
 } from '@opensumi/ide-core-common';
@@ -41,14 +42,14 @@ export class IntelligentCompletionsHandler extends Disposable {
     return this.editor.monacoEditor;
   }
 
-  public async fetchProvider(): Promise<IIntelligentCompletionsResult | undefined> {
+  public async fetchProvider(bean: IAICompletionOption): Promise<IIntelligentCompletionsResult | undefined> {
     const provider = this.intelligentCompletionsRegistry.getProvider();
     if (!provider) {
       return;
     }
 
     const position = this.monacoEditor.getPosition()!;
-    const intelligentCompletionModel = await provider(this.monacoEditor, position, this.cancelIndicator.token);
+    const intelligentCompletionModel = await provider(this.monacoEditor, position, bean, this.cancelIndicator.token);
 
     return intelligentCompletionModel;
   }

packages/ai-native/src/browser/types.ts

@@ -203,6 +203,7 @@ export interface ITerminalProviderRegistry {
 export type IIntelligentCompletionProvider = (
   editor: ICodeEditor,
   position: IPosition,
+  contextBean: IAICompletionOption,
   token: CancellationToken,
 ) => MaybePromise<IIntelligentCompletionsResult>;
 export interface IIntelligentCompletionsRegistry {

packages/startup/entry/sample-modules/ai-native/ai-native.contribution.ts

@@ -407,7 +407,7 @@ export class AINativeContribution implements AINativeCoreContribution {
   }
 
   registerIntelligentCompletionFeature(registry: IIntelligentCompletionsRegistry): void {
-    registry.registerIntelligentCompletionProvider(async (editor, position, token) => {
+    registry.registerIntelligentCompletionProvider(async (editor, position, bean, token) => {
       const model = editor.getModel()!;
       const value = model.getValueInRange({
         startLineNumber: position.lineNumber,