Skip to content

Commit e6754df

Browse files
committed
Requested changes: added comments explaining title truncation
1 parent 669f939 commit e6754df

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

server/api/views/uploadFile/test_title.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,4 +96,5 @@ def test_truncates_long_openai_title(self, mock_openAI):
9696

9797
result = title.generate_title(doc)
9898

99+
# Ensure the title is truncated to fit the UploadFile model's title field (max_length=255), since OpenAI responses may exceed this limit
99100
self.assertLessEqual(len(result), 255)

server/api/views/uploadFile/title.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,4 +59,5 @@ def summarize_pdf(pdf: fitz.Document) -> str:
5959
response = openAIServices.openAI(
6060
first_page_content, prompt, model='gpt-4o', temp=0.0)
6161
title = response.choices[0].message.content.strip().strip('"').strip("'")
62+
# Truncate to fit UploadFile model's max_length=255 title field as a final safeguard
6263
return title[:255]

0 commit comments

Comments
 (0)