Fix tests

harishmohanraj · harishmohanraj · commit eea617bff9a0 · 2024-12-19T22:59:27.000Z
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 number	Diff line change
`@@ -226,7 +226,7 @@ def test_retrieve_utils(self):`
`226`	`226`	`dir_path="./website/docs",`
`227`	`227`	`client=client,`
`228`	`228`	`collection_name="autogen-docs",`
`229`		`- custom_text_types=["txt", "md", "rtf", "rst"],`
	`229`	`+ custom_text_types=["txt", "md", "rtf", "rst", "mdx"],`
`230`	`230`	`get_or_create=True,`
`231`	`231`	`)`
`232`	`232`	`results = query_vector_db(`