From eea617bff9a00965faf27907ff6b934e14ba866a Mon Sep 17 00:00:00 2001 From: Harish Mohan Raj Date: Thu, 19 Dec 2024 22:59:27 +0000 Subject: [PATCH] Fix tests --- autogen/retrieve_utils.py | 1 + test/test_retrieve_utils.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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(