Dockery/Trilium
1
0
Fork 0
mirror of https://github.com/zadam/trilium.git synced 2026-06-26 17:30:38 +02:00
Code Issues Packages Projects Releases Activity

feat(llm): improve model name display

Browse Source
This commit is contained in:
Elian Doran
2026-03-30 17:40:57 +03:00
parent af0eb9551a
commit 12797293f0
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
apps/server/src/routes/api/llm_chat.ts
View File

@@ -52,10 +52,11 @@ async function streamChat(req: Request, res: Response) {
const provider = getProviderByType(config.provider || "anthropic");
const result = provider.chat(messages, config);
// Get pricing from provider for cost calculation
const model = config.model || "claude-sonnet-4-20250514";
const pricing = provider.getModelPricing(model);
for await (const chunk of streamToChunks(result, { model, pricing })) {
// Get pricing and display name for the model
const modelId = config.model || "claude-sonnet-4-6";
const pricing = provider.getModelPricing(modelId);
const modelDisplayName = provider.getAvailableModels().find(m => m.id === modelId)?.name || modelId;
for await (const chunk of streamToChunks(result, { model: modelDisplayName, pricing })) {
res.write(`data: ${JSON.stringify(chunk)}\n\n`);
// Flush immediately to ensure real-time streaming
if (typeof flushableRes.flush === "function") {