add AutoEmbedding to init

safevideo · Jan 12, 2024 · 607797d · 607797d
1 parent 4fa30c1
commit 607797d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/autollm/__init__.py b/autollm/__init__.py
@@ -8,6 +8,7 @@
 __author__ = 'safevideo'
 __license__ = 'AGPL-3.0'
 
+from autollm.auto.embedding import AutoEmbedding
 from autollm.auto.fastapi_app import AutoFastAPI
 from autollm.auto.llm import AutoLiteLLM
 from autollm.auto.query_engine import AutoQueryEngine
@@ -22,6 +23,6 @@
 
 __all__ = [
     'AutoLiteLLM', 'AutoServiceContext', 'AutoVectorStoreIndex', 'AutoQueryEngine', 'AutoFastAPI',
-    'read_files_as_documents', 'read_github_repo_as_documents', 'read_webpage_as_documents',
+    'AutoEmbedding', 'read_files_as_documents', 'read_github_repo_as_documents', 'read_webpage_as_documents',
     'read_website_as_documents'
 ]