diff --git a/pageindex/config.yaml b/pageindex/config.yaml index e1d4666..6b57144 100644 --- a/pageindex/config.yaml +++ b/pageindex/config.yaml @@ -1,7 +1,7 @@ -model: gpt-4o-2024-11-20 +model: "gpt-4o-2024-11-20" toc_check_page_num: 20 max_page_num_each_node: 10 max_token_num_each_node: 20000 -if_add_node_id: yes -if_add_node_summary: no -if_add_doc_description: yes \ No newline at end of file +if_add_node_id: "yes" +if_add_node_summary: "no" +if_add_doc_description: "yes" \ No newline at end of file diff --git a/pageindex/page_index.py b/pageindex/page_index.py index 8c44268..e3e6330 100644 --- a/pageindex/page_index.py +++ b/pageindex/page_index.py @@ -1033,7 +1033,7 @@ def page_index_main(doc, opt=None): def page_index(doc, model=None, toc_check_page_num=None, max_page_num_each_node=None, max_token_num_each_node=None, - f_add_node_id=None, if_add_node_summary=None, if_add_doc_description=None): + if_add_node_id=None, if_add_node_summary=None, if_add_doc_description=None): user_opt = { arg: value for arg, value in locals().items()