apunkt/PageIndex
1
0
Fork 0
mirror of https://github.com/VectifyAI/PageIndex.git synced 2026-05-03 20:12:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge pull request #65 from luojiyin1987/fix/extract-toc-infinite-loop

Browse source 
fix: prevent infinite loop in extract_toc_content
This commit is contained in:
Bukely_ 2026-03-16 13:34:05 +08:00 • committed by GitHub
commit 599d2ce497
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pageindex/page_index.py
View file

@ -180,7 +180,14 @@ def extract_toc_content(content, model=None):
response = response + new_response response = response + new_response
if_complete = check_if_toc_transformation_is_complete(content, response, model) if_complete = check_if_toc_transformation_is_complete(content, response, model)
attempt = 0
max_attempts = 5
while not (if_complete == "yes" and finish_reason == "finished"): while not (if_complete == "yes" and finish_reason == "finished"):
attempt += 1
if attempt > max_attempts:
raise Exception('Failed to complete table of contents after maximum retries')
chat_history = [ chat_history = [
{"role": "user", "content": prompt}, {"role": "user", "content": prompt},
{"role": "assistant", "content": response}, {"role": "assistant", "content": response},
@ -190,10 +197,6 @@ def extract_toc_content(content, model=None):
response = response + new_response response = response + new_response
if_complete = check_if_toc_transformation_is_complete(content, response, model) if_complete = check_if_toc_transformation_is_complete(content, response, model)
# Optional: Add a maximum retry limit to prevent infinite loops
if len(chat_history) > 5: # Arbitrary limit of 10 attempts
raise Exception('Failed to complete table of contents after maximum retries')
return response return response
def detect_page_index(toc_content, model=None): def detect_page_index(toc_content, model=None):