Skip to content

Commit ef6012a

Browse files
author
Bob Strahan
committed
remove enforcement of required placeholders
1 parent 1815ae2 commit ef6012a

File tree

2 files changed

+5
-9
lines changed

2 files changed

+5
-9
lines changed

lib/idp_common_pkg/idp_common/classification/service.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,7 @@ def _build_content_with_image_placeholder(
292292
"DOCUMENT_TEXT": document_text,
293293
"CLASS_NAMES_AND_DESCRIPTIONS": class_names_and_descriptions,
294294
},
295-
required_placeholders=[], # Don't enforce required placeholders for partial templates
295+
required_placeholders=[],
296296
)
297297

298298
after_image = self._prepare_prompt_from_template(
@@ -301,7 +301,7 @@ def _build_content_with_image_placeholder(
301301
"DOCUMENT_TEXT": document_text,
302302
"CLASS_NAMES_AND_DESCRIPTIONS": class_names_and_descriptions,
303303
},
304-
required_placeholders=[], # Don't enforce required placeholders for partial templates
304+
required_placeholders=[],
305305
)
306306

307307
# Build content array with image in the middle
@@ -355,7 +355,7 @@ def _build_content_without_image_placeholder(
355355
"DOCUMENT_TEXT": document_text,
356356
"CLASS_NAMES_AND_DESCRIPTIONS": class_names_and_descriptions,
357357
},
358-
required_placeholders=["DOCUMENT_TEXT", "CLASS_NAMES_AND_DESCRIPTIONS"],
358+
required_placeholders=[],
359359
)
360360

361361
content = [{"text": task_prompt}]
@@ -1406,7 +1406,7 @@ def holistic_classify_document(self, document: Document) -> Document:
14061406
"DOCUMENT_TEXT": doc_text,
14071407
"CLASS_NAMES_AND_DESCRIPTIONS": classes_table,
14081408
},
1409-
required_placeholders=["DOCUMENT_TEXT", "CLASS_NAMES_AND_DESCRIPTIONS"],
1409+
required_placeholders=[],
14101410
)
14111411

14121412
# Invoke Bedrock to get the holistic classification

lib/idp_common_pkg/idp_common/extraction/service.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -257,11 +257,7 @@ def _build_content_without_image_placeholder(
257257
"DOCUMENT_CLASS": class_label,
258258
"ATTRIBUTE_NAMES_AND_DESCRIPTIONS": attribute_descriptions,
259259
},
260-
required_placeholders=[
261-
"DOCUMENT_TEXT",
262-
"DOCUMENT_CLASS",
263-
"ATTRIBUTE_NAMES_AND_DESCRIPTIONS",
264-
],
260+
required_placeholders=[],
265261
)
266262

267263
content = [{"text": task_prompt}]

0 commit comments

Comments
 (0)