27 lines
596 B
TypeScript
27 lines
596 B
TypeScript
import { OAIEngine } from './OAIEngine'
|
|
|
|
/**
|
|
* Base OAI Remote Inference Provider
|
|
* Added the implementation of loading and unloading model (applicable to local inference providers)
|
|
*/
|
|
export abstract class RemoteOAIEngine extends OAIEngine {
|
|
// The inference engine
|
|
abstract apiKey: string
|
|
/**
|
|
* On extension load, subscribe to events.
|
|
*/
|
|
override onLoad() {
|
|
super.onLoad()
|
|
}
|
|
|
|
/**
|
|
* Headers for the inference request
|
|
*/
|
|
override headers(): HeadersInit {
|
|
return {
|
|
'Authorization': `Bearer ${this.apiKey}`,
|
|
'api-key': `${this.apiKey}`,
|
|
}
|
|
}
|
|
}
|