feat: bare-bones 'style' api update support

This just brings the testing up to the point of only failing on the unsupported/not yet added endpoints.

See Haidra-Org/AI-Horde#465 for more information.

Author: tazlin

docs/horde_sdk/ai_horde_api/apimodels/_styles.md

@@ -0,0 +1,2 @@
+# _styles
+::: horde_sdk.ai_horde_api.apimodels._styles

docs/response_field_names_and_descriptions.json

@@ -444,6 +444,13 @@
                 "int",
                 "None"
             ]
+        },
+        "styles": {
+            "description": "The styles this user has created.",
+            "types": [
+                "list[horde_sdk.ai_horde_api.apimodels._styles.ResponseModelStylesUser]",
+                "None"
+            ]
         }
     },
     "HordePerformanceResponse": {
@@ -726,6 +733,13 @@
                 "list[str]",
                 "None"
             ]
+        },
+        "ttl": {
+            "description": "The amount of seconds before this job is considered stale and aborted.",
+            "types": [
+                "int",
+                "None"
+            ]
         }
     },
     "JobSubmitResponse": {
@@ -1308,6 +1322,13 @@
                 "str",
                 "None"
             ]
+        },
+        "ttl": {
+            "description": "The amount of seconds before this job is considered stale and aborted.",
+            "types": [
+                "int",
+                "None"
+            ]
         }
     },
     "TextStatsModelResponse": {

horde_sdk/ai_horde_api/apimodels/__init__.py

@@ -44,6 +44,7 @@
     NewsRequest,
     NewsResponse,
 )
+from horde_sdk.ai_horde_api.apimodels._styles import ResponseModelStylesUser
 from horde_sdk.ai_horde_api.apimodels._users import (
     ActiveGenerations,
     ContributionsDetails,
@@ -210,6 +211,7 @@
     "Newspiece",
     "NewsRequest",
     "NewsResponse",
+    "ResponseModelStylesUser",
     "ListUsersDetailsRequest",
     "ListUsersDetailsResponse",
     "ModifyUser",

horde_sdk/ai_horde_api/apimodels/_styles.py

@@ -0,0 +1,19 @@
+from enum import auto
+
+from pydantic import Field
+from strenum import StrEnum
+
+from horde_sdk.generic_api.apimodels import HordeAPIDataObject
+
+
+class StyleType(StrEnum):
+    """An enum representing the different types of styles."""
+
+    image = auto()
+    text = auto()
+
+
+class ResponseModelStylesUser(HordeAPIDataObject):
+    name: str
+    id_: str = Field(alias="id")
+    type_: StyleType = Field(alias="type")

horde_sdk/ai_horde_api/apimodels/_users.py

@@ -3,6 +3,7 @@
 from pydantic import Field, RootModel
 from typing_extensions import override
 
+from horde_sdk.ai_horde_api.apimodels._styles import ResponseModelStylesUser
 from horde_sdk.ai_horde_api.apimodels.base import BaseAIHordeRequest
 from horde_sdk.ai_horde_api.endpoints import AI_HORDE_API_ENDPOINT_SUBPATH
 from horde_sdk.ai_horde_api.fields import UUID_Identifier
@@ -52,6 +53,9 @@ class UserKudosDetails(HordeAPIDataObject):
     recurring: float | None = Field(0)
     """The amount of Kudos this user has received from recurring rewards."""
 
+    styled: float | None = Field(0)
+    """The amount of Kudos this user has received from styling images."""
+
 
 class MonthlyKudos(HordeAPIDataObject):
     amount: int | None = Field(default=None)
@@ -85,6 +89,7 @@ class UserRecords(HordeAPIDataObject):
     fulfillment: UserAmountRecords | None = None
     request: UserAmountRecords | None = None
     usage: UserThingRecords | None = None
+    style: UserAmountRecords | None = None
 
 
 class UsageDetails(HordeAPIDataObject):
@@ -264,6 +269,9 @@ def get_api_model_name(cls) -> str | None:
     """Whether this user has been invited to join a worker to the horde and how many of them.
     When 0, this user cannot add (new) workers to the horde."""
 
+    styles: list[ResponseModelStylesUser] | None = None
+    """The styles this user has created."""
+
 
 @Unhashable
 @Unequatable

horde_sdk/ai_horde_api/apimodels/generate/_pop.py

@@ -232,6 +232,8 @@ class ImageGenerateJobPopResponse(
     """(Obsolete) The r2 upload link to use to upload this image."""
     r2_uploads: list[str] | None = None
     """The r2 upload links for each this image. Each index matches the ID in self.ids"""
+    ttl: int | None = None
+    """The amount of seconds before this job is considered stale and aborted."""
 
     @field_validator("source_processing")
     def source_processing_must_be_known(cls, v: str | KNOWN_SOURCE_PROCESSING) -> str | KNOWN_SOURCE_PROCESSING:

horde_sdk/ai_horde_api/apimodels/generate/text/_pop.py

@@ -60,6 +60,8 @@ class TextGenerateJobPopResponse(
     """The soft prompt requested for this generation."""
     model: str | None = Field(default=None)
     """The model requested for this generation."""
+    ttl: int | None = None
+    """The amount of seconds before this job is considered stale and aborted."""
 
     @field_validator("id_", mode="before")
     def validate_id(cls, v: str | JobID) -> JobID | str:

horde_sdk/generic_api/utils/swagger.py

@@ -449,7 +449,12 @@ def get_endpoint_method_examples(
         # Iterate through each defined response for the HTTP method.
         for http_status_code_str, response_definition in endpoint_method_definition.responses.items():
             # Convert the HTTP status code string to an HTTPStatusCode object.
-            http_status_code_object = HTTPStatusCode(int(http_status_code_str))
+            try:
+                http_status_code_object = HTTPStatusCode(int(http_status_code_str))
+            except ValueError as e:
+                logger.error(f"Failed to convert {http_status_code_str} to an HTTPStatusCode object: {e}")
+                logger.error(f"response_definition: {response_definition}")
+                raise
 
             # Get the response example for this HTTP status code.
             example_response = self.get_response_example(response_definition)

tests/test_data/ai_horde_api/example_payloads/_v2_collections_collection_id_patch.json

@@ -0,0 +1,8 @@
+{
+    "name": "My Awesome Collection",
+    "info": "Collection of optimistic styles",
+    "public": true,
+    "styles": [
+        "a"
+    ]
+}

tests/test_data/ai_horde_api/example_payloads/_v2_collections_post.json

@@ -0,0 +1,8 @@
+{
+    "name": "My Awesome Collection",
+    "info": "Collection of optimistic styles",
+    "public": true,
+    "styles": [
+        "a"
+    ]
+}