Need an explanation of how root_map works with WildcardManager/RandomPromptGenerator

[ryanwalder]

I have a test yaml file:

test/test.yaml

test:
  foo:
    - one
    - two
  bar:
    - three
    - four

and an object with the same data:

wc = {"test": {"foo": ["one", "two"], "bar": ["three", "four"]}}

When I use the file, it can correctly generate a random prompt:

wm = WildcardManager(Path("test"))
generator = RandomPromptGenerator(wildcard_manager=wm)
foo = generator.generate("__test/foo__")

Which returns a random result (['two']) as expected.

However if I pass in the wcobject as a root_map I get an error:

wm = WildcardManager(root_map=wc)
generator = RandomPromptGenerator(wildcard_manager=wm)
foo = generator.generate("__test/foo__")

Traceback (most recent call last):
  File "/home/ryan/git/ipg/./reactor.py", line 78, in <module>
    main()
    ~~~~^^
  File "/home/ryan/git/ipg/./reactor.py", line 64, in main
    foo = generator.generate("__test/foo__")
  File "/home/ryan/git/ipg/venv/lib/python3.13/site-packages/dynamicprompts/generators/randomprompt.py", line 74, in generate
    str(p) for p in self._context.sample_prompts(template, num_images)
                    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^
  File "/home/ryan/git/ipg/venv/lib/python3.13/site-packages/dynamicprompts/samplers/random.py", line 119, in _get_wildcard
    values = context.wildcard_manager.get_values(wildcard_path)
  File "/home/ryan/git/ipg/venv/lib/python3.13/site-packages/dynamicprompts/wildcards/wildcard_manager.py", line 163, in get_values
    values = self._get_values(wildcard)
  File "/home/ryan/git/ipg/venv/lib/python3.13/site-packages/dynamicprompts/wildcards/wildcard_manager.py", line 178, in _get_values
    for f in self.match_collections(wildcard):
             ~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^
  File "/home/ryan/git/ipg/venv/lib/python3.13/site-packages/dynamicprompts/wildcards/wildcard_manager.py", line 135, in match_collections
    return self.tree.match_collections(wildcard)
           ^^^^^^^^^
  File "/home/ryan/git/ipg/venv/lib/python3.13/site-packages/dynamicprompts/wildcards/wildcard_manager.py", line 116, in tree
    self._tree = build_tree_from_root_map(self._root_map)
                 ~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^
  File "/home/ryan/git/ipg/venv/lib/python3.13/site-packages/dynamicprompts/wildcards/tree/build.py", line 82, in build_tree_from_root_map
    for name, collection_or_list in _mappingify_item(item).items():
                                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'str' object has no attribute 'items'

Am I missing something here?

[akx]

The root map is meant for loading wildcards from the filesystem (and multiple roots therein).

If you have wildcard values in code, you would need to build a WildcardTree by hand, like in the test

dynamicprompts/tests/wildcard/test_tree.py

Lines 5 to 13 in 58815af

	def test_synthetic_tree():
	# Silly test for an entirely synthetic tree...
	root = WildcardTreeNode()
	root.collections["mike"] = ListWildcardCollection(["one", "two"])
	root.collections["joe"] = ListWildcardCollection(["coffee", "regular"])
	root.child_nodes["jane"] = WildcardTreeNode("jane")
	root.child_nodes["jane"].collections["doe"] = ListWildcardCollection(["homer"])
	wt = WildcardTree(root=root)
	assert set(wt.map) == {"mike", "joe", "jane/doe"}

,

and then if you want to use a WildcardManager with it, assign the generated tree to the private property _tree. (There is currently no public API to create a wildcard manager for a programmatic tree.)

[ryanwalder]

Thanks for the reply. If I raised a feature request to allow passing in a dict of wildcards (vs. loading from a Path) would it be considered? Would be a little less hacky than the approach above. Obviously cool if not in scope of the lib.