Skip to content

Commit 7ed20cb

Browse files
committed
fixed sdoc metadata issue
1 parent edb43c9 commit 7ed20cb

File tree

1 file changed

+8
-20
lines changed

1 file changed

+8
-20
lines changed

backend/src/modules/doc_processing/image/image_caption_job.py

Lines changed: 8 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,7 @@
22

33
from common.doc_type import DocType
44
from common.job_type import JobType
5-
from common.meta_type import MetaType
6-
from core.metadata.project_metadata_crud import crud_project_meta
75
from core.metadata.source_document_metadata_crud import crud_sdoc_meta
8-
from core.metadata.source_document_metadata_dto import SourceDocumentMetadataCreate
96
from modules.doc_processing.doc_processing_dto import SdocProcessingJobInput
107
from modules.llm_assistant.prompts.image_captioning_prompt import (
118
IMG_CAPTION_USER_PROMPT,
@@ -46,25 +43,16 @@ def handle_image_caption_job(
4643

4744
with sqlr.db_session() as db:
4845
# Store caption in the database
49-
lang_project_metadata = (
50-
crud_project_meta.read_by_project_and_key_and_metatype_and_doctype(
51-
db=db,
52-
project_id=payload.project_id,
53-
key="caption",
54-
metatype=MetaType.STRING,
55-
doctype=DocType.image,
56-
)
57-
)
58-
assert lang_project_metadata is not None, "Language metadata does not exist!"
59-
crud_sdoc_meta.create(
46+
crud_sdoc_meta.update_multi_with_doctype(
6047
db=db,
61-
create_dto=SourceDocumentMetadataCreate.with_metatype(
62-
value=caption,
63-
source_document_id=payload.sdoc_id,
64-
project_metadata_id=lang_project_metadata.id,
65-
metatype=MetaType.STRING,
66-
),
48+
project_id=payload.project_id,
49+
sdoc_id=payload.sdoc_id,
50+
doctype=DocType.image,
51+
keys=["caption"],
52+
values=[caption],
53+
manual_commit=True,
6754
)
55+
6856
return ImageCaptionJobOutput(
6957
text=caption, html=f"<html><body><p>{caption}</p></body></html>"
7058
)

0 commit comments

Comments
 (0)