opensource/lms-cli
1
0
Fork 0
mirror of https://github.com/lmstudio-ai/lms.git synced 2025-09-19 03:30:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix lms-key-2 fetch logic when llmster has never run (#307)

Browse Source 
* Fix lms-key-2 fetch logic when llmster has never run

* Fix

* Fix dummy

* Fix comment
This commit is contained in:
ryan-the-crayon
2025-08-28 17:59:40 -04:00
committed by GitHub
parent 20babaecd7
commit aa470615ec
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/createClient.ts
View File

@ -5,6 +5,7 @@ import chalk from "chalk";
import { spawn } from "child_process"; import { spawn } from "child_process";
import { randomBytes } from "crypto"; import { randomBytes } from "crypto";
import { readFile } from "fs/promises"; import { readFile } from "fs/promises";
import { exists } from "./exists.js";
import { appInstallLocationFilePath, lmsKey2Path } from "./lmstudioPaths.js"; import { appInstallLocationFilePath, lmsKey2Path } from "./lmstudioPaths.js";
import { type LogLevelArgs } from "./logLevel.js"; import { type LogLevelArgs } from "./logLevel.js";
import { createRefinedNumberParser } from "./types/refinedNumber.js"; import { createRefinedNumberParser } from "./types/refinedNumber.js";
@ -178,11 +179,20 @@ export async function createClient(
clientIdentifier: "lms-cli-dev", clientIdentifier: "lms-cli-dev",
}; };
} else { } else {
if (await exists(lmsKey2Path)) {
const lmsKey2 = (await readFile(lmsKey2Path, "utf-8")).trim(); const lmsKey2 = (await readFile(lmsKey2Path, "utf-8")).trim();
auth = { auth = {
clientIdentifier: "lms-cli", clientIdentifier: "lms-cli",
clientPasskey: lmsKey + lmsKey2, clientPasskey: lmsKey + lmsKey2,
}; };
} else {
// This case will happen when the CLI is the production build, yet the local LM Studio has
// not been run yet (so no lms-key-2 file). In this case, we will just use a dummy client
// identifier as we will soon try to wake up the service and refetch the key.
auth = {
clientIdentifier: "lms-cli",
};
}
} }
} }
if (port === undefined && host === "127.0.0.1") { if (port === undefined && host === "127.0.0.1") {