26 lines
858 B
TypeScript
26 lines
858 B
TypeScript
import { EmbeddingsProviderName, EmbedOptions } from "../..";
|
|
import { CONTROL_PLANE_URL } from "../../control-plane/client";
|
|
import OpenAIEmbeddingsProvider from "./OpenAIEmbeddingsProvider";
|
|
|
|
class ContinueProxyEmbeddingsProvider extends OpenAIEmbeddingsProvider {
|
|
static providerName: EmbeddingsProviderName = "continue-proxy";
|
|
static defaultOptions: Partial<EmbedOptions> | undefined = {
|
|
apiBase: new URL("/model-proxy/v1", CONTROL_PLANE_URL).toString(),
|
|
};
|
|
|
|
private _workOsAccessToken: string | undefined = undefined;
|
|
|
|
get workOsAccessToken(): string | undefined {
|
|
return this._workOsAccessToken;
|
|
}
|
|
|
|
set workOsAccessToken(value: string | undefined) {
|
|
if (this._workOsAccessToken !== value) {
|
|
this._workOsAccessToken = value;
|
|
this.options.apiKey = value;
|
|
}
|
|
}
|
|
}
|
|
|
|
export default ContinueProxyEmbeddingsProvider;
|