From aa57dd3b39ea9044ad6a46214d2627d59b0d0f94 Mon Sep 17 00:00:00 2001
From: db0 <mail@dbzer0.com>
Date: Tue, 5 Nov 2024 09:23:09 +0100
Subject: [PATCH] fix: missing style fields

---
 CHANGELOG.md                   | 6 ++++++
 horde/apis/models/kobold_v2.py | 1 +
 horde/apis/models/stable_v2.py | 1 +
 horde/apis/models/v2.py        | 1 +
 horde/classes/base/style.py    | 1 +
 horde/consts.py                | 2 +-
 6 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index dcc29d72..ec545590 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,9 +6,15 @@ SPDX-License-Identifier: AGPL-3.0-or-later
 
 # Changelog
 
+# 4.44.1
+
+* Various fixes around styles
+* Added `/.well-known/serviceinfo`
+
 # 4.44.0
 
 * Adds styles
+* Adds TabbyAPI as approved LLM backend
 
 # 4.43.9
 
diff --git a/horde/apis/models/kobold_v2.py b/horde/apis/models/kobold_v2.py
index 875b5890..e9e15276 100644
--- a/horde/apis/models/kobold_v2.py
+++ b/horde/apis/models/kobold_v2.py
@@ -571,6 +571,7 @@ def __init__(self, api):
                     description="The UUID of the style. Use this to use the style or retrieve its information in the future.",
                     example="00000000-0000-0000-0000-000000000000",
                 ),
+                "use_count": fields.Integer(description="The amount of times this style has been used in generations."),
                 "creator": fields.String(description="The alias of the user to whom this style belongs to.", example="db0#1"),
             },
         )
diff --git a/horde/apis/models/stable_v2.py b/horde/apis/models/stable_v2.py
index 56f8c096..d6f53ac7 100644
--- a/horde/apis/models/stable_v2.py
+++ b/horde/apis/models/stable_v2.py
@@ -1149,6 +1149,7 @@ def __init__(self, api):
                     description="The UUID of the style. Use this to use the style or retrieve its information in the future.",
                     example="00000000-0000-0000-0000-000000000000",
                 ),
+                "use_count": fields.Integer(description="The amount of times this style has been used in generations."),
                 "creator": fields.String(description="The alias of the user to whom this style belongs to.", example="db0#1"),
                 "examples": fields.List(fields.Nested(self.response_model_style_example, skip_none=True)),
             },
diff --git a/horde/apis/models/v2.py b/horde/apis/models/v2.py
index 7a3d90cb..7d94811b 100644
--- a/horde/apis/models/v2.py
+++ b/horde/apis/models/v2.py
@@ -1862,5 +1862,6 @@ def __init__(self, api):
                     ),
                 ),
                 "styles": fields.List(fields.Nested(self.response_model_styles_short)),
+                "use_count": fields.Integer(description="The amount of times this collection has been used in generations."),
             },
         )
diff --git a/horde/classes/base/style.py b/horde/classes/base/style.py
index ddad7db5..8c524adf 100644
--- a/horde/classes/base/style.py
+++ b/horde/classes/base/style.py
@@ -195,6 +195,7 @@ def get_details(self, details_privilege=0):
         """We display these in the styles list json"""
         ret_dict = {
             "name": self.name,
+            "info": self.info,
             "id": self.id,
             "params": self.params,
             "prompt": self.prompt,
diff --git a/horde/consts.py b/horde/consts.py
index 536b8ac7..fe54318b 100644
--- a/horde/consts.py
+++ b/horde/consts.py
@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: AGPL-3.0-or-later
 
-HORDE_VERSION = "4.44.0"
+HORDE_VERSION = "4.44.1"
 HORDE_API_VERSION = "2.5"
 
 WHITELISTED_SERVICE_IPS = {