Skip to content

Commit 52aa22b

Browse files
committed
bugfix(completion): fix bug with choosing wrong server for processing requests and small improvement log system
1 parent a70a882 commit 52aa22b

File tree

3 files changed

+20
-16
lines changed

3 files changed

+20
-16
lines changed

src/common/completion/index.ts

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,9 +87,7 @@ export const getInlineCompletionProvider = (
8787
parameters,
8888
abortController,
8989
loggerCompletion.uuid(),
90-
triggerAuto
91-
? servers["base-small"].serverUrl
92-
: servers["base-small"].serverUrl
90+
servers[modelType].serverUrl
9391
);
9492

9593
if (completion === null) {

src/common/download/index.ts

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ export const downloadModel = async (typeModel: TypeModel) => {
264264
throw new Error("Server file info not found");
265265
}
266266

267-
Logger.info("Got model file info", {
267+
Logger.info(`Got model ${typeModel} file info`, {
268268
component: "download>model",
269269
sendTelemetry: true,
270270
});
@@ -275,13 +275,16 @@ export const downloadModel = async (typeModel: TypeModel) => {
275275
const checksum = await getChecksum(modelPath);
276276

277277
if (checksum === modelFileInfo.checksum) {
278-
Logger.info("Checksum model is correct, just return model path", {
279-
component: "download>model",
280-
sendTelemetry: true,
281-
});
278+
Logger.info(
279+
`Checksum model ${typeModel} is correct, just return model path`,
280+
{
281+
component: "download>model",
282+
sendTelemetry: true,
283+
}
284+
);
282285
return modelPath;
283286
}
284-
Logger.info("Checksum model mismatch", {
287+
Logger.info(`Checksum model ${typeModel} mismatch`, {
285288
component: "download>model",
286289
sendTelemetry: true,
287290
});
@@ -291,7 +294,7 @@ export const downloadModel = async (typeModel: TypeModel) => {
291294
await fsPromise.unlink(modelPath);
292295
}
293296

294-
Logger.info("Started download model", {
297+
Logger.info(`Started download model ${typeModel}`, {
295298
component: "download>model",
296299
sendTelemetry: true,
297300
});
@@ -304,7 +307,7 @@ export const downloadModel = async (typeModel: TypeModel) => {
304307
`Downloading model ${typeModel}: ${downloaded} / ${total}`
305308
);
306309

307-
Logger.info("Finish download model", {
310+
Logger.info(`Finish download model ${typeModel}`, {
308311
component: "download>model",
309312
sendTelemetry: true,
310313
});

src/common/server/index.ts

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -82,17 +82,20 @@ class Server {
8282
sendTelemetry: true,
8383
});
8484
FirecoderTelemetrySenderInstance.sendErrorData(error as Error, {
85-
step: "Error while downloading server or model",
85+
step: `Error while downloading server or model ${this.typeModel}`,
8686
});
8787
stopTask();
8888
throw error;
8989
}
9090

9191
if (!serverPath || !modelPath) {
92-
Logger.error("Server is not started. Don't have server or model path.", {
93-
component: "server",
94-
sendTelemetry: true,
95-
});
92+
Logger.error(
93+
`Server ${this.typeModel} is not started. Don't have server or model path.`,
94+
{
95+
component: "server",
96+
sendTelemetry: true,
97+
}
98+
);
9699

97100
stopTask();
98101
return false;

0 commit comments

Comments
 (0)