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