50 lines
2.6 KiB
TypeScript
50 lines
2.6 KiB
TypeScript
// *****************************************************************************
|
|
// Copyright (C) 2024 EclipseSource GmbH.
|
|
//
|
|
// This program and the accompanying materials are made available under the
|
|
// terms of the Eclipse Public License v. 2.0 which is available at
|
|
// http://www.eclipse.org/legal/epl-2.0.
|
|
//
|
|
// This Source Code may also be made available under the following Secondary
|
|
// Licenses when the conditions for such availability set forth in the Eclipse
|
|
// Public License v. 2.0 are satisfied: GNU General Public License, version 2
|
|
// with the GNU Classpath Exception which is available at
|
|
// https://www.gnu.org/software/classpath/license.html.
|
|
//
|
|
// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
|
|
// *****************************************************************************
|
|
|
|
import { CancellationToken } from '@theia/core';
|
|
import {
|
|
LanguageModelMetaData, LanguageModelParsedResponse, LanguageModelRequest, LanguageModelStreamResponsePart,
|
|
LanguageModelTextResponse, ToolCallResult
|
|
} from './language-model';
|
|
|
|
export const LanguageModelDelegateClient = Symbol('LanguageModelDelegateClient');
|
|
export interface LanguageModelDelegateClient {
|
|
toolCall(requestId: string, toolId: string, args_string: string, toolCallId?: string): Promise<ToolCallResult>;
|
|
send(id: string, token: LanguageModelStreamResponsePart | undefined): void;
|
|
error(id: string, error: Error): void;
|
|
}
|
|
export const LanguageModelRegistryFrontendDelegate = Symbol('LanguageModelRegistryFrontendDelegate');
|
|
export interface LanguageModelRegistryFrontendDelegate {
|
|
getLanguageModelDescriptions(): Promise<LanguageModelMetaData[]>;
|
|
}
|
|
|
|
export interface LanguageModelStreamResponseDelegate {
|
|
streamId: string;
|
|
}
|
|
export const isLanguageModelStreamResponseDelegate = (obj: unknown): obj is LanguageModelStreamResponseDelegate =>
|
|
!!(obj && typeof obj === 'object' && 'streamId' in obj && typeof (obj as { streamId: unknown }).streamId === 'string');
|
|
|
|
export type LanguageModelResponseDelegate = LanguageModelTextResponse | LanguageModelParsedResponse | LanguageModelStreamResponseDelegate;
|
|
|
|
export const LanguageModelFrontendDelegate = Symbol('LanguageModelFrontendDelegate');
|
|
export interface LanguageModelFrontendDelegate {
|
|
cancel(requestId: string): void;
|
|
request(modelId: string, request: LanguageModelRequest, requestId: string, cancellationToken?: CancellationToken): Promise<LanguageModelResponseDelegate>;
|
|
}
|
|
|
|
export const languageModelRegistryDelegatePath = '/services/languageModelRegistryDelegatePath';
|
|
export const languageModelDelegatePath = '/services/languageModelDelegatePath';
|