From c41ff09b2f62b8c715ec9beb2fdac987b25892d9 Mon Sep 17 00:00:00 2001 From: Grail Finder Date: Thu, 16 Jan 2025 17:53:35 +0300 Subject: Fix: specv2 cards wrapper case --- bot.go | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'bot.go') diff --git a/bot.go b/bot.go index e8ded26..38f62b6 100644 --- a/bot.go +++ b/bot.go @@ -77,6 +77,31 @@ func fetchModelName() { logger.Warn("failed to decode resp", "link", api, "error", err) return } + if resp.StatusCode != 200 { + currentModel = "none" + return + } + currentModel = path.Base(llmModel.Data[0].ID) + updateStatusLine() +} + +func fetchProps() { + api := "http://localhost:8080/props" + resp, err := httpClient.Get(api) + if err != nil { + logger.Warn("failed to get model", "link", api, "error", err) + return + } + defer resp.Body.Close() + llmModel := models.LLMModels{} + if err := json.NewDecoder(resp.Body).Decode(&llmModel); err != nil { + logger.Warn("failed to decode resp", "link", api, "error", err) + return + } + if resp.StatusCode != 200 { + currentModel = "none" + return + } currentModel = path.Base(llmModel.Data[0].ID) updateStatusLine() } -- cgit v1.2.3