Show token bar at bottom of chat input (#33)
This commit is contained in:
@@ -19,6 +19,8 @@ import type {
|
||||
SystemDebugInfo,
|
||||
LocalModel,
|
||||
LocalModelListResponse,
|
||||
TokenCountParams,
|
||||
TokenCountResult,
|
||||
} from "./ipc_types";
|
||||
import type { CodeProposal, ProposalResult } from "@/lib/schemas";
|
||||
import { showError } from "@/lib/toast";
|
||||
@@ -747,4 +749,17 @@ export class IpcClient {
|
||||
this.ipcRenderer.removeListener("deep-link-received", listener);
|
||||
};
|
||||
}
|
||||
|
||||
// Count tokens for a chat and input
|
||||
public async countTokens(
|
||||
params: TokenCountParams
|
||||
): Promise<TokenCountResult> {
|
||||
try {
|
||||
const result = await this.ipcRenderer.invoke("chat:count-tokens", params);
|
||||
return result as TokenCountResult;
|
||||
} catch (error) {
|
||||
showError(error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user