From ac672cc1e120abab0ee6a17bb99f4555320dae41 Mon Sep 17 00:00:00 2001 From: db0 Date: Tue, 26 Dec 2023 17:30:32 +0100 Subject: [PATCH] feat: support for lora is_version --- horde_sdk/ai_horde_api/apimodels/base.py | 2 ++ .../ai_horde_api/example_payloads/_v2_generate_async_post.json | 3 ++- .../example_responses/_v2_generate_pop_post_200.json | 3 ++- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/horde_sdk/ai_horde_api/apimodels/base.py b/horde_sdk/ai_horde_api/apimodels/base.py index fab9ebc..551ca8d 100644 --- a/horde_sdk/ai_horde_api/apimodels/base.py +++ b/horde_sdk/ai_horde_api/apimodels/base.py @@ -74,6 +74,8 @@ class LorasPayloadEntry(BaseModel): """The strength of the LoRa against the clip model.""" inject_trigger: str | None = Field(default=None, min_length=1, max_length=30) """Any trigger required to activate the LoRa model.""" + is_version: bool = Field(default=False) + """If true, will treat the lora name as a version ID.""" class TIPayloadEntry(BaseModel): diff --git a/tests/test_data/ai_horde_api/example_payloads/_v2_generate_async_post.json b/tests/test_data/ai_horde_api/example_payloads/_v2_generate_async_post.json index 0a1f4b8..7adc708 100644 --- a/tests/test_data/ai_horde_api/example_payloads/_v2_generate_async_post.json +++ b/tests/test_data/ai_horde_api/example_payloads/_v2_generate_async_post.json @@ -24,7 +24,8 @@ "name": "GlowingRunesAIV6", "model": 1.0, "clip": 1.0, - "inject_trigger": "a" + "inject_trigger": "a", + "is_version": false } ], "tis": [ diff --git a/tests/test_data/ai_horde_api/example_responses/_v2_generate_pop_post_200.json b/tests/test_data/ai_horde_api/example_responses/_v2_generate_pop_post_200.json index 139498a..ea18405 100644 --- a/tests/test_data/ai_horde_api/example_responses/_v2_generate_pop_post_200.json +++ b/tests/test_data/ai_horde_api/example_responses/_v2_generate_pop_post_200.json @@ -23,7 +23,8 @@ "name": "GlowingRunesAIV6", "model": 1.0, "clip": 1.0, - "inject_trigger": "a" + "inject_trigger": "a", + "is_version": false } ], "tis": [