jason/ThunderAI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added an optional api key. see #144

Browse source
This commit is contained in:
mic 2024-10-01 20:45:15 +02:00
parent 57c59914e3
commit 1730922c04
1 changed files with 15 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
js/api/openai_comp.js
View file

@ -23,21 +23,26 @@ export class OpenAIComp {
host = '';
model = '';
apiKey = '';
stream = false;
constructor(host, model, stream = false) {
constructor(host, model, apiKey = '', stream = false) {
this.host = host.trim().replace(/\/+$/, "");
this.model = model;
this.stream = stream;
this.apiKey = apiKey;
}
fetchModels = async () => {
const curr_headers = {
"Content-Type": "application/json",
};
if(this.apiKey !== '') curr_headers["Authorization"] = "Bearer "+ this.apiKey;
const response = await fetch(this.host + "/v1/models", {
method: "GET",
headers: {
"Content-Type": "application/json",
},
headers: curr_headers,
});
if (!response.ok) {
@ -59,12 +64,15 @@ export class OpenAIComp {
}
fetchResponse = async (messages, maxTokens = 0) => {
const curr_headers = {
"Content-Type": "application/json",
};
if(this.apiKey !== '') curr_headers["Authorization"] = "Bearer "+ this.apiKey;
try {
const response = await fetch(this.host + "/v1/chat/completions", {
method: "POST",
headers: {
"Content-Type": "application/json",
},
headers: curr_headers,
body: JSON.stringify({
model: this.model,
messages: messages,