diff --git a/cognee/tests/integration/documents/AudioDocument_test.py b/cognee/tests/integration/documents/AudioDocument_test.py index 2903844fe..c079e2122 100644 --- a/cognee/tests/integration/documents/AudioDocument_test.py +++ b/cognee/tests/integration/documents/AudioDocument_test.py @@ -29,6 +29,7 @@ def test_AudioDocument(): id=uuid.uuid4(), name="audio-dummy-test", raw_data_location="", + foreign_metadata="", mime_type="", ) with patch.object(AudioDocument, "create_transcript", return_value=TEST_TEXT): diff --git a/cognee/tests/integration/documents/ImageDocument_test.py b/cognee/tests/integration/documents/ImageDocument_test.py index 8bac287f2..c7e3ee82d 100644 --- a/cognee/tests/integration/documents/ImageDocument_test.py +++ b/cognee/tests/integration/documents/ImageDocument_test.py @@ -18,6 +18,7 @@ def test_ImageDocument(): id=uuid.uuid4(), name="image-dummy-test", raw_data_location="", + foreign_metadata="", mime_type="", ) with patch.object(ImageDocument, "transcribe_image", return_value=TEST_TEXT): diff --git a/cognee/tests/integration/documents/TextDocument_test.py b/cognee/tests/integration/documents/TextDocument_test.py index 51e2aabcf..6c0c42880 100644 --- a/cognee/tests/integration/documents/TextDocument_test.py +++ b/cognee/tests/integration/documents/TextDocument_test.py @@ -32,6 +32,7 @@ def test_TextDocument(input_file, chunk_size): id=uuid.uuid4(), name=input_file, raw_data_location=test_file_path, + foreign_metadata="", mime_type="", ) diff --git a/cognee/tests/integration/documents/UnstructuredDocument_test.py b/cognee/tests/integration/documents/UnstructuredDocument_test.py index 3dea34a6a..a619b745e 100644 --- a/cognee/tests/integration/documents/UnstructuredDocument_test.py +++ b/cognee/tests/integration/documents/UnstructuredDocument_test.py @@ -39,6 +39,7 @@ def test_UnstructuredDocument(): id=uuid.uuid4(), name="example.pptx", raw_data_location=pptx_file_path, + foreign_metadata="", mime_type="application/vnd.openxmlformats-officedocument.presentationml.presentation", ) @@ -46,6 +47,7 @@ def test_UnstructuredDocument(): id=uuid.uuid4(), name="example.docx", raw_data_location=docx_file_path, + foreign_metadata="", mime_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document", ) @@ -53,6 +55,7 @@ def test_UnstructuredDocument(): id=uuid.uuid4(), name="example.csv", raw_data_location=csv_file_path, + foreign_metadata="", mime_type="text/csv", ) @@ -60,6 +63,7 @@ def test_UnstructuredDocument(): id=uuid.uuid4(), name="example.xlsx", raw_data_location=xlsx_file_path, + foreign_metadata="", mime_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", )