From 43fbd8a834ec8ba6e95911c009c3ae27cbd9649d Mon Sep 17 00:00:00 2001 From: moosecrap Date: Tue, 22 Sep 2026 00:39:44 -0700 Subject: [PATCH] Tag wiki fix --- config.py | 2 +- tools/search_tags.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/config.py b/config.py index d0d02ac..b6bfb6c 100644 --- a/config.py +++ b/config.py @@ -20,7 +20,7 @@ MODEL_PRESETS_PATH = str(ROOT_DIR / "model_presets.toml") RES_PRESETS_PATH = str(ROOT_DIR / "resolution_presets.toml") TAG_DATABASE_PATH = str(ROOT_DIR / "danbooru.csv") TAG_SEARCH_LIMIT = 20 -ENABLE_TAG_WIKI = False +ENABLE_TAG_WIKI = True DANBOORU_LOGIN = "" # Your Danbooru username (Optional) DANBOORU_API_KEY = "" # Your Danbooru API key (Optional) diff --git a/tools/search_tags.py b/tools/search_tags.py index 96d75fe..a66309d 100644 --- a/tools/search_tags.py +++ b/tools/search_tags.py @@ -67,7 +67,7 @@ async def _fetch_wiki_info(tag_name: str) -> Optional[str]: if resp.status_code == 200: data = resp.json() - wiki_body = data.get("wiki_page", {}).get("body") + wiki_body = data.get("body") if wiki_body: # Strip HTML tags for the LLM return re.sub(r'<[^>]*>', '', wiki_body).strip()