diff --git a/tree_of_thoughts/__init__.py b/tree_of_thoughts/__init__.py
index e69de29..4e59216 100644
--- a/tree_of_thoughts/__init__.py
+++ b/tree_of_thoughts/__init__.py
@@ -0,0 +1,5 @@
+from tree_of_thoughts.models.openai_models import OpenAILanguageModel, OptimizedOpenAILanguageModel
+from tree_of_thoughts.treeofthoughts import TreeofThoughts, MonteCarloTreeofThoughts, TreeofThoughtsBFS, TreeofThoughtsDFS, TreeofThoughtsBEST, TreeofThoughtsASearch
+from tree_of_thoughts.models.abstract_language_model import AbstractLanguageModel
+
+from tree_of_thoughts.models.huggingface_model import HuggingLanguageModel, HFPipelineModel