dan/dots-hyprland
1
0
Fork 0
mirror of https://github.com/danbulant/dots-hyprland synced 2026-05-24 12:22:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ai: add usage metadata for openai and mistral

Browse source
This commit is contained in:
end-4 2025-08-07 21:53:37 +07:00
parent f581fd4821
commit 733a792610
2 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
.config/quickshell/ii/services/ai/MistralApiStrategy.qml
View file

@ -100,6 +100,17 @@ ApiStrategy {
message.content += newContent;
message.rawContent += newContent;
// Usage metadata
if (dataJson.usage) {
return {
tokenUsage: {
input: dataJson.usage.prompt_tokens ?? -1,
output: dataJson.usage.completion_tokens ?? -1,
total: dataJson.usage.total_tokens ?? -1
}
};
}
if (`dataJson`.done) {
return { finished: true };
}

12
.config/quickshell/ii/services/ai/OpenAiApiStrategy.qml
View file

@ -46,6 +46,7 @@ ApiStrategy {
// Real stuff
try {
const dataJson = JSON.parse(cleanData);
console.log(JSON.stringify(dataJson, null, 2));
let newContent = "";
const responseContent = dataJson.choices[0]?.delta?.content || dataJson.message?.content;
@ -72,6 +73,17 @@ ApiStrategy {
message.content += newContent;
message.rawContent += newContent;
// Usage metadata
if (dataJson.usage) {
return {
tokenUsage: {
input: dataJson.usage.prompt_tokens ?? -1,
output: dataJson.usage.completion_tokens ?? -1,
total: dataJson.usage.total_tokens ?? -1
}
};
}
if (dataJson.done) {
return { finished: true };
}