|
2 | 2 |
|
3 | 3 | from common.doc_type import DocType |
4 | 4 | from common.job_type import JobType |
5 | | -from common.meta_type import MetaType |
6 | | -from core.metadata.project_metadata_crud import crud_project_meta |
7 | 5 | from core.metadata.source_document_metadata_crud import crud_sdoc_meta |
8 | | -from core.metadata.source_document_metadata_dto import SourceDocumentMetadataCreate |
9 | 6 | from modules.doc_processing.doc_processing_dto import SdocProcessingJobInput |
10 | 7 | from modules.llm_assistant.prompts.image_captioning_prompt import ( |
11 | 8 | IMG_CAPTION_USER_PROMPT, |
@@ -46,25 +43,16 @@ def handle_image_caption_job( |
46 | 43 |
|
47 | 44 | with sqlr.db_session() as db: |
48 | 45 | # 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( |
60 | 47 | 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, |
67 | 54 | ) |
| 55 | + |
68 | 56 | return ImageCaptionJobOutput( |
69 | 57 | text=caption, html=f"<html><body><p>{caption}</p></body></html>" |
70 | 58 | ) |
0 commit comments