This commit is contained in:
Hk-Gosuto 2024-02-20 10:25:30 +08:00
parent 6dd0ad628e
commit 8363fd26c5
2 changed files with 7 additions and 1 deletions

View File

@ -225,7 +225,8 @@ export class AgentApi {
const serverConfig = getServerSideConfig(); const serverConfig = getServerSideConfig();
// const reqBody: RequestBody = await req.json(); // const reqBody: RequestBody = await req.json();
const isAzure = reqBody.isAzure || serverConfig.isAzure; // ui set azure model provider
const isAzure = reqBody.isAzure;
const authHeaderName = isAzure ? "api-key" : "Authorization"; const authHeaderName = isAzure ? "api-key" : "Authorization";
const authToken = req.headers.get(authHeaderName) ?? ""; const authToken = req.headers.get(authHeaderName) ?? "";
const token = authToken.trim().replaceAll("Bearer ", "").trim(); const token = authToken.trim().replaceAll("Bearer ", "").trim();

View File

@ -91,6 +91,11 @@ function createEmptySession(): ChatSession {
} }
function getSummarizeModel(currentModel: string) { function getSummarizeModel(currentModel: string) {
// if the current model does not exist in the default model
// example azure services cannot use SUMMARIZE_MODEL
const model = DEFAULT_MODELS.find((m) => m.name === currentModel);
console.log("model", model);
if (!model) return currentModel;
// if it is using gpt-* models, force to use 3.5 to summarize // if it is using gpt-* models, force to use 3.5 to summarize
return currentModel.startsWith("gpt") ? SUMMARIZE_MODEL : currentModel; return currentModel.startsWith("gpt") ? SUMMARIZE_MODEL : currentModel;
} }