Fix tests

ag2ai · Dec 19, 2024 · eea617b · eea617b
1 parent 7de1810
commit eea617b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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(
-Original file line number
+Diff line change
@@ Expand Up / @@ -52,6 +52,7 @@ @@
         "yaml",
         "yml",
         "pdf",
+        "mdx",
     ]
     UNSTRUCTURED_FORMATS = [
         "doc",
@@ Expand Down @@