apunkt/PageIndex
1
0
Fork 0
mirror of https://github.com/VectifyAI/PageIndex.git synced 2026-05-31 19:15:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix model

Browse source
This commit is contained in:
Ray 2025-08-28 13:07:15 +08:00
parent 6d1b505541
commit 4b4ae4d51d
5 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pageindex/page_index.py
View file

@ -496,7 +496,7 @@ def remove_first_physical_index_section(text):
return text
### add verify completeness
def generate_toc_continue(toc_content, part, model="gpt-4.1"):
def generate_toc_continue(toc_content, part, model="gpt-4o-2024-11-20"):
print('start generate_toc_continue')
prompt = """
You are an expert in extracting hierarchical tree structure.
@ -729,7 +729,7 @@ def check_toc(page_list, opt=None):
################### fix incorrect toc #########################################################
def single_toc_item_index_fixer(section_title, content, model="gpt-4.1"):
def single_toc_item_index_fixer(section_title, content, model="gpt-4o-2024-11-20"):
tob_extractor_prompt = """
You are given a section title and several pages of a document, your job is to find the physical index of the start page of the section in the partial document.