Fix: specv2 cards wrapper case

1 files changed, 25 insertions, 0 deletions

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()
 }