mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-05-04 20:42:36 +02:00
fix structure
This commit is contained in:
parent
802f149dd1
commit
34ed3fbc60
2 changed files with 36 additions and 3 deletions
|
|
@ -606,6 +606,24 @@ def generate_doc_description(structure, model=None):
|
|||
return response
|
||||
|
||||
|
||||
def reorder_dict(data, key_order):
|
||||
if not key_order:
|
||||
return data
|
||||
return {key: data[key] for key in key_order if key in data}
|
||||
|
||||
|
||||
def format_structure(structure, order=None):
|
||||
if isinstance(structure, dict):
|
||||
if 'nodes' in structure:
|
||||
structure['nodes'] = format_structure(structure['nodes'], order)
|
||||
if not structure.get('nodes'):
|
||||
structure.pop('nodes', None)
|
||||
structure = reorder_dict(structure, order)
|
||||
elif isinstance(structure, list):
|
||||
structure = [format_structure(item, order) for item in structure]
|
||||
return structure
|
||||
|
||||
|
||||
class ConfigLoader:
|
||||
def __init__(self, default_path: str = None):
|
||||
if default_path is None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue