feat: support news title and more_info_urls fields

horde/apis/models/v2.py

@@ -1168,6 +1168,13 @@ def __init__(self, api):
                     example="Information",
                     description="How critical this piece of news is.",
                 ),
+                "tags": fields.List(
+                    fields.String(description="Tags for this newspiece."),
+                ),
+                "title": fields.String(description="The title of this newspiece."),
+                "more_info_urls": fields.List(
+                    fields.String(description="URLs for more information about this newspiece."),
+                ),
             },
         )
 

requirements.dev.txt

@@ -1,6 +1,6 @@
 pytest==8.0.2
-black==24.2.0
-ruff==0.3.1
+black==24.4.2
+ruff==0.4.2
 tox~=4.14.1
 horde_sdk>=0.7.29
 

tests/conftest.py

@@ -4,17 +4,17 @@
 import requests
 
 
-@pytest.fixture
+@pytest.fixture(scope="session")
 def CIVERSION() -> str:
     return "0.1.1"
 
 
-@pytest.fixture
+@pytest.fixture(scope="session")
 def HORDE_URL() -> str:
     return "localhost:7001"
 
 
-@pytest.fixture
+@pytest.fixture(scope="session")
 def api_key() -> str:
     key_file = pathlib.Path(__file__).parent / "apikey.txt"
     if key_file.exists():
@@ -23,7 +23,7 @@ def api_key() -> str:
     raise ValueError("No api key file found")
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture(autouse=True, scope="session")
 def increase_kudos(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
     headers = {"apikey": api_key, "Client-Agent": f"aihorde_ci_client:{CIVERSION}:(discord)db0#1625", "user_id": "1"}
 

tests/test_image.py

@@ -1,3 +1,5 @@
+import json
+
 import requests
 
 TEST_MODELS = ["Fustercluck", "AlbedoBase XL (SDXL)"]
@@ -31,10 +33,12 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
     async_results = async_req.json()
     req_id = async_results["id"]
     # print(async_results)
+    print(async_results)
     pop_dict = {
         "name": "CICD Fake Dreamer",
         "models": TEST_MODELS,
-        "bridge_agent": "AI Horde Worker reGen:4.1.0-citests:https://github.com/Haidra-Org/horde-worker-reGen",
+        "bridge_agent": "AI Horde Worker reGen:8.0.1-citests:https://github.com/Haidra-Org/horde-worker-reGen",
+        "nsfw": True,
         "amount": 10,
         "max_pixels": 4194304,
         "allow_img2img": True,
@@ -47,13 +51,14 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
     }
     pop_req = requests.post(f"{protocol}://{HORDE_URL}/api/v2/generate/pop", json=pop_dict, headers=headers)
     try:
+        print(pop_req.text)
         assert pop_req.ok, pop_req.text
     except AssertionError as err:
         requests.delete(f"{protocol}://{HORDE_URL}/api/v2/generate/status/{req_id}", headers=headers)
         print("Request cancelled")
         raise err
     pop_results = pop_req.json()
-    # print(json.dumps(pop_results, indent=4))
+    print(json.dumps(pop_results, indent=4))
 
     job_id = pop_results["id"]
     try:
@@ -75,7 +80,7 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
     retrieve_req = requests.get(f"{protocol}://{HORDE_URL}/api/v2/generate/status/{req_id}", headers=headers)
     assert retrieve_req.ok, retrieve_req.text
     retrieve_results = retrieve_req.json()
-    # print(json.dumps(retrieve_results,indent=4))
+    print(json.dumps(retrieve_results, indent=4))
     assert len(retrieve_results["generations"]) == 1
     gen = retrieve_results["generations"][0]
     assert len(gen["gen_metadata"]) == 0

tests/test_image_extra_sources.py

@@ -55,7 +55,8 @@ def test_simple_image_gen(api_key: str, HORDE_URL: str, CIVERSION: str) -> None:
     pop_dict = {
         "name": "CICD Fake Dreamer",
         "models": TEST_MODELS,
-        "bridge_agent": "AI Horde Worker reGen:5.3.0-citests:https://github.com/Haidra-Org/horde-worker-reGen",
+        "bridge_agent": "AI Horde Worker reGen:8.0.1-citests:https://github.com/Haidra-Org/horde-worker-reGen",
+        "nsfw": True,
         "amount": 10,
         "max_pixels": 4194304,
         "allow_img2img": True,