diff --git a/autogen/retrieve_utils.py b/autogen/retrieve_utils.py index e20fec9656..774e5e57d3 100644 --- a/autogen/retrieve_utils.py +++ b/autogen/retrieve_utils.py @@ -52,6 +52,7 @@ "yaml", "yml", "pdf", + "mdx", ] UNSTRUCTURED_FORMATS = [ "doc", diff --git a/test/test_retrieve_utils.py b/test/test_retrieve_utils.py index 18bb1ea23f..f9e8d71b74 100755 --- a/test/test_retrieve_utils.py +++ b/test/test_retrieve_utils.py @@ -226,7 +226,7 @@ def test_retrieve_utils(self): dir_path="./website/docs", client=client, collection_name="autogen-docs", - custom_text_types=["txt", "md", "rtf", "rst"], + custom_text_types=["txt", "md", "rtf", "rst", "mdx"], get_or_create=True, ) results = query_vector_db(