diff --git a/arcaflow_plugin_opensearch/opensearch_schema.py b/arcaflow_plugin_opensearch/opensearch_schema.py index 70499c9..86537e5 100644 --- a/arcaflow_plugin_opensearch/opensearch_schema.py +++ b/arcaflow_plugin_opensearch/opensearch_schema.py @@ -27,7 +27,7 @@ class BulkUploadOperationMeta: @dataclass -class BulkUploadObject: +class BulkUploadObject: operation: typing.Annotated[ typing.Dict[Operation, BulkUploadOperationMeta], schema.name("operation"), @@ -39,6 +39,7 @@ class BulkUploadObject: schema.description("The JSON data document to upload to your index."), ] + bulk_upload_object_schema = plugin.build_object_schema(BulkUploadObject) @@ -94,7 +95,6 @@ class DocumentRequest: @dataclass class SuccessOutput: message: str - document_ids: typing.List[str]