From 3314856ee3aec4ab79c27d130179964ad2a79096 Mon Sep 17 00:00:00 2001 From: Taras Glek Date: Mon, 8 Jul 2024 10:25:18 +0200 Subject: [PATCH] make model listing llama.cpp friendly --- src/lib/ChatCraftModel.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lib/ChatCraftModel.ts b/src/lib/ChatCraftModel.ts index 5ca7e458..16d9108b 100644 --- a/src/lib/ChatCraftModel.ts +++ b/src/lib/ChatCraftModel.ts @@ -12,9 +12,9 @@ export class ChatCraftModel { this.id = model; const parts = model.split("/"); // Default to "openai" if we don't get a vendor name - this.vendor = parts.length > 1 ? parts[0] : "openai"; + this.vendor = parts.at(-2) || "openai"; // If we get a vendor, use the second part, otherwise the whole thing is the model name - this.name = parts.length > 1 ? parts[1] : parts[0]; + this.name = parts.at(-1) || model; } get logoUrl() {