Skip to content

Commit b96f9c1

Browse files
pamelafoxCopilot
andauthored
Apply suggestions from code review
Co-authored-by: Copilot <[email protected]>
1 parent db9dc7e commit b96f9c1

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

app/backend/prepdocslib/blobmanager.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -432,7 +432,8 @@ async def upload_blob(self, file: File) -> str:
432432
blob_client = await container_client.upload_blob(blob_name, reopened_file, overwrite=True)
433433
file.url = blob_client.url
434434

435-
assert file.url is not None, "file.url must be set after upload"
435+
if file.url is None:
436+
raise ValueError("file.url must be set after upload")
436437
return unquote(file.url)
437438

438439
async def upload_document_image(

app/backend/prepdocslib/page.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def from_skill_payload(cls, data: dict[str, Any]) -> tuple["ImageOnPage", str]:
6767
if filename is None:
6868
raise ValueError("filename is required")
6969
if figure_id is None:
70-
raise ValueError("figure_id is required")
70+
raise ValueError("figure_id is required for ImageOnPage deserialization")
7171

7272
# Generate placeholder if not provided
7373
if placeholder is None:

tests/test_searchmanager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -694,7 +694,7 @@ async def mock_upload_documents(self, documents):
694694
description="Test image",
695695
figure_id="fig1",
696696
page_num=0,
697-
placeholder="<figure id='fig1'></figure>", # required positional arg
697+
placeholder="<figure id='fig1'></figure>",
698698
url="http://example.com/img1.png",
699699
embedding=[0.01, 0.02],
700700
)

0 commit comments

Comments
 (0)