Split out monolith node into smaller nodes

Author: jfcantu

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ComfyUI-Prompt-Companion"
-version = "0.0.6"
+version = "0.0.7"
 description = "A node that lets you save and reuse parts of prompts (embeddings, quality keywords, and so on.)"
 authors = [
   {name = "John Cantu", email = "jfcantu@gmail.com"}
@@ -21,11 +21,13 @@ dependencies = [
 
 [project.optional-dependencies]
 dev = [
+    "aiohttp",  # for testing API handlers
     "bump-my-version",
     "coverage",  # testing
     "mypy",  # linting
     "pre-commit", # runs linting on commit
     "pytest",  # testing
+    "pytest-asyncio",  # for async test support
     "ruff",  # linting
 ]
 

pytest.ini

@@ -3,6 +3,7 @@ testpaths = tests
 python_files = test_*.py
 python_classes = Test*
 python_functions = test_*
+pythonpath = ../../
 addopts = -v --tb=short --strict-markers
 markers =
     asyncio: marks tests as async tests

src/api_handlers.py

@@ -129,7 +129,6 @@ async def get_prompt_additions(request: web.Request) -> web.Response:
         JSON response with prompt additions and groups data
     """
     try:
-        print(f"[ComfyUI-Prompt-Companion] get_prompt_additions called, PROMPT_ADDITIONS: {PROMPT_ADDITIONS}")
         if PROMPT_ADDITIONS is None:
             return create_error_response(
                 "Prompt additions not initialized",
@@ -143,7 +142,6 @@ async def get_prompt_additions(request: web.Request) -> web.Response:
         )
     except Exception as e:
         logging.error(f"Server error retrieving prompt additions: {e}")
-        print(f"[ComfyUI-Prompt-Companion] Exception in get_prompt_additions: {e}")
         return create_error_response(
             "Failed to retrieve prompt additions",
             ["An unexpected error occurred"],
@@ -174,12 +172,12 @@ async def write_prompt_addition(request: web.Request) -> web.Response:
     # Input validation
     is_valid, message, errors = validate_request_json(prompt_addition_data)
     if not is_valid:
-        return create_error_response(message, errors, status=400)
+        return create_error_response(message or "Validation error", errors or [], status=400)
 
     # Validate required fields
     is_valid, message, errors = validate_name_field(prompt_addition_data)
     if not is_valid:
-        return create_error_response(message, errors, status=400)
+        return create_error_response(message or "Validation error", errors or [], status=400)
 
     name = prompt_addition_data["name"].strip()
 
@@ -311,12 +309,12 @@ async def write_prompt_group(request: web.Request) -> web.Response:
     # Input validation
     is_valid, message, errors = validate_request_json(prompt_group_data)
     if not is_valid:
-        return create_error_response(message, errors, status=400)
+        return create_error_response(message or "Validation error", errors or [], status=400)
 
     # Validate required fields
     is_valid, message, errors = validate_name_field(prompt_group_data)
     if not is_valid:
-        return create_error_response(message, errors, status=400)
+        return create_error_response(message or "Validation error", errors or [], status=400)
 
     name = prompt_group_data["name"].strip()
 

src/extension_config.py

@@ -264,8 +264,5 @@ def prompt_groups(self) -> dict[int, PromptGroup]:
 if os.path.exists(CONFIG_PATH):
     with open(CONFIG_PATH, "r") as config_file:
         config_data = json.load(config_file)
-        print(f"[ComfyUI-Prompt-Companion] Loaded config from: {CONFIG_PATH}")
-else:
-    print(f"[ComfyUI-Prompt-Companion] No existing config found, will create new config at: {CONFIG_PATH}")
 
 PROMPT_ADDITIONS = ExtensionConfig(config_data)

src/nodes.py

@@ -19,7 +19,6 @@
 # Register API endpoints only when running in ComfyUI
 try:
     from server import PromptServer
-    print("[ComfyUI-Prompt-Companion] PromptServer imported successfully")
 
     # Import API handlers from separate module
     from api_handlers import (
@@ -30,11 +29,9 @@
         write_prompt_group,
         delete_prompt_group
     )
-    print("[ComfyUI-Prompt-Companion] API handlers imported successfully")
 
     # Get the routes instance for registering API endpoints
     routes = PromptServer.instance.routes
-    print(f"[ComfyUI-Prompt-Companion] Got routes instance: {routes}")
 
     # Register API endpoints using decorator pattern
     PromptServer.instance.routes.get("/prompt-companion/prompt-addition")(get_prompt_additions)
@@ -44,13 +41,11 @@
     PromptServer.instance.routes.post("/prompt-companion/prompt-group")(write_prompt_group)
     PromptServer.instance.routes.delete("/prompt-companion/prompt-group/{prompt_group_id}")(delete_prompt_group)
 
-    print(f"[ComfyUI-Prompt-Companion] API routes registered successfully - Total routes: {len(routes)}")
 
 except ImportError as e:
     # Running outside ComfyUI - routes won't be registered
-    print(f"[ComfyUI-Prompt-Companion] Running outside ComfyUI - API routes not registered: {e}")
+    pass
 except Exception as e:
-    print(f"[ComfyUI-Prompt-Companion] Error during route registration: {e}")
     import traceback
     traceback.print_exc()