From bd2ae9948972fbeaf06b663cbc47b9203feceea0 Mon Sep 17 00:00:00 2001
From: ChihYu Yeh <chihyu.jimmy.yeh@gmail.com>
Date: Mon, 1 Jul 2024 14:37:43 +0800
Subject: [PATCH] fix test

---
 wren-ai-service/tests/pytest/providers/test_loader.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/wren-ai-service/tests/pytest/providers/test_loader.py b/wren-ai-service/tests/pytest/providers/test_loader.py
index 7598a5c691..175487dad3 100644
--- a/wren-ai-service/tests/pytest/providers/test_loader.py
+++ b/wren-ai-service/tests/pytest/providers/test_loader.py
@@ -3,7 +3,7 @@
 
 def test_import_mods():
     loader.import_mods("src.providers")
-    assert len(loader.PROVIDERS) == 6
+    assert len(loader.PROVIDERS) == 7
 
 
 def test_get_provider():
@@ -29,3 +29,6 @@ def test_get_provider():
 
     provider = loader.get_provider("wren-ibis")
     assert provider.__name__ == "WrenIbis"
+
+    provider = loader.get_provider("wren-engine")
+    assert provider.__name__ == "WrenEngine"