Vito
4c9cb7f5ea
* refactor: remove dead and test-only code and the write-only prompt registry (#620) Removes helpers nothing calls, test-only wrappers and their tests, the unused config writer, three unused images, and the prompt registry the nodes wrote to but nothing read, which grew with every run. * fix: network-share readers, model unload off the loop, bad save requests (#620) Every read-only SQLite open now uses one URI helper, so a library on a Windows network share works in backups, restores, moves and history reads. Unloading a model no longer blocks the server. A malformed /save body gets a 400, not a 500. * feat(library): add Recently used, Most used and Name Z-A sorting (#620) The old Dashboard's sorts the community asked for, on the Prompt library, in saved views, and with paging through the whole list. * feat(workspace): stop linking the old pages (#620) They show a removal notice since 3.5.1. System no longer links them, the load failure offers a reload and ComfyUI instead, and the README says they are going away. The pages still work at their URLs.
61 lines
2.6 KiB
Python
61 lines
2.6 KiB
Python
"""Focused shares encoding and tracking while exposing composition sockets."""
|
|
|
|
import logging
|
|
import unittest
|
|
import unittest.mock as mock
|
|
|
|
from prompt_manager import PromptManager, PromptManagerFocused
|
|
from utils.prompt_graph import positive_prompt_nodes, run_prompt_nodes
|
|
|
|
|
|
class TestFocusedNode(unittest.TestCase):
|
|
def test_focused_keeps_the_execution_contract_and_visible_string_inputs(self):
|
|
self.assertTrue(issubclass(PromptManagerFocused, PromptManager))
|
|
self.assertEqual(PromptManagerFocused.RETURN_TYPES, PromptManager.RETURN_TYPES)
|
|
self.assertEqual(PromptManagerFocused.FUNCTION, PromptManager.FUNCTION)
|
|
focused = PromptManagerFocused.INPUT_TYPES()
|
|
for name in ("prepend_text", "append_text"):
|
|
self.assertTrue(focused["optional"][name][1]["forceInput"])
|
|
self.assertNotIn(
|
|
"forceInput", PromptManager.INPUT_TYPES()["optional"][name][1]
|
|
)
|
|
self.assertEqual(focused["hidden"]["unique_id"], "UNIQUE_ID")
|
|
|
|
def test_focused_encodes_composition_and_tracks_main_text_and_graph(self):
|
|
node = PromptManagerFocused.__new__(PromptManagerFocused)
|
|
node.logger = logging.getLogger("test.focused")
|
|
node._inject_lora_trigger_words = lambda text: text
|
|
node._track_prompt_execution = mock.Mock(return_value=42)
|
|
clip = mock.Mock()
|
|
graph = {"1": {"class_type": "PromptManagerFocused", "inputs": {"text": "cat"}}}
|
|
conditioning, text = node.encode_prompt(
|
|
clip,
|
|
"cat",
|
|
prepend_text="trigger",
|
|
append_text="detailed",
|
|
prompt=graph,
|
|
unique_id="1",
|
|
)
|
|
self.assertEqual(text, "trigger cat detailed")
|
|
clip.tokenize.assert_called_once_with(text)
|
|
self.assertIs(conditioning, clip.encode_from_tokens_scheduled.return_value)
|
|
tracked = node._track_prompt_execution.call_args.kwargs
|
|
self.assertEqual(tracked["text"], "cat")
|
|
self.assertEqual(tracked["prompt_graph"], graph)
|
|
|
|
def test_focused_is_traced_as_positive_and_not_as_the_negative_run(self):
|
|
graph = {
|
|
"1": {"class_type": "PromptManagerFocused", "inputs": {"text": "cat"}},
|
|
"2": {"class_type": "PromptManagerFocused", "inputs": {"text": "blur"}},
|
|
"3": {
|
|
"class_type": "KSampler",
|
|
"inputs": {
|
|
"model": ["4", 0],
|
|
"positive": ["1", 0],
|
|
"negative": ["2", 0],
|
|
},
|
|
},
|
|
}
|
|
self.assertEqual(positive_prompt_nodes(graph), ["1"])
|
|
self.assertEqual(run_prompt_nodes(graph), ["1"])
|