sketch up lm-extention that allows for function calling (#214334)

This commit is contained in:
Johannes Rieken
2024-06-05 16:32:08 +02:00
committed by GitHub
parent dee8eaf526
commit ab9a1286f2
9 changed files with 274 additions and 43 deletions

View File

@@ -1448,6 +1448,10 @@ export function createApiFactoryAndRegisterActors(accessor: ServicesAccessor): I
}
return extHostLanguageModels.onDidChangeProviders(listener, thisArgs, disposables);
},
registerChatModelProvider: (id, provider, metadata) => {
checkProposedApiEnabled(extension, 'chatProvider');
return extHostLanguageModels.registerLanguageModel(extension, id, provider, metadata);
},
// --- embeddings
get embeddingModels() {
checkProposedApiEnabled(extension, 'embeddings');
@@ -1729,6 +1733,9 @@ export function createApiFactoryAndRegisterActors(accessor: ServicesAccessor): I
ChatLocation: extHostTypes.ChatLocation,
LanguageModelChatMessageRole: extHostTypes.LanguageModelChatMessageRole,
LanguageModelChatMessage: extHostTypes.LanguageModelChatMessage,
LanguageModelChatMessageFunctionResultPart: extHostTypes.LanguageModelFunctionResultPart,
LanguageModelChatResponseTextPart: extHostTypes.LanguageModelTextPart,
LanguageModelChatResponseFunctionUsePart: extHostTypes.LanguageModelFunctionUsePart,
LanguageModelChatMessage2: extHostTypes.LanguageModelChatMessage, // TODO@jrieken REMOVE
LanguageModelChatSystemMessage: extHostTypes.LanguageModelChatSystemMessage,// TODO@jrieken REMOVE
LanguageModelChatUserMessage: extHostTypes.LanguageModelChatUserMessage,// TODO@jrieken REMOVE