RelatedInformation & EmbeddingVector API (#191006)

This commit is contained in:
Tyler James Leonhardt
2023-08-22 12:22:53 -07:00
committed by GitHub
parent eccdef295e
commit f058931630
15 changed files with 638 additions and 23 deletions

View File

@@ -107,6 +107,8 @@ import { ExtHostShare } from 'vs/workbench/api/common/extHostShare';
import { ExtHostChatProvider } from 'vs/workbench/api/common/extHostChatProvider';
import { ExtHostChatSlashCommands } from 'vs/workbench/api/common/extHostChatSlashCommand';
import { ExtHostChatVariables } from 'vs/workbench/api/common/extHostChatVariables';
import { ExtHostRelatedInformation } from 'vs/workbench/api/common/extHostAiRelatedInformation';
import { ExtHostAiEmbeddingVector } from 'vs/workbench/api/common/extHostEmbeddingVector';
export interface IExtensionRegistries {
mine: ExtensionDescriptionRegistry;
@@ -211,6 +213,8 @@ export function createApiFactoryAndRegisterActors(accessor: ServicesAccessor): I
const extHostChatVariables = rpcProtocol.set(ExtHostContext.ExtHostChatVariables, new ExtHostChatVariables(rpcProtocol));
const extHostChat = rpcProtocol.set(ExtHostContext.ExtHostChat, new ExtHostChat(rpcProtocol, extHostLogService));
const extHostSemanticSimilarity = rpcProtocol.set(ExtHostContext.ExtHostSemanticSimilarity, new ExtHostSemanticSimilarity(rpcProtocol));
const extHostAiRelatedInformation = rpcProtocol.set(ExtHostContext.ExtHostAiRelatedInformation, new ExtHostRelatedInformation(rpcProtocol));
const extHostAiEmbeddingVector = rpcProtocol.set(ExtHostContext.ExtHostAiEmbeddingVector, new ExtHostAiEmbeddingVector(rpcProtocol));
const extHostIssueReporter = rpcProtocol.set(ExtHostContext.ExtHostIssueReporter, new ExtHostIssueReporter(rpcProtocol));
const extHostStatusBar = rpcProtocol.set(ExtHostContext.ExtHostStatusBar, new ExtHostStatusBar(rpcProtocol, extHostCommands.converter));
@@ -1323,6 +1327,18 @@ export function createApiFactoryAndRegisterActors(accessor: ServicesAccessor): I
registerSemanticSimilarityProvider(provider: vscode.SemanticSimilarityProvider) {
checkProposedApiEnabled(extension, 'semanticSimilarity');
return extHostSemanticSimilarity.registerSemanticSimilarityProvider(extension, provider);
},
getRelatedInformation(query: string, types: vscode.RelatedInformationType[]): Thenable<vscode.RelatedInformationResult[]> {
checkProposedApiEnabled(extension, 'aiRelatedInformation');
return extHostAiRelatedInformation.getRelatedInformation(extension, query, types);
},
registerRelatedInformationProvider(types: vscode.RelatedInformationType[], provider: vscode.RelatedInformationProvider) {
checkProposedApiEnabled(extension, 'aiRelatedInformation');
return extHostAiRelatedInformation.registerRelatedInformationProvider(extension, types, provider);
},
registerEmbeddingVectorProvider(model: string, provider: vscode.EmbeddingVectorProvider) {
checkProposedApiEnabled(extension, 'aiRelatedInformation');
return extHostAiEmbeddingVector.registerEmbeddingVectorProvider(extension, model, provider);
}
};
@@ -1560,7 +1576,8 @@ export function createApiFactoryAndRegisterActors(accessor: ServicesAccessor): I
InteractiveEditorResponseFeedbackKind: extHostTypes.InteractiveEditorResponseFeedbackKind,
StackFrameFocus: extHostTypes.StackFrameFocus,
ThreadFocus: extHostTypes.ThreadFocus,
NotebookCodeActionKind: extHostTypes.NotebookCodeActionKind
NotebookCodeActionKind: extHostTypes.NotebookCodeActionKind,
RelatedInformationType: extHostTypes.RelatedInformationType
};
};
}